wiki/backref.go

98 lines
1.6 KiB
Go
Raw Normal View History

2020-05-07 13:20:36 +00:00
package main
import (
"encoding/json"
"fmt"
2020-05-07 13:20:36 +00:00
"os"
"path/filepath"
"strings"
)
type Reference struct {
Link ParsedLink
Name string
}
type Refs map[string][]Reference
2020-05-07 13:20:36 +00:00
func processBackrefs(fp *FilePages) error {
pages, err := fp.AllPages()
refs := make(Refs)
if err != nil {
return err
}
for _, page := range pages {
content := page.Content
var listItems []struct {
Id int
Indented int
Text string
}
err = json.NewDecoder(strings.NewReader(content)).Decode(&listItems)
if err == nil {
pageText := ""
for _, item := range listItems {
pageText += strings.Repeat(" ", item.Indented) + "* " + item.Text + "\n"
}
content = pageText
}
links, err := ParseLinks(content)
2020-05-07 13:20:36 +00:00
if err != nil {
return err
}
for _, link := range links {
fmt.Println(link.Line)
refs[link.PageName] = append(refs[link.PageName], Reference{link, page.Name})
2020-05-07 13:20:36 +00:00
}
}
f, err := os.Create(filepath.Join(fp.dirname, "backrefs.json"))
if err != nil {
return err
}
defer f.Close()
err = json.NewEncoder(f).Encode(&refs)
return err
}
2020-05-08 15:21:39 +00:00
func loadBackrefs(fp *FilePages, p string) ([]Backref, error) {
2020-05-07 13:20:36 +00:00
refs := make(Refs)
p = strings.Replace(p, " ", "_", -1)
f, err := os.Open(filepath.Join(fp.dirname, "backrefs.json"))
if err != nil {
return nil, err
}
defer f.Close()
err = json.NewDecoder(f).Decode(&refs)
if err != nil {
return nil, err
}
2020-05-08 15:21:39 +00:00
var result []Backref
for _, ref := range refs[p] {
title := strings.Replace(ref.Name, "_", " ", -1)
2020-05-08 15:21:39 +00:00
result = append(result, Backref{
Name: ref.Name,
2020-05-08 15:21:39 +00:00
Title: title,
Line: strings.TrimLeft(ref.Link.Line, " *"),
2020-05-08 15:21:39 +00:00
})
}
return result, nil
2020-05-07 13:20:36 +00:00
}