Files
pybug/ui/view.go
2026-03-28 17:41:54 +01:00

84 lines
2.1 KiB
Go

package ui
import (
"bytes"
"fmt"
"slices"
"strings"
"github.com/alecthomas/chroma/v2/formatters"
"github.com/alecthomas/chroma/v2/lexers"
"github.com/alecthomas/chroma/v2/styles"
"github.com/charmbracelet/lipgloss"
)
var panelStyle = lipgloss.NewStyle().Border(lipgloss.NormalBorder()).Padding(0, 1)
func flattenDict(m map[string]interface{}, indent int) []string {
lines := []string{}
prefix := strings.Repeat(" ", indent)
for k, v := range m {
switch val := v.(type) {
case map[string]interface{}:
lines = append(lines, fmt.Sprintf("%s%s:", prefix, k))
lines = append(lines, flattenDict(val, indent+1)...)
default:
lines = append(lines, fmt.Sprintf("%s%s: %v", prefix, k, val))
}
}
return lines
}
func (m Model) View() string {
var buf bytes.Buffer
lexer := lexers.Get("python")
style := styles.Get("monokai")
formatter := formatters.Get("terminal16m")
iterator, _ := lexer.Tokenise(nil, m.text)
formatter.Format(&buf, style, iterator)
var out strings.Builder
lines := strings.Split(buf.String(), "\n")
breakpoints := m.breakpoints[m.currentFile]
for i, line := range lines {
breakpoint := " "
cursor := " "
executor := " "
if slices.Contains(breakpoints, i+1) {
breakpoint = "O"
}
if i+1 == m.currExecutionPoint.Line {
executor = "!"
}
if i == m.cursor {
cursor = ">"
}
fmt.Fprintf(&out, "%-4d%s%s%s %s\n", i+1, breakpoint, executor, cursor, line)
}
m.codeViewer.SetContent(out.String())
hFrame, wFrame := panelStyle.GetFrameSize()
topPanel := panelStyle.
Height(m.codeViewer.Height - hFrame).
Width(m.codeViewer.Width - wFrame).
Render(m.codeViewer.View())
bottomLeftPanel := panelStyle.Height(m.stdoutOutput.Height - hFrame).
Width(m.stdoutOutput.Width - wFrame).
Render(m.stdoutOutput.View())
bottomRightPanel := panelStyle.Height(m.stdoutOutput.Height - hFrame).
Width(m.stdoutOutput.Width - wFrame).
Render(m.localsViewer.View())
bottomPanel := lipgloss.JoinHorizontal(lipgloss.Top, bottomLeftPanel, bottomRightPanel)
return lipgloss.JoinVertical(lipgloss.Top, topPanel, bottomPanel)
}