]> git.lizzy.rs Git - micro.git/blobdiff - cmd/micro/buffer.go
Merge pull request #507 from NicolaiSoeborg/master
[micro.git] / cmd / micro / buffer.go
index 67175de56830e0cbb3c881b389b57365c813ad78..fdcc37ad2d43cf3f8e71773c43d749b174f48165 100644 (file)
@@ -3,13 +3,19 @@ package main
 import (
        "bytes"
        "encoding/gob"
+       "io"
        "io/ioutil"
        "os"
        "os/exec"
        "os/signal"
        "path/filepath"
+       "regexp"
+       "strconv"
+       "strings"
        "time"
        "unicode/utf8"
+
+       "github.com/mitchellh/go-homedir"
 )
 
 // Buffer stores the text for files that are loaded into the text editor
@@ -18,15 +24,19 @@ import (
 type Buffer struct {
        // The eventhandler for undo/redo
        *EventHandler
+       // This stores all the text in the buffer as an array of lines
        *LineArray
 
        Cursor Cursor
 
        // Path to the file on disk
        Path string
+       // Absolute path to the file on disk
+       AbsPath string
        // Name of the buffer on the status line
-       Name string
+       name string
 
+       // Whether or not the buffer has been modified since it was opened
        IsModified bool
 
        // Stores the last modification time of the file the buffer is pointing to
@@ -36,34 +46,57 @@ type Buffer struct {
 
        // Syntax highlighting rules
        rules []SyntaxRule
-       // The buffer's filetype
-       FileType string
+
+       // Buffer local settings
+       Settings map[string]interface{}
 }
 
 // The SerializedBuffer holds the types that get serialized when a buffer is saved
+// These are used for the savecursor and saveundo options
 type SerializedBuffer struct {
        EventHandler *EventHandler
        Cursor       Cursor
        ModTime      time.Time
 }
 
-// NewBuffer creates a new buffer from `txt` with path and name `path`
-func NewBuffer(txt []byte, path string) *Buffer {
-       b := new(Buffer)
-       b.LineArray = NewLineArray(txt)
+func NewBufferFromString(text, path string) *Buffer {
+       return NewBuffer(strings.NewReader(text), path)
+}
 
-       b.Path = path
-       b.Name = path
+// NewBuffer creates a new buffer from a given reader with a given path
+func NewBuffer(reader io.Reader, path string) *Buffer {
+       if path != "" {
+               for _, tab := range tabs {
+                       for _, view := range tab.views {
+                               if view.Buf.Path == path {
+                                       return view.Buf
+                               }
+                       }
+               }
+       }
+
+       b := new(Buffer)
+       b.LineArray = NewLineArray(reader)
 
-       if path == "" {
-               b.Name = "No name"
+       b.Settings = DefaultLocalSettings()
+       for k, v := range globalSettings {
+               if _, ok := b.Settings[k]; ok {
+                       b.Settings[k] = v
+               }
        }
 
+       absPath, _ := filepath.Abs(path)
+
+       b.Path = path
+       b.AbsPath = absPath
+
+       // The last time this file was modified
        b.ModTime, _ = GetModTime(b.Path)
 
        b.EventHandler = NewEventHandler(b)
 
        b.Update()
+       b.FindFileType()
        b.UpdateRules()
 
        if _, err := os.Stat(configDir + "/buffers/"); os.IsNotExist(err) {
@@ -71,17 +104,46 @@ func NewBuffer(txt []byte, path string) *Buffer {
        }
 
        // Put the cursor at the first spot
+       cursorStartX := 0
+       cursorStartY := 0
+       // If -startpos LINE,COL was passed, use start position LINE,COL
+       if len(*flagStartPos) > 0 {
+               positions := strings.Split(*flagStartPos, ",")
+               if len(positions) == 2 {
+                       lineNum, errPos1 := strconv.Atoi(positions[0])
+                       colNum, errPos2 := strconv.Atoi(positions[1])
+                       if errPos1 == nil && errPos2 == nil {
+                               cursorStartX = colNum
+                               cursorStartY = lineNum - 1
+                               // Check to avoid line overflow
+                               if cursorStartY > b.NumLines {
+                                       cursorStartY = b.NumLines - 1
+                               } else if cursorStartY < 0 {
+                                       cursorStartY = 0
+                               }
+                               // Check to avoid column overflow
+                               if cursorStartX > len(b.Line(cursorStartY)) {
+                                       cursorStartX = len(b.Line(cursorStartY))
+                               } else if cursorStartX < 0 {
+                                       cursorStartX = 0
+                               }
+                       }
+               }
+       }
        b.Cursor = Cursor{
                Loc: Loc{
-                       X: 0,
-                       Y: 0,
+                       X: cursorStartX,
+                       Y: cursorStartY,
                },
                buf: b,
        }
 
-       if settings["savecursor"].(bool) || settings["saveundo"].(bool) {
-               absPath, _ := filepath.Abs(b.Path)
-               file, err := os.Open(configDir + "/buffers/" + EscapePath(absPath))
+       InitLocalSettings(b)
+
+       if b.Settings["savecursor"].(bool) || b.Settings["saveundo"].(bool) {
+               // If either savecursor or saveundo is turned on, we need to load the serialized information
+               // from ~/.config/micro/buffers
+               file, err := os.Open(configDir + "/buffers/" + EscapePath(b.AbsPath))
                if err == nil {
                        var buffer SerializedBuffer
                        decoder := gob.NewDecoder(file)
@@ -90,13 +152,13 @@ func NewBuffer(txt []byte, path string) *Buffer {
                        if err != nil {
                                TermMessage(err.Error(), "\n", "You may want to remove the files in ~/.config/micro/buffers (these files store the information for the 'saveundo' and 'savecursor' options) if this problem persists.")
                        }
-                       if settings["savecursor"].(bool) {
+                       if b.Settings["savecursor"].(bool) {
                                b.Cursor = buffer.Cursor
                                b.Cursor.buf = b
                                b.Cursor.Relocate()
                        }
 
-                       if settings["saveundo"].(bool) {
+                       if b.Settings["saveundo"].(bool) {
                                // We should only use last time's eventhandler if the file wasn't by someone else in the meantime
                                if b.ModTime == buffer.ModTime {
                                        b.EventHandler = buffer.EventHandler
@@ -110,10 +172,38 @@ func NewBuffer(txt []byte, path string) *Buffer {
        return b
 }
 
+func (b *Buffer) GetName() string {
+       if b.name == "" {
+               if b.Path == "" {
+                       return "No name"
+               }
+               return b.Path
+       }
+       return b.name
+}
+
 // UpdateRules updates the syntax rules and filetype for this buffer
 // This is called when the colorscheme changes
 func (b *Buffer) UpdateRules() {
-       b.rules, b.FileType = GetRules(b)
+       b.rules = GetRules(b)
+}
+
+// FindFileType identifies this buffer's filetype based on the extension or header
+func (b *Buffer) FindFileType() {
+       b.Settings["filetype"] = FindFileType(b)
+}
+
+// FileType returns the buffer's filetype
+func (b *Buffer) FileType() string {
+       return b.Settings["filetype"].(string)
+}
+
+// IndentString returns a string representing one level of indentation
+func (b *Buffer) IndentString() string {
+       if b.Settings["tabstospaces"].(bool) {
+               return Spaces(int(b.Settings["tabsize"].(float64)))
+       }
+       return "\t"
 }
 
 // CheckModTime makes sure that the file this buffer points to hasn't been updated
@@ -171,9 +261,8 @@ func (b *Buffer) SaveWithSudo() error {
 
 // Serialize serializes the buffer to configDir/buffers
 func (b *Buffer) Serialize() error {
-       if settings["savecursor"].(bool) || settings["saveundo"].(bool) {
-               absPath, _ := filepath.Abs(b.Path)
-               file, err := os.Create(configDir + "/buffers/" + EscapePath(absPath))
+       if b.Settings["savecursor"].(bool) || b.Settings["saveundo"].(bool) {
+               file, err := os.Create(configDir + "/buffers/" + EscapePath(b.AbsPath))
                if err == nil {
                        enc := gob.NewEncoder(file)
                        gob.Register(TextEvent{})
@@ -182,7 +271,6 @@ func (b *Buffer) Serialize() error {
                                b.Cursor,
                                b.ModTime,
                        })
-                       // err = enc.Encode(b.Cursor)
                }
                file.Close()
                return err
@@ -192,24 +280,45 @@ func (b *Buffer) Serialize() error {
 
 // SaveAs saves the buffer to a specified path (filename), creating the file if it does not exist
 func (b *Buffer) SaveAs(filename string) error {
+       b.FindFileType()
        b.UpdateRules()
-       b.Name = filename
-       b.Path = filename
-       data := []byte(b.String())
+       dir, _ := homedir.Dir()
+       b.Path = strings.Replace(filename, "~", dir, 1)
+       if b.Settings["rmtrailingws"].(bool) {
+               r, _ := regexp.Compile(`[ \t]+$`)
+               for lineNum, line := range b.Lines(0, b.NumLines) {
+                       indices := r.FindStringIndex(line)
+                       if indices == nil {
+                               continue
+                       }
+                       startLoc := Loc{indices[0], lineNum}
+                       b.deleteToEnd(startLoc)
+               }
+               b.Cursor.Relocate()
+       }
+       if b.Settings["eofnewline"].(bool) {
+               end := b.End()
+               if b.RuneAt(Loc{end.X - 1, end.Y}) != '\n' {
+                       b.Insert(end, "\n")
+               }
+       }
+       str := b.String()
+       data := []byte(str)
        err := ioutil.WriteFile(filename, data, 0644)
        if err == nil {
                b.IsModified = false
                b.ModTime, _ = GetModTime(filename)
                return b.Serialize()
        }
+       b.ModTime, _ = GetModTime(filename)
        return err
 }
 
 // SaveAsWithSudo is the same as SaveAs except it uses a neat trick
 // with tee to use sudo so the user doesn't have to reopen micro with sudo
 func (b *Buffer) SaveAsWithSudo(filename string) error {
+       b.FindFileType()
        b.UpdateRules()
-       b.Name = filename
        b.Path = filename
 
        // The user may have already used sudo in which case we won't need the password
@@ -266,6 +375,11 @@ func (b *Buffer) remove(start, end Loc) string {
        b.Update()
        return sub
 }
+func (b *Buffer) deleteToEnd(start Loc) {
+       b.IsModified = true
+       b.LineArray.DeleteToEnd(start)
+       b.Update()
+}
 
 // Start returns the location of the first character in the buffer
 func (b *Buffer) Start() Loc {
@@ -277,8 +391,20 @@ func (b *Buffer) End() Loc {
        return Loc{utf8.RuneCount(b.lines[b.NumLines-1]), b.NumLines - 1}
 }
 
+// RuneAt returns the rune at a given location in the buffer
+func (b *Buffer) RuneAt(loc Loc) rune {
+       line := []rune(b.Line(loc.Y))
+       if len(line) > 0 {
+               return line[loc.X]
+       }
+       return '\n'
+}
+
 // Line returns a single line
 func (b *Buffer) Line(n int) string {
+       if n >= len(b.lines) {
+               return ""
+       }
        return string(b.lines[n])
 }
 
@@ -296,3 +422,48 @@ func (b *Buffer) Lines(start, end int) []string {
 func (b *Buffer) Len() int {
        return Count(b.String())
 }
+
+// MoveLinesUp moves the range of lines up one row
+func (b *Buffer) MoveLinesUp(start int, end int) {
+       // 0 < start < end <= len(b.lines)
+       if start < 1 || start >= end || end > len(b.lines) {
+               return // what to do? FIXME
+       }
+       if end == len(b.lines) {
+               b.Insert(
+                       Loc{
+                               utf8.RuneCount(b.lines[end-1]),
+                               end - 1,
+                       },
+                       "\n"+b.Line(start-1),
+               )
+       } else {
+               b.Insert(
+                       Loc{0, end},
+                       b.Line(start-1)+"\n",
+               )
+       }
+       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) {
+       // 0 <= start < end < len(b.lines)
+       // if end == len(b.lines), we can't do anything here because the
+       // last line is unaccessible, FIXME
+       if start < 0 || start >= end || end >= len(b.lines)-1 {
+               return // what to do? FIXME
+       }
+       b.Insert(
+               Loc{0, start},
+               b.Line(end)+"\n",
+       )
+       end++
+       b.Remove(
+               Loc{0, end},
+               Loc{0, end + 1},
+       )
+}