124 lines
2.6 KiB
Go
124 lines
2.6 KiB
Go
package codeviewer
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/alecthomas/chroma/v2/formatters"
|
|
"github.com/alecthomas/chroma/v2/lexers"
|
|
"github.com/alecthomas/chroma/v2/styles"
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
)
|
|
|
|
type CodeViewer struct {
|
|
lines []string
|
|
|
|
Width, Height int
|
|
cursor int
|
|
offset int
|
|
}
|
|
|
|
func NewCodeViewer(text string) CodeViewer {
|
|
cv := CodeViewer{
|
|
Width: 0,
|
|
Height: 0,
|
|
cursor: 0,
|
|
offset: 0,
|
|
}
|
|
|
|
return cv.colorize(text)
|
|
}
|
|
|
|
func (c CodeViewer) Init() tea.Cmd {
|
|
return nil
|
|
}
|
|
|
|
func (c CodeViewer) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
|
switch msg := msg.(type) {
|
|
case tea.KeyMsg:
|
|
return c.handleKeyMsg(msg)
|
|
// case tea.WindowSizeMsg:
|
|
// return c.UpdateWindowSize(msg)
|
|
}
|
|
|
|
return c, nil
|
|
}
|
|
|
|
func (c CodeViewer) handleKeyMsg(key tea.KeyMsg) (tea.Model, tea.Cmd) {
|
|
switch key.String() {
|
|
case "k":
|
|
c.cursor = max(0, c.cursor-1)
|
|
topThreshold := c.offset + int(float64(c.Height)*0.1)
|
|
if c.cursor < topThreshold && c.offset > 0 {
|
|
c.offset -= 1
|
|
}
|
|
case "j":
|
|
c.cursor = min(len(c.lines)-1, c.cursor+1)
|
|
bottomThreshold := c.offset + int(float64(c.Height)*0.9)
|
|
if c.cursor > bottomThreshold && c.offset < len(c.lines) {
|
|
c.offset += 1
|
|
}
|
|
|
|
// case "b":
|
|
// lineNumber := c.cursor + 1
|
|
//
|
|
// c.bridge.Breakpoint(m.currentFile, lineNumber)
|
|
// if file, ok := m.breakpoints[m.currentFile]; ok {
|
|
// m.breakpoints[m.currentFile] = append(file, lineNumber)
|
|
// } else {
|
|
// m.breakpoints[m.currentFile] = []int{
|
|
// lineNumber,
|
|
// }
|
|
// }
|
|
// case "s":
|
|
// m.bridge.Step()
|
|
// case "c":
|
|
// m.bridge.Continue()
|
|
// case "r":
|
|
// m.messages = make([]string, 0)
|
|
// m.stdoutOutput.SetContent("")
|
|
// err := m.bridge.Start()
|
|
// if err != nil {
|
|
// slog.Error("could not start brige", "error", err)
|
|
// }
|
|
}
|
|
|
|
return c, nil
|
|
}
|
|
|
|
func (c CodeViewer) colorize(text string) CodeViewer {
|
|
var buf bytes.Buffer
|
|
lexer := lexers.Get("python")
|
|
style := styles.Get("monokai")
|
|
formatter := formatters.Get("terminal16m")
|
|
|
|
iterator, _ := lexer.Tokenise(nil, text)
|
|
formatter.Format(&buf, style, iterator)
|
|
|
|
c.lines = strings.Split(buf.String(), "\n")
|
|
return c
|
|
}
|
|
|
|
func (c CodeViewer) View() string {
|
|
var out strings.Builder
|
|
|
|
lines := c.lines[max(0, c.offset):min(c.offset+c.Height, len(c.lines))]
|
|
for i, line := range lines {
|
|
breakpoint := " "
|
|
cursor := " "
|
|
executor := " "
|
|
if c.offset+i == c.cursor {
|
|
cursor = ">"
|
|
}
|
|
fmt.Fprintf(&out, "%-4d%s%s%s %s\n", c.offset+i+1, breakpoint, executor, cursor, line)
|
|
}
|
|
if len(lines)-c.offset < c.Height {
|
|
for range c.Height - len(lines) {
|
|
fmt.Fprintf(&out, "\n")
|
|
}
|
|
}
|
|
|
|
return out.String()
|
|
}
|