chore: cli refactor

cmd/root.go

@@ -0,0 +1,108 @@
+package cmd
+
+import (
+	"context"
+	"github.com/abelith/spellbook/internal/models"
+	"github.com/olekukonko/tablewriter"
+	"github.com/spf13/cobra"
+	"io"
+	"os"
+	"strings"
+)
+
+type Application interface {
+	AddSpell(ctx context.Context, spell models.Spell) error
+	GetSpell(ctx context.Context, spellName string) (models.Spell, error)
+	GetSpells(ctx context.Context, query models.SearchQuery) ([]models.Spell, error)
+	DeleteSpell(ctx context.Context, spellName string) error
+	CastSpell(ctx context.Context, spellCast models.SpellCast) (models.CastResult, error)
+	ExportSpells(ctx context.Context, exp string, w io.Writer) error
+	ImportSpells(ctx context.Context, exp string, r io.Reader) error
+}
+
+func rootCmd(app Application) func(*cobra.Command, []string) error {
+	return func(cmd *cobra.Command, args []string) error {
+		ctx := cmd.Context()
+
+		if len(args) < 1 {
+			return listCommands(ctx, cmd, app)
+		}
+
+		return runCommand(ctx, app, args)
+	}
+}
+
+func runCommand(ctx context.Context, app Application, args []string) error {
+	spell, err := app.GetSpell(ctx, args[0])
+	if err != nil {
+		return err
+	}
+
+	m := make(map[string]string, len(args[1:]))
+	for _, pair := range args[1:] {
+		keyVal := strings.Split(pair, "=")
+		m[keyVal[0]] = keyVal[1]
+	}
+
+	_, err = app.CastSpell(ctx, models.SpellCast{
+		Spell:  spell,
+		Args:   m,
+		Stdin:  os.Stdin,
+		Stdout: os.Stdout,
+		Stderr: os.Stderr,
+	})
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func listCommands(ctx context.Context, cmd *cobra.Command, app Application) error {
+	spells, err := app.GetSpells(ctx, models.SearchQuery{})
+	if err != nil {
+		return err
+	}
+
+	if len(spells) == 0 {
+		cmd.Println("Spell book is empty. Try `spellbook add` to add your first spell")
+		return nil
+	}
+
+	return printTable(cmd, spells)
+}
+
+func printTable(cmd *cobra.Command, spells []models.Spell) error {
+	table := tablewriter.NewTable(cmd.OutOrStdout())
+
+	rows := make([][]string, 0, len(spells))
+	for _, spell := range spells {
+		row := []string{
+			spell.Name, spell.Spell, spell.ShortDesc, strings.Join(spell.Tags, ", "),
+		}
+		rows = append(rows, row)
+	}
+
+	table.Header("Name", "Spell", "Short description", "Tags")
+	_ = table.Bulk(rows)
+	if err := table.Render(); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func NewRootCmd(ver string, app Application) *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "spellbook",
+		Short: "CLI tool to store, manage and execute command templates",
+		Long: `Spellbook is a lightweight binary command manager written in Go.
+It allows you to save complex shell commands as parametrized templates (spells)
+and execute them instantly with custom arguments from anywhere in your terminal`,
+		RunE:    rootCmd(app),
+		Args:    cobra.ArbitraryArgs,
+		Version: ver,
+	}
+
+	return cmd
+}