1 files changed, 105 insertions, 0 deletions
diff --git a/cmd/ui/ui.go b/cmd/ui/ui.go
new file mode 100644
index 0000000..82c0bc5
--- /dev/null
+++ b/cmd/ui/ui.go
@@ -0,0 +1,105 @@
+package ui
+
+import (
+	"context"
+	"fmt"
+	"log/slog"
+
+	"github.com/gdamore/tcell/v2"
+	"github.com/rivo/tview"
+	"github.com/urfave/cli/v2"
+
+	"git.gabrielgio.me/dict/db"
+)
+
+const (
+	memory = ":memory:"
+)
+
+var UICommand = &cli.Command{
+	Name:  "ui",
+	Usage: "interactive dictionary",
+	Flags: []cli.Flag{
+		&cli.StringFlag{
+			Name:  "filename",
+			Value: "main.dict",
+			Usage: "Dictionary database location",
+		},
+	},
+	Action: func(cCtx *cli.Context) error {
+		name := cCtx.String("lang")
+		return Run(context.Background(), name)
+	},
+}
+
+func Run(ctx context.Context, name string) error {
+	db, err := db.Open(memory)
+	if err != nil {
+		return err
+	}
+
+	err = db.Restore(ctx, name)
+	if err != nil {
+		return err
+	}
+
+	textView := tview.NewTextView().
+		SetDynamicColors(true).
+		SetRegions(true)
+
+	input := tview.NewInputField().
+		SetChangedFunc(func(v string) {
+			textView.Clear()
+
+			words, err := db.SelectDict(ctx, v, 100)
+			if err != nil {
+				return
+			}
+
+			lastWord := ""
+			for _, w := range words {
+
+				if lastWord == w.Word {
+					fmt.Fprintf(textView, "%s\n", w.Line)
+				} else if lastWord == "" {
+					fmt.Fprintf(textView, "[bold]%s[normal]\n", w.Word)
+					fmt.Fprintf(textView, "%s\n", w.Line)
+				} else {
+					fmt.Fprintf(textView, "\n[bold]%s[normal]\n", w.Word)
+					fmt.Fprintf(textView, "%s\n", w.Line)
+				}
+
+				lastWord = w.Word
+			}
+		}).
+		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) {
+		textView.Clear()
+		input.SetText("")
+	})
+
+	grid := tview.NewGrid().
+		SetRows(1, 0, 3).
+		AddItem(input, 0, 0, 1, 3, 0, 0, false).
+		AddItem(textView, 1, 0, 1, 3, 0, 0, false)
+
+	err = tview.NewApplication().
+		SetRoot(grid, true).
+		SetFocus(input).
+		Run()
+
+	return err
+}