add initial stuff

2 files changed, 126 insertions, 56 deletions

diff --git a/cmd/root.go b/cmd/root.go
new file mode 100644
index 000000000..e91444860
--- /dev/null
+++ b/cmd/root.go
@@ -0,0 +1,126 @@
+/*
+Copyright © 2025 NAME HERE <EMAIL ADDRESS>
+*/
+package cmd
+
+import (
+	"context"
+	"os"
+	"sync"
+
+	tea "github.com/charmbracelet/bubbletea"
+	"github.com/kujtimiihoxha/termai/internal/app"
+	"github.com/kujtimiihoxha/termai/internal/db"
+	"github.com/kujtimiihoxha/termai/internal/tui"
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+)
+
+var rootCmd = &cobra.Command{
+	Use:   "termai",
+	Short: "A terminal ai assistant",
+	Long:  `A terminal ai assistant`,
+	RunE: func(cmd *cobra.Command, args []string) error {
+		if cmd.Flag("help").Changed {
+			cmd.Help()
+			return nil
+		}
+		debug, _ := cmd.Flags().GetBool("debug")
+		viper.Set("debug", debug)
+		if debug {
+			viper.Set("log.level", "debug")
+		}
+
+		conn, err := db.Connect()
+		if err != nil {
+			return err
+		}
+		ctx := context.Background()
+
+		app := app.New(ctx, conn)
+		app.Logger.Info("Starting termai...")
+		tui := tea.NewProgram(
+			tui.New(app),
+			tea.WithAltScreen(),
+		)
+		app.Logger.Info("Setting up subscriptions...")
+		ch, unsub := setupSubscriptions(app)
+		defer unsub()
+
+		go func() {
+			for msg := range ch {
+				tui.Send(msg)
+			}
+		}()
+		if _, err := tui.Run(); err != nil {
+			return err
+		}
+		return nil
+	},
+}
+
+func setupSubscriptions(app *app.App) (chan tea.Msg, func()) {
+	ch := make(chan tea.Msg)
+	wg := sync.WaitGroup{}
+	ctx, cancel := context.WithCancel(app.Context)
+
+	if viper.GetBool("debug") {
+		sub := app.Logger.Subscribe(ctx)
+		wg.Add(1)
+		go func() {
+			for ev := range sub {
+				ch <- ev
+			}
+			wg.Done()
+		}()
+	}
+	{
+		sub := app.Sessions.Subscribe(ctx)
+		wg.Add(1)
+		go func() {
+			for ev := range sub {
+				ch <- ev
+			}
+			wg.Done()
+		}()
+	}
+	return ch, func() {
+		cancel()
+		wg.Wait()
+		close(ch)
+	}
+}
+
+// Execute adds all child commands to the root command and sets flags appropriately.
+// This is called by main.main(). It only needs to happen once to the rootCmd.
+func Execute() {
+	err := rootCmd.Execute()
+	if err != nil {
+		os.Exit(1)
+	}
+}
+
+func loadConfig() {
+	viper.SetConfigName(".termai")
+	viper.SetConfigType("yaml")
+	viper.AddConfigPath("$HOME")
+	viper.AddConfigPath("$XDG_CONFIG_HOME/termai")
+	viper.AddConfigPath(".")
+	viper.SetEnvPrefix("TERMAI")
+	// SET DEFAULTS
+	viper.SetDefault("log.level", "info")
+	viper.SetDefault("data.dir", ".termai")
+
+	//
+	viper.ReadInConfig()
+}
+
+func init() {
+	loadConfig()
+	// Here you will define your flags and configuration settings.
+	// Cobra supports persistent flags, which, if defined here,
+	// will be global for your application.
+
+	rootCmd.Flags().BoolP("help", "h", false, "Help")
+	rootCmd.Flags().BoolP("debug", "d", false, "Help")
+}
diff --git a/cmd/termai/main.go b/cmd/termai/main.go
deleted file mode 100644
index e1e506b86..000000000
--- a/cmd/termai/main.go
+++ /dev/null
@@ -1,56 +0,0 @@
-package main
-
-import (
-	"context"
-	"sync"
-
-	tea "github.com/charmbracelet/bubbletea"
-	"github.com/kujtimiihoxha/termai/internal/logging"
-	"github.com/kujtimiihoxha/termai/internal/tui"
-)
-
-var log = logging.Get()
-
-func main() {
-	log.Info("Starting termai...")
-	ctx := context.Background()
-
-	app := tea.NewProgram(
-		tui.New(),
-		tea.WithAltScreen(),
-	)
-	log.Info("Setting up subscriptions...")
-	ch, unsub := setupSubscriptions(ctx)
-	defer unsub()
-
-	go func() {
-		for msg := range ch {
-			app.Send(msg)
-		}
-	}()
-	if _, err := app.Run(); err != nil {
-		panic(err)
-	}
-}
-
-func setupSubscriptions(ctx context.Context) (chan tea.Msg, func()) {
-	ch := make(chan tea.Msg)
-	wg := sync.WaitGroup{}
-	ctx, cancel := context.WithCancel(ctx)
-
-	{
-		sub := log.Subscribe(ctx)
-		wg.Add(1)
-		go func() {
-			for ev := range sub {
-				ch <- ev
-			}
-			wg.Done()
-		}()
-	}
-	return ch, func() {
-		cancel()
-		wg.Wait()
-		close(ch)
-	}
-}