195 lines
3 KiB
Go
195 lines
3 KiB
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)
|
|
}
|