openedFiles.go

package langd

import (
	"fmt"

	"github.com/object88/rope"
)

// OpenedFiles is a collection of files opened across the caravan
type OpenedFiles struct {
	ropes map[string]*rope.Rope
}

// NewOpenedFiles creates a new OpenedFiles instance
func NewOpenedFiles() *OpenedFiles {
	return &OpenedFiles{
		ropes: map[string]*rope.Rope{},
	}
}

// EnsureOpened will create a new rope for a file that's not previously opened
func (of *OpenedFiles) EnsureOpened(absFilepath, text string) error {
	if _, ok := of.ropes[absFilepath]; ok {
		return fmt.Errorf("File %s is already opened", absFilepath)
	}
	of.ropes[absFilepath] = rope.CreateRope(text)
	return nil
}

// Close will remove a rope from the collection
func (of *OpenedFiles) Close(absFilepath string) error {
	_, ok := of.ropes[absFilepath]
	if !ok {
		return fmt.Errorf("openedFiles.Close:: File %s is not opened", absFilepath)
	}

	delete(of.ropes, absFilepath)
	return nil
}

// Get returns the rope for a file
func (of *OpenedFiles) Get(absFilepath string) (*rope.Rope, error) {
	buf, ok := of.ropes[absFilepath]
	if !ok {
		return nil, fmt.Errorf("openedFiles.Get:: File %s is not opened", absFilepath)
	}

	return buf, nil
}

// Replace will replace an existing rope with a completely new rope based on
// the provided text
func (of *OpenedFiles) Replace(absFilepath, text string) error {
	_, ok := of.ropes[absFilepath]
	if !ok {
		return fmt.Errorf("File %s is not opened", absFilepath)
	}

	// Replace the entire document
	buf := rope.CreateRope(text)
	of.ropes[absFilepath] = buf

	return nil
}