f829b37d0c
I ran pprof to find what was causing stuttering, and found it to be the getTabCountInLineAtCol function in TextEdit, because it was iterating many bytes of the buffer, for each rune rendered. Replaced it with a more optimal system. Also changed the architecture of the highlighting system to use a single RegexpRange structure for all regular expressions. This allows for optimizations and multiline matches in the future.
375 lines
9.7 KiB
Go
375 lines
9.7 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"io/fs"
|
|
"io/ioutil"
|
|
"os"
|
|
"runtime"
|
|
"runtime/pprof"
|
|
|
|
"github.com/fivemoreminix/qedit/ui"
|
|
"github.com/gdamore/tcell/v2"
|
|
)
|
|
|
|
var (
|
|
cpuprofile = flag.String("cpuprofile", "", "write cpu profile to `file`")
|
|
memprofile = flag.String("memprofile", "", "write memory profile to `file`")
|
|
)
|
|
|
|
var theme = ui.Theme{
|
|
"StatusBar": tcell.Style{}.Foreground(tcell.ColorBlack).Background(tcell.ColorSilver),
|
|
}
|
|
|
|
var (
|
|
menuBar *ui.MenuBar
|
|
tabContainer *ui.TabContainer
|
|
dialog ui.Component // nil if not present (has exclusive focus)
|
|
|
|
focusedComponent ui.Component = nil
|
|
)
|
|
|
|
func changeFocus(to ui.Component) {
|
|
if focusedComponent != nil {
|
|
focusedComponent.SetFocused(false)
|
|
}
|
|
focusedComponent = to
|
|
to.SetFocused(true)
|
|
}
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
if *cpuprofile != "" {
|
|
f, err := os.Create(*cpuprofile)
|
|
if err != nil {
|
|
log.Fatal("Could not create CPU profile: ", err)
|
|
}
|
|
defer f.Close()
|
|
if err := pprof.StartCPUProfile(f); err != nil {
|
|
log.Fatal("Could not start CPU profile: ", err)
|
|
}
|
|
defer pprof.StopCPUProfile()
|
|
}
|
|
|
|
s, e := tcell.NewScreen()
|
|
if e != nil {
|
|
fmt.Fprintf(os.Stderr, "%v\n", e)
|
|
os.Exit(1)
|
|
}
|
|
if e := s.Init(); e != nil {
|
|
fmt.Fprintf(os.Stderr, "%v\n", e)
|
|
os.Exit(1)
|
|
}
|
|
defer s.Fini() // Useful for handling panics
|
|
|
|
var closing bool
|
|
sizex, sizey := s.Size()
|
|
|
|
tabContainer = ui.NewTabContainer(&theme)
|
|
tabContainer.SetPos(0, 1)
|
|
tabContainer.SetSize(sizex, sizey-2)
|
|
|
|
_, err := ClipInitialize(ClipExternal)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
// Open files from command-line arguments
|
|
if len(os.Args) > 1 {
|
|
for i := 1; i < len(os.Args); i++ {
|
|
if os.Args[i] == "-cpuprofile" || os.Args[i] == "-memprofile" {
|
|
i++
|
|
continue
|
|
}
|
|
|
|
_, err := os.Stat(os.Args[i])
|
|
|
|
var dirty bool
|
|
var bytes []byte
|
|
|
|
if errors.Is(err, os.ErrNotExist) { // If the file does not exist...
|
|
dirty = true
|
|
} else { // If the file exists...
|
|
file, err := os.Open(os.Args[i])
|
|
if err != nil {
|
|
panic("File could not be opened at path " + os.Args[i])
|
|
}
|
|
defer file.Close()
|
|
|
|
bytes, err = ioutil.ReadAll(file)
|
|
if err != nil {
|
|
panic("Could not read all of " + os.Args[i])
|
|
}
|
|
}
|
|
|
|
textEdit := ui.NewTextEdit(&s, os.Args[i], bytes, &theme)
|
|
textEdit.Dirty = dirty
|
|
tabContainer.AddTab(os.Args[i], textEdit)
|
|
}
|
|
}
|
|
|
|
menuBar = ui.NewMenuBar(&theme)
|
|
|
|
fileMenu := ui.NewMenu("File", 0, &theme)
|
|
|
|
fileMenu.AddItems([]ui.Item{&ui.ItemEntry{Name: "New File", Shortcut: "Ctrl+N", Callback: func() {
|
|
textEdit := ui.NewTextEdit(&s, "", []byte{}, &theme) // No file path, no contents
|
|
tabContainer.AddTab("noname", textEdit)
|
|
}}, &ui.ItemEntry{Name: "Open...", Shortcut: "Ctrl+O", Callback: func() {
|
|
callback := func(filePaths []string) {
|
|
for _, path := range filePaths {
|
|
file, err := os.Open(path)
|
|
if err != nil {
|
|
panic("Could not open file at path " + path)
|
|
}
|
|
defer file.Close()
|
|
|
|
bytes, err := ioutil.ReadAll(file)
|
|
if err != nil {
|
|
panic("Could not read all of file")
|
|
}
|
|
|
|
textEdit := ui.NewTextEdit(&s, path, bytes, &theme)
|
|
tabContainer.AddTab(path, textEdit)
|
|
}
|
|
// TODO: free the dialog instead?
|
|
dialog = nil // Hide the file selector
|
|
|
|
changeFocus(tabContainer)
|
|
}
|
|
dialog = ui.NewFileSelectorDialog(
|
|
&s,
|
|
"Comma-separated files or a directory",
|
|
true,
|
|
&theme,
|
|
callback,
|
|
func() { // Dialog is canceled
|
|
dialog = nil
|
|
changeFocus(tabContainer)
|
|
},
|
|
)
|
|
changeFocus(dialog)
|
|
}}, &ui.ItemEntry{Name: "Save", Shortcut: "Ctrl+S", Callback: func() {
|
|
if tabContainer.GetTabCount() > 0 {
|
|
tab := tabContainer.GetTab(tabContainer.GetSelectedTabIdx())
|
|
te := tab.Child.(*ui.TextEdit)
|
|
if len(te.FilePath) > 0 {
|
|
f, err := os.OpenFile(te.FilePath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, fs.ModePerm)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer f.Close()
|
|
|
|
_, err = te.Buffer.WriteTo(f) // TODO: check count
|
|
if err != nil {
|
|
panic(fmt.Sprintf("Error occurred while writing buffer to file: %v", err))
|
|
}
|
|
|
|
te.Dirty = false
|
|
}
|
|
changeFocus(tabContainer)
|
|
}
|
|
}}, &ui.ItemEntry{Name: "Save As...", QuickChar: 5, Callback: func() {
|
|
// TODO: implement a "Save as" dialog system, and show that when trying to save noname files
|
|
callback := func(filePaths []string) {
|
|
dialog = nil // Hide the file selector
|
|
changeFocus(tabContainer)
|
|
}
|
|
|
|
dialog = ui.NewFileSelectorDialog(
|
|
&s,
|
|
"Select a file to overwrite",
|
|
false,
|
|
&theme,
|
|
callback,
|
|
func() { // Dialog canceled
|
|
dialog = nil
|
|
changeFocus(tabContainer)
|
|
},
|
|
)
|
|
changeFocus(dialog)
|
|
}}, &ui.ItemSeparator{}, &ui.ItemEntry{Name: "Close", Shortcut: "Ctrl+Q", Callback: func() {
|
|
if tabContainer.GetTabCount() > 0 {
|
|
tabContainer.RemoveTab(tabContainer.GetSelectedTabIdx())
|
|
} else { // No tabs open; close the editor
|
|
closing = true
|
|
}
|
|
}}})
|
|
|
|
panelMenu := ui.NewMenu("Panel", 0, &theme)
|
|
|
|
panelMenu.AddItems([]ui.Item{&ui.ItemEntry{Name: "Focus Up", QuickChar: -1, Shortcut: "Alt+Up", Callback: func() {
|
|
|
|
}}, &ui.ItemEntry{Name: "Focus Down", QuickChar: -1, Shortcut: "Alt+Down", Callback: func() {
|
|
|
|
}}, &ui.ItemEntry{Name: "Focus Left", QuickChar: -1, Shortcut: "Alt+Left", Callback: func() {
|
|
|
|
}}, &ui.ItemEntry{Name: "Focus Right", QuickChar: -1, Shortcut: "Alt+Right", Callback: func() {
|
|
|
|
}}, &ui.ItemSeparator{}, &ui.ItemEntry{Name: "Split Top", QuickChar: 6, Callback: func() {
|
|
|
|
}}, &ui.ItemEntry{Name: "Split Bottom", QuickChar: 6, Callback: func() {
|
|
|
|
}}, &ui.ItemEntry{Name: "Split Left", QuickChar: 6, Callback: func() {
|
|
|
|
}}, &ui.ItemEntry{Name: "Split Right", QuickChar: 6, Callback: func() {
|
|
|
|
}}, &ui.ItemSeparator{}, &ui.ItemEntry{Name: "Move", Shortcut: "Ctrl+M", Callback: func() {
|
|
|
|
}}, &ui.ItemEntry{Name: "Resize", Shortcut: "Ctrl+R", Callback: func() {
|
|
|
|
}}, &ui.ItemEntry{Name: "Float", Callback: func() {
|
|
|
|
}}})
|
|
|
|
editMenu := ui.NewMenu("Edit", 0, &theme)
|
|
|
|
editMenu.AddItems([]ui.Item{&ui.ItemEntry{Name: "Cut", Shortcut: "Ctrl+X", Callback: func() {
|
|
if tabContainer.GetTabCount() > 0 {
|
|
tab := tabContainer.GetTab(tabContainer.GetSelectedTabIdx())
|
|
te := tab.Child.(*ui.TextEdit)
|
|
bytes := te.GetSelectedBytes()
|
|
if len(bytes) > 0 { // If something is selected...
|
|
te.Delete(false) // Delete the selection
|
|
// TODO: better error handling within editor
|
|
_ = ClipWrite(string(bytes)) // Add the selectedStr to clipboard
|
|
}
|
|
changeFocus(tabContainer)
|
|
}
|
|
}}, &ui.ItemEntry{Name: "Copy", Shortcut: "Ctrl+C", Callback: func() {
|
|
if tabContainer.GetTabCount() > 0 {
|
|
tab := tabContainer.GetTab(tabContainer.GetSelectedTabIdx())
|
|
te := tab.Child.(*ui.TextEdit)
|
|
bytes := te.GetSelectedBytes()
|
|
if len(bytes) > 0 { // If there is something selected...
|
|
_ = ClipWrite(string(bytes)) // Add selectedStr to clipboard
|
|
}
|
|
changeFocus(tabContainer)
|
|
}
|
|
}}, &ui.ItemEntry{Name: "Paste", Shortcut: "Ctrl+V", Callback: func() {
|
|
if tabContainer.GetTabCount() > 0 {
|
|
tab := tabContainer.GetTab(tabContainer.GetSelectedTabIdx())
|
|
te := tab.Child.(*ui.TextEdit)
|
|
|
|
contents, err := ClipRead()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
te.Insert(contents)
|
|
|
|
changeFocus(tabContainer)
|
|
}
|
|
}}, &ui.ItemSeparator{}, &ui.ItemEntry{Name: "Select All", QuickChar: 7, Shortcut: "Ctrl+A", Callback: func() {
|
|
|
|
}}, &ui.ItemEntry{Name: "Select Line", QuickChar: 7, Callback: func() {
|
|
|
|
}}})
|
|
|
|
searchMenu := ui.NewMenu("Search", 0, &theme)
|
|
|
|
searchMenu.AddItems([]ui.Item{&ui.ItemEntry{Name: "New", Callback: func() {
|
|
s.Beep()
|
|
}}})
|
|
|
|
menuBar.AddMenu(fileMenu)
|
|
menuBar.AddMenu(panelMenu)
|
|
menuBar.AddMenu(editMenu)
|
|
menuBar.AddMenu(searchMenu)
|
|
|
|
changeFocus(tabContainer) // TabContainer is focused by default
|
|
|
|
for !closing {
|
|
s.Clear()
|
|
|
|
// Draw background (grey and black checkerboard)
|
|
ui.DrawRect(s, 0, 0, sizex, sizey, '▚', tcell.Style{}.Foreground(tcell.ColorGrey).Background(tcell.ColorBlack))
|
|
|
|
if tabContainer.GetTabCount() > 0 { // Draw the tab container only if a tab is open
|
|
tabContainer.Draw(s)
|
|
}
|
|
menuBar.Draw(s) // Always draw the menu bar
|
|
|
|
if dialog != nil {
|
|
// Update fileSelector dialog pos and size
|
|
diagMinX, diagMinY := dialog.GetMinSize()
|
|
dialog.SetSize(diagMinX, diagMinY)
|
|
dialog.SetPos(sizex/2-diagMinX/2, sizey/2-diagMinY/2) // Center
|
|
|
|
dialog.Draw(s)
|
|
}
|
|
|
|
// Draw statusbar
|
|
ui.DrawRect(s, 0, sizey-1, sizex, 1, ' ', theme["StatusBar"])
|
|
if tabContainer.GetTabCount() > 0 {
|
|
focusedTab := tabContainer.GetTab(tabContainer.GetSelectedTabIdx())
|
|
te := focusedTab.Child.(*ui.TextEdit)
|
|
|
|
var delim string
|
|
if te.IsCRLF {
|
|
delim = "CRLF"
|
|
} else {
|
|
delim = "LF"
|
|
}
|
|
|
|
line, col := te.GetLineCol()
|
|
|
|
var tabs string
|
|
if te.UseHardTabs {
|
|
tabs = "Tabs: Hard"
|
|
} else {
|
|
tabs = "Tabs: Spaces"
|
|
}
|
|
|
|
str := fmt.Sprintf(" Filetype: %s %d, %d %s %s", "None", line+1, col+1, delim, tabs)
|
|
ui.DrawStr(s, 0, sizey-1, str, theme["StatusBar"])
|
|
}
|
|
|
|
s.Show()
|
|
|
|
switch ev := s.PollEvent().(type) {
|
|
case *tcell.EventResize:
|
|
sizex, sizey = s.Size()
|
|
|
|
menuBar.SetSize(sizex, 1)
|
|
tabContainer.SetSize(sizex, sizey-2)
|
|
|
|
s.Sync() // Redraw everything
|
|
case *tcell.EventKey:
|
|
// On Escape, we change focus between editor and the MenuBar.
|
|
if dialog == nil {
|
|
if ev.Key() == tcell.KeyEscape {
|
|
if focusedComponent == tabContainer {
|
|
changeFocus(menuBar)
|
|
} else {
|
|
changeFocus(tabContainer)
|
|
}
|
|
}
|
|
|
|
if ev.Modifiers() & tcell.ModCtrl != 0 {
|
|
handled := menuBar.HandleEvent(ev)
|
|
if handled {
|
|
continue // Avoid passing the event to the focusedComponent
|
|
}
|
|
}
|
|
}
|
|
|
|
focusedComponent.HandleEvent(ev)
|
|
}
|
|
}
|
|
|
|
if *memprofile != "" {
|
|
f, err := os.Create(*memprofile)
|
|
if err != nil {
|
|
log.Fatal("Could not create memory profile: ", err)
|
|
}
|
|
defer f.Close()
|
|
runtime.GC() // Get updated statistics
|
|
if err := pprof.WriteHeapProfile(f); err != nil {
|
|
log.Fatal("Could not write memory profile: ", err)
|
|
}
|
|
}
|
|
}
|