104 lines
2.3 KiB
Go
104 lines
2.3 KiB
Go
package models
|
|
|
|
import (
|
|
"slices"
|
|
"strings"
|
|
)
|
|
|
|
type Node struct {
|
|
ID string `json:"id"`
|
|
Title string `json:"title"`
|
|
Content string `json:"content,omitempty"`
|
|
DueDate string `json:"due_date,omitempty"`
|
|
CreatedAt string `json:"created_at"`
|
|
UpdatedAt string `json:"updated_at"`
|
|
Tags []string `json:"tags,omitempty"`
|
|
Relations map[string][]string `json:"relations,omitempty"`
|
|
}
|
|
|
|
func NewNode() *Node {
|
|
return &Node{
|
|
Relations: make(map[string][]string),
|
|
}
|
|
}
|
|
|
|
func (n *Node) AddTag(tag string) {
|
|
if tag == "" {
|
|
return
|
|
}
|
|
// If it's a property (name::value format), replace any existing tag with the same prefix.
|
|
if idx := strings.Index(tag, "::"); idx >= 0 {
|
|
prefix := tag[:idx+2]
|
|
var newTags []string
|
|
for _, t := range n.Tags {
|
|
if !strings.HasPrefix(t, prefix) {
|
|
newTags = append(newTags, t)
|
|
}
|
|
}
|
|
n.Tags = newTags
|
|
}
|
|
if !slices.Contains(n.Tags, tag) {
|
|
n.Tags = append(n.Tags, tag)
|
|
}
|
|
}
|
|
|
|
func (n *Node) RemoveTag(tag string) {
|
|
var newTags []string
|
|
for _, t := range n.Tags {
|
|
if t != tag {
|
|
newTags = append(newTags, t)
|
|
}
|
|
}
|
|
n.Tags = newTags
|
|
}
|
|
|
|
func (n *Node) AddRelation(relType RelType, target string) {
|
|
if n.Relations == nil {
|
|
n.Relations = make(map[string][]string)
|
|
}
|
|
if relType == RelAssignee || relType == RelCreated || relType == RelInNamespace {
|
|
n.Relations[string(relType)] = []string{target}
|
|
return
|
|
}
|
|
if !slices.Contains(n.Relations[string(relType)], target) {
|
|
n.Relations[string(relType)] = append(n.Relations[string(relType)], target)
|
|
}
|
|
}
|
|
|
|
func (n *Node) RemoveRelation(relType RelType, target string) {
|
|
if n.Relations == nil {
|
|
return
|
|
}
|
|
var newTgts []string
|
|
for _, tgt := range n.Relations[string(relType)] {
|
|
if tgt != target {
|
|
newTgts = append(newTgts, tgt)
|
|
}
|
|
}
|
|
if len(newTgts) == 0 {
|
|
delete(n.Relations, string(relType))
|
|
} else {
|
|
n.Relations[string(relType)] = newTgts
|
|
}
|
|
}
|
|
|
|
func (n *Node) GetProperty(k string) string {
|
|
prefix := "_" + k + "::"
|
|
for _, t := range n.Tags {
|
|
if strings.HasPrefix(t, prefix) {
|
|
return strings.TrimPrefix(t, prefix)
|
|
}
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func (n *Node) GetDisplayTags() []string {
|
|
var tags []string
|
|
for _, t := range n.Tags {
|
|
if !strings.HasPrefix(t, "_") {
|
|
tags = append(tags, t)
|
|
}
|
|
}
|
|
return tags
|
|
}
|