github.com/xyproto/orbiton/v2@v2.65.12-0.20240516144430-e10a419274ec/portal.go

package main

import (
	"bytes"
	"errors"
	"fmt"
	"os"
	"path/filepath"
	"strconv"
	"strings"
	"time"

	"github.com/xyproto/env/v2"
	"github.com/xyproto/files"
)

var portalFilename = env.ExpandUser(filepath.Join(tempDir, env.Str("LOGNAME", "o")+"_portal.txt"))

var errPortalTimedOut = errors.New("portal timed out")

// Portal is a filename and a line number, for pulling text from
type Portal struct {
	timestamp   time.Time
	absFilename string
	lineNumber  LineNumber
}

// NewPortal returns a new portal to this filename and line number,
// but does not save the new portal. Use the Save() method for that.
func (e *Editor) NewPortal() (*Portal, error) {
	absFilename, err := e.AbsFilename()
	if err != nil {
		return nil, err
	}
	return &Portal{time.Now(), absFilename, e.LineNumber()}, nil
}

// SameFile checks if the portal exist in the same file as the editor is editing
func (p *Portal) SameFile(e *Editor) bool {
	absFilename, err := e.AbsFilename()
	if err != nil {
		return false
	}
	return absFilename == p.absFilename
}

// MoveDown is useful when using portals within the same file.
func (p *Portal) MoveDown() {
	// PopLine handles overflows.
	p.lineNumber++
}

// ClosePortal will clear the portal by removing the portal file
func (e *Editor) ClosePortal() error {
	e.sameFilePortal = nil
	return os.Remove(portalFilename)
}

// ClearPortal will clear the portal by removing the portal file
func ClearPortal() error {
	return os.Remove(portalFilename)
}

// HasPortal checks if a portal is currently active
func HasPortal() bool {
	return files.Exists(portalFilename)
}

// LoadPortal will load a filename + line number from the portal.txt file
func LoadPortal(maxPortalAge time.Duration) (*Portal, error) {
	//logf("Loading %s\n", portalFilename)
	data, err := os.ReadFile(portalFilename)
	if err != nil {
		return nil, err
	}
	if !bytes.Contains(data, []byte{'\n'}) {
		return nil, errors.New(portalFilename + " does not have a newline, it's not a portal file")
	}
	lines := strings.Split(strings.TrimSpace(string(data)), "\n")
	lineCounter := 0
	timestamp := time.Now()
	switch len(lines) {
	case 3: // optional timestamp on the first line, if there are 3 lines
		timestampInt, err := strconv.ParseInt(lines[lineCounter], 10, 64)
		if err != nil {
			return nil, err
		}
		lineCounter++
		timestamp = time.Unix(timestampInt, 0)
		fallthrough
	case 2: // 2 lines without a timestamp, use time.Now() from above
		absFilename, err := filepath.Abs(lines[lineCounter])
		if err != nil {
			return nil, err
		}
		lineCounter++
		lineInt, err := strconv.Atoi(lines[lineCounter])
		if err != nil {
			return nil, err
		}
		lineNumber := LineNumber(lineInt)
		portalAge := time.Since(timestamp)
		// Check if the portal was created for too long ago to be used for the current session
		if portalAge > maxPortalAge {
			return nil, errPortalTimedOut
		}
		return &Portal{timestamp, absFilename, lineNumber}, nil
	default:
		//logf("%s contains too few lines!\n%s\n", portalFilename, strings.Join(lines, ";"))
		return nil, errors.New(portalFilename + " contains too few lines")
	}
}

// LineIndex returns the current line index that the portal points to
func (p *Portal) LineIndex() LineIndex {
	return p.lineNumber.LineIndex()
}

// Save will save the portal
func (p *Portal) Save() error {
	//logf("Saving %s\n", portalFilename)
	s := fmt.Sprintf("%d\n%s\n%s\n", p.timestamp.Unix(), p.absFilename, p.lineNumber)
	// Anyone can read this file
	if err := os.WriteFile(portalFilename, []byte(s), 0o600); err != nil {
		return err
	}
	return os.Chmod(portalFilename, 0o666)
}

// String returns the current portal (filename + line number) as a colon separated string
func (p *Portal) String() string {
	return filepath.Base(p.absFilename) + ":" + p.lineNumber.String()
}

// NewLineInserted reacts when the editor inserts a new line in the same file,
// and moves the portal source one line down, if needed.
func (p *Portal) NewLineInserted(y LineIndex) {
	if y < p.LineIndex() {
		p.MoveDown()
	}
}

// PopLine removes (!) a line from the portal file, then removes that line
func (p *Portal) PopLine(e *Editor, removeLine bool) (string, error) {
	// popping a line from the same file is a special case
	if p == e.sameFilePortal {
		if removeLine {
			return "", errors.New("not implemented") // not implemented and currently not in use
		}
		// The line moving is done by the editor InsertAbove and InsertBelow functions
		return e.Line(p.LineIndex()), nil
	}
	data, err := os.ReadFile(p.absFilename)
	if err != nil {
		return "", err
	}
	lines := strings.Split(string(data), "\n")
	foundLine := ""
	found := false
	if removeLine {
		modifiedLines := make([]string, 0, len(lines)-1)
		for i, line := range lines {
			if LineIndex(i) == p.lineNumber.LineIndex() {
				foundLine = line
				found = true
			} else {
				modifiedLines = append(modifiedLines, line)
			}
		}
		if !found {
			return "", errors.New("Could not teleport line " + p.String())
		}
		data = []byte(strings.Join(modifiedLines, "\n"))
		if err = os.WriteFile(p.absFilename, data, 0o600); err != nil {
			return "", err
		}
	} else {
		for i, line := range lines {
			if LineIndex(i) == p.lineNumber.LineIndex() {
				foundLine = line
				found = true
				break
			}
		}
		if !found {
			return "", errors.New("Could not teleport line " + p.String())
		}
		// Now move the line number +1
		p.lineNumber++
		// And save the new portal
		if err := p.Save(); err != nil {
			return foundLine, err
		}
	}
	return foundLine, nil
}