package ui import ( "context" "fmt" "log/slog" "sync" "github.com/gdamore/tcell/v2" "github.com/rivo/tview" "github.com/urfave/cli/v2" "git.gabrielgio.me/dict/db" ) var UICommand = &cli.Command{ Name: "ui", Usage: "interactive dictionary", Flags: []cli.Flag{ &cli.StringFlag{ Name: "database", Value: "main.dict", Usage: "Dictionary database location", }, }, Action: func(cCtx *cli.Context) error { name := cCtx.String("database") return Run(context.Background(), name) }, } func Run(ctx context.Context, name string) error { db, err := db.Open(name) if err != nil { return err } textView := tview.NewTextView(). SetDynamicColors(true). SetRegions(true) app := tview.NewApplication() search := "" tx := sync.Mutex{} input := tview.NewInputField(). SetChangedFunc(func(v string) { tx.Lock() defer tx.Unlock() search = v go func(v string) { words, err := db.SelectDict(ctx, v, 20) if err != nil { return } tx.Lock() if search != v { tx.Unlock() return } tx.Unlock() textView.Clear() lastWord := "" for _, w := range words { w.Word = tview.Escape(w.Word) w.Line = tview.Escape(w.Line) if lastWord == w.Word { fmt.Fprintf(textView, "%s\n", w.Line) } else if lastWord == "" { fmt.Fprintf(textView, "[::bu]%s[-:-:-]\n", w.Word) fmt.Fprintf(textView, "%s\n", w.Line) } else { fmt.Fprintf(textView, "\n[::bu]%s[-:-:-]\n", w.Word) fmt.Fprintf(textView, "%s\n", w.Line) } lastWord = w.Word } app.Draw() }(v) }). SetAutocompleteFunc(func(v string) []string { if len(v) == 0 { return []string{} } vs, err := db.SelectSpell(ctx, v) if err != nil { slog.Error("Error select spelling", "error", err) return []string{} } return vs }) input.SetDoneFunc(func(key tcell.Key) { if key == tcell.KeyEscape { textView.Clear() input.SetText("") } }) grid := tview.NewGrid(). SetRows(1, 0). AddItem(input, 0, 0, 1, 3, 0, 0, false). AddItem(textView, 1, 0, 1, 3, 0, 0, false) return app. SetRoot(grid, true). SetFocus(input). Run() }