77 lines
1.3 KiB
Go
77 lines
1.3 KiB
Go
|
package note
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"git.sr.ht/~ewintr/shitty-ssg/pkg/adoc"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Kind string
|
||
|
|
|
||
|
|
type Tag string
|
||
|
|
|
||
|
|
const (
|
||
|
|
KIND_NOTE = Kind("note")
|
||
|
|
KIND_PRIVATE_NOTE = Kind("private_note")
|
||
|
|
KIND_WORK_NOTE = Kind("work_note")
|
||
|
|
KIND_INVALID = Kind("")
|
||
|
|
)
|
||
|
|
|
||
|
|
func mapKind(akind adoc.Kind) Kind {
|
||
|
|
nkind, ok := map[adoc.Kind]Kind{
|
||
|
|
adoc.KIND_NOTE: KIND_NOTE,
|
||
|
|
adoc.KIND_PRIVATE_NOTE: KIND_PRIVATE_NOTE,
|
||
|
|
adoc.KIND_WORK_NOTE: KIND_WORK_NOTE,
|
||
|
|
}[akind]
|
||
|
|
if !ok {
|
||
|
|
return KIND_INVALID
|
||
|
|
}
|
||
|
|
|
||
|
|
return nkind
|
||
|
|
}
|
||
|
|
|
||
|
|
type Note struct {
|
||
|
|
doc *adoc.ADoc
|
||
|
|
Title string
|
||
|
|
Kind Kind
|
||
|
|
Tags []Tag
|
||
|
|
Content string
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewNote(doc *adoc.ADoc) *Note {
|
||
|
|
var paragraphs []string
|
||
|
|
for _, be := range doc.Content {
|
||
|
|
paragraphs = append(paragraphs, be.Text())
|
||
|
|
}
|
||
|
|
content := strings.Join(paragraphs, "\n\n")
|
||
|
|
|
||
|
|
var tags []Tag
|
||
|
|
for _, t := range doc.Tags {
|
||
|
|
tags = append(tags, Tag(t))
|
||
|
|
}
|
||
|
|
|
||
|
|
return &Note{
|
||
|
|
doc: doc,
|
||
|
|
Kind: mapKind(doc.Kind),
|
||
|
|
Title: doc.Title,
|
||
|
|
Tags: tags,
|
||
|
|
Content: content,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (n *Note) Contains(term string) bool {
|
||
|
|
for _, t := range n.Tags {
|
||
|
|
if strings.ToLower(string(t)) == strings.ToLower(term) {
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, w := range strings.Split(n.Content, " ") {
|
||
|
|
if strings.ToLower(w) == strings.ToLower(term) {
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return false
|
||
|
|
}
|