Init

Signed-off-by: Hadi <112569860+anotherhadi@users.noreply.github.com>
2026-05-20 01:32:33 +02:00 · 2026-05-12 19:12:29 +02:00
commit e8e64eff12
101 changed files with 10081 additions and 0 deletions
@@ -0,0 +1,137 @@
+package app
+
+import (
+	"log"
+	"os"
+	"path/filepath"
+	"strconv"
+	"time"
+
+	tea "charm.land/bubbletea/v2"
+	"github.com/anotherhadi/spilltea/internal/config"
+	"github.com/anotherhadi/spilltea/internal/db"
+	"github.com/anotherhadi/spilltea/internal/intercept"
+	"github.com/anotherhadi/spilltea/internal/plugins"
+	proxyPkg "github.com/anotherhadi/spilltea/internal/proxy"
+	copyasUI "github.com/anotherhadi/spilltea/internal/ui/components/copyas"
+	notificationsUI "github.com/anotherhadi/spilltea/internal/ui/components/notifications"
+	diffUI "github.com/anotherhadi/spilltea/internal/ui/diff"
+	docsUI "github.com/anotherhadi/spilltea/internal/ui/docs"
+	findingsUI "github.com/anotherhadi/spilltea/internal/ui/findings"
+	historyUI "github.com/anotherhadi/spilltea/internal/ui/history"
+	interceptUI "github.com/anotherhadi/spilltea/internal/ui/intercept"
+	pluginsUI "github.com/anotherhadi/spilltea/internal/ui/plugins"
+	replayUI "github.com/anotherhadi/spilltea/internal/ui/replay"
+	scopeUI "github.com/anotherhadi/spilltea/internal/ui/scope"
+	"github.com/sirupsen/logrus"
+)
+
+const tickInterval = 2 * time.Second
+
+type tickMsg struct{}
+
+func tickCmd() tea.Cmd {
+	return func() tea.Msg {
+		time.Sleep(tickInterval)
+		return tickMsg{}
+	}
+}
+
+var sidebarEntries = pageRegistry
+
+var pageShortcuts = func() map[string]page {
+	m := make(map[string]page, len(sidebarEntries))
+	for i, e := range sidebarEntries {
+		m[strconv.Itoa(i+1)] = e.id
+	}
+	return m
+}()
+
+type Model struct {
+	broker        *intercept.Broker
+	page          page
+	projectName   string
+	projectPath   string
+	database      *db.DB
+	logFile       *os.File
+	pluginManager *plugins.Manager
+
+	width         int
+	height        int
+	sidebarState  sidebarState
+	intercept     interceptUI.Model
+	history       historyUI.Model
+	replay        replayUI.Model
+	diff          diffUI.Model
+	docs          docsUI.Model
+	scope         scopeUI.Model
+	pluginsPage   pluginsUI.Model
+	findingsPage  findingsUI.Model
+	copyAs        copyasUI.Model
+	notifications notificationsUI.Model
+}
+
+func New(broker *intercept.Broker, name, path string) Model {
+	cfg := config.Global
+	mgr := plugins.NewManager(broker)
+
+	m := Model{
+		broker:        broker,
+		page:          pageIntercept,
+		projectName:   name,
+		projectPath:   path,
+		pluginManager: mgr,
+		intercept:     interceptUI.New(broker),
+		history:       historyUI.New(),
+		replay:        replayUI.New(),
+		diff:          diffUI.New(),
+		docs:          docsUI.New(),
+		scope:         scopeUI.New(name, path),
+		pluginsPage:   pluginsUI.New(mgr),
+		findingsPage:  findingsUI.New(),
+		copyAs:        copyasUI.New(),
+		notifications: notificationsUI.New(),
+		sidebarState:  sidebarState(cfg.TUI.DefaultSidebarState),
+	}
+
+	if d, err := db.Open(path); err == nil {
+		m.database = d
+		broker.SetDB(d)
+		m.history.SetDB(d)
+		m.replay.SetDB(d)
+		m.findingsPage.SetDB(d)
+		mgr.SetDB(d)
+		if wl, bl, err := d.LoadScope(); err == nil {
+			broker.SetScope(wl, bl)
+			m.scope.SetScope(wl, bl)
+		}
+	}
+
+	pluginsDir := config.ExpandPath(cfg.App.PluginsDir)
+	if err := mgr.LoadFromDir(pluginsDir); err != nil {
+		log.Printf("plugins: %v", err)
+	}
+	m.pluginsPage.Refresh()
+
+	if lf, err := os.OpenFile(filepath.Join(filepath.Dir(path), "logs.log"), os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0o600); err == nil {
+		m.logFile = lf
+		log.SetOutput(lf)
+		logrus.SetOutput(lf)
+	}
+
+	return m
+}
+
+func (m Model) Init() tea.Cmd {
+	mgr := m.pluginManager
+	return tea.Batch(
+		intercept.WaitForRequest(m.broker),
+		intercept.WaitForResponse(m.broker),
+		tickCmd(),
+		proxyPkg.StartCmd(m.broker, mgr),
+		plugins.WaitForNotif(mgr),
+		plugins.WaitForQuit(mgr),
+		findingsUI.RefreshCmd(m.database),
+		func() tea.Msg { mgr.RunOnStart(); return nil },
+	)
+}