125 lines
2.6 KiB
Go
125 lines
2.6 KiB
Go
package models
|
|
|
|
import (
|
|
"strings"
|
|
)
|
|
|
|
var PropertyPrefixes = []string{"_type::", "_status::", "_prio::"}
|
|
|
|
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) HasTag(tag string) bool {
|
|
for _, t := range n.Tags {
|
|
if t == tag {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (n *Node) AddTag(tag string) {
|
|
if tag == "" {
|
|
return
|
|
}
|
|
for _, prefix := range PropertyPrefixes {
|
|
if strings.HasPrefix(tag, prefix) {
|
|
var newTags []string
|
|
for _, t := range n.Tags {
|
|
if !strings.HasPrefix(t, prefix) {
|
|
newTags = append(newTags, t)
|
|
}
|
|
}
|
|
n.Tags = newTags
|
|
break
|
|
}
|
|
}
|
|
if !n.HasTag(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) HasRelation(relType RelType, target string) bool {
|
|
for _, tgt := range n.Relations[string(relType)] {
|
|
if tgt == target {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
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 !n.HasRelation(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 {
|
|
for _, t := range n.Tags {
|
|
if strings.HasPrefix(t, "_") {
|
|
if p := strings.SplitN(t[1:], "::", 2); len(p) == 2 && p[0] == k {
|
|
return p[1]
|
|
}
|
|
}
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func (n *Node) GetDisplayTags() []string {
|
|
var tags []string
|
|
for _, t := range n.Tags {
|
|
if !strings.HasPrefix(t, "_") {
|
|
tags = append(tags, t)
|
|
}
|
|
}
|
|
return tags
|
|
}
|