package main

import (
	"github.com/vinzmay/go-rope"
	"io/ioutil"
	"regexp"
	"strings"
)

// Buffer stores the text for files that are loaded into the text editor
// It uses a rope to efficiently store the string and contains some
// simple functions for saving and wrapper functions for modifying the rope
type Buffer struct {
	// Stores the text of the buffer
	r *rope.Rope

	// Path to the file on disk
	path string
	// Name of the buffer on the status line
	name string

	// This is the text stored every time the buffer is saved to check if the buffer is modified
	savedText string

	// Provide efficient and easy access to text and lines so the rope String does not
	// need to be constantly recalculated
	// These variables are updated in the update() function
	text  string
	lines []string

	// Syntax highlighting rules
	rules []SyntaxRule
	// File type of the buffer
	filetype string
}

// NewBuffer creates a new buffer from `txt` with path and name `path`
func NewBuffer(txt, path string) *Buffer {
	b := new(Buffer)
	if txt == "" {
		b.r = new(rope.Rope)
	} else {
		b.r = rope.New(txt)
	}
	b.path = path
	b.name = path
	b.savedText = txt

	b.Update()
	b.UpdateRules()

	return b
}

// 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)
}

// Update fetches the string from the rope and updates the `text` and `lines` in the buffer
func (b *Buffer) Update() {
	if b.r.Len() == 0 {
		b.text = ""
	} else {
		b.text = b.r.String()
	}
	b.lines = strings.Split(b.text, "\n")
}

// Save saves the buffer to its default path
func (b *Buffer) Save() error {
	return b.SaveAs(b.path)
}

// 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.UpdateRules()
	err := ioutil.WriteFile(filename, []byte(b.text), 0644)
	if err == nil {
		b.savedText = b.text
	}
	return err
}

// IsDirty returns whether or not the buffer has been modified compared to the one on disk
func (b *Buffer) IsDirty() bool {
	return b.savedText != b.text
}

// Insert a string into the rope
func (b *Buffer) Insert(idx int, value string) {
	b.r = b.r.Insert(idx, value)
	b.Update()
}

// Remove a slice of the rope from start to end (exclusive)
// Returns the string that was removed
func (b *Buffer) Remove(start, end int) string {
	if start < 0 {
		start = 0
	}
	if end > b.Len() {
		end = b.Len()
	}
	removed := b.text[start:end]
	// The rope implenentation I am using wants indicies starting at 1 instead of 0
	start++
	end++
	b.r = b.r.Delete(start, end-start)
	b.Update()
	return removed
}

func (b *Buffer) Replace(search, replace string) error {
	re, err := regexp.Compile(search)
	if err != nil {
		return err
	}
	text := re.ReplaceAllString(b.text, replace)
	if text == "" {
		b.r = new(rope.Rope)
	} else {
		b.r = rope.New(text)
	}
	b.Update()
	return nil
}

// Len gives the length of the buffer
func (b *Buffer) Len() int {
	return b.r.Len()
}