chore: scaffold project structure for mvp
This commit is contained in:
parent
12a2e78e37
commit
2bfa8d5b63
17
cmd/difi/main.go
Normal file
17
cmd/difi/main.go
Normal file
|
|
@ -0,0 +1,17 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
|
||||||
|
tea "github.com/charmbracelet/bubbletea"
|
||||||
|
"github.com/oug-t/difi/internal/ui"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
p := tea.NewProgram(ui.NewModel(), tea.WithAltScreen())
|
||||||
|
if _, err := p.Run(); err != nil {
|
||||||
|
fmt.Printf("Error: %v\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
}
|
||||||
2
go.mod
2
go.mod
|
|
@ -1,4 +1,4 @@
|
||||||
module github.com/oug-t/stagi
|
module github.com/oug-t/difi
|
||||||
|
|
||||||
go 1.25.6
|
go 1.25.6
|
||||||
|
|
||||||
|
|
|
||||||
48
internal/git/client.go
Normal file
48
internal/git/client.go
Normal file
|
|
@ -0,0 +1,48 @@
|
||||||
|
package git
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"os/exec"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
tea "github.com/charmbracelet/bubbletea"
|
||||||
|
)
|
||||||
|
|
||||||
|
func ListChangedFiles(targetBranch string) ([]string, error) {
|
||||||
|
cmd := exec.Command("git", "diff", "--name-only", targetBranch)
|
||||||
|
out, err := cmd.Output()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
files := strings.Split(strings.TrimSpace(string(out)), "\n")
|
||||||
|
if len(files) == 1 && files[0] == "" {
|
||||||
|
return []string{}, nil
|
||||||
|
}
|
||||||
|
return files, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func DiffCmd(targetBranch, path string) tea.Cmd {
|
||||||
|
return func() tea.Msg {
|
||||||
|
out, err := exec.Command("git", "diff", "--color=always", targetBranch, "--", path).Output()
|
||||||
|
if err != nil {
|
||||||
|
return DiffMsg{Content: "Error fetching diff: " + err.Error()}
|
||||||
|
}
|
||||||
|
return DiffMsg{Content: string(out)}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func OpenEditorCmd(path string) tea.Cmd {
|
||||||
|
editor := os.Getenv("EDITOR")
|
||||||
|
if editor == "" {
|
||||||
|
editor = "vim"
|
||||||
|
}
|
||||||
|
c := exec.Command(editor, path)
|
||||||
|
c.Stdin, c.Stdout, c.Stderr = os.Stdin, os.Stdout, os.Stderr
|
||||||
|
return tea.ExecProcess(c, func(err error) tea.Msg {
|
||||||
|
return EditorFinishedMsg{Err: err}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
type DiffMsg struct{ Content string }
|
||||||
|
type EditorFinishedMsg struct{ Err error }
|
||||||
118
internal/tree/builder.go
Normal file
118
internal/tree/builder.go
Normal file
|
|
@ -0,0 +1,118 @@
|
||||||
|
package tree
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"path/filepath"
|
||||||
|
"sort"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/charmbracelet/bubbles/list"
|
||||||
|
)
|
||||||
|
|
||||||
|
type TreeItem struct {
|
||||||
|
Path string
|
||||||
|
FullPath string
|
||||||
|
IsDir bool
|
||||||
|
Depth int
|
||||||
|
}
|
||||||
|
|
||||||
|
func (i TreeItem) FilterValue() string { return i.FullPath }
|
||||||
|
func (i TreeItem) Description() string { return "" }
|
||||||
|
func (i TreeItem) Title() string {
|
||||||
|
indent := strings.Repeat(" ", i.Depth)
|
||||||
|
icon := getIcon(i.Path, i.IsDir)
|
||||||
|
return fmt.Sprintf("%s%s %s", indent, icon, i.Path)
|
||||||
|
}
|
||||||
|
|
||||||
|
func Build(paths []string) []list.Item {
|
||||||
|
root := &node{children: make(map[string]*node)}
|
||||||
|
|
||||||
|
for _, path := range paths {
|
||||||
|
parts := strings.Split(path, "/")
|
||||||
|
current := root
|
||||||
|
for i, part := range parts {
|
||||||
|
if _, exists := current.children[part]; !exists {
|
||||||
|
isDir := i < len(parts)-1
|
||||||
|
fullPath := strings.Join(parts[:i+1], "/")
|
||||||
|
current.children[part] = &node{
|
||||||
|
name: part,
|
||||||
|
fullPath: fullPath,
|
||||||
|
children: make(map[string]*node),
|
||||||
|
isDir: isDir,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
current = current.children[part]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var items []list.Item
|
||||||
|
flatten(root, 0, &items)
|
||||||
|
return items
|
||||||
|
}
|
||||||
|
|
||||||
|
type node struct {
|
||||||
|
name string
|
||||||
|
fullPath string
|
||||||
|
children map[string]*node
|
||||||
|
isDir bool
|
||||||
|
}
|
||||||
|
|
||||||
|
func flatten(n *node, depth int, items *[]list.Item) {
|
||||||
|
keys := make([]string, 0, len(n.children))
|
||||||
|
for k := range n.children {
|
||||||
|
keys = append(keys, k)
|
||||||
|
}
|
||||||
|
|
||||||
|
sort.Slice(keys, func(i, j int) bool {
|
||||||
|
a, b := n.children[keys[i]], n.children[keys[j]]
|
||||||
|
if a.isDir && !b.isDir {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
if !a.isDir && b.isDir {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return a.name < b.name
|
||||||
|
})
|
||||||
|
|
||||||
|
for _, k := range keys {
|
||||||
|
child := n.children[k]
|
||||||
|
*items = append(*items, TreeItem{
|
||||||
|
Path: child.name,
|
||||||
|
FullPath: child.fullPath,
|
||||||
|
IsDir: child.isDir,
|
||||||
|
Depth: depth,
|
||||||
|
})
|
||||||
|
if child.isDir {
|
||||||
|
flatten(child, depth+1, items)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func getIcon(name string, isDir bool) string {
|
||||||
|
if isDir {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
ext := filepath.Ext(name)
|
||||||
|
switch strings.ToLower(ext) {
|
||||||
|
case ".go":
|
||||||
|
return ""
|
||||||
|
case ".js", ".ts":
|
||||||
|
return ""
|
||||||
|
case ".md":
|
||||||
|
return ""
|
||||||
|
case ".json":
|
||||||
|
return ""
|
||||||
|
case ".yml", ".yaml":
|
||||||
|
return ""
|
||||||
|
case ".html":
|
||||||
|
return ""
|
||||||
|
case ".css":
|
||||||
|
return ""
|
||||||
|
case ".git":
|
||||||
|
return ""
|
||||||
|
case ".dockerfile":
|
||||||
|
return ""
|
||||||
|
default:
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
}
|
||||||
134
internal/ui/model.go
Normal file
134
internal/ui/model.go
Normal file
|
|
@ -0,0 +1,134 @@
|
||||||
|
package ui
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
|
||||||
|
"github.com/charmbracelet/bubbles/list"
|
||||||
|
"github.com/charmbracelet/bubbles/viewport"
|
||||||
|
tea "github.com/charmbracelet/bubbletea"
|
||||||
|
"github.com/charmbracelet/lipgloss"
|
||||||
|
|
||||||
|
"github.com/oug-t/difi/internal/git"
|
||||||
|
"github.com/oug-t/difi/internal/tree"
|
||||||
|
)
|
||||||
|
|
||||||
|
const TargetBranch = "main"
|
||||||
|
|
||||||
|
type Model struct {
|
||||||
|
fileTree list.Model
|
||||||
|
diffViewport viewport.Model
|
||||||
|
selectedPath string
|
||||||
|
width, height int
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewModel() Model {
|
||||||
|
files, _ := git.ListChangedFiles(TargetBranch)
|
||||||
|
items := tree.Build(files)
|
||||||
|
|
||||||
|
l := list.New(items, listDelegate{}, 0, 0)
|
||||||
|
l.SetShowTitle(false)
|
||||||
|
l.SetShowHelp(false)
|
||||||
|
l.SetShowStatusBar(false)
|
||||||
|
l.SetFilteringEnabled(false)
|
||||||
|
|
||||||
|
m := Model{
|
||||||
|
fileTree: l,
|
||||||
|
diffViewport: viewport.New(0, 0),
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(items) > 0 {
|
||||||
|
if first, ok := items[0].(tree.TreeItem); ok {
|
||||||
|
m.selectedPath = first.FullPath
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return m
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m Model) Init() tea.Cmd {
|
||||||
|
if m.selectedPath != "" {
|
||||||
|
return git.DiffCmd(TargetBranch, m.selectedPath)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||||||
|
var cmd tea.Cmd
|
||||||
|
var cmds []tea.Cmd
|
||||||
|
|
||||||
|
switch msg := msg.(type) {
|
||||||
|
case tea.WindowSizeMsg:
|
||||||
|
m.width = msg.Width
|
||||||
|
m.height = msg.Height
|
||||||
|
treeWidth := int(float64(m.width) * 0.25)
|
||||||
|
m.fileTree.SetSize(treeWidth, m.height)
|
||||||
|
m.diffViewport.Width = m.width - treeWidth - 2
|
||||||
|
m.diffViewport.Height = m.height
|
||||||
|
|
||||||
|
case tea.KeyMsg:
|
||||||
|
switch msg.String() {
|
||||||
|
case "q", "ctrl+c":
|
||||||
|
return m, tea.Quit
|
||||||
|
case "up", "k", "down", "j":
|
||||||
|
m.fileTree, cmd = m.fileTree.Update(msg)
|
||||||
|
cmds = append(cmds, cmd)
|
||||||
|
if item, ok := m.fileTree.SelectedItem().(tree.TreeItem); ok && !item.IsDir {
|
||||||
|
if item.FullPath != m.selectedPath {
|
||||||
|
m.selectedPath = item.FullPath
|
||||||
|
cmds = append(cmds, git.DiffCmd(TargetBranch, m.selectedPath))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
case "e":
|
||||||
|
if m.selectedPath != "" {
|
||||||
|
return m, git.OpenEditorCmd(m.selectedPath)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
case git.DiffMsg:
|
||||||
|
m.diffViewport.SetContent(msg.Content)
|
||||||
|
|
||||||
|
case git.EditorFinishedMsg:
|
||||||
|
if msg.Err != nil {
|
||||||
|
}
|
||||||
|
return m, git.DiffCmd(TargetBranch, m.selectedPath)
|
||||||
|
}
|
||||||
|
|
||||||
|
return m, tea.Batch(cmds...)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m Model) View() string {
|
||||||
|
if m.width == 0 {
|
||||||
|
return "Loading..."
|
||||||
|
}
|
||||||
|
|
||||||
|
treeView := PaneStyle.Copy().
|
||||||
|
Width(m.fileTree.Width()).
|
||||||
|
Height(m.fileTree.Height()).
|
||||||
|
Render(m.fileTree.View())
|
||||||
|
|
||||||
|
diffView := DiffStyle.Copy().
|
||||||
|
Width(m.diffViewport.Width).
|
||||||
|
Height(m.diffViewport.Height).
|
||||||
|
Render(m.diffViewport.View())
|
||||||
|
|
||||||
|
return lipgloss.JoinHorizontal(lipgloss.Top, treeView, diffView)
|
||||||
|
}
|
||||||
|
|
||||||
|
type listDelegate struct{}
|
||||||
|
|
||||||
|
func (d listDelegate) Height() int { return 1 }
|
||||||
|
func (d listDelegate) Spacing() int { return 0 }
|
||||||
|
func (d listDelegate) Update(msg tea.Msg, m *list.Model) tea.Cmd { return nil }
|
||||||
|
func (d listDelegate) Render(w io.Writer, m list.Model, index int, item list.Item) {
|
||||||
|
i, ok := item.(tree.TreeItem)
|
||||||
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
str := i.Title()
|
||||||
|
if index == m.Index() {
|
||||||
|
fmt.Fprint(w, SelectedItemStyle.Render("│ "+str))
|
||||||
|
} else {
|
||||||
|
fmt.Fprint(w, ItemStyle.Render(str))
|
||||||
|
}
|
||||||
|
}
|
||||||
25
internal/ui/styles.go
Normal file
25
internal/ui/styles.go
Normal file
|
|
@ -0,0 +1,25 @@
|
||||||
|
package ui
|
||||||
|
|
||||||
|
import "github.com/charmbracelet/lipgloss"
|
||||||
|
|
||||||
|
var (
|
||||||
|
ColorBorder = lipgloss.AdaptiveColor{Light: "#D9DCCF", Dark: "#383838"}
|
||||||
|
ColorSelected = lipgloss.AdaptiveColor{Light: "#1F1F1F", Dark: "#F8F8F2"}
|
||||||
|
ColorInactive = lipgloss.AdaptiveColor{Light: "#A8A8A8", Dark: "#626262"}
|
||||||
|
ColorAccent = lipgloss.AdaptiveColor{Light: "#00BCF0", Dark: "#00BCF0"} // Cyan
|
||||||
|
|
||||||
|
PaneStyle = lipgloss.NewStyle().
|
||||||
|
Border(lipgloss.NormalBorder(), false, true, false, false).
|
||||||
|
BorderForeground(ColorBorder).
|
||||||
|
MarginRight(1)
|
||||||
|
|
||||||
|
DiffStyle = lipgloss.NewStyle().
|
||||||
|
Padding(0, 1)
|
||||||
|
|
||||||
|
ItemStyle = lipgloss.NewStyle().PaddingLeft(1)
|
||||||
|
|
||||||
|
SelectedItemStyle = lipgloss.NewStyle().
|
||||||
|
PaddingLeft(0).
|
||||||
|
Foreground(ColorSelected).
|
||||||
|
Bold(true)
|
||||||
|
)
|
||||||
346
main.go
346
main.go
|
|
@ -1,346 +0,0 @@
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
"io"
|
|
||||||
"os"
|
|
||||||
"os/exec"
|
|
||||||
"path/filepath"
|
|
||||||
"sort"
|
|
||||||
"strings"
|
|
||||||
|
|
||||||
"github.com/charmbracelet/bubbles/list"
|
|
||||||
"github.com/charmbracelet/bubbles/viewport"
|
|
||||||
tea "github.com/charmbracelet/bubbletea"
|
|
||||||
"github.com/charmbracelet/lipgloss"
|
|
||||||
)
|
|
||||||
|
|
||||||
const targetBranch = "main"
|
|
||||||
|
|
||||||
// --- STYLES ---
|
|
||||||
|
|
||||||
var (
|
|
||||||
// Modern, Clean Theme
|
|
||||||
colorBorder = lipgloss.AdaptiveColor{Light: "#D9DCCF", Dark: "#383838"}
|
|
||||||
colorSelected = lipgloss.AdaptiveColor{Light: "#1F1F1F", Dark: "#F8F8F2"}
|
|
||||||
colorInactive = lipgloss.AdaptiveColor{Light: "#A8A8A8", Dark: "#626262"}
|
|
||||||
colorAccent = lipgloss.AdaptiveColor{Light: "#00BCF0", Dark: "#00BCF0"} // Dash-like Cyan
|
|
||||||
|
|
||||||
// Panes
|
|
||||||
treeStyle = lipgloss.NewStyle().
|
|
||||||
Border(lipgloss.NormalBorder(), false, true, false, false). // Right border only
|
|
||||||
BorderForeground(colorBorder).
|
|
||||||
MarginRight(1)
|
|
||||||
|
|
||||||
diffStyle = lipgloss.NewStyle().
|
|
||||||
Padding(0, 1)
|
|
||||||
|
|
||||||
// Tree Items
|
|
||||||
itemStyle = lipgloss.NewStyle().PaddingLeft(1)
|
|
||||||
selectedItemStyle = lipgloss.NewStyle().
|
|
||||||
PaddingLeft(0).
|
|
||||||
Foreground(colorSelected).
|
|
||||||
Bold(true)
|
|
||||||
)
|
|
||||||
|
|
||||||
// --- DATA STRUCTURES ---
|
|
||||||
|
|
||||||
type node struct {
|
|
||||||
name string
|
|
||||||
fullPath string
|
|
||||||
children map[string]*node
|
|
||||||
isDir bool
|
|
||||||
}
|
|
||||||
|
|
||||||
type treeItem struct {
|
|
||||||
path string
|
|
||||||
fullPath string
|
|
||||||
isDir bool
|
|
||||||
depth int
|
|
||||||
}
|
|
||||||
|
|
||||||
func (i treeItem) FilterValue() string { return i.fullPath }
|
|
||||||
func (i treeItem) Description() string { return "" }
|
|
||||||
|
|
||||||
// Title renders the Nerd Font icon + filename
|
|
||||||
func (i treeItem) Title() string {
|
|
||||||
indent := strings.Repeat(" ", i.depth)
|
|
||||||
icon := getIcon(i.path, i.isDir)
|
|
||||||
return fmt.Sprintf("%s%s %s", indent, icon, i.path)
|
|
||||||
}
|
|
||||||
|
|
||||||
// --- NERD FONT ICON MAPPING ---
|
|
||||||
|
|
||||||
func getIcon(name string, isDir bool) string {
|
|
||||||
if isDir {
|
|
||||||
return "" // Folder icon
|
|
||||||
}
|
|
||||||
|
|
||||||
ext := filepath.Ext(name)
|
|
||||||
switch strings.ToLower(ext) {
|
|
||||||
case ".go":
|
|
||||||
return "" // Go Gopher
|
|
||||||
case ".js", ".ts":
|
|
||||||
return "" // JS/Node
|
|
||||||
case ".md":
|
|
||||||
return "" // Markdown
|
|
||||||
case ".json":
|
|
||||||
return "" // JSON
|
|
||||||
case ".yml", ".yaml":
|
|
||||||
return "" // Settings/Config
|
|
||||||
case ".html":
|
|
||||||
return ""
|
|
||||||
case ".css":
|
|
||||||
return ""
|
|
||||||
case ".git":
|
|
||||||
return "" // Git
|
|
||||||
case ".dockerfile", "dockerfile":
|
|
||||||
return "" // Docker
|
|
||||||
default:
|
|
||||||
return "" // Default File
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// --- DELEGATE (CUSTOM RENDERER) ---
|
|
||||||
|
|
||||||
type delegate struct{}
|
|
||||||
|
|
||||||
func (d delegate) Height() int { return 1 }
|
|
||||||
func (d delegate) Spacing() int { return 0 }
|
|
||||||
func (d delegate) Update(msg tea.Msg, m *list.Model) tea.Cmd { return nil }
|
|
||||||
func (d delegate) Render(w io.Writer, m list.Model, index int, item list.Item) {
|
|
||||||
i, ok := item.(treeItem)
|
|
||||||
if !ok {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
str := i.Title()
|
|
||||||
|
|
||||||
// If selected, add a border or indicator
|
|
||||||
if index == m.Index() {
|
|
||||||
fmt.Fprint(w, selectedItemStyle.Render("│ "+str))
|
|
||||||
} else {
|
|
||||||
// Render unselected items with subtle gray
|
|
||||||
fmt.Fprint(w, itemStyle.Render(str))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// --- MODEL ---
|
|
||||||
|
|
||||||
type model struct {
|
|
||||||
fileTree list.Model
|
|
||||||
diffViewport viewport.Model
|
|
||||||
selectedPath string
|
|
||||||
width, height int
|
|
||||||
}
|
|
||||||
|
|
||||||
func initialModel() model {
|
|
||||||
// 1. Fetch changed files
|
|
||||||
// Compares HEAD against targetBranch
|
|
||||||
cmd := exec.Command("git", "diff", "--name-only", targetBranch)
|
|
||||||
out, _ := cmd.Output()
|
|
||||||
filePaths := strings.Split(strings.TrimSpace(string(out)), "\n")
|
|
||||||
|
|
||||||
// Handle empty diff case
|
|
||||||
if len(filePaths) == 1 && filePaths[0] == "" {
|
|
||||||
filePaths = []string{}
|
|
||||||
}
|
|
||||||
|
|
||||||
// 2. Build Tree
|
|
||||||
items := buildTree(filePaths)
|
|
||||||
|
|
||||||
// 3. Configure List
|
|
||||||
l := list.New(items, delegate{}, 0, 0)
|
|
||||||
l.SetShowTitle(false)
|
|
||||||
l.SetShowHelp(false)
|
|
||||||
l.SetShowStatusBar(false)
|
|
||||||
l.SetFilteringEnabled(false)
|
|
||||||
|
|
||||||
vp := viewport.New(0, 0)
|
|
||||||
|
|
||||||
m := model{
|
|
||||||
fileTree: l,
|
|
||||||
diffViewport: vp,
|
|
||||||
}
|
|
||||||
|
|
||||||
// Select first file if available
|
|
||||||
if len(items) > 0 {
|
|
||||||
if first, ok := items[0].(treeItem); ok {
|
|
||||||
m.selectedPath = first.fullPath
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return m
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m model) Init() tea.Cmd {
|
|
||||||
if m.selectedPath != "" {
|
|
||||||
return fetchDiff(m.selectedPath)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// --- UPDATE ---
|
|
||||||
|
|
||||||
func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
|
||||||
var cmd tea.Cmd
|
|
||||||
var cmds []tea.Cmd
|
|
||||||
|
|
||||||
switch msg := msg.(type) {
|
|
||||||
case tea.WindowSizeMsg:
|
|
||||||
m.width = msg.Width
|
|
||||||
m.height = msg.Height
|
|
||||||
|
|
||||||
// Layout: 25% Tree (Fixed Left), Rest Diff
|
|
||||||
treeWidth := int(float64(m.width) * 0.25)
|
|
||||||
diffWidth := m.width - treeWidth - 2
|
|
||||||
|
|
||||||
m.fileTree.SetSize(treeWidth, m.height)
|
|
||||||
m.diffViewport.Width = diffWidth
|
|
||||||
m.diffViewport.Height = m.height
|
|
||||||
|
|
||||||
case tea.KeyMsg:
|
|
||||||
switch msg.String() {
|
|
||||||
case "q", "ctrl+c":
|
|
||||||
return m, tea.Quit
|
|
||||||
|
|
||||||
case "up", "k", "down", "j":
|
|
||||||
m.fileTree, cmd = m.fileTree.Update(msg)
|
|
||||||
cmds = append(cmds, cmd)
|
|
||||||
|
|
||||||
// Fetch diff only if actual file selected
|
|
||||||
if item, ok := m.fileTree.SelectedItem().(treeItem); ok && !item.isDir {
|
|
||||||
if item.fullPath != m.selectedPath {
|
|
||||||
m.selectedPath = item.fullPath
|
|
||||||
cmds = append(cmds, fetchDiff(m.selectedPath))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
case "e":
|
|
||||||
// Edit in NVIM
|
|
||||||
if m.selectedPath != "" {
|
|
||||||
return m, openEditor(m.selectedPath)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
case diffMsg:
|
|
||||||
m.diffViewport.SetContent(string(msg))
|
|
||||||
|
|
||||||
case editorFinishedMsg:
|
|
||||||
return m, fetchDiff(m.selectedPath)
|
|
||||||
}
|
|
||||||
|
|
||||||
return m, tea.Batch(cmds...)
|
|
||||||
}
|
|
||||||
|
|
||||||
// --- VIEW ---
|
|
||||||
|
|
||||||
func (m model) View() string {
|
|
||||||
if m.width == 0 {
|
|
||||||
return "Loading..."
|
|
||||||
}
|
|
||||||
|
|
||||||
treeView := treeStyle.Copy().
|
|
||||||
Width(m.fileTree.Width()).
|
|
||||||
Height(m.fileTree.Height()).
|
|
||||||
Render(m.fileTree.View())
|
|
||||||
|
|
||||||
diffView := diffStyle.Copy().
|
|
||||||
Width(m.diffViewport.Width).
|
|
||||||
Height(m.diffViewport.Height).
|
|
||||||
Render(m.diffViewport.View())
|
|
||||||
|
|
||||||
return lipgloss.JoinHorizontal(lipgloss.Top, treeView, diffView)
|
|
||||||
}
|
|
||||||
|
|
||||||
// --- COMMANDS ---
|
|
||||||
|
|
||||||
type diffMsg string
|
|
||||||
type editorFinishedMsg struct{ err error }
|
|
||||||
|
|
||||||
func fetchDiff(path string) tea.Cmd {
|
|
||||||
return func() tea.Msg {
|
|
||||||
// Use --color=always to let git handle syntax highlighting
|
|
||||||
out, _ := exec.Command("git", "diff", "--color=always", targetBranch, "--", path).Output()
|
|
||||||
return diffMsg(out)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func openEditor(path string) tea.Cmd {
|
|
||||||
editor := os.Getenv("EDITOR")
|
|
||||||
if editor == "" {
|
|
||||||
editor = "vim"
|
|
||||||
}
|
|
||||||
c := exec.Command(editor, path)
|
|
||||||
c.Stdin, c.Stdout, c.Stderr = os.Stdin, os.Stdout, os.Stderr
|
|
||||||
return tea.ExecProcess(c, func(err error) tea.Msg {
|
|
||||||
return editorFinishedMsg{err}
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
// --- TREE ALGORITHMS ---
|
|
||||||
|
|
||||||
func buildTree(paths []string) []list.Item {
|
|
||||||
root := &node{children: make(map[string]*node)}
|
|
||||||
|
|
||||||
for _, path := range paths {
|
|
||||||
parts := strings.Split(path, "/")
|
|
||||||
current := root
|
|
||||||
for i, part := range parts {
|
|
||||||
if _, exists := current.children[part]; !exists {
|
|
||||||
isDir := i < len(parts)-1
|
|
||||||
fullPath := strings.Join(parts[:i+1], "/")
|
|
||||||
current.children[part] = &node{
|
|
||||||
name: part,
|
|
||||||
fullPath: fullPath,
|
|
||||||
children: make(map[string]*node),
|
|
||||||
isDir: isDir,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
current = current.children[part]
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
var items []list.Item
|
|
||||||
flatten(root, 0, &items)
|
|
||||||
return items
|
|
||||||
}
|
|
||||||
|
|
||||||
func flatten(n *node, depth int, items *[]list.Item) {
|
|
||||||
keys := make([]string, 0, len(n.children))
|
|
||||||
for k := range n.children {
|
|
||||||
keys = append(keys, k)
|
|
||||||
}
|
|
||||||
|
|
||||||
sort.Slice(keys, func(i, j int) bool {
|
|
||||||
a, b := n.children[keys[i]], n.children[keys[j]]
|
|
||||||
if a.isDir && !b.isDir {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
if !a.isDir && b.isDir {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
return a.name < b.name
|
|
||||||
})
|
|
||||||
|
|
||||||
for _, k := range keys {
|
|
||||||
child := n.children[k]
|
|
||||||
*items = append(*items, treeItem{
|
|
||||||
path: child.name,
|
|
||||||
fullPath: child.fullPath,
|
|
||||||
isDir: child.isDir,
|
|
||||||
depth: depth,
|
|
||||||
})
|
|
||||||
if child.isDir {
|
|
||||||
flatten(child, depth+1, items)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func main() {
|
|
||||||
p := tea.NewProgram(initialModel(), tea.WithAltScreen())
|
|
||||||
if _, err := p.Run(); err != nil {
|
|
||||||
fmt.Println(err)
|
|
||||||
os.Exit(1)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Loading…
Reference in New Issue
Block a user