zet/internal/lib/zet.go

package lib

import (
	"bufio"
	"fmt"
	"log"
	"os"
	"os/exec"
	"path"
	"regexp"
	"sort"
	"strconv"
	"strings"
)

func CreateZet(title string) {
	zid := newZid()

	fmt.Println(title, zid)

	path := path.Join(GetZetDir(), strconv.Itoa(zid))

	os.Mkdir(path, 0750)

	filePath := fmt.Sprintf("%s/index.adoc", path)

	file, err := os.Create(filePath)

	if err != nil {
		log.Fatal(err)
	}

	defer file.Close()

	_, err = fmt.Fprintf(file, "= %s", title)

	if err != nil {
		log.Fatal(err)
	}

	EditFile(filePath)

	onSave(zid)
}

func EditZet(id int) {
	zetPath := path.Join(GetZetDir(), strconv.Itoa(id), "index.adoc")

	EditFile(zetPath)

	onSave(id)
}

func GetAllZets() []int {
	zets, err := execGitCommand("ls-files")

	if err != nil {
		log.Println(err)
	}

	re := regexp.MustCompile(`[0-9]+`)
	matches := re.FindAllString(zets, -1)

	sort.Strings(matches)

	ids := make(map[int]struct{})
	for _, id := range matches {
		num, err := strconv.Atoi(id)

		if err == nil {
			ids[num] = struct{}{}
		}
	}

	var sorted []int
	for num := range ids {
		sorted = append(sorted, num)
	}

	sort.Ints(sorted)

	return sorted
}

func SearchZets(query string) []int {
	zets, err := execGitCommand("grep", "-i", "--name-only", "--word-regex", query)

	if err != nil {
		fmt.Printf("No matches found for %s.\n", query)

		os.Exit(1)
	}

	re := regexp.MustCompile(`[0-9]+`)
	matches := re.FindAllString(zets, -1)

	sort.Strings(matches)

	ids := make(map[int]struct{})
	for _, id := range matches {
		num, err := strconv.Atoi(id)

		if err == nil {
			ids[num] = struct{}{}
		}
	}

	var sorted []int
	for num := range ids {
		sorted = append(sorted, num)
	}

	sort.Ints(sorted)

	return sorted
}

func ParseZetList(ids []int) []string {
	var lines []string

	green := "\033[32m"
	reset := "\033[0m"

	for _, num := range ids {
		line := fmt.Sprintf("%s%s%s %s", green, strconv.Itoa(num), reset, getTitle(num))

		fmt.Println(line)

		lines = append(lines, line)
	}

	return lines
}

func ViewZet(id int) string {
	zetPath := path.Join(GetZetDir(), strconv.Itoa(id), "index.adoc")

	return ViewFile(zetPath)
}

func getTitle(id int) string {
	return getTitleFromFile(path.Join(strconv.Itoa(id), "index.adoc"))
}

func getTitleFromFile(filePath string) string {
	filePath = path.Join(GetZetDir(), filePath)

	file, err := os.Open(filePath)

	if err != nil {
		fmt.Println("Error opening file:", err)

		return ""
	}

	defer file.Close()

	scanner := bufio.NewScanner(file)

	if scanner.Scan() {
		text := scanner.Text()

		return strings.TrimPrefix(text, "= ")
	}

	if err := scanner.Err(); err != nil {
		fmt.Println("Error reading file:", err)
	}

	return ""
}

func newZid() int {
	cmd := exec.Command("ls", GetZetDir())
	output, _ := cmd.CombinedOutput()

	zets := strings.Split(string(output), "\n")

	var zetCount int

	for _, zet := range zets {
		num, err := strconv.Atoi(zet)
		if err == nil && num > zetCount {
			zetCount = num
		}
	}

	return zetCount + 1
}

func onSave(id int) {
	title := getTitle(id)

	CommitZettel(id, title)
}