refactor: replace explicit fields with tag-based property system

This commit is contained in:
2026-03-27 02:11:46 +01:00
parent 2d4cff717b
commit b2225cff7b
17 changed files with 485 additions and 825 deletions

View File

@@ -12,46 +12,38 @@ import (
var aliasList bool var aliasList bool
var aliasCmd = &cobra.Command{ var aliasCmd = &cobra.Command{
Use: "alias [name] [command]", Use: "alias [name] [command]", Short: "Manage aliases", Args: cobra.MaximumNArgs(2),
Short: "Manage aliases",
Args: cobra.MaximumNArgs(2),
Run: func(cmd *cobra.Command, args []string) { Run: func(cmd *cobra.Command, args []string) {
d, err := db.GetDB() d, err := db.GetDB()
if err != nil { if err != nil {
fmt.Fprintln(os.Stderr, err) fmt.Fprintln(os.Stderr, err)
return return
} }
if aliasList { w := cmd.OutOrStdout()
aliases, err := d.ListAliases()
if err != nil { if aliasList || len(args) == 0 {
return if aliases, err := d.ListAliases(); err == nil {
output.PrintAliases(w, aliases, jsonFlag)
} }
output.PrintAliases(cmd.OutOrStdout(), aliases, jsonFlag)
return
}
if len(args) == 0 {
aliases, _ := d.ListAliases()
output.PrintAliases(cmd.OutOrStdout(), aliases, jsonFlag)
return return
} }
if len(args) == 1 { if len(args) == 1 {
a, err := d.GetAlias(args[0]) if a, err := d.GetAlias(args[0]); err != nil {
if err != nil {
fmt.Fprintln(os.Stderr, "alias not found:", args[0]) fmt.Fprintln(os.Stderr, "alias not found:", args[0])
return } else {
}
fmt.Println(a.Command) fmt.Println(a.Command)
}
return return
} }
if err := d.SetAlias(args[0], args[1]); err != nil { if err := d.SetAlias(args[0], args[1]); err != nil {
fmt.Fprintln(os.Stderr, "failed to set alias:", err) fmt.Fprintln(os.Stderr, "failed to set alias:", err)
return } else {
output.PrintAction(w, "Alias set", args[0], false)
} }
output.PrintSuccess(cmd.OutOrStdout(), "Alias '%s' set", args[0])
}, },
} }
func init() { func init() {
rootCmd.AddCommand(aliasCmd) rootCmd.AddCommand(aliasCmd)
aliasCmd.Flags().BoolVar(&aliasList, "list", false, "list all aliases") aliasCmd.Flags().BoolVar(&aliasList, "list", false, "")
} }

View File

@@ -6,63 +6,61 @@ import (
"axolotl/output" "axolotl/output"
"fmt" "fmt"
"os" "os"
"slices"
"strings"
"github.com/spf13/cobra" "github.com/spf13/cobra"
) )
var createType, createStatus, createPrio, createNamespace string = "issue", "open", "", "" var cDue, cContent, cDummy string
var createDue, createContent string var cTags, cRels []string
var createTags, createRels []string
var createCmd = &cobra.Command{ var createCmd = &cobra.Command{
Use: "create <title>", Use: "create <title>", Short: "Create a new node", Args: cobra.ExactArgs(1),
Short: "Create a new node",
Args: cobra.ExactArgs(1),
Run: func(cmd *cobra.Command, args []string) { Run: func(cmd *cobra.Command, args []string) {
d, err := db.GetDB() d, err := db.GetDB()
if err != nil { if err != nil {
fmt.Fprintln(os.Stderr, err) fmt.Fprintln(os.Stderr, err)
return return
} }
if createType == "issue" && createStatus == "" {
createStatus = "open" if !slices.ContainsFunc(cTags, func(e string) bool { return strings.HasPrefix(e, "_type::") }) {
cTags = append(cTags, "_type::issue")
} }
if slices.Contains(cTags, "_type::issue") && !slices.ContainsFunc(cTags, func(e string) bool { return strings.HasPrefix(e, "_status::") }) {
cTags = append(cTags, "_status::open")
}
if !slices.ContainsFunc(cTags, func(e string) bool { return strings.HasPrefix(e, "_namespace::") }) {
cTags = append(cTags, "_namespace::" + db.GetCurrentUser())
}
rels := make(map[models.RelType][]string) rels := make(map[models.RelType][]string)
for _, r := range createRels { for _, r := range cRels {
relType, target, err := db.ParseRelFlag(r) rt, tgt, err := db.ParseRelFlag(r)
if err != nil { if err != nil {
fmt.Fprintln(os.Stderr, err) fmt.Fprintln(os.Stderr, err)
return return
} }
rels[relType] = append(rels[relType], target) rels[rt] = append(rels[rt], tgt)
} }
n, err := d.CreateNode(db.CreateParams{
Title: args[0], if n, err := d.CreateNode(db.CreateParams{Title: args[0], Content: cContent, DueDate: cDue, Tags: cTags, Rels: rels}); err != nil {
Content: createContent,
DueDate: createDue,
Type: createType,
Status: createStatus,
Priority: createPrio,
Namespace: createNamespace,
Tags: createTags,
Rels: rels,
})
if err != nil {
fmt.Fprintln(os.Stderr, "failed to create:", err) fmt.Fprintln(os.Stderr, "failed to create:", err)
return } else {
}
output.PrintNode(cmd.OutOrStdout(), n, jsonFlag) output.PrintNode(cmd.OutOrStdout(), n, jsonFlag)
}
}, },
} }
func init() { func init() {
rootCmd.AddCommand(createCmd) rootCmd.AddCommand(createCmd)
createCmd.Flags().StringVar(&createType, "type", "issue", "node type (issue, note, user, namespace)") f := createCmd.Flags()
createCmd.Flags().StringVar(&createStatus, "status", "", "status (open, done)") f.StringVar(&cDummy, "type", "issue", "")
createCmd.Flags().StringVar(&createPrio, "prio", "", "priority (high, medium, low)") f.StringVar(&cDummy, "status", "", "")
createCmd.Flags().StringVar(&createNamespace, "namespace", "", "namespace") f.StringVar(&cDummy, "prio", "", "")
createCmd.Flags().StringVar(&createDue, "due", "", "due date") f.StringVar(&cDummy, "namespace", "", "")
createCmd.Flags().StringVar(&createContent, "content", "", "content") f.StringVar(&cDue, "due", "", "")
createCmd.Flags().StringArrayVar(&createTags, "tag", nil, "tags (repeatable)") f.StringVar(&cContent, "content", "", "")
createCmd.Flags().StringArrayVar(&createRels, "rel", nil, "relations (type:id, repeatable)") f.StringArrayVar(&cTags, "tag", nil, "")
f.StringArrayVar(&cRels, "rel", nil, "")
} }

View File

@@ -11,43 +11,39 @@ import (
"github.com/spf13/cobra" "github.com/spf13/cobra"
) )
var deleteForce bool var dForce bool
var deleteCmd = &cobra.Command{ var deleteCmd = &cobra.Command{
Use: "delete <id>", Use: "delete <id>", Short: "Delete a node", Args: cobra.ExactArgs(1),
Short: "Delete a node",
Args: cobra.ExactArgs(1),
Run: func(cmd *cobra.Command, args []string) { Run: func(cmd *cobra.Command, args []string) {
d, err := db.GetDB() d, err := db.GetDB()
if err != nil { if err != nil {
fmt.Fprintln(os.Stderr, err) fmt.Fprintln(os.Stderr, err)
return return
} }
id := args[0] n, err := d.NodeByID(args[0])
n, err := d.NodeByID(id)
if err != nil { if err != nil {
fmt.Fprintln(os.Stderr, " node not found:", id) fmt.Fprintln(os.Stderr, " node not found:", args[0])
return return
} }
if !deleteForce {
fmt.Printf(" Delete %s '%s'? [y/N]: ", n.GetType(), n.Title) if !dForce {
reader := bufio.NewReader(os.Stdin) fmt.Printf("Delete %s '%s'? [y/N]: ", n.GetProperty("type"), n.Title)
resp, _ := reader.ReadString('\n') r, _ := bufio.NewReader(os.Stdin).ReadString('\n')
resp = strings.TrimSpace(strings.ToLower(resp)) if r = strings.TrimSpace(strings.ToLower(r)); r != "y" && r != "yes" {
if resp != "y" && resp != "yes" {
fmt.Println("Cancelled.") fmt.Println("Cancelled.")
return return
} }
} }
if err := d.DeleteNode(id); err != nil {
if err := d.DeleteNode(args[0]); err != nil {
fmt.Fprintln(os.Stderr, "failed to delete: ", err) fmt.Fprintln(os.Stderr, "failed to delete: ", err)
return } else {
output.PrintAction(cmd.OutOrStdout(), "Deleted", args[0], true)
} }
output.PrintDeleted(cmd.OutOrStdout(), id)
}, },
} }
func init() { func init() {
rootCmd.AddCommand(deleteCmd) rootCmd.AddCommand(deleteCmd)
deleteCmd.Flags().BoolVarP(&deleteForce, "force", "f", false, "skip confirmation") deleteCmd.Flags().BoolVarP(&dForce, "force", "f", false, "")
} }

View File

@@ -11,21 +11,19 @@ import (
) )
var editCmd = &cobra.Command{ var editCmd = &cobra.Command{
Use: "edit <id>", Use: "edit <id>", Short: "Edit node content in $EDITOR", Args: cobra.ExactArgs(1),
Short: "Edit node content in $EDITOR",
Args: cobra.ExactArgs(1),
Run: func(cmd *cobra.Command, args []string) { Run: func(cmd *cobra.Command, args []string) {
d, err := db.GetDB() d, err := db.GetDB()
if err != nil { if err != nil {
fmt.Fprintln(os.Stderr, err) fmt.Fprintln(os.Stderr, err)
return return
} }
id := args[0] n, err := d.NodeByID(args[0])
n, err := d.NodeByID(id)
if err != nil { if err != nil {
fmt.Fprintln(os.Stderr, "node not found:", id) fmt.Fprintln(os.Stderr, "node not found:", args[0])
return return
} }
tmp, err := os.CreateTemp("", "ax-*.md") tmp, err := os.CreateTemp("", "ax-*.md")
if err != nil { if err != nil {
fmt.Fprintln(os.Stderr, "failed to create temp file:", err) fmt.Fprintln(os.Stderr, "failed to create temp file:", err)
@@ -40,24 +38,22 @@ var editCmd = &cobra.Command{
editor = "vi" editor = "vi"
} }
c := exec.Command(editor, tmp.Name()) c := exec.Command(editor, tmp.Name())
c.Stdin = os.Stdin c.Stdin, c.Stdout, c.Stderr = os.Stdin, os.Stdout, os.Stderr
c.Stdout = os.Stdout
c.Stderr = os.Stderr
if err := c.Run(); err != nil { if err := c.Run(); err != nil {
fmt.Fprintln(os.Stderr, "editor failed:", err) fmt.Fprintln(os.Stderr, "editor failed:", err)
return return
} }
content, err := os.ReadFile(tmp.Name())
if err != nil { if content, err := os.ReadFile(tmp.Name()); err == nil {
fmt.Fprintln(os.Stderr, "failed to read temp file:", err) if err := d.UpdateNode(args[0], db.UpdateParams{Content: string(content)}); err != nil {
return
}
if err := d.UpdateNode(id, db.UpdateParams{Content: string(content)}); err != nil {
fmt.Fprintln(os.Stderr, "failed to update:", err) fmt.Fprintln(os.Stderr, "failed to update:", err)
return return
} }
n, _ = d.NodeByID(id) n, _ = d.NodeByID(args[0])
output.PrintNode(cmd.OutOrStdout(), n, jsonFlag) output.PrintNode(cmd.OutOrStdout(), n, jsonFlag)
} else {
fmt.Fprintln(os.Stderr, "failed to read temp file:", err)
}
}, },
} }

View File

@@ -3,23 +3,23 @@ package cmd
import ( import (
"axolotl/db" "axolotl/db"
"axolotl/output" "axolotl/output"
"fmt"
"os"
"github.com/spf13/cobra" "github.com/spf13/cobra"
) )
var inboxCmd = &cobra.Command{ var inboxCmd = &cobra.Command{
Use: "inbox", Use: "inbox", Short: "Show your inbox",
Short: "Show your inbox",
Run: func(cmd *cobra.Command, args []string) { Run: func(cmd *cobra.Command, args []string) {
d, err := db.GetDB() d, err := db.GetDB()
user := db.GetCurrentUser()
nodes, err := d.ListNodes(db.ListFilter{
Inbox: user,
})
if err != nil { if err != nil {
fmt.Fprintln(os.Stderr, err)
return return
} }
if nodes, err := d.ListNodes(db.ListFilter{TagPrefixes: []string{"_inbox::" + db.GetCurrentUser()}}); err == nil {
output.PrintNodes(cmd.OutOrStdout(), nodes, jsonFlag) output.PrintNodes(cmd.OutOrStdout(), nodes, jsonFlag)
}
}, },
} }

View File

@@ -11,28 +11,23 @@ import (
) )
var initCmd = &cobra.Command{ var initCmd = &cobra.Command{
Use: "init [path]", Use: "init [path]", Short: "Initialize a new database", Args: cobra.MaximumNArgs(1),
Short: "Initialize a new database",
Args: cobra.MaximumNArgs(1),
Run: func(cmd *cobra.Command, args []string) { Run: func(cmd *cobra.Command, args []string) {
path := "." p := "."
if len(args) > 0 { if len(args) > 0 {
path = args[0] p = args[0]
} }
dbPath := filepath.Join(path, ".ax.db") dbPath := filepath.Join(p, ".ax.db")
if _, err := os.Stat(dbPath); err == nil { if _, err := os.Stat(dbPath); err == nil {
fmt.Fprintln(os.Stderr, "database already exists:", dbPath) fmt.Fprintln(os.Stderr, "database already exists:", dbPath)
os.Exit(1) os.Exit(1)
} }
err := db.Init(dbPath) if err := db.Init(dbPath); err != nil {
if err != nil {
fmt.Fprintln(os.Stderr, "failed to initialize:", err) fmt.Fprintln(os.Stderr, "failed to initialize:", err)
os.Exit(1) os.Exit(1)
} }
output.PrintCreated(cmd.OutOrStdout(), dbPath) output.PrintAction(cmd.OutOrStdout(), "Created", dbPath, false)
}, },
} }
func init() { func init() { rootCmd.AddCommand(initCmd) }
rootCmd.AddCommand(initCmd)
}

View File

@@ -9,40 +9,32 @@ import (
"github.com/spf13/cobra" "github.com/spf13/cobra"
) )
var listType, listStatus, listPrio, listNamespace, listTag, listInbox, listAssignee string var lDummy, lAssignee string
var lTags []string
var listCmd = &cobra.Command{ var listCmd = &cobra.Command{
Use: "list", Use: "list", Short: "List nodes",
Short: "List nodes",
Run: func(cmd *cobra.Command, args []string) { Run: func(cmd *cobra.Command, args []string) {
d, err := db.GetDB() d, err := db.GetDB()
if err != nil { if err != nil {
fmt.Fprintln(os.Stderr, err) fmt.Fprintln(os.Stderr, err)
return return
} }
nodes, err := d.ListNodes(db.ListFilter{ if nodes, err := d.ListNodes(db.ListFilter{TagPrefixes: lTags, Assignee: lAssignee}); err == nil {
Type: listType,
Status: listStatus,
Priority: listPrio,
Namespace: listNamespace,
Tag: listTag,
Inbox: listInbox,
Assignee: listAssignee,
})
if err != nil {
return
}
output.PrintNodes(cmd.OutOrStdout(), nodes, jsonFlag) output.PrintNodes(cmd.OutOrStdout(), nodes, jsonFlag)
} else {
fmt.Fprintln(os.Stderr, "err: %v", err)
}
}, },
} }
func init() { func init() {
rootCmd.AddCommand(listCmd) rootCmd.AddCommand(listCmd)
listCmd.Flags().StringVar(&listType, "type", "", "filter by type") f := listCmd.Flags()
listCmd.Flags().StringVar(&listStatus, "status", "", "filter by status") f.StringVar(&lDummy, "type", "", "")
listCmd.Flags().StringVar(&listPrio, "prio", "", "filter by priority") f.StringVar(&lDummy, "status", "", "")
listCmd.Flags().StringVar(&listNamespace, "namespace", "", "filter by namespace") f.StringVar(&lDummy, "prio", "", "")
listCmd.Flags().StringVar(&listTag, "tag", "", "filter by tag") f.StringVar(&lDummy, "namespace", "", "")
listCmd.Flags().StringVar(&listInbox, "inbox", "", "filter by inbox user") f.StringVar(&lAssignee, "assignee", "", "")
listCmd.Flags().StringVar(&listAssignee, "assignee", "", "filter by assignee") f.StringArrayVar(&lTags, "tag", nil, "")
} }

View File

@@ -2,23 +2,46 @@ package cmd
import ( import (
"os" "os"
"strings"
"github.com/spf13/cobra" "github.com/spf13/cobra"
) )
var jsonFlag bool var jsonFlag bool
var rootCmd = &cobra.Command{Use: "ax", Short: "The axolotl issue tracker"}
var rootCmd = &cobra.Command{
Use: "ax",
Short: "The axolotl issue tracker",
}
func Execute() { func Execute() {
rootCmd.SetArgs(transformArgs(os.Args[1:]))
if err := rootCmd.Execute(); err != nil { if err := rootCmd.Execute(); err != nil {
os.Exit(1) os.Exit(1)
} }
} }
func init() { func transformArgs(args []string) []string {
rootCmd.PersistentFlags().BoolVar(&jsonFlag, "json", false, "output as JSON") aliases := map[string]string{
"--status": "_status",
"--prio": "_prio",
"--type": "_type",
"--namespace": "_namespace",
}
result := []string{}
for i := 0; i < len(args); i++ {
if idx := strings.Index(args[i], "="); idx != -1 {
if prop, ok := aliases[args[i][:idx]]; ok {
result = append(result, "--tag", prop+"::"+args[i][idx+1:])
continue
}
}
if prop, ok := aliases[args[i]]; ok && i+1 < len(args) {
result, i = append(result, "--tag", prop+"::"+args[i+1]), i+1
continue
}
result = append(result, args[i])
}
return result
}
func init() {
rootCmd.PersistentFlags().BoolVar(&jsonFlag, "json", false, "")
} }

View File

@@ -10,21 +10,18 @@ import (
) )
var showCmd = &cobra.Command{ var showCmd = &cobra.Command{
Use: "show <id>", Use: "show <id>", Short: "Show node details", Args: cobra.ExactArgs(1),
Short: "Show node details",
Args: cobra.ExactArgs(1),
Run: func(cmd *cobra.Command, args []string) { Run: func(cmd *cobra.Command, args []string) {
d, err := db.GetDB() d, err := db.GetDB()
if err != nil { if err != nil {
fmt.Fprintln(os.Stderr, err) fmt.Fprintln(os.Stderr, err)
return return
} }
n, err := d.NodeByID(args[0]) if n, err := d.NodeByID(args[0]); err == nil {
if err != nil {
fmt.Fprintln(os.Stderr, "node not found:", args[0])
return
}
output.PrintNode(cmd.OutOrStdout(), n, jsonFlag) output.PrintNode(cmd.OutOrStdout(), n, jsonFlag)
} else {
fmt.Fprintln(os.Stderr, "node not found:", args[0])
}
}, },
} }

View File

@@ -6,45 +6,45 @@ import (
"axolotl/output" "axolotl/output"
"fmt" "fmt"
"os" "os"
"slices"
"strings"
"github.com/spf13/cobra" "github.com/spf13/cobra"
) )
var updateTitle, updateContent, updateStatus, updatePrio, updateDue string var (
var updateClearDue bool uTitle, uContent, uDue, dummy string
var updateAddTags, updateRemoveTags, updateAddRels, updateRemoveRels []string uClearDue bool
uAddTags, uRmTags, uAddRels, uRmRels []string
)
var updateCmd = &cobra.Command{ var updateCmd = &cobra.Command{
Use: "update <id>", Use: "update <id>", Short: "Update a node", Args: cobra.ExactArgs(1),
Short: "Update a node",
Args: cobra.ExactArgs(1),
Run: func(cmd *cobra.Command, args []string) { Run: func(cmd *cobra.Command, args []string) {
d, err := db.GetDB() d, err := db.GetDB()
if err != nil { if err != nil {
fmt.Fprintln(os.Stderr, err) fmt.Fprintln(os.Stderr, err)
return return
} }
id := args[0] addRels, rmRels := make(map[models.RelType][]string), make(map[models.RelType][]string)
addRels := make(map[models.RelType][]string)
removeRels := make(map[models.RelType][]string) parseRel := func(src []string, dst map[models.RelType][]string) bool {
for _, r := range updateAddRels { for _, r := range src {
relType, target, err := db.ParseRelFlag(r) rt, tgt, err := db.ParseRelFlag(r)
if err != nil { if err != nil {
fmt.Fprintln(os.Stderr, err) fmt.Fprintln(os.Stderr, err)
return false
}
dst[rt] = append(dst[rt], tgt)
}
return true
}
if !parseRel(uAddRels, addRels) || !parseRel(uRmRels, rmRels) {
return return
} }
addRels[relType] = append(addRels[relType], target)
} if slices.Contains(uAddTags, "_status::done") {
for _, r := range updateRemoveRels { ok, blockers, err := d.CanClose(args[0])
relType, target, err := db.ParseRelFlag(r)
if err != nil {
fmt.Fprintln(os.Stderr, err)
return
}
removeRels[relType] = append(removeRels[relType], target)
}
if updateStatus == "done" {
ok, blockers, err := d.CanClose(id)
if err != nil { if err != nil {
fmt.Fprintln(os.Stderr, "failed to check blockers:", err) fmt.Fprintln(os.Stderr, "failed to check blockers:", err)
return return
@@ -53,42 +53,39 @@ var updateCmd = &cobra.Command{
fmt.Fprintf(os.Stderr, "cannot close: blocked by %v\n", blockers) fmt.Fprintf(os.Stderr, "cannot close: blocked by %v\n", blockers)
return return
} }
uRmTags = append(uRmTags, "_status::open")
} else if slices.Contains(uAddTags, "_status::open") {
uRmTags = append(uRmTags, "_status::done")
} }
err = d.UpdateNode(id, db.UpdateParams{ if slices.ContainsFunc(uAddTags, func(e string) bool { return strings.HasPrefix(e, "_prio") }) {
Title: updateTitle, uRmTags = append(uRmTags, "_prio::low", "_prio::medium", "_prio::high")
Content: updateContent, }
DueDate: updateDue,
ClearDue: updateClearDue, uParams := db.UpdateParams{Title: uTitle, Content: uContent, DueDate: uDue, ClearDue: uClearDue,
Status: updateStatus, AddTags: uAddTags, RemoveTags: uRmTags, AddRels: addRels, RemoveRels: rmRels}
Priority: updatePrio, if err := d.UpdateNode(args[0], uParams); err != nil {
AddTags: updateAddTags,
RemoveTags: updateRemoveTags,
AddRels: addRels,
RemoveRels: removeRels,
})
if err != nil {
fmt.Fprintln(os.Stderr, "failed to update:", err) fmt.Fprintln(os.Stderr, "failed to update:", err)
return return
} }
n, err := d.NodeByID(id) if n, err := d.NodeByID(args[0]); err == nil {
if err != nil {
fmt.Fprintln(os.Stderr, "failed to fetch node:", err)
return
}
output.PrintNode(cmd.OutOrStdout(), n, jsonFlag) output.PrintNode(cmd.OutOrStdout(), n, jsonFlag)
} else {
fmt.Fprintln(os.Stderr, "failed to fetch node:", err)
}
}, },
} }
func init() { func init() {
rootCmd.AddCommand(updateCmd) rootCmd.AddCommand(updateCmd)
updateCmd.Flags().StringVar(&updateTitle, "title", "", "new title") f := updateCmd.Flags()
updateCmd.Flags().StringVar(&updateContent, "content", "", "new content") f.StringVar(&uTitle, "title", "", "")
updateCmd.Flags().StringVar(&updateStatus, "status", "", "status (open, done)") f.StringVar(&uContent, "content", "", "")
updateCmd.Flags().StringVar(&updatePrio, "prio", "", "priority (high, medium, low)") f.StringVar(&uDue, "due", "", "")
updateCmd.Flags().StringVar(&updateDue, "due", "", "due date") f.BoolVar(&uClearDue, "clear-due", false, "")
updateCmd.Flags().BoolVar(&updateClearDue, "clear-due", false, "clear due date") f.StringVar(&dummy, "status", "", "")
updateCmd.Flags().StringArrayVar(&updateAddTags, "tag", nil, "add tags (repeatable)") f.StringVar(&dummy, "prio", "", "")
updateCmd.Flags().StringArrayVar(&updateRemoveTags, "tag-remove", nil, "remove tags (repeatable)") f.StringArrayVar(&uAddTags, "tag", nil, "")
updateCmd.Flags().StringArrayVar(&updateAddRels, "rel", nil, "add relations (type:id, repeatable)") f.StringArrayVar(&uRmTags, "tag-remove", nil, "")
updateCmd.Flags().StringArrayVar(&updateRemoveRels, "rel-remove", nil, "remove relations (type:id, repeatable)") f.StringArrayVar(&uAddRels, "rel", nil, "")
f.StringArrayVar(&uRmRels, "rel-remove", nil, "")
} }

View File

@@ -2,35 +2,25 @@ package db
import "errors" import "errors"
type Alias struct { type Alias struct{ Name, Command string }
Name string
Command string
}
func (db *DB) GetAlias(name string) (*Alias, error) { func (db *DB) GetAlias(n string) (*Alias, error) {
a := &Alias{} a := &Alias{}
err := db.QueryRow("SELECT name, command FROM aliases WHERE name = ?", name).Scan(&a.Name, &a.Command) err := db.QueryRow("SELECT name, command FROM aliases WHERE name = ?", n).Scan(&a.Name, &a.Command)
if err != nil { return a, err
return nil, err
}
return a, nil
} }
func (db *DB) SetAlias(name, command string) error { func (db *DB) SetAlias(n, c string) error {
_, err := db.Exec( _, err := db.Exec("INSERT INTO aliases (name, command) VALUES (?, ?) ON CONFLICT(name) DO UPDATE SET command = excluded.command", n, c)
"INSERT INTO aliases (name, command) VALUES (?, ?) ON CONFLICT(name) DO UPDATE SET command = excluded.command",
name, command,
)
return err return err
} }
func (db *DB) DeleteAlias(name string) error { func (db *DB) DeleteAlias(n string) error {
res, err := db.Exec("DELETE FROM aliases WHERE name = ?", name) res, err := db.Exec("DELETE FROM aliases WHERE name = ?", n)
if err != nil { if err != nil {
return err return err
} }
n, _ := res.RowsAffected() if a, _ := res.RowsAffected(); a == 0 {
if n == 0 {
return errors.New("alias not found") return errors.New("alias not found")
} }
return nil return nil
@@ -42,7 +32,6 @@ func (db *DB) ListAliases() ([]*Alias, error) {
return nil, err return nil, err
} }
defer rows.Close() defer rows.Close()
var aliases []*Alias var aliases []*Alias
for rows.Next() { for rows.Next() {
a := &Alias{} a := &Alias{}

111
db/db.go
View File

@@ -8,8 +8,6 @@ import (
"os/user" "os/user"
"path/filepath" "path/filepath"
"axolotl/models"
_ "modernc.org/sqlite" _ "modernc.org/sqlite"
) )
@@ -18,38 +16,16 @@ type DB struct {
path string path string
} }
var database *DB var (
var migrations = []string{ database *DB
`CREATE TABLE IF NOT EXISTS nodes ( migrations = []string{
id TEXT PRIMARY KEY, `CREATE TABLE IF NOT EXISTS nodes (id TEXT PRIMARY KEY, title TEXT NOT NULL, content TEXT, due_date TEXT, created_at TEXT NOT NULL DEFAULT CURRENT_TIMESTAMP, updated_at TEXT NOT NULL DEFAULT CURRENT_TIMESTAMP)`,
title TEXT NOT NULL, `CREATE TABLE IF NOT EXISTS tags (node_id TEXT NOT NULL, tag TEXT NOT NULL, PRIMARY KEY (node_id, tag), FOREIGN KEY (node_id) REFERENCES nodes(id) ON DELETE CASCADE)`,
content TEXT, `CREATE TABLE IF NOT EXISTS rels (from_id TEXT NOT NULL, to_id TEXT NOT NULL, rel_type TEXT NOT NULL, PRIMARY KEY (from_id, to_id, rel_type), FOREIGN KEY (from_id) REFERENCES nodes(id) ON DELETE CASCADE, FOREIGN KEY (to_id) REFERENCES nodes(id) ON DELETE CASCADE)`,
due_date TEXT, `CREATE TABLE IF NOT EXISTS aliases (name TEXT PRIMARY KEY, command TEXT NOT NULL)`,
created_at TEXT NOT NULL DEFAULT CURRENT_TIMESTAMP, `CREATE INDEX IF NOT EXISTS idx_tags_tag ON tags(tag)`, `CREATE INDEX IF NOT EXISTS idx_rels_from ON rels(from_id)`, `CREATE INDEX IF NOT EXISTS idx_rels_to ON rels(to_id)`,
updated_at TEXT NOT NULL DEFAULT CURRENT_TIMESTAMP
)`,
`CREATE TABLE IF NOT EXISTS tags (
node_id TEXT NOT NULL,
tag TEXT NOT NULL,
PRIMARY KEY (node_id, tag),
FOREIGN KEY (node_id) REFERENCES nodes(id) ON DELETE CASCADE
)`,
`CREATE TABLE IF NOT EXISTS rels (
from_id TEXT NOT NULL,
to_id TEXT NOT NULL,
rel_type TEXT NOT NULL,
PRIMARY KEY (from_id, to_id, rel_type),
FOREIGN KEY (from_id) REFERENCES nodes(id) ON DELETE CASCADE,
FOREIGN KEY (to_id) REFERENCES nodes(id) ON DELETE CASCADE
)`,
`CREATE TABLE IF NOT EXISTS aliases (
name TEXT PRIMARY KEY,
command TEXT NOT NULL
)`,
`CREATE INDEX IF NOT EXISTS idx_tags_tag ON tags(tag)`,
`CREATE INDEX IF NOT EXISTS idx_rels_from ON rels(from_id)`,
`CREATE INDEX IF NOT EXISTS idx_rels_to ON rels(to_id)`,
} }
)
func GetDB() (*DB, error) { func GetDB() (*DB, error) {
if database != nil { if database != nil {
@@ -60,25 +36,22 @@ func GetDB() (*DB, error) {
return nil, err return nil, err
} }
for { for {
path := filepath.Join(dir, ".ax.db") if _, err := os.Stat(filepath.Join(dir, ".ax.db")); err == nil {
if _, err := os.Stat(path); err == nil { if database, err = Open(filepath.Join(dir, ".ax.db")); err != nil {
database, err = Open(path)
if err != nil {
return nil, fmt.Errorf("failed to open database: %w", err) return nil, fmt.Errorf("failed to open database: %w", err)
} }
return database, nil return database, nil
} }
parent := filepath.Dir(dir) if parent := filepath.Dir(dir); parent == dir {
if parent == dir {
return nil, errors.New("no .ax.db found (run 'ax init' first)") return nil, errors.New("no .ax.db found (run 'ax init' first)")
} else {
dir = parent
} }
path = parent
} }
} }
func Init(path string) (error) { func Init(path string) error {
dir := filepath.Dir(path) if err := os.MkdirAll(filepath.Dir(path), 0755); err != nil {
if err := os.MkdirAll(dir, 0755); err != nil {
return err return err
} }
var err error var err error
@@ -87,19 +60,16 @@ func Init(path string) (error) {
} }
func Open(path string) (*DB, error) { func Open(path string) (*DB, error) {
database, err := sql.Open("sqlite", path) db, err := sql.Open("sqlite", path)
if err != nil { if err != nil {
return nil, err return nil, err
} }
database.Exec("PRAGMA journal_mode=WAL") for _, q := range append([]string{"PRAGMA journal_mode=WAL", "PRAGMA busy_timeout=5000", "PRAGMA foreign_keys=ON"}, migrations...) {
database.Exec("PRAGMA busy_timeout=5000") if _, err := db.Exec(q); err != nil {
database.Exec("PRAGMA foreign_keys=ON")
for _, m := range migrations {
if _, err := database.Exec(m); err != nil {
return nil, err return nil, err
} }
} }
return &DB{DB: database, path: path}, nil return &DB{DB: db, path: path}, nil
} }
func GetCurrentUser() string { func GetCurrentUser() string {
@@ -111,44 +81,3 @@ func GetCurrentUser() string {
} }
return "unknown" return "unknown"
} }
func (db *DB) NodeExists(id string) (bool, error) {
var exists bool
err := db.QueryRow("SELECT EXISTS(SELECT 1 FROM nodes WHERE id = ?)", id).Scan(&exists)
return exists, err
}
func (db *DB) NodeByID(id string) (*models.Node, error) {
n := &models.Node{Relations: make(map[string][]string)}
err := db.QueryRow(
"SELECT id, title, COALESCE(content, ''), COALESCE(due_date, ''), created_at, updated_at FROM nodes WHERE id = ?",
id,
).Scan(&n.ID, &n.Title, &n.Content, &n.DueDate, &n.CreatedAt, &n.UpdatedAt)
if err != nil {
return nil, err
}
rows, err := db.Query("SELECT tag FROM tags WHERE node_id = ?", id)
if err != nil {
return nil, err
}
defer rows.Close()
for rows.Next() {
var tag string
rows.Scan(&tag)
n.Tags = append(n.Tags, tag)
}
rows, err = db.Query("SELECT to_id, rel_type FROM rels WHERE from_id = ?", id)
if err != nil {
return nil, err
}
defer rows.Close()
for rows.Next() {
var toID, relType string
rows.Scan(&toID, &relType)
n.Relations[relType] = append(n.Relations[relType], toID)
}
return n, nil
}

View File

@@ -10,38 +10,46 @@ import (
"time" "time"
) )
func generateID() string { func genID() string {
const chars = "abcdefghijklmnopqrstuvwxyz"
b := make([]byte, 5) b := make([]byte, 5)
for i := range b { for i := range b {
b[i] = chars[rand.Intn(26)] b[i] = "abcdefghijklmnopqrstuvwxyz"[rand.Intn(26)]
} }
return string(b) return string(b)
} }
func ParseRelFlag(s string) (models.RelType, string, error) {
if p := strings.SplitN(s, ":", 2); len(p) == 2 {
return models.RelType(p[0]), p[1], nil
}
return "", "", fmt.Errorf("invalid relation format: %s (expected type:id)", s)
}
func (db *DB) generateUniqueID() string { func (db *DB) generateUniqueID() string {
//TODO: Check if all ids are reserved
for { for {
id := generateID() id := genID()
var exists bool e, _ := db.NodeExists(id)
db.QueryRow("SELECT EXISTS(SELECT 1 FROM nodes WHERE id = ?)", id).Scan(&exists) if !e {
if !exists {
return id return id
} }
} }
} }
type CreateParams struct { type CreateParams struct {
Title string Title, Content, DueDate string
Content string
DueDate string
Type string
Status string
Priority string
Namespace string
Tags []string Tags []string
Rels map[models.RelType][]string Rels map[models.RelType][]string
} }
type UpdateParams struct {
Title, Content, DueDate string
ClearDue bool
AddTags, RemoveTags []string
AddRels, RemoveRels map[models.RelType][]string
}
type ListFilter struct {
TagPrefixes []string
Assignee string
}
func (db *DB) CreateNode(p CreateParams) (*models.Node, error) { func (db *DB) CreateNode(p CreateParams) (*models.Node, error) {
tx, err := db.Begin() tx, err := db.Begin()
@@ -49,78 +57,35 @@ func (db *DB) CreateNode(p CreateParams) (*models.Node, error) {
return nil, err return nil, err
} }
defer tx.Rollback() defer tx.Rollback()
now, id := time.Now().UTC().Format(time.RFC3339), db.generateUniqueID()
now := time.Now().UTC().Format(time.RFC3339) _, err = tx.Exec("INSERT INTO nodes (id, title, content, due_date, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?)",
id := db.generateUniqueID() id, p.Title, p.Content, p.DueDate, now, now)
_, err = tx.Exec(
"INSERT INTO nodes (id, title, content, due_date, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?)",
id, p.Title, p.Content, p.DueDate, now, now,
)
if err != nil { if err != nil {
return nil, err return nil, err
} }
for _, t := range append(p.Tags, parse.Mentions(p.Title+" "+p.Content)...) {
tags := p.Tags if !strings.HasPrefix(t, "_") && strings.HasPrefix(t, "@") {
if p.Type != "" { if _, err = tx.Exec("INSERT INTO tags (node_id, tag) VALUES (?, ?)", id, "_inbox::"+t[1:]); err != nil {
tags = append(tags, models.PropertyTag("type", p.Type)) return nil, err
} else {
tags = append(tags, models.PropertyTag("type", "issue"))
} }
if p.Status != "" { } else if _, err := tx.Exec("INSERT INTO tags (node_id, tag) VALUES (?, ?)", id, t); err != nil {
tags = append(tags, models.PropertyTag("status", p.Status))
}
if p.Priority != "" {
tags = append(tags, models.PropertyTag("prio", p.Priority))
}
ns := p.Namespace
if ns == "" {
ns = GetCurrentUser()
}
tags = append(tags, models.PropertyTag("namespace", ns))
for _, tag := range tags {
if _, err := tx.Exec("INSERT INTO tags (node_id, tag) VALUES (?, ?)", id, tag); err != nil {
return nil, err return nil, err
} }
} }
for rt, tgts := range p.Rels {
for relType, targets := range p.Rels { for _, tgt := range tgts {
for _, target := range targets { if _, err := tx.Exec("INSERT INTO rels (from_id, to_id, rel_type) VALUES (?, ?, ?)", id, tgt, rt); err != nil {
if _, err := tx.Exec(
"INSERT INTO rels (from_id, to_id, rel_type) VALUES (?, ?, ?)",
id, target, relType,
); err != nil {
return nil, err return nil, err
} }
} }
} }
allText := p.Title + " " + p.Content
for _, u := range parse.Mentions(allText) {
tx.Exec("INSERT INTO tags (node_id, tag) VALUES (?, ?)", id, models.PropertyTag("inbox", u))
}
if err := tx.Commit(); err != nil { if err := tx.Commit(); err != nil {
return nil, err return nil, err
} }
return db.NodeByID(id) return db.NodeByID(id)
} }
type UpdateParams struct {
Title string
Content string
DueDate string
ClearDue bool
Status string
Priority string
AddTags []string
RemoveTags []string
AddRels map[models.RelType][]string
RemoveRels map[models.RelType][]string
}
func (db *DB) UpdateNode(id string, p UpdateParams) error { func (db *DB) UpdateNode(id string, p UpdateParams) error {
tx, err := db.Begin() tx, err := db.Begin()
if err != nil { if err != nil {
@@ -128,231 +93,197 @@ func (db *DB) UpdateNode(id string, p UpdateParams) error {
} }
defer tx.Rollback() defer tx.Rollback()
now := time.Now().UTC().Format(time.RFC3339) upd := func(col, val string) error {
_, err = tx.Exec("UPDATE nodes SET updated_at = ? WHERE id = ?", now, id) _, err := tx.Exec("UPDATE nodes SET "+col+" = ? WHERE id = ?", val, id)
if err != nil {
return err return err
} }
//TODO: does it make sense to check for emtpy sting? should it be possible to set the content to ""?
if p.Title != "" { if p.Title != "" {
_, err = tx.Exec("UPDATE nodes SET title = ? WHERE id = ?", p.Title, id) if err := upd("title", p.Title); err != nil {
if err != nil {
return err return err
} }
} }
if p.Content != "" { if p.Content != "" {
_, err = tx.Exec("UPDATE nodes SET content = ? WHERE id = ?", p.Content, id) if err := upd("content", p.Content); err != nil {
if err != nil {
return err return err
} }
} }
if p.DueDate != "" { if p.DueDate != "" {
_, err = tx.Exec("UPDATE nodes SET due_date = ? WHERE id = ?", p.DueDate, id) if err := upd("due_date", p.DueDate); err != nil {
if err != nil {
return err return err
} }
} }
if p.ClearDue { if p.ClearDue {
_, err = tx.Exec("UPDATE nodes SET due_date = NULL WHERE id = ?", id) if _, err := tx.Exec("UPDATE nodes SET due_date = NULL WHERE id = ?", id); err != nil {
if err != nil {
return err return err
} }
} }
for _, tag := range p.AddTags { for _, t := range p.RemoveTags {
tx.Exec("INSERT OR IGNORE INTO tags (node_id, tag) VALUES (?, ?)", id, tag) tx.Exec("DELETE FROM tags WHERE node_id = ? AND tag = ?", id, t)
} }
for _, tag := range p.RemoveTags { for _, t := range p.AddTags {
tx.Exec("DELETE FROM tags WHERE node_id = ? AND tag = ?", id, tag) tx.Exec("INSERT OR IGNORE INTO tags (node_id, tag) VALUES (?, ?)", id, t)
} }
for rt, tgts := range p.RemoveRels {
if p.Status != "" { for _, tgt := range tgts {
tx.Exec("DELETE FROM tags WHERE node_id = ? AND tag LIKE '_status::%'", id) tx.Exec("DELETE FROM rels WHERE from_id = ? AND to_id = ? AND rel_type = ?", id, tgt, rt)
tx.Exec("INSERT INTO tags (node_id, tag) VALUES (?, ?)", id, models.PropertyTag("status", p.Status))
}
if p.Priority != "" {
tx.Exec("DELETE FROM tags WHERE node_id = ? AND tag LIKE '_prio::%'", id)
tx.Exec("INSERT INTO tags (node_id, tag) VALUES (?, ?)", id, models.PropertyTag("prio", p.Priority))
}
for relType, targets := range p.AddRels {
for _, target := range targets {
tx.Exec("INSERT OR IGNORE INTO rels (from_id, to_id, rel_type) VALUES (?, ?, ?)", id, target, relType)
} }
} }
for relType, targets := range p.RemoveRels { for rt, tgts := range p.AddRels {
for _, target := range targets { for _, tgt := range tgts {
tx.Exec("DELETE FROM rels WHERE from_id = ? AND to_id = ? AND rel_type = ?", id, target, relType) tx.Exec("INSERT OR IGNORE INTO rels (from_id, to_id, rel_type) VALUES (?, ?, ?)", id, tgt, rt)
} }
} }
if _, err = tx.Exec("UPDATE nodes SET updated_at = ? WHERE id = ?", time.Now().UTC().Format(time.RFC3339), id); err != nil {
return err
}
return tx.Commit() return tx.Commit()
} }
func (db *DB) DeleteNode(id string) error { func (db *DB) DeleteNode(id string) error {
//TODO: check if this delete propagates?
_, err := db.Exec("DELETE FROM nodes WHERE id = ?", id) _, err := db.Exec("DELETE FROM nodes WHERE id = ?", id)
return err return err
} }
type ListFilter struct { func (db *DB) NodeExists(id string) (bool, error) {
Type string var e bool
Status string err := db.QueryRow("SELECT EXISTS(SELECT 1 FROM nodes WHERE id = ?)", id).Scan(&e)
Priority string return e, err
Namespace string }
Tag string
Inbox string func (db *DB) NodeByID(id string) (*models.Node, error) {
Assignee string n := &models.Node{Relations: make(map[string][]string)}
q := db.QueryRow("SELECT id, title, COALESCE(content, ''), COALESCE(due_date, ''), created_at, updated_at FROM nodes WHERE id = ?", id)
if err := q.Scan(&n.ID, &n.Title, &n.Content, &n.DueDate, &n.CreatedAt, &n.UpdatedAt); err != nil {
return nil, err
}
if rows, err := db.Query("SELECT tag FROM tags WHERE node_id = ?", id); err == nil {
defer rows.Close()
for rows.Next() {
var tag string
rows.Scan(&tag)
n.Tags = append(n.Tags, tag)
}
} else {
return nil, err
}
if rows, err := db.Query("SELECT to_id, rel_type FROM rels WHERE from_id = ?", id); err == nil {
defer rows.Close()
for rows.Next() {
var toID, relType string
rows.Scan(&toID, &relType)
n.Relations[relType] = append(n.Relations[relType], toID)
}
} else {
return nil, err
}
return n, nil
} }
func (db *DB) ListNodes(f ListFilter) ([]*models.Node, error) { func (db *DB) ListNodes(f ListFilter) ([]*models.Node, error) {
q := "SELECT DISTINCT n.id FROM nodes n" q, args, joins, conds := "SELECT DISTINCT n.id FROM nodes n", []any{}, []string{}, []string{}
args := []interface{}{} if len(f.TagPrefixes) == 0 {
joins := []string{} f.TagPrefixes = append(f.TagPrefixes, "")
conds := []string{} }
if f.Type != "" {
joins = append(joins, "JOIN tags t_type ON n.id = t_type.node_id")
conds = append(conds, "t_type.tag = ?")
args = append(args, models.PropertyTag("type", f.Type))
}
if f.Status != "" {
joins = append(joins, "JOIN tags t_status ON n.id = t_status.node_id")
conds = append(conds, "t_status.tag = ?")
args = append(args, models.PropertyTag("status", f.Status))
}
if f.Priority != "" {
joins = append(joins, "JOIN tags t_prio ON n.id = t_prio.node_id")
conds = append(conds, "t_prio.tag = ?")
args = append(args, models.PropertyTag("prio", f.Priority))
}
if f.Namespace != "" {
joins = append(joins, "JOIN tags t_ns ON n.id = t_ns.node_id")
conds = append(conds, "t_ns.tag = ?")
args = append(args, models.PropertyTag("namespace", f.Namespace))
}
if f.Tag != "" {
joins = append(joins, "JOIN tags t_tag ON n.id = t_tag.node_id") joins = append(joins, "JOIN tags t_tag ON n.id = t_tag.node_id")
conds = append(conds, "t_tag.tag = ?") cond := ""
args = append(args, f.Tag) for _, t := range f.TagPrefixes {
} cond += "t_tag.tag LIKE ? || '%' OR "
if f.Inbox != "" { args = append(args, t)
joins = append(joins, "JOIN tags t_inbox ON n.id = t_inbox.node_id")
conds = append(conds, "t_inbox.tag = ?")
args = append(args, models.PropertyTag("inbox", f.Inbox))
} }
conds = append(conds, "SUM(CASE WHEN "+cond[:len(cond)-4]+" THEN 1 ELSE 0 END) >= ?")
args = append(args, len(f.TagPrefixes))
if f.Assignee != "" { if f.Assignee != "" {
joins = append(joins, "JOIN rels r_assign ON n.id = r_assign.from_id") joins, conds, args = append(joins, "JOIN rels r_assign ON n.id = r_assign.from_id"), append(conds, "r_assign.to_id = ? AND r_assign.rel_type = ?"), append(args, f.Assignee, models.RelAssignee)
conds = append(conds, "r_assign.to_id = ? AND r_assign.rel_type = ?")
args = append(args, f.Assignee, models.RelAssignee)
} }
if len(joins) > 0 { if len(joins) > 0 {
q += " " + strings.Join(joins, " ") q += " " + strings.Join(joins, " ") + " "
} }
q += "GROUP BY n.id"
if len(conds) > 0 { if len(conds) > 0 {
q += " WHERE " + strings.Join(conds, " AND ") q += " HAVING " + strings.Join(conds, " AND ")
} }
q += " ORDER BY n.created_at DESC"
rows, err := db.Query(q, args...) rows, err := db.Query(q+" ORDER BY n.created_at DESC", args...)
if err != nil { if err != nil {
return nil, err return nil, err
} }
defer rows.Close() defer rows.Close()
var nodes []*models.Node var nodes []*models.Node
for rows.Next() { for rows.Next() {
var id string var id string
if err := rows.Scan(&id); err != nil { if err := rows.Scan(&id); err != nil {
return nil, err return nil, err
} }
n, err := db.NodeByID(id) if n, err := db.NodeByID(id); err == nil {
if err != nil { nodes = append(nodes, n)
} else {
return nil, err return nil, err
} }
nodes = append(nodes, n)
} }
return nodes, nil return nodes, nil
} }
func (db *DB) CanClose(id string) (bool, []string, error) { func (db *DB) CanClose(id string) (bool, []string, error) {
rows, err := db.Query( rows, err := db.Query("SELECT to_id FROM rels WHERE from_id = ? AND rel_type = ?", id, models.RelBlocks)
"SELECT to_id FROM rels WHERE from_id = ? AND rel_type = ?",
id, models.RelBlocks,
)
if err != nil { if err != nil {
return false, nil, err return false, nil, err
} }
defer rows.Close() defer rows.Close()
var blocking []string var blocking []string
for rows.Next() { for rows.Next() {
var blockerID string var bID, tag string
if err := rows.Scan(&blockerID); err != nil { if err := rows.Scan(&bID); err != nil {
return false, nil, err return false, nil, err
} }
var tag string if err := db.QueryRow("SELECT tag FROM tags WHERE node_id = ? AND tag LIKE '_status::%'", bID).Scan(&tag); err == sql.ErrNoRows {
err := db.QueryRow(
"SELECT tag FROM tags WHERE node_id = ? AND tag LIKE '_status::%'",
blockerID,
).Scan(&tag)
if err == sql.ErrNoRows {
blocking = append(blocking, blockerID)
continue continue
} } else if err != nil {
if err != nil {
return false, nil, err return false, nil, err
} }
if strings.HasSuffix(tag, "::open") { if strings.HasSuffix(tag, "::open") {
blocking = append(blocking, blockerID) blocking = append(blocking, bID)
} }
} }
return len(blocking) == 0, blocking, nil return len(blocking) == 0, blocking, nil
} }
func (db *DB) GetAllUsers() ([]string, error) { func (db *DB) GetAllUsers() ([]string, error) {
rows, err := db.Query("SELECT n.id FROM nodes n JOIN tags t ON n.id = t.node_id WHERE t.tag = ?", models.PropertyTag("type", "user")) //TODO: Seperate sql query for performance
nodes, err := db.ListNodes(ListFilter{TagPrefixes: []string{"_type::user"}})
if err != nil { if err != nil {
return nil, err return nil, err
} }
defer rows.Close() users := make([]string, len(nodes))
for i, n := range nodes {
var users []string users[i] = n.ID
for rows.Next() {
var id string
rows.Scan(&id)
users = append(users, id)
} }
return users, nil return users, nil
} }
func (db *DB) GetSubtasks(parentID string) ([]*models.Node, error) { func (db *DB) GetSubtasks(parentID string) ([]*models.Node, error) {
rows, err := db.Query( rows, err := db.Query("SELECT to_id FROM rels WHERE from_id = ? AND rel_type = ?", parentID, models.RelSubtask)
"SELECT to_id FROM rels WHERE from_id = ? AND rel_type = ?",
parentID, models.RelSubtask,
)
if err != nil { if err != nil {
return nil, err return nil, err
} }
defer rows.Close() defer rows.Close()
var nodes []*models.Node var nodes []*models.Node
for rows.Next() { for rows.Next() {
var id string var id string
rows.Scan(&id) rows.Scan(&id)
n, err := db.NodeByID(id) if n, err := db.NodeByID(id); err == nil {
if err != nil { nodes = append(nodes, n)
} else {
return nil, err return nil, err
} }
nodes = append(nodes, n)
} }
return nodes, nil return nodes, nil
} }
func ParseRelFlag(s string) (models.RelType, string, error) {
parts := strings.SplitN(s, ":", 2)
if len(parts) != 2 {
return "", "", fmt.Errorf("invalid relation format: %s (expected type:id)", s)
}
return models.RelType(parts[0]), parts[1], nil
}

View File

@@ -2,56 +2,33 @@ package db
import "axolotl/models" import "axolotl/models"
func (db *DB) AddRel(fromID, toID string, relType models.RelType) error { func (db *DB) AddRel(f, t string, r models.RelType) error {
_, err := db.Exec( _, err := db.Exec("INSERT OR IGNORE INTO rels (from_id, to_id, rel_type) VALUES (?, ?, ?)", f, t, r)
"INSERT OR IGNORE INTO rels (from_id, to_id, rel_type) VALUES (?, ?, ?)", return err
fromID, toID, relType, }
) func (db *DB) RemoveRel(f, t string, r models.RelType) error {
_, err := db.Exec("DELETE FROM rels WHERE from_id = ? AND to_id = ? AND rel_type = ?", f, t, r)
return err return err
} }
func (db *DB) RemoveRel(fromID, toID string, relType models.RelType) error { func getIDs(db *DB, q, id string, r models.RelType) ([]string, error) {
_, err := db.Exec( rows, err := db.Query(q, id, r)
"DELETE FROM rels WHERE from_id = ? AND to_id = ? AND rel_type = ?",
fromID, toID, relType,
)
return err
}
func (db *DB) GetRelated(id string, relType models.RelType) ([]string, error) {
rows, err := db.Query(
"SELECT to_id FROM rels WHERE from_id = ? AND rel_type = ?",
id, relType,
)
if err != nil { if err != nil {
return nil, err return nil, err
} }
defer rows.Close() defer rows.Close()
var ids []string var ids []string
for rows.Next() { for rows.Next() {
var target string var i string
rows.Scan(&target) rows.Scan(&i)
ids = append(ids, target) ids = append(ids, i)
} }
return ids, nil return ids, nil
} }
func (db *DB) GetIncomingRels(id string, relType models.RelType) ([]string, error) { func (db *DB) GetRelated(id string, r models.RelType) ([]string, error) {
rows, err := db.Query( return getIDs(db, "SELECT to_id FROM rels WHERE from_id = ? AND rel_type = ?", id, r)
"SELECT from_id FROM rels WHERE to_id = ? AND rel_type = ?",
id, relType,
)
if err != nil {
return nil, err
} }
defer rows.Close() func (db *DB) GetIncomingRels(id string, r models.RelType) ([]string, error) {
return getIDs(db, "SELECT from_id FROM rels WHERE to_id = ? AND rel_type = ?", id, r)
var ids []string
for rows.Next() {
var source string
rows.Scan(&source)
ids = append(ids, source)
}
return ids, nil
} }

View File

@@ -23,40 +23,13 @@ const (
RelAssignee RelType = "assignee" RelAssignee RelType = "assignee"
) )
func ParseTag(s string) (key, value string, isProperty bool) { func (n *Node) GetProperty(k string) string {
if strings.HasPrefix(s, "_") { for _, t := range n.Tags {
if parts := strings.SplitN(s[1:], "::", 2); len(parts) == 2 { if strings.HasPrefix(t, "_") {
return parts[0], parts[1], true if p := strings.SplitN(t[1:], "::", 2); len(p) == 2 && p[0] == k {
return p[1]
} }
} }
return "", s, false
}
func PropertyTag(key, value string) string {
return "_" + key + "::" + value
}
func (n *Node) GetProperty(key string) string {
for _, tag := range n.Tags {
if k, v, ok := ParseTag(tag); ok && k == key {
return v
}
} }
return "" return ""
} }
func (n *Node) GetType() string {
if t := n.GetProperty("type"); t != "" {
return t
}
return "issue"
}
func (n *Node) HasTag(tag string) bool {
for _, t := range n.Tags {
if t == tag {
return true
}
}
return false
}

View File

@@ -12,283 +12,161 @@ import (
"github.com/fatih/color" "github.com/fatih/color"
) )
type RenderMap map[string]struct {
s string
l string
c *color.Color
}
var ( var (
idColor = color.New(color.FgYellow) cPrimary = color.New(color.FgCyan)
titleColor = color.New(color.FgWhite, color.Bold) cSecond = color.New(color.FgMagenta)
statusOpen = color.New(color.FgYellow) cDim = color.New(color.FgHiBlack)
statusDone = color.New(color.FgHiBlack) cText = color.New(color.FgWhite)
prioHigh = color.New(color.FgHiRed) cTitle = color.New(color.FgWhite, color.Bold)
prioMedium = color.New(color.FgYellow) cGood = color.New(color.FgGreen)
prioLow = color.New(color.FgYellow, color.Faint) cWarn = color.New(color.FgYellow)
tagColor = color.New(color.FgCyan) cBad = color.New(color.FgRed)
nsColor = color.New(color.FgYellow)
dimColor = color.New(color.FgHiBlack) typeRM = RenderMap{
contentColor = color.New(color.FgWhite) "issue": {" ", "\uf188 issue", cSecond},
labelColor = color.New(color.FgYellow) "note": {"\uf15c", "\uf15c note", cPrimary},
typeIssue = color.New(color.FgMagenta) "user": {"\uf007", "\uf007 user", cGood},
typeNote = color.New(color.FgHiBlue) "namespace": {"\uf07b", "\uf07b namespace", cWarn},
typeUser = color.New(color.FgHiGreen) "": {" ", "n/a", cDim},
typeNs = color.New(color.FgHiYellow) }
statusRM = RenderMap{
"open": {"●", "● open", cPrimary},
"done": {"○", "○ done", cDim},
"": {"—", "n/a", cDim},
}
prioRM = RenderMap{
"high": {"\uf0e7", "high", cBad},
"medium": {"\uf0e7", "medium", cWarn},
"low": {" ", "low", cDim},
"": {" ", "n/a", cDim},
}
relIcons = map[string]string{"blocks": "\uf068", "subtask": "\uf0da", "related": "\uf0c1", "assignee": "\uf007"}
prioRanks = map[string]int{"high": 3, "medium": 2, "low": 1}
statusRanks = map[string]int{"open": 2, "": 1, "done": 0}
) )
type iconSet struct { const (
Issue, Note, User, Namespace string iconCalendar = "\uf133"
Blocks, Subtask, Related string iconCheck = "\uf00c"
Assignee, Created string iconCross = "\uf00d"
Tag, Calendar string iconNamespace = "\uf07b"
Check, Cross string
}
var icons = iconSet{
Issue: "\uf188", // bug icon
Note: "\uf15c", // file-text
User: "\uf007", // user
Namespace: "\uf07b", // folder
Blocks: "\uf068", // minus
Subtask: "\uf0da", // caret-right
Related: "\uf0c1", // link
Assignee: "\uf007", // user
Created: "\uf007", // user
Tag: "\uf02b", // tag
Calendar: "\uf133", // calendar
Check: "\uf00c", // check
Cross: "\uf00d", // times
}
func typeIcon(t string) string {
switch t {
case "issue":
return icons.Issue
case "note":
return icons.Note
case "user":
return icons.User
case "namespace":
return icons.Namespace
default:
return icons.Issue
}
}
func typeIconColored(t string) string {
switch t {
case "issue":
return typeIssue.Sprint(icons.Issue)
case "note":
return typeNote.Sprint(icons.Note)
case "user":
return typeUser.Sprint(icons.User)
case "namespace":
return typeNs.Sprint(icons.Namespace)
default:
return typeIssue.Sprint(icons.Issue)
}
}
func statusIcon(s string) string {
if s == "done" {
return icons.Check
}
return icons.Cross
}
func statusIconColored(s string) string {
if s == "" {
return " "
}
if s == "done" {
return dimColor.Sprint("○")
}
return dimColor.Sprint("●")
}
func statusColored(s string) string {
if s == "" {
return dimColor.Sprint("—")
}
if s == "done" {
return statusDone.Sprint("○ done")
}
return statusOpen.Sprint("● open")
}
func prioColored(p string) string {
switch p {
case "high":
return prioHigh.Sprint("● high")
case "medium":
return prioMedium.Sprint("◆ medium")
case "low":
return prioLow.Sprint("○ low")
default:
return dimColor.Sprint("—")
}
}
func PrintNodes(w io.Writer, nodes []*models.Node, jsonOut bool) error {
if jsonOut {
return json.NewEncoder(w).Encode(nodes)
}
fmt.Fprintln(w)
if len(nodes) == 0 {
fmt.Fprintln(w, dimColor.Sprint(" No results."))
return nil
}
sort.Slice(nodes, func(i, j int) bool {
si, sj := nodes[i].GetProperty("status"), nodes[j].GetProperty("status")
if si != sj {
return statusRank(si) > statusRank(sj)
}
pi, pj := nodes[i].GetProperty("prio"), nodes[j].GetProperty("prio")
return prioRank(pi) > prioRank(pj)
})
for _, n := range nodes {
tags := getDisplayTags(n)
nodeType := n.GetType()
var typeAndStatus string
if nodeType == "issue" {
typeAndStatus = statusIconColored(n.GetProperty("status"))
} else {
typeAndStatus = typeIconColored(nodeType) + " " + statusIconColored(n.GetProperty("status"))
}
fmt.Fprintf(w, " %s %s %s %s %s",
idColor.Sprintf("%s", n.ID),
prioColoredShort(n.GetProperty("prio")),
typeAndStatus,
titleColor.Sprint(truncate(n.Title, 35)),
dimColor.Sprint("["+n.GetProperty("namespace")+"]"),
) )
if len(tags) > 0 {
var hashTags []string
for _, t := range tags {
hashTags = append(hashTags, "#"+t)
}
fmt.Fprintf(w, " %s", tagColor.Sprint(strings.Join(hashTags, " ")))
}
fmt.Fprintln(w)
}
fmt.Fprintln(w)
return nil
}
func prioColoredShort(p string) string { func render(rm RenderMap, key string, short bool) string {
switch p { v, ok := rm[key]
case "high": if !ok {
return prioHigh.Sprint("\uf0e7 ") v, ok = rm[""]
case "medium": if !ok {
return prioMedium.Sprint("\uf12a ")
case "low":
return prioLow.Sprint("\uf068 ")
default:
return "" return ""
} }
} }
if short {
func prioRank(p string) int { return v.c.Sprint(v.s)
switch p {
case "high":
return 3
case "medium":
return 2
case "low":
return 1
default:
return 0
}
}
func statusRank(s string) int {
switch s {
case "open":
return 2
case "done":
return 0
default:
return 1
} }
return v.c.Sprint(v.l)
} }
func getDisplayTags(n *models.Node) []string { func getDisplayTags(n *models.Node) []string {
var tags []string var tags []string
for _, t := range n.Tags { for _, t := range n.Tags {
if _, _, ok := models.ParseTag(t); !ok { if !strings.HasPrefix(t, "_") {
tags = append(tags, t) tags = append(tags, t)
} }
} }
return tags return tags
} }
func PrintNodes(w io.Writer, nodes []*models.Node, jsonOut bool) error {
if jsonOut {
return json.NewEncoder(w).Encode(nodes)
}
if len(nodes) == 0 {
fmt.Fprintln(w, cDim.Sprint("No results."))
return nil
}
fmt.Fprintln(w)
sort.Slice(nodes, func(i, j int) bool {
si, sj := nodes[i].GetProperty("status"), nodes[j].GetProperty("status")
if si != sj {
return statusRanks[si] > statusRanks[sj]
}
return prioRanks[nodes[i].GetProperty("prio")] > prioRanks[nodes[j].GetProperty("prio")]
})
for _, n := range nodes {
tags := getDisplayTags(n)
fmt.Fprintf(w, " %s %s %s %s %s %s",
cDim.Sprint(n.ID),
render(prioRM, n.GetProperty("prio"), true),
render(statusRM, n.GetProperty("status"), true),
render(typeRM, n.GetProperty("type"), true),
cTitle.Sprint(truncate(n.Title, 35)),
cDim.Sprint("["+n.GetProperty("namespace")+"]"),
)
if len(tags) > 0 {
fmt.Fprintf(w, " %s", cPrimary.Sprint("#"+strings.Join(tags, " #")))
}
fmt.Fprintln(w)
}
fmt.Fprintln(w)
return nil
}
func PrintNode(w io.Writer, n *models.Node, jsonOut bool) error { func PrintNode(w io.Writer, n *models.Node, jsonOut bool) error {
if jsonOut { if jsonOut {
return json.NewEncoder(w).Encode(n) return json.NewEncoder(w).Encode(n)
} }
icon := typeIcon(n.GetType())
nodeType := strings.Title(n.GetType())
fmt.Fprintln(w) fmt.Fprintln(w)
fmt.Fprintf(w, " %s %s %s %s\n", fmt.Fprintf(w, " %s %s %s\n", render(typeRM, n.GetProperty("type"), false), cDim.Sprint(n.ID), cTitle.Sprint(n.Title))
icon, fmt.Fprintln(w, cDim.Sprint(" ───────────────────────────────"))
idColor.Sprint(n.ID), fmt.Fprintf(w, " Status: %s\n", render(statusRM, n.GetProperty("status"), false))
titleColor.Sprint(n.Title), fmt.Fprintf(w, " Priority: %s\n", render(prioRM, n.GetProperty("prio"), false))
dimColor.Sprint("["+nodeType+"]"), fmt.Fprintf(w, " Namespace: %s\n", cWarn.Sprint(n.GetProperty("namespace")))
)
fmt.Fprintln(w, dimColor.Sprint(" ───────────────────────────"))
fmt.Fprintf(w, " Status: %s\n", statusColored(n.GetProperty("status")))
fmt.Fprintf(w, " Priority: %s\n", prioColored(n.GetProperty("prio")))
fmt.Fprintf(w, " Namespace: %s\n", nsColor.Sprint(n.GetProperty("namespace")))
if n.DueDate != "" { if n.DueDate != "" {
fmt.Fprintf(w, " Due: %s %s\n", icons.Calendar, n.DueDate) fmt.Fprintf(w, " Due: %s %s\n", iconCalendar, n.DueDate)
} }
fmt.Fprintf(w, " Created: %s\n", dimColor.Sprint(n.CreatedAt)) fmt.Fprintf(w, " Created: %s\n", cDim.Sprint(n.CreatedAt))
fmt.Fprintf(w, " Updated: %s\n", cDim.Sprint(n.UpdatedAt))
if n.Content != "" { if n.Content != "" {
fmt.Fprintln(w) fmt.Fprintln(w, "\n"+cPrimary.Sprint(" Content:"))
fmt.Fprintln(w, labelColor.Sprint(" Content:")) for i, line := range strings.Split(n.Content, "\n") {
for _, line := range strings.Split(n.Content, "\n") { if i > 5 {
fmt.Fprintf(w, dimColor.Sprint(" │ ")+contentColor.Sprint("%s\n"), line) fmt.Fprintf(w, "%s ...\n", cDim.Sprint(" │ "))
break
}
fmt.Fprintf(w, "%s%s\n", cDim.Sprint(" │ "), cText.Sprint(line))
} }
} }
if len(n.Tags) > 0 { if tags := getDisplayTags(n); len(tags) > 0 {
var tags []string fmt.Fprintf(w, "\n tags: %s\n\n", cPrimary.Sprint(strings.Join(tags, " • ")))
for _, t := range n.Tags {
if _, _, ok := models.ParseTag(t); !ok {
tags = append(tags, t)
}
}
if len(tags) > 0 {
fmt.Fprintln(w)
fmt.Fprintf(w, " Tags: %s\n", tagColor.Sprint(strings.Join(tags, " • ")))
}
} }
if db, err := db.GetDB(); err == nil {
if len(n.Relations) > 0 { if len(n.Relations) > 0 {
fmt.Fprintln(w)
fmt.Fprintln(w, labelColor.Sprint(" Relations:"))
for relType, ids := range n.Relations { for relType, ids := range n.Relations {
if relType == "created" { fmt.Fprintf(w, " %s\n", string(relType))
continue if relIcon, ok := relIcons[string(relType)]; ok && relType != "created" {
for _, id := range ids {
node, err := db.NodeByID(id)
if err == nil {
fmt.Fprintf(w, " %s %s\n", relIcon, node.Title)
} }
relIcon := ""
switch relType {
case "blocks":
relIcon = icons.Blocks
case "subtask":
relIcon = icons.Subtask
case "related":
relIcon = icons.Related
case "assignee":
relIcon = icons.Assignee
} }
coloredIDs := make([]string, len(ids))
for i, id := range ids {
coloredIDs[i] = idColor.Sprint(id)
} }
fmt.Fprintf(w, " %s %s %s\n", relIcon, strings.Title(string(relType)), strings.Join(coloredIDs, " "))
} }
} }
} else {
fmt.Fprintf(w, "failed to attach to db: %v", err)
}
fmt.Fprintln(w) fmt.Fprintln(w)
return nil return nil
@@ -299,30 +177,27 @@ func PrintAliases(w io.Writer, aliases []*db.Alias, jsonOut bool) error {
return json.NewEncoder(w).Encode(aliases) return json.NewEncoder(w).Encode(aliases)
} }
if len(aliases) == 0 { if len(aliases) == 0 {
fmt.Fprintln(w, dimColor.Sprint(" No aliases defined.")) fmt.Fprintln(w, cDim.Sprint("No aliases defined."))
return nil return nil
} }
fmt.Fprintln(w) fmt.Fprintln(w)
for _, a := range aliases { for _, a := range aliases {
fmt.Fprintf(w, " %s %s\n", fmt.Fprintf(w, " %s %s\n", cPrimary.Sprint(a.Name), cDim.Sprint(a.Command))
idColor.Sprint(a.Name),
dimColor.Sprint(a.Command),
)
} }
fmt.Fprintln(w) fmt.Fprintln(w)
return nil return nil
} }
func PrintSuccess(w io.Writer, format string, args ...interface{}) { func PrintAction(w io.Writer, action, detail string, isError bool) {
fmt.Fprintf(w, icons.Check+" "+format+"\n", args...) if isError {
fmt.Fprintln(w, cBad.Sprint(iconCross+" "+action+" ")+cDim.Sprint(detail))
return
} }
icon := iconCheck
func PrintDeleted(w io.Writer, id string) { if action == "Created" {
fmt.Fprintf(w, " "+icons.Cross+" Deleted %s\n", idColor.Sprint(id)) icon = iconNamespace
} }
fmt.Fprintln(w, cGood.Sprint(icon+" "+action+" ")+cDim.Sprint(detail))
func PrintCreated(w io.Writer, dbPath string) {
fmt.Fprintf(w, " "+icons.Namespace+" Created %s\n", dimColor.Sprint(dbPath))
} }
func truncate(s string, max int) string { func truncate(s string, max int) string {

View File

@@ -6,11 +6,11 @@ import (
"slices" "slices"
) )
var mentionRegex = regexp.MustCompile(`@([a-z0-9_]+)`) var r = regexp.MustCompile(`@([a-z0-9_]+)`)
func Mentions(text string) []string { func Mentions(t string) []string {
seen := make(map[string]bool) seen := make(map[string]bool)
for _, m := range mentionRegex.FindAllStringSubmatch(text, -1) { for _, m := range r.FindAllStringSubmatch(t, -1) {
seen[m[1]] = true seen[m[1]] = true
} }
return slices.Collect(maps.Keys(seen)) return slices.Collect(maps.Keys(seen))