mend/main.go

package main

import (
	"fmt"
	"os"

	tea "github.com/charmbracelet/bubbletea"
	"github.com/charmbracelet/lipgloss"
)

/*
Quick notes to self:

# How bubbletea works?
Bubbletea is a
Init ( once for sync/async opts both )
-> Update ( model ) ( againc an be sync/async )
-> View loop
init creates initial model ( model is the global ui state and just a struct )
update modifies model based on an immediate or deferred compute
view is purely for rendering based on model

notes:
- model struct next -> global ui model
- createModel is the initial sync model state population
I anyways need to create the model struct even if blank
- Init func is for bubbletea to call once at start so both are needed
- all widths and heights are character count based
- async updates are via cmds (tea.Cmd ) that are no arg funcs that return
a tea.Msg ( that is basically a struct and hence has the data needed )
this data in msg when returned to the update func is used to update the model
*/

type model struct {
	width          int
	terminalWidth  int
	terminalHeight int
	fsTreeWidth    int
	noteViewWidth  int
	tree           *FsTree
	rootPath       string // path to load the tree from
	loading        bool
	noteView       *NoteView
}

func NewModel(rootPath string) *model {
	return &model{
		rootPath: rootPath,
		loading:  true,
		noteView: NewNoteView(),
	}
}

// =================== bubbletea ui fns ===================
// these need to be on the "model" ( duck typing "implements" interface )

type treeLoadedMsg struct {
	tree *FsTree
}

func (m *model) loadTreeCmd(path string) tea.Cmd {
	return func() tea.Msg {
		var targetPath string
		if path == "" {
			cwd, err := os.Getwd()
			if err != nil {
				fmt.Println("Error getting cwd:", err)
				os.Exit(1)
			}
			targetPath = cwd
		} else {
			targetPath = path
		}
		return treeLoadedMsg{tree: NewFsTree(targetPath)}
	}
}

func (m *model) layout(width, height int) {
	m.terminalWidth = width
	m.terminalHeight = height
	m.fsTreeWidth = width / 4
	m.noteViewWidth = width - m.fsTreeWidth
}

func (m *model) Init() tea.Cmd {
	return m.loadTreeCmd(m.rootPath)
}

func (m *model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
	switch msg := msg.(type) {
	case tea.WindowSizeMsg:
		m.layout(msg.Width, msg.Height)

		var cmds []tea.Cmd
		if m.tree != nil {
			_, cmd := m.tree.Update(tea.WindowSizeMsg{
				Width:  m.fsTreeWidth,
				Height: m.terminalHeight,
			})
			cmds = append(cmds, cmd)
		}
		_, cmd := m.noteView.Update(tea.WindowSizeMsg{
			Width:  m.noteViewWidth,
			Height: m.terminalHeight,
		})
		cmds = append(cmds, cmd)
		return m, tea.Batch(cmds...)

	case treeLoadedMsg:
		m.tree = msg.tree
		m.loading = false
		_, cmd := m.tree.Update(tea.WindowSizeMsg{
			Height: m.terminalHeight,
			Width:  m.fsTreeWidth,
		})
		return m, cmd

	case nodeSelected:
		// Forward node selection to noteView
		_, cmd := m.noteView.Update(loadNote{path: msg.path})
		return m, cmd

	case loadedNote:
		// Forward loaded note to noteView
		_, cmd := m.noteView.Update(msg)
		return m, cmd

	case tea.KeyMsg:
		switch msg.String() {
		case "q", "ctrl+c":
			return m, tea.Quit
		}
		// Forward keyboard input to tree
		var cmd tea.Cmd
		if m.tree != nil {
			_, cmd = m.tree.Update(msg)
		}
		return m, cmd

	case tea.MouseMsg:
		// Forward mouse input to tree
		var cmd tea.Cmd
		if m.tree != nil {
			_, cmd = m.tree.Update(msg)
		}
		return m, cmd
	}

	return m, nil
}

func (m model) View() string {
	if m.loading {
		return "Loading files..."
	}

	tree := m.tree.View()
	tree = lipgloss.NewStyle().
		Border(lipgloss.NormalBorder()).
		Width(40).
		Align(lipgloss.Left).
		PaddingRight(2).
		BorderLeft(false).
		BorderTop(false).
		BorderBottom(false).
		Render(tree)
	notes := m.noteView.View()

	full := lipgloss.JoinHorizontal(
		lipgloss.Top,
		tree,
		notes,
	)
	return full
}

// =================== bubbletea ui fns ===================

func main() {
	var rootPath string
	if len(os.Args) > 1 {
		rootPath = os.Args[1]
	}
	// if rootPath is empty, createModel will use cwd
	p := tea.NewProgram(
		NewModel(rootPath),
		tea.WithAltScreen(), // full screen tui
		tea.WithMouseAllMotion(),
	)
	if _, err := p.Run(); err != nil {
		fmt.Printf("Error: %v", err)
		os.Exit(1)
	}
}