Merge pull request #10 from jackyzha0/draft-support

2024-06-14 11:42:35 +03:00 · 2021-12-27 17:02:47 -05:00 · 2021-12-27 17:02:47 -05:00 · 1e937f6a83
commit 1e937f6a83
parent d9c0db0468 9a48624656
7 changed files with 276 additions and 203 deletions
--- a/go.mod
+++ b/go.mod
@ -3,6 +3,7 @@ module github.com/jackyzha0/hugo-obsidian
 go 1.16

 require (
+	github.com/BurntSushi/toml v0.4.1 // indirect
 	github.com/PuerkitoBio/goquery v1.8.0
 	github.com/abhinav/goldmark-wikilink v0.3.0
 	github.com/gernest/front v0.0.0-20210301115436-8a0b0a782d0a
--- a/go.sum
+++ b/go.sum
@ -1,3 +1,5 @@
+github.com/BurntSushi/toml v0.4.1 h1:GaI7EiDXDRfa8VshkTj7Fym7ha+y8/XxIgD2okUIjLw=
+github.com/BurntSushi/toml v0.4.1/go.mod h1:CxXYINrC8qIiEnFrOxCa7Jy5BFHlXnUU2pbicEuybxQ=
 github.com/PuerkitoBio/goquery v1.8.0 h1:PJTF7AmFCFKk1N6V6jmKfrNH9tV5pNE6lZMkG0gta/U=
 github.com/PuerkitoBio/goquery v1.8.0/go.mod h1:ypIiRMtY7COPGk+I/YbZLbxsxn9g5ejnI2HSMtkjZvI=
 github.com/abhinav/goldmark-wikilink v0.3.0 h1:ry8CBaULn410PKCSkwLz/WVI2f/g7EB+yqY7LKHDcPQ=
--- a/main.go
+++ b/main.go
@ -1,24 +1,17 @@
 package main

 import (
-	"bytes"
 	"flag"
-	"fmt"
-	"github.com/gernest/front"
-	"gopkg.in/yaml.v3"
-	"io/fs"
-	"io/ioutil"
-	"path"
-	"path/filepath"
-	"strings"
-
-	"github.com/PuerkitoBio/goquery"
+	"github.com/BurntSushi/toml"
 	wikilink "github.com/abhinav/goldmark-wikilink"
 	"github.com/yuin/goldmark"
+	"io/ioutil"
+	"path/filepath"
+	"time"
 )

-
 var md goldmark.Markdown
+
 func init() {
 	md = goldmark.New(
 		goldmark.WithExtensions(&wikilink.Extender{}),
@ -40,217 +33,47 @@ type Index struct {
 type Content struct {
 	Title   string
 	Content string
+	LastModified time.Time
 }

 type ContentIndex = map[string]Content

-func trim(source, prefix, suffix string) string {
-	return strings.TrimPrefix(strings.TrimSuffix(source, suffix), prefix)
+type ConfigTOML struct {
+	IgnoredFiles []string `toml:"ignoreFiles"`
 }

-func hugoPathTrim(source string) string {
-	return strings.TrimSuffix(strings.TrimSuffix(source, "/index"), "_index")
-}
+func getIgnoredFiles(base string) (res map[string]struct{}) {
+	res = make(map[string]struct{})

-func processTarget(source string) string {
-	if !isInternal(source) {
-		return source
-	}
-	if strings.HasPrefix(source, "/") {
-		return strings.TrimSuffix(source, ".md")
-	}
-	return "/" + strings.TrimSuffix(strings.TrimSuffix(source, ".html"), ".md")
-}
-
-func isInternal(link string) bool {
-	return !strings.HasPrefix(link, "http")
-}
-
-// parse single file for links
-func parse(dir, pathPrefix string) []Link {
-	// read file
-	source, err := ioutil.ReadFile(dir)
+	source, err := ioutil.ReadFile(base + "/config.toml")
 	if err != nil {
-		panic(err)
-	}
-
-	// parse md
-	var links []Link
-	fmt.Printf("[Parsing note] %s\n", trim(dir, pathPrefix, ".md"))
-
-	var buf bytes.Buffer
-	if err := md.Convert(source, &buf); err != nil {
-		panic(err)
-	}
-
-	doc, err := goquery.NewDocumentFromReader(&buf)
-	var n int
-	doc.Find("a").Each(func(i int, s *goquery.Selection) {
-		text := strings.TrimSpace(s.Text())
-		target, ok := s.Attr("href")
-		if !ok {
-			target = "#"
-		}
-
-		target = strings.Replace(target, "%20", " ", -1)
-		target = strings.Split(processTarget(target), "#")[0]
-		target = strings.TrimSpace(target)
-		target = strings.Replace(target, " ", "-", -1)
-
-		fmt.Printf("  '%s' => %s\n", text, target)
-		links = append(links, Link{
-			Source: filepath.ToSlash(hugoPathTrim(trim(dir, pathPrefix, ".md"))),
-			Target: target,
-			Text: text,
-		})
-		n++
-	})
-	fmt.Printf(":: %d links\n", n)
-
-	return links
-}
-
-func getText(dir string) string {
-	// read file
-	bytes, err := ioutil.ReadFile(dir)
-	if err != nil {
-		panic(err)
-	}
-
-	return string(bytes)
-}
-
-// recursively walk directory and return all files with given extension
-func walk(root, ext string, index bool) (res []Link, i ContentIndex) {
-	println(root)
-	i = make(ContentIndex)
-
-	m := front.NewMatter()
-	m.Handle("---", front.YAMLHandler)
-	nPrivate := 0
-
-	err := filepath.WalkDir(root, func(s string, d fs.DirEntry, e error) error {
-		if e != nil {
-			return e
-		}
-		if filepath.Ext(d.Name()) == ext {
-			res = append(res, parse(s, root)...)
-			if index {
-				text := getText(s)
-
-				frontmatter, body, err := m.Parse(strings.NewReader(text))
-				if err != nil {
-					frontmatter = map[string]interface{}{}
-					body = text
-				}
-
-				var title string
-				if parsedTitle, ok := frontmatter["title"]; ok {
-					title = parsedTitle.(string)
-				} else {
-					title = "Untitled Page"
-				}
-
-				// check if page is private
-				if parsedPrivate, ok := frontmatter["draft"]; !ok || !parsedPrivate.(bool) {
-					adjustedPath := strings.Replace(hugoPathTrim(trim(s, root, ".md")), " ", "-", -1)
-					i[adjustedPath] = Content{
-						Title: title,
-						Content: body,
-					}
-				} else {
-					nPrivate++
-				}
-			}
-		}
-		return nil
-	})
-	if err != nil {
-		panic(err)
-	}
-	fmt.Printf("Ignored %d private files \n", nPrivate)
-	fmt.Printf("Parsed %d total links \n", len(res))
-	return res, i
-}
-
-// filter out certain links (e.g. to media)
-func filter(links []Link) (res []Link) {
-	for _, l := range links {
-		// filter external and non-md
-		isMarkdown := filepath.Ext(l.Target) == "" || filepath.Ext(l.Target) == ".md"
-		if isInternal(l.Target) && isMarkdown {
-			res = append(res, l)
-		}
-	}
-	fmt.Printf("Removed %d external and non-markdown links\n", len(links) - len(res))
 		return res
-}
-
-// constructs index from links
-func index(links []Link) (index Index) {
-	linkMap := make(map[string][]Link)
-	backlinkMap := make(map[string][]Link)
-	for _, l := range links {
-		// backlink (only if internal)
-		if _, ok := backlinkMap[l.Target]; ok {
-			backlinkMap[l.Target] = append(backlinkMap[l.Target], l)
-		} else {
-			backlinkMap[l.Target] = []Link{l}
 	}

-		// regular link
-		if _, ok := linkMap[l.Source]; ok {
-			linkMap[l.Source] = append(linkMap[l.Source], l)
-		} else {
-			linkMap[l.Source] = []Link{l}
-		}
-	}
-	index.Links = linkMap
-	index.Backlinks = backlinkMap
-	return index
-}
-
-const message = "# THIS FILE WAS GENERATED USING github.com/jackyzha0/hugo-obsidian\n# DO NOT EDIT\n"
-func write(links []Link, contentIndex ContentIndex, toIndex bool, out string) error {
-	index := index(links)
-	resStruct := struct{
-		Index Index
-		Links []Link
-	}{
-		Index: index,
-		Links: links,
-	}
-	marshalledIndex, mErr := yaml.Marshal(&resStruct)
-	if mErr != nil {
-		return mErr
+	var config ConfigTOML
+	if _, err := toml.Decode(string(source), &config); err != nil {
+		return res
 	}

-	writeErr := ioutil.WriteFile(path.Join(out, "linkIndex.yaml"), append([]byte(message), marshalledIndex...), 0644)
-	if writeErr != nil {
-		return writeErr
-	}
-
-	if toIndex {
-		marshalledContentIndex, mcErr := yaml.Marshal(&contentIndex)
-		if mcErr != nil {
-			return mcErr
-		}
-
-		writeErr = ioutil.WriteFile(path.Join(out, "contentIndex.yaml"), append([]byte(message), marshalledContentIndex...), 0644)
-		if writeErr != nil {
-			return writeErr
+	for _, glb := range config.IgnoredFiles {
+		matches, _ := filepath.Glob(base + glb)
+		for _, match := range matches {
+			res[match] = struct{}{}
 		}
 	}

-	return nil
+	return res
 }

 func main() {
 	in := flag.String("input", ".", "Input Directory")
 	out := flag.String("output", ".", "Output Directory")
+	root := flag.String("root", "..", "Root Directory (for config parsing)")
 	index := flag.Bool("index", false, "Whether to index the content")
 	flag.Parse()
-	l, i := walk(*in, ".md", *index)
+
+	ignoreBlobs := getIgnoredFiles(*root)
+	l, i := walk(*in, ".md", *index, ignoreBlobs)
 	f := filter(l)
 	err := write(f, i, *index, *out)
 	if err != nil {
--- a/parse.go
+++ b/parse.go
@ -0,0 +1,54 @@
+package main
+
+import (
+	"bytes"
+	"fmt"
+	"github.com/PuerkitoBio/goquery"
+	"io/ioutil"
+	"path/filepath"
+	"strings"
+)
+
+// parse single file for links
+func parse(dir, pathPrefix string) []Link {
+	// read file
+	source, err := ioutil.ReadFile(dir)
+	if err != nil {
+		panic(err)
+	}
+
+	// parse md
+	var links []Link
+	fmt.Printf("[Parsing note] %s\n", trim(dir, pathPrefix, ".md"))
+
+	var buf bytes.Buffer
+	if err := md.Convert(source, &buf); err != nil {
+		panic(err)
+	}
+
+	doc, err := goquery.NewDocumentFromReader(&buf)
+	var n int
+	doc.Find("a").Each(func(i int, s *goquery.Selection) {
+		text := strings.TrimSpace(s.Text())
+		target, ok := s.Attr("href")
+		if !ok {
+			target = "#"
+		}
+
+		target = strings.Replace(target, "%20", " ", -1)
+		target = strings.Split(processTarget(target), "#")[0]
+		target = strings.TrimSpace(target)
+		target = strings.Replace(target, " ", "-", -1)
+
+		fmt.Printf("  '%s' => %s\n", text, target)
+		links = append(links, Link{
+			Source: filepath.ToSlash(hugoPathTrim(trim(dir, pathPrefix, ".md"))),
+			Target: target,
+			Text:   text,
+		})
+		n++
+	})
+	fmt.Printf("  Found: %d links\n", n)
+
+	return links
+}
--- a/util.go
+++ b/util.go
@ -0,0 +1,43 @@
+package main
+
+import (
+	"fmt"
+	"path/filepath"
+	"strings"
+)
+
+func trim(source, prefix, suffix string) string {
+	return strings.TrimPrefix(strings.TrimSuffix(source, suffix), prefix)
+}
+
+func hugoPathTrim(source string) string {
+	return strings.TrimSuffix(strings.TrimSuffix(source, "/index"), "_index")
+}
+
+func processTarget(source string) string {
+	if !isInternal(source) {
+		return source
+	}
+	if strings.HasPrefix(source, "/") {
+		return strings.TrimSuffix(source, ".md")
+	}
+	return "/" + strings.TrimSuffix(strings.TrimSuffix(source, ".html"), ".md")
+}
+
+func isInternal(link string) bool {
+	return !strings.HasPrefix(link, "http")
+}
+
+// filter out certain links (e.g. to media)
+func filter(links []Link) (res []Link) {
+	for _, l := range links {
+		// filter external and non-md
+		isMarkdown := filepath.Ext(l.Target) == "" || filepath.Ext(l.Target) == ".md"
+		if isInternal(l.Target) && isMarkdown {
+			res = append(res, l)
+		}
+	}
+	fmt.Printf("Removed %d external and non-markdown links\n", len(links)-len(res))
+	return res
+}
+
--- a/walk.go
+++ b/walk.go
@ -0,0 +1,81 @@
+package main
+
+import (
+	"fmt"
+	"github.com/gernest/front"
+	"io/fs"
+	"io/ioutil"
+	"os"
+	"path/filepath"
+	"strings"
+)
+
+// recursively walk directory and return all files with given extension
+func walk(root, ext string, index bool, ignorePaths map[string]struct{}) (res []Link, i ContentIndex) {
+	fmt.Printf("Scraping %s\n", root)
+	i = make(ContentIndex)
+
+	m := front.NewMatter()
+	m.Handle("---", front.YAMLHandler)
+	nPrivate := 0
+
+	err := filepath.WalkDir(root, func(s string, d fs.DirEntry, e error) error {
+		if e != nil {
+			return e
+		}
+		if _, ignored := ignorePaths[s]; ignored {
+			fmt.Printf("[Ignored] %s\n", d.Name())
+			nPrivate++
+		} else if filepath.Ext(d.Name()) == ext {
+			res = append(res, parse(s, root)...)
+			if index {
+				text := getText(s)
+
+				frontmatter, body, err := m.Parse(strings.NewReader(text))
+				if err != nil {
+					frontmatter = map[string]interface{}{}
+					body = text
+				}
+
+				var title string
+				if parsedTitle, ok := frontmatter["title"]; ok {
+					title = parsedTitle.(string)
+				} else {
+					title = "Untitled Page"
+				}
+
+				// check if page is private
+				if parsedPrivate, ok := frontmatter["draft"]; !ok || !parsedPrivate.(bool) {
+					info, _ := os.Stat(s)
+					adjustedPath := strings.Replace(hugoPathTrim(trim(s, root, ".md")), " ", "-", -1)
+					i[adjustedPath] = Content{
+						LastModified: info.ModTime(),
+						Title:   title,
+						Content: body,
+					}
+				} else {
+					fmt.Printf("[Ignored] %s\n", d.Name())
+					nPrivate++
+				}
+			}
+		}
+		return nil
+	})
+	if err != nil {
+		panic(err)
+	}
+	fmt.Printf("Ignored %d private files \n", nPrivate)
+	fmt.Printf("Parsed %d total links \n", len(res))
+	return res, i
+}
+
+func getText(dir string) string {
+	// read file
+	fileBytes, err := ioutil.ReadFile(dir)
+	if err != nil {
+		panic(err)
+	}
+
+	return string(fileBytes)
+}
+
--- a/write.go
+++ b/write.go
@ -0,0 +1,69 @@
+package main
+
+import (
+	"gopkg.in/yaml.v3"
+	"io/ioutil"
+	"path"
+)
+
+const message = "# THIS FILE WAS GENERATED USING github.com/jackyzha0/hugo-obsidian\n# DO NOT EDIT\n"
+func write(links []Link, contentIndex ContentIndex, toIndex bool, out string) error {
+	index := index(links)
+	resStruct := struct {
+		Index Index
+		Links []Link
+	}{
+		Index: index,
+		Links: links,
+	}
+	marshalledIndex, mErr := yaml.Marshal(&resStruct)
+	if mErr != nil {
+		return mErr
+	}
+
+	writeErr := ioutil.WriteFile(path.Join(out, "linkIndex.yaml"), append([]byte(message), marshalledIndex...), 0644)
+	if writeErr != nil {
+		return writeErr
+	}
+
+	if toIndex {
+		marshalledContentIndex, mcErr := yaml.Marshal(&contentIndex)
+		if mcErr != nil {
+			return mcErr
+		}
+
+		writeErr = ioutil.WriteFile(path.Join(out, "contentIndex.yaml"), append([]byte(message), marshalledContentIndex...), 0644)
+		if writeErr != nil {
+			return writeErr
+		}
+	}
+
+	return nil
+}
+
+// constructs index from links
+func index(links []Link) (index Index) {
+	linkMap := make(map[string][]Link)
+	backlinkMap := make(map[string][]Link)
+	for _, l := range links {
+		// backlink (only if internal)
+		if _, ok := backlinkMap[l.Target]; ok {
+			backlinkMap[l.Target] = append(backlinkMap[l.Target], l)
+		} else {
+			backlinkMap[l.Target] = []Link{l}
+		}
+
+		// regular link
+		if _, ok := linkMap[l.Source]; ok {
+			linkMap[l.Source] = append(linkMap[l.Source], l)
+		} else {
+			linkMap[l.Source] = []Link{l}
+		}
+	}
+	index.Links = linkMap
+	index.Backlinks = backlinkMap
+	return index
+}
+
+
+