micro/internal/buffer/buffer.go

1057 lines
25 KiB
Go
Raw Normal View History

2018-08-27 22:53:10 +03:00
package buffer
2016-03-18 00:27:57 +03:00
import (
2020-02-09 23:36:31 +03:00
"bufio"
2019-01-15 06:29:24 +03:00
"bytes"
"crypto/md5"
"errors"
"io"
2016-03-18 00:27:57 +03:00
"io/ioutil"
"log"
"os"
"path/filepath"
2019-06-16 01:22:36 +03:00
"strconv"
"strings"
2020-01-29 04:54:14 +03:00
"sync"
"time"
2016-06-07 18:43:28 +03:00
"unicode/utf8"
2016-12-13 16:58:08 +03:00
2019-03-20 01:28:51 +03:00
luar "layeh.com/gopher-luar"
2020-02-10 22:49:08 +03:00
dmp "github.com/sergi/go-diff/diffmatchpatch"
2019-02-04 07:17:24 +03:00
"github.com/zyedidia/micro/internal/config"
2019-03-20 01:28:51 +03:00
ulua "github.com/zyedidia/micro/internal/lua"
2019-02-04 07:17:24 +03:00
"github.com/zyedidia/micro/internal/screen"
"github.com/zyedidia/micro/internal/util"
2019-03-20 01:28:51 +03:00
"github.com/zyedidia/micro/pkg/highlight"
"golang.org/x/text/encoding/htmlindex"
"golang.org/x/text/encoding/unicode"
"golang.org/x/text/transform"
2016-03-18 00:27:57 +03:00
)
const backupTime = 8000
2019-12-22 03:55:23 +03:00
var (
// OpenBuffers is a list of the currently open buffers
OpenBuffers []*Buffer
// LogBuf is a reference to the log buffer which can be opened with the
// `> log` command
LogBuf *Buffer
)
2018-08-26 06:06:44 +03:00
// The BufType defines what kind of buffer this is
type BufType struct {
Kind int
2019-01-19 23:37:59 +03:00
Readonly bool // The buffer cannot be edited
Scratch bool // The buffer cannot be saved
2019-01-01 06:07:01 +03:00
Syntax bool // Syntax highlighting is enabled
2018-08-26 06:06:44 +03:00
}
var (
// BTDefault is a default buffer
2019-01-01 06:07:01 +03:00
BTDefault = BufType{0, false, false, true}
// BTHelp is a help buffer
BTHelp = BufType{1, true, true, true}
// BTLog is a log buffer
BTLog = BufType{2, true, true, false}
// BTScratch is a buffer that cannot be saved (for scratch work)
2019-01-01 06:07:01 +03:00
BTScratch = BufType{3, false, true, false}
// BTRaw is is a buffer that shows raw terminal events
BTRaw = BufType{4, false, true, false}
// BTInfo is a buffer for inputting information
BTInfo = BufType{5, false, true, false}
2019-01-14 05:06:58 +03:00
// ErrFileTooLarge is returned when the file is too large to hash
// (fastdirty is automatically enabled)
2019-01-14 05:06:58 +03:00
ErrFileTooLarge = errors.New("File is too large to hash")
)
// SharedBuffer is a struct containing info that is shared among buffers
// that have the same file open
2019-01-15 00:52:25 +03:00
type SharedBuffer struct {
*LineArray
// Stores the last modification time of the file the buffer is pointing to
ModTime time.Time
// Type of the buffer (e.g. help, raw, scratch etc..)
Type BufType
isModified bool
2019-01-25 02:09:57 +03:00
// Whether or not suggestions can be autocompleted must be shared because
// it changes based on how the buffer has changed
HasSuggestions bool
// Modifications is the list of modified regions for syntax highlighting
Modifications []Loc
2019-01-15 00:52:25 +03:00
}
func (b *SharedBuffer) insert(pos Loc, value []byte) {
b.isModified = true
2019-01-25 02:09:57 +03:00
b.HasSuggestions = false
2019-01-15 00:52:25 +03:00
b.LineArray.insert(pos, value)
// b.Modifications is cleared every screen redraw so it's
// ok to append duplicates
b.Modifications = append(b.Modifications, Loc{pos.Y, pos.Y + bytes.Count(value, []byte{'\n'})})
2019-01-15 00:52:25 +03:00
}
func (b *SharedBuffer) remove(start, end Loc) []byte {
b.isModified = true
2019-01-25 02:09:57 +03:00
b.HasSuggestions = false
b.Modifications = append(b.Modifications, Loc{start.Y, start.Y})
2019-01-15 00:52:25 +03:00
return b.LineArray.remove(start, end)
}
2020-02-08 10:56:24 +03:00
const (
DSUnchanged = 0
DSAdded = 1
DSModified = 2
DSDeletedAbove = 3
)
type DiffStatus byte
2018-12-31 22:46:04 +03:00
// Buffer stores the main information about a currently open file including
// the actual text (in a LineArray), the undo/redo stack (in an EventHandler)
// all the cursors, the syntax highlighting info, the settings for the buffer
// and some misc info about modification time and path location.
// The syntax highlighting info must be stored with the buffer because the syntax
// highlighter attaches information to each line of the buffer for optimization
// purposes so it doesn't have to rehighlight everything on every update.
2016-03-18 00:27:57 +03:00
type Buffer struct {
2018-08-26 06:06:44 +03:00
*EventHandler
2019-01-15 00:52:25 +03:00
*SharedBuffer
2018-08-27 22:53:10 +03:00
cursors []*Cursor
2019-01-03 01:39:50 +03:00
curCursor int
2018-08-27 22:53:10 +03:00
StartCursor Loc
2016-03-18 00:27:57 +03:00
// Path to the file on disk
2016-04-25 19:48:43 +03:00
Path string
// Absolute path to the file on disk
AbsPath string
2016-03-18 00:27:57 +03:00
// Name of the buffer on the status line
2016-11-20 03:07:51 +03:00
name string
2016-03-18 00:27:57 +03:00
// SyntaxDef represents the syntax highlighting definition being used
// This stores the highlighting rules and filetype detection info
SyntaxDef *highlight.Def
// The Highlighter struct actually performs the highlighting
Highlighter *highlight.Highlighter
2020-01-29 04:54:14 +03:00
HighlightLock sync.Mutex
2016-08-25 02:55:44 +03:00
// Hash of the original buffer -- empty if fastdirty is on
origHash [md5.Size]byte
2018-08-26 06:06:44 +03:00
// Settings customized by the user
2016-08-25 02:55:44 +03:00
Settings map[string]interface{}
2016-03-18 00:27:57 +03:00
2019-01-25 02:09:57 +03:00
Suggestions []string
Completions []string
CurSuggestion int
2019-01-14 01:22:11 +03:00
Messages []*Message
2019-12-22 03:55:23 +03:00
2020-02-08 10:56:24 +03:00
updateDiffTimer *time.Timer
diffBase []byte
diffBaseLineCount int
diffLock sync.RWMutex
diff map[int]DiffStatus
2019-12-22 03:55:23 +03:00
// counts the number of edits
// resets every backupTime edits
2019-12-22 03:55:23 +03:00
lastbackup time.Time
2016-05-29 18:02:56 +03:00
}
// NewBufferFromFile opens a new buffer using the given path
// It will also automatically handle `~`, and line/column with filename:l:c
// It will return an empty buffer if the path does not exist
// and an error if the file is a directory
2019-01-01 06:07:01 +03:00
func NewBufferFromFile(path string, btype BufType) (*Buffer, error) {
2018-08-26 06:06:44 +03:00
var err error
filename, cursorPos := util.GetPathAndCursorPosition(path)
filename, err = util.ReplaceHome(filename)
2018-08-26 06:06:44 +03:00
if err != nil {
return nil, err
}
file, err := os.Open(filename)
fileInfo, _ := os.Stat(filename)
if err == nil && fileInfo.IsDir() {
return nil, errors.New("Error: " + filename + " is a directory and cannot be opened")
}
defer file.Close()
2019-06-16 05:23:19 +03:00
cursorLoc, cursorerr := ParseCursorLocation(cursorPos)
if cursorerr != nil {
2019-06-16 01:22:36 +03:00
cursorLoc = Loc{-1, -1}
}
var buf *Buffer
if err != nil {
// File does not exist -- create an empty buffer with that name
2019-01-01 06:07:01 +03:00
buf = NewBufferFromString("", filename, btype)
} else {
buf = NewBuffer(file, util.FSize(file), filename, cursorLoc, btype)
}
return buf, nil
}
// NewBufferFromString creates a new buffer containing the given string
2019-01-01 06:07:01 +03:00
func NewBufferFromString(text, path string, btype BufType) *Buffer {
2019-06-16 01:22:36 +03:00
return NewBuffer(strings.NewReader(text), int64(len(text)), path, Loc{-1, -1}, btype)
}
// NewBuffer creates a new buffer from a given reader with a given path
2018-08-26 06:06:44 +03:00
// Ensure that ReadSettings and InitGlobalSettings have been called before creating
// a new buffer
// Places the cursor at startcursor. If startcursor is -1, -1 places the
// cursor at an autodetected location (based on savecursor or :LINE:COL)
2019-06-16 01:22:36 +03:00
func NewBuffer(r io.Reader, size int64, path string, startcursor Loc, btype BufType) *Buffer {
2019-01-14 05:06:58 +03:00
absPath, _ := filepath.Abs(path)
2016-03-18 00:27:57 +03:00
b := new(Buffer)
2019-08-04 03:12:23 +03:00
b.Settings = config.DefaultCommonSettings()
2018-08-27 22:53:10 +03:00
for k, v := range config.GlobalSettings {
if _, ok := b.Settings[k]; ok {
b.Settings[k] = v
}
2016-08-25 02:55:44 +03:00
}
config.InitLocalSettings(b.Settings, path)
2016-08-25 02:55:44 +03:00
enc, err := htmlindex.Get(b.Settings["encoding"].(string))
if err != nil {
enc = unicode.UTF8
b.Settings["encoding"] = "utf-8"
}
2020-02-09 23:36:31 +03:00
reader := bufio.NewReader(transform.NewReader(r, enc.NewDecoder()))
found := false
2019-01-14 08:57:39 +03:00
if len(path) > 0 {
for _, buf := range OpenBuffers {
2019-06-15 22:50:37 +03:00
if buf.AbsPath == absPath && buf.Type != BTInfo {
2019-01-14 08:57:39 +03:00
found = true
2019-01-15 00:52:25 +03:00
b.SharedBuffer = buf.SharedBuffer
2019-01-14 08:57:39 +03:00
b.EventHandler = buf.EventHandler
}
}
}
2019-12-23 02:05:23 +03:00
b.Path = path
b.AbsPath = absPath
2019-12-22 03:55:23 +03:00
2019-12-23 02:05:23 +03:00
if !found {
2019-01-15 00:52:25 +03:00
b.SharedBuffer = new(SharedBuffer)
b.Type = btype
2019-12-22 03:55:23 +03:00
2019-12-23 02:05:23 +03:00
hasBackup := b.ApplyBackup(size)
2019-12-22 03:55:23 +03:00
2019-12-23 02:05:23 +03:00
if !hasBackup {
2019-12-22 03:55:23 +03:00
b.LineArray = NewLineArray(uint64(size), FFAuto, reader)
}
2019-01-15 00:52:25 +03:00
b.EventHandler = NewEventHandler(b.SharedBuffer, b.cursors)
}
if b.Settings["readonly"].(bool) && b.Type == BTDefault {
2019-06-18 00:45:38 +03:00
b.Type.Readonly = true
}
// The last time this file was modified
b.UpdateModTime()
2019-01-15 07:24:49 +03:00
switch b.Endings {
case FFUnix:
b.Settings["fileformat"] = "unix"
case FFDos:
b.Settings["fileformat"] = "dos"
}
2016-03-26 17:54:18 +03:00
b.UpdateRules()
config.InitLocalSettings(b.Settings, b.Path)
2018-08-27 22:53:10 +03:00
2020-02-11 21:09:17 +03:00
if _, err := os.Stat(filepath.Join(config.ConfigDir, "buffers")); os.IsNotExist(err) {
os.Mkdir(filepath.Join(config.ConfigDir, "buffers"), os.ModePerm)
2018-08-27 22:53:10 +03:00
}
2019-06-16 01:22:36 +03:00
if startcursor.X != -1 && startcursor.Y != -1 {
b.StartCursor = startcursor
} else {
if b.Settings["savecursor"].(bool) || b.Settings["saveundo"].(bool) {
err := b.Unserialize()
if err != nil {
screen.TermMessage(err)
}
2018-08-27 22:53:10 +03:00
}
}
b.AddCursor(NewCursor(b, b.StartCursor))
2019-06-16 01:22:36 +03:00
b.GetActiveCursor().Relocate()
if !b.Settings["fastdirty"].(bool) {
if size > LargeFileThreshold {
2018-08-26 06:06:44 +03:00
// If the file is larger than LargeFileThreshold fastdirty needs to be on
b.Settings["fastdirty"] = true
} else {
calcHash(b, &b.origHash)
}
}
2019-03-20 01:28:51 +03:00
err = config.RunPluginFn("onBufferOpen", luar.New(ulua.L, b))
if err != nil {
screen.TermMessage(err)
}
b.Modifications = make([]Loc, 0, 10)
2019-01-14 05:06:58 +03:00
OpenBuffers = append(OpenBuffers, b)
2016-03-18 00:27:57 +03:00
return b
}
2019-01-14 05:06:58 +03:00
// Close removes this buffer from the list of open buffers
func (b *Buffer) Close() {
for i, buf := range OpenBuffers {
if b == buf {
2019-06-15 22:50:37 +03:00
b.Fini()
2019-01-14 05:06:58 +03:00
copy(OpenBuffers[i:], OpenBuffers[i+1:])
OpenBuffers[len(OpenBuffers)-1] = nil
OpenBuffers = OpenBuffers[:len(OpenBuffers)-1]
return
}
}
}
2019-06-15 22:50:37 +03:00
// Fini should be called when a buffer is closed and performs
// some cleanup
func (b *Buffer) Fini() {
if !b.Modified() {
b.Serialize()
}
2019-12-22 03:55:23 +03:00
b.RemoveBackup()
2019-06-15 22:50:37 +03:00
}
// GetName returns the name that should be displayed in the statusline
// for this buffer
2016-11-20 03:07:51 +03:00
func (b *Buffer) GetName() string {
if b.name == "" {
if b.Path == "" {
return "No name"
}
2016-11-20 03:07:51 +03:00
return b.Path
}
return b.name
}
2019-01-15 00:52:25 +03:00
//SetName changes the name for this buffer
func (b *Buffer) SetName(s string) {
b.name = s
}
// Insert inserts the given string of text at the start location
func (b *Buffer) Insert(start Loc, text string) {
2019-01-19 23:37:59 +03:00
if !b.Type.Readonly {
b.EventHandler.cursors = b.cursors
b.EventHandler.active = b.curCursor
b.EventHandler.Insert(start, text)
2019-12-22 03:55:23 +03:00
2019-12-22 07:26:53 +03:00
go b.Backup(true)
2019-01-19 23:37:59 +03:00
}
2019-01-15 00:56:10 +03:00
}
// Remove removes the characters between the start and end locations
2019-01-15 00:56:10 +03:00
func (b *Buffer) Remove(start, end Loc) {
2019-01-19 23:37:59 +03:00
if !b.Type.Readonly {
b.EventHandler.cursors = b.cursors
b.EventHandler.active = b.curCursor
b.EventHandler.Remove(start, end)
2019-12-22 03:55:23 +03:00
2019-12-22 07:26:53 +03:00
go b.Backup(true)
2019-01-19 23:37:59 +03:00
}
2019-01-15 00:56:10 +03:00
}
// ClearModifications clears the list of modified lines in this buffer
// The list of modified lines is used for syntax highlighting so that
// we can selectively highlight only the necessary lines
// This function should be called every time this buffer is drawn to
// the screen
func (b *Buffer) ClearModifications() {
// clear slice without resetting the cap
b.Modifications = b.Modifications[:0]
}
// FileType returns the buffer's filetype
func (b *Buffer) FileType() string {
return b.Settings["filetype"].(string)
2016-03-26 17:54:18 +03:00
}
// ExternallyModified returns whether the file being edited has
// been modified by some external process
func (b *Buffer) ExternallyModified() bool {
modTime, err := util.GetModTime(b.Path)
if err == nil {
return modTime != b.ModTime
}
return false
}
// UpdateModTime updates the modtime of this file
func (b *Buffer) UpdateModTime() (err error) {
b.ModTime, err = util.GetModTime(b.Path)
return
}
2016-05-31 00:48:33 +03:00
// ReOpen reloads the current buffer from disk
2018-08-26 06:06:44 +03:00
func (b *Buffer) ReOpen() error {
file, err := os.Open(b.Path)
if err != nil {
return err
}
enc, err := htmlindex.Get(b.Settings["encoding"].(string))
if err != nil {
return err
}
2020-02-09 23:36:31 +03:00
reader := bufio.NewReader(transform.NewReader(file, enc.NewDecoder()))
data, err := ioutil.ReadAll(reader)
2016-05-31 00:48:33 +03:00
txt := string(data)
if err != nil {
2018-08-26 06:06:44 +03:00
return err
2016-05-31 00:48:33 +03:00
}
b.EventHandler.ApplyDiff(txt)
2016-05-31 00:48:33 +03:00
err = b.UpdateModTime()
2019-01-15 00:52:25 +03:00
b.isModified = false
2019-01-25 02:25:59 +03:00
b.RelocateCursors()
return err
}
// RelocateCursors relocates all cursors (makes sure they are in the buffer)
2019-01-25 02:25:59 +03:00
func (b *Buffer) RelocateCursors() {
2019-01-03 01:39:50 +03:00
for _, c := range b.cursors {
c.Relocate()
}
2016-03-18 00:27:57 +03:00
}
2018-08-26 06:06:44 +03:00
// RuneAt returns the rune at a given location in the buffer
func (b *Buffer) RuneAt(loc Loc) rune {
line := b.LineBytes(loc.Y)
if len(line) > 0 {
i := 0
for len(line) > 0 {
r, size := utf8.DecodeRune(line)
line = line[size:]
i++
if i == loc.X {
return r
}
}
2016-06-07 18:43:28 +03:00
}
2018-08-26 06:06:44 +03:00
return '\n'
2016-05-07 17:57:40 +03:00
}
2018-08-26 06:06:44 +03:00
// Modified returns if this buffer has been modified since
// being opened
func (b *Buffer) Modified() bool {
2019-01-19 23:37:59 +03:00
if b.Type.Scratch {
return false
}
2018-08-26 06:06:44 +03:00
if b.Settings["fastdirty"].(bool) {
2019-01-15 00:52:25 +03:00
return b.isModified
}
2018-08-26 06:06:44 +03:00
var buff [md5.Size]byte
2018-08-26 06:06:44 +03:00
calcHash(b, &buff)
return buff != b.origHash
}
2018-08-26 06:06:44 +03:00
// calcHash calculates md5 hash of all lines in the buffer
2019-01-14 05:06:58 +03:00
func calcHash(b *Buffer, out *[md5.Size]byte) error {
2018-08-26 06:06:44 +03:00
h := md5.New()
2019-01-14 05:06:58 +03:00
size := 0
2018-08-26 06:06:44 +03:00
if len(b.lines) > 0 {
2019-01-14 05:06:58 +03:00
n, e := h.Write(b.lines[0].data)
if e != nil {
return e
}
size += n
2018-08-26 06:06:44 +03:00
for _, l := range b.lines[1:] {
2019-01-14 05:06:58 +03:00
n, e = h.Write([]byte{'\n'})
if e != nil {
return e
}
size += n
n, e = h.Write(l.data)
if e != nil {
return e
}
size += n
2018-08-26 06:06:44 +03:00
}
}
2019-01-14 05:06:58 +03:00
if size > LargeFileThreshold {
return ErrFileTooLarge
}
2018-08-26 06:06:44 +03:00
h.Sum((*out)[:0])
2019-01-14 05:06:58 +03:00
return nil
}
2018-08-26 06:06:44 +03:00
// UpdateRules updates the syntax rules and filetype for this buffer
// This is called when the colorscheme changes
func (b *Buffer) UpdateRules() {
2019-01-01 06:07:01 +03:00
if !b.Type.Syntax {
return
}
ft := b.Settings["filetype"].(string)
if ft == "off" {
return
}
syntaxFile := ""
var header *highlight.Header
for _, f := range config.ListRuntimeFiles(config.RTSyntaxHeader) {
2018-08-26 06:06:44 +03:00
data, err := f.Data()
if err != nil {
screen.TermMessage("Error loading syntax header file " + f.Name() + ": " + err.Error())
continue
}
header, err = highlight.MakeHeader(data)
if err != nil {
screen.TermMessage("Error reading syntax header file", f.Name(), err)
continue
}
if ft == "unknown" || ft == "" {
if highlight.MatchFiletype(header.FtDetect, b.Path, b.lines[0].data) {
syntaxFile = f.Name()
break
}
} else if header.FileType == ft {
syntaxFile = f.Name()
break
}
}
if syntaxFile == "" {
// search for the syntax file in the user's custom syntax files
for _, f := range config.ListRealRuntimeFiles(config.RTSyntax) {
log.Println("real runtime file", f.Name())
data, err := f.Data()
2018-08-26 06:06:44 +03:00
if err != nil {
2019-01-14 05:06:58 +03:00
screen.TermMessage("Error loading syntax file " + f.Name() + ": " + err.Error())
2018-08-26 06:06:44 +03:00
continue
}
header, err = highlight.MakeHeaderYaml(data)
file, err := highlight.ParseFile(data)
2018-08-26 06:06:44 +03:00
if err != nil {
screen.TermMessage("Error parsing syntax file " + f.Name() + ": " + err.Error())
2018-08-26 06:06:44 +03:00
continue
}
if ((ft == "unknown" || ft == "") && highlight.MatchFiletype(header.FtDetect, b.Path, b.lines[0].data)) || header.FileType == ft {
syndef, err := highlight.ParseDef(file, header)
if err != nil {
screen.TermMessage("Error parsing syntax file " + f.Name() + ": " + err.Error())
continue
}
b.SyntaxDef = syndef
break
}
}
} else {
for _, f := range config.ListRuntimeFiles(config.RTSyntax) {
if f.Name() == syntaxFile {
data, err := f.Data()
if err != nil {
screen.TermMessage("Error loading syntax file " + f.Name() + ": " + err.Error())
continue
}
file, err := highlight.ParseFile(data)
if err != nil {
screen.TermMessage("Error parsing syntax file " + f.Name() + ": " + err.Error())
continue
}
syndef, err := highlight.ParseDef(file, header)
if err != nil {
screen.TermMessage("Error parsing syntax file " + f.Name() + ": " + err.Error())
continue
}
b.SyntaxDef = syndef
break
}
2018-08-26 06:06:44 +03:00
}
}
2019-12-29 05:57:03 +03:00
if b.SyntaxDef != nil && highlight.HasIncludes(b.SyntaxDef) {
includes := highlight.GetIncludes(b.SyntaxDef)
var files []*highlight.File
for _, f := range config.ListRuntimeFiles(config.RTSyntax) {
data, err := f.Data()
if err != nil {
screen.TermMessage("Error parsing syntax file " + f.Name() + ": " + err.Error())
continue
}
header, err := highlight.MakeHeaderYaml(data)
if err != nil {
screen.TermMessage("Error parsing syntax file " + f.Name() + ": " + err.Error())
continue
}
2019-12-29 05:57:03 +03:00
for _, i := range includes {
if header.FileType == i {
file, err := highlight.ParseFile(data)
if err != nil {
screen.TermMessage("Error parsing syntax file " + f.Name() + ": " + err.Error())
continue
}
2019-12-29 05:57:03 +03:00
files = append(files, file)
break
}
}
if len(files) >= len(includes) {
break
}
}
highlight.ResolveIncludes(b.SyntaxDef, files)
}
2018-08-26 06:06:44 +03:00
if b.Highlighter == nil || syntaxFile != "" {
2020-01-29 04:54:14 +03:00
if b.SyntaxDef != nil {
b.Settings["filetype"] = b.SyntaxDef.FileType
}
} else {
b.SyntaxDef = &highlight.EmptyDef
}
if b.SyntaxDef != nil {
b.Highlighter = highlight.NewHighlighter(b.SyntaxDef)
if b.Settings["syntax"].(bool) {
2020-01-29 04:54:14 +03:00
go func() {
b.Highlighter.HighlightStates(b)
b.Highlighter.HighlightMatches(b, 0, b.End().Y)
screen.Redraw()
2020-01-29 04:54:14 +03:00
}()
}
}
}
2018-09-03 23:54:56 +03:00
2019-01-14 05:06:58 +03:00
// ClearMatches clears all of the syntax highlighting for the buffer
func (b *Buffer) ClearMatches() {
for i := range b.lines {
b.SetMatch(i, nil)
b.SetState(i, nil)
}
}
2018-12-31 22:46:04 +03:00
// IndentString returns this buffer's indent method (a tabstop or n spaces
// depending on the settings)
func (b *Buffer) IndentString(tabsize int) string {
2018-09-03 23:54:56 +03:00
if b.Settings["tabstospaces"].(bool) {
return util.Spaces(tabsize)
2018-09-03 23:54:56 +03:00
}
return "\t"
2018-09-03 23:54:56 +03:00
}
2019-01-03 01:39:50 +03:00
2019-01-03 07:26:40 +03:00
// SetCursors resets this buffer's cursors to a new list
func (b *Buffer) SetCursors(c []*Cursor) {
b.cursors = c
2019-01-15 08:24:53 +03:00
b.EventHandler.cursors = b.cursors
2019-01-17 01:52:30 +03:00
b.EventHandler.active = b.curCursor
2019-01-03 07:26:40 +03:00
}
// AddCursor adds a new cursor to the list
func (b *Buffer) AddCursor(c *Cursor) {
b.cursors = append(b.cursors, c)
2019-01-15 08:24:53 +03:00
b.EventHandler.cursors = b.cursors
2019-01-17 01:52:30 +03:00
b.EventHandler.active = b.curCursor
2019-01-03 07:26:40 +03:00
b.UpdateCursors()
}
// SetCurCursor sets the current cursor
func (b *Buffer) SetCurCursor(n int) {
b.curCursor = n
}
// GetActiveCursor returns the main cursor in this buffer
func (b *Buffer) GetActiveCursor() *Cursor {
return b.cursors[b.curCursor]
}
// GetCursor returns the nth cursor
func (b *Buffer) GetCursor(n int) *Cursor {
return b.cursors[n]
}
// GetCursors returns the list of cursors in this buffer
func (b *Buffer) GetCursors() []*Cursor {
return b.cursors
}
// NumCursors returns the number of cursors
func (b *Buffer) NumCursors() int {
return len(b.cursors)
}
2019-01-03 01:39:50 +03:00
// MergeCursors merges any cursors that are at the same position
// into one cursor
func (b *Buffer) MergeCursors() {
var cursors []*Cursor
for i := 0; i < len(b.cursors); i++ {
c1 := b.cursors[i]
if c1 != nil {
for j := 0; j < len(b.cursors); j++ {
c2 := b.cursors[j]
if c2 != nil && i != j && c1.Loc == c2.Loc {
b.cursors[j] = nil
}
}
cursors = append(cursors, c1)
}
}
b.cursors = cursors
for i := range b.cursors {
b.cursors[i].Num = i
}
if b.curCursor >= len(b.cursors) {
b.curCursor = len(b.cursors) - 1
}
2019-01-15 08:24:53 +03:00
b.EventHandler.cursors = b.cursors
2019-01-17 01:52:30 +03:00
b.EventHandler.active = b.curCursor
2019-01-03 01:39:50 +03:00
}
// UpdateCursors updates all the cursors indicies
func (b *Buffer) UpdateCursors() {
2019-01-17 01:52:30 +03:00
b.EventHandler.cursors = b.cursors
b.EventHandler.active = b.curCursor
2019-01-03 01:39:50 +03:00
for i, c := range b.cursors {
c.Num = i
}
}
2019-01-03 07:26:40 +03:00
func (b *Buffer) RemoveCursor(i int) {
copy(b.cursors[i:], b.cursors[i+1:])
b.cursors[len(b.cursors)-1] = nil
b.cursors = b.cursors[:len(b.cursors)-1]
b.curCursor = util.Clamp(b.curCursor, 0, len(b.cursors)-1)
2019-01-17 01:52:30 +03:00
b.UpdateCursors()
2019-01-03 07:26:40 +03:00
}
2019-01-03 01:39:50 +03:00
// ClearCursors removes all extra cursors
func (b *Buffer) ClearCursors() {
for i := 1; i < len(b.cursors); i++ {
b.cursors[i] = nil
}
b.cursors = b.cursors[:1]
b.UpdateCursors()
b.curCursor = 0
b.GetActiveCursor().ResetSelection()
}
2019-01-15 06:38:59 +03:00
// MoveLinesUp moves the range of lines up one row
func (b *Buffer) MoveLinesUp(start int, end int) {
if start < 1 || start >= end || end > len(b.lines) {
return
}
l := string(b.LineBytes(start - 1))
2019-01-15 06:38:59 +03:00
if end == len(b.lines) {
b.Insert(
Loc{
utf8.RuneCount(b.lines[end-1].data),
end - 1,
},
"\n"+l,
2019-01-15 06:38:59 +03:00
)
} else {
b.Insert(
Loc{0, end},
l+"\n",
2019-01-15 06:38:59 +03:00
)
}
b.Remove(
Loc{0, start - 1},
Loc{0, start},
)
}
// MoveLinesDown moves the range of lines down one row
func (b *Buffer) MoveLinesDown(start int, end int) {
if start < 0 || start >= end || end >= len(b.lines)-1 {
return
}
l := string(b.LineBytes(end))
2019-01-15 06:38:59 +03:00
b.Insert(
Loc{0, start},
l+"\n",
2019-01-15 06:38:59 +03:00
)
end++
b.Remove(
Loc{0, end},
Loc{0, end + 1},
)
}
var BracePairs = [][2]rune{
{'(', ')'},
{'{', '}'},
{'[', ']'},
}
// FindMatchingBrace returns the location in the buffer of the matching bracket
// It is given a brace type containing the open and closing character, (for example
// '{' and '}') as well as the location to match from
// TODO: maybe can be more efficient with utf8 package
2019-08-05 01:11:09 +03:00
// returns the location of the matching brace
// if the boolean returned is true then the original matching brace is one character left
// of the starting location
func (b *Buffer) FindMatchingBrace(braceType [2]rune, start Loc) (Loc, bool) {
2019-01-15 06:38:59 +03:00
curLine := []rune(string(b.LineBytes(start.Y)))
2019-08-05 01:11:09 +03:00
startChar := ' '
if start.X >= 0 && start.X < len(curLine) {
startChar = curLine[start.X]
}
leftChar := ' '
if start.X-1 >= 0 && start.X-1 < len(curLine) {
leftChar = curLine[start.X-1]
}
2019-01-15 06:38:59 +03:00
var i int
2019-08-05 01:11:09 +03:00
if startChar == braceType[0] || leftChar == braceType[0] {
2019-01-15 06:38:59 +03:00
for y := start.Y; y < b.LinesNum(); y++ {
l := []rune(string(b.LineBytes(y)))
xInit := 0
if y == start.Y {
2019-08-05 01:11:09 +03:00
if startChar == braceType[0] {
xInit = start.X
} else {
xInit = start.X - 1
}
2019-01-15 06:38:59 +03:00
}
for x := xInit; x < len(l); x++ {
r := l[x]
if r == braceType[0] {
i++
} else if r == braceType[1] {
i--
if i == 0 {
2019-08-05 01:11:09 +03:00
if startChar == braceType[0] {
return Loc{x, y}, false
}
return Loc{x, y}, true
2019-01-15 06:38:59 +03:00
}
}
}
}
2019-08-05 01:11:09 +03:00
} else if startChar == braceType[1] || leftChar == braceType[1] {
2019-01-15 06:38:59 +03:00
for y := start.Y; y >= 0; y-- {
l := []rune(string(b.lines[y].data))
xInit := len(l) - 1
if y == start.Y {
2019-08-05 01:11:09 +03:00
if leftChar == braceType[1] {
xInit = start.X - 1
} else {
xInit = start.X
}
2019-01-15 06:38:59 +03:00
}
for x := xInit; x >= 0; x-- {
r := l[x]
if r == braceType[0] {
i--
if i == 0 {
2019-08-05 01:11:09 +03:00
if leftChar == braceType[1] {
return Loc{x, y}, true
}
return Loc{x, y}, false
2019-01-15 06:38:59 +03:00
}
} else if r == braceType[1] {
i++
}
}
}
}
2019-08-05 01:11:09 +03:00
return start, true
2019-01-15 06:38:59 +03:00
}
2019-01-15 06:44:06 +03:00
// Retab changes all tabs to spaces or vice versa
func (b *Buffer) Retab() {
toSpaces := b.Settings["tabstospaces"].(bool)
tabsize := util.IntOpt(b.Settings["tabsize"])
2019-01-15 06:44:06 +03:00
dirty := false
for i := 0; i < b.LinesNum(); i++ {
l := b.LineBytes(i)
ws := util.GetLeadingWhitespace(l)
2019-01-15 06:44:06 +03:00
if len(ws) != 0 {
if toSpaces {
ws = bytes.Replace(ws, []byte{'\t'}, bytes.Repeat([]byte{' '}, tabsize), -1)
} else {
ws = bytes.Replace(ws, bytes.Repeat([]byte{' '}, tabsize), []byte{'\t'}, -1)
}
}
l = bytes.TrimLeft(l, " \t")
b.lines[i].data = append(ws, l...)
2020-02-10 22:49:08 +03:00
b.Modifications = append(b.Modifications, Loc{i, i})
2019-01-15 06:44:06 +03:00
dirty = true
}
b.isModified = dirty
}
2019-06-16 01:22:36 +03:00
// ParseCursorLocation turns a cursor location like 10:5 (LINE:COL)
// into a loc
func ParseCursorLocation(cursorPositions []string) (Loc, error) {
startpos := Loc{0, 0}
var err error
// if no positions are available exit early
if cursorPositions == nil {
return startpos, errors.New("No cursor positions were provided.")
}
startpos.Y, err = strconv.Atoi(cursorPositions[0])
startpos.Y--
2019-06-16 01:22:36 +03:00
if err == nil {
if len(cursorPositions) > 1 {
startpos.X, err = strconv.Atoi(cursorPositions[1])
if startpos.X > 0 {
startpos.X--
2019-06-16 01:22:36 +03:00
}
}
}
return startpos, err
}
2019-08-03 00:48:59 +03:00
// Line returns the string representation of the given line number
2019-08-03 00:48:59 +03:00
func (b *Buffer) Line(i int) string {
return string(b.LineBytes(i))
}
2019-08-06 06:43:34 +03:00
func (b *Buffer) Write(bytes []byte) (n int, err error) {
b.EventHandler.InsertBytes(b.End(), bytes)
return len(bytes), nil
}
2020-02-08 10:56:24 +03:00
func (b *Buffer) updateDiffSync() {
b.diffLock.Lock()
defer b.diffLock.Unlock()
b.diff = make(map[int]DiffStatus)
if b.diffBase == nil {
return
}
differ := dmp.New()
baseRunes, bufferRunes, _ := differ.DiffLinesToRunes(string(b.diffBase), string(b.Bytes()))
diffs := differ.DiffMainRunes(baseRunes, bufferRunes, false)
lineN := 0
for _, diff := range diffs {
lineCount := len([]rune(diff.Text))
switch diff.Type {
case dmp.DiffEqual:
lineN += lineCount
case dmp.DiffInsert:
var status DiffStatus
if b.diff[lineN] == DSDeletedAbove {
status = DSModified
} else {
status = DSAdded
}
for i := 0; i < lineCount; i++ {
b.diff[lineN] = status
lineN++
}
case dmp.DiffDelete:
b.diff[lineN] = DSDeletedAbove
}
}
}
// UpdateDiff computes the diff between the diff base and the buffer content.
// The update may be performed synchronously or asynchronously.
// UpdateDiff calls the supplied callback when the update is complete.
// The argument passed to the callback is set to true if and only if
// the update was performed synchronously.
// If an asynchronous update is already pending when UpdateDiff is called,
// UpdateDiff does not schedule another update, in which case the callback
// is not called.
func (b *Buffer) UpdateDiff(callback func(bool)) {
if b.updateDiffTimer != nil {
return
}
lineCount := b.LinesNum()
if b.diffBaseLineCount > lineCount {
lineCount = b.diffBaseLineCount
}
if lineCount < 1000 {
b.updateDiffSync()
callback(true)
} else if lineCount < 30000 {
b.updateDiffTimer = time.AfterFunc(500*time.Millisecond, func() {
b.updateDiffTimer = nil
b.updateDiffSync()
callback(false)
})
} else {
// Don't compute diffs for very large files
b.diffLock.Lock()
b.diff = make(map[int]DiffStatus)
b.diffLock.Unlock()
callback(true)
}
}
// SetDiffBase sets the text that is used as the base for diffing the buffer content
func (b *Buffer) SetDiffBase(diffBase []byte) {
b.diffBase = diffBase
if diffBase == nil {
b.diffBaseLineCount = 0
} else {
b.diffBaseLineCount = strings.Count(string(diffBase), "\n")
}
b.UpdateDiff(func(synchronous bool) {
screen.Redraw()
2020-02-08 10:56:24 +03:00
})
}
// DiffStatus returns the diff status for a line in the buffer
func (b *Buffer) DiffStatus(lineN int) DiffStatus {
b.diffLock.RLock()
defer b.diffLock.RUnlock()
// Note that the zero value for DiffStatus is equal to DSUnchanged
return b.diff[lineN]
}
// WriteLog writes a string to the log buffer
2019-08-06 06:43:34 +03:00
func WriteLog(s string) {
LogBuf.EventHandler.Insert(LogBuf.End(), s)
}
// GetLogBuf returns the log buffer
func GetLogBuf() *Buffer {
return LogBuf
}