update to write json instead of yaml

This commit is contained in:
Jacky Zhao 2022-02-15 19:36:14 -05:00
parent 31b4ffca89
commit 4691c1c600
2 changed files with 16 additions and 16 deletions

17
main.go
View File

@ -19,21 +19,22 @@ func init() {
} }
type Link struct { type Link struct {
Source string Source string `json:"source"`
Target string Target string `json:"target"`
Text string Text string `json:"text"`
} }
type LinkTable = map[string][]Link type LinkTable = map[string][]Link
type Index struct { type Index struct {
Links LinkTable Links LinkTable `json:"links"`
Backlinks LinkTable Backlinks LinkTable `json:"backlinks"`
} }
type Content struct { type Content struct {
Title string Title string `json:"title"`
Content string Content string `json:"content"`
LastModified time.Time LastModified time.Time `json:"lastmodified"`
Tags []string `json:"tags"`
} }
type ContentIndex = map[string]Content type ContentIndex = map[string]Content

View File

@ -1,38 +1,37 @@
package main package main
import ( import (
"gopkg.in/yaml.v3" "encoding/json"
"io/ioutil" "io/ioutil"
"path" "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 { func write(links []Link, contentIndex ContentIndex, toIndex bool, out string) error {
index := index(links) index := index(links)
resStruct := struct { resStruct := struct {
Index Index Index Index `json:"index"`
Links []Link Links []Link `json:"links"`
}{ }{
Index: index, Index: index,
Links: links, Links: links,
} }
marshalledIndex, mErr := yaml.Marshal(&resStruct) marshalledIndex, mErr := json.MarshalIndent(&resStruct, "", " ")
if mErr != nil { if mErr != nil {
return mErr return mErr
} }
writeErr := ioutil.WriteFile(path.Join(out, "linkIndex.yaml"), append([]byte(message), marshalledIndex...), 0644) writeErr := ioutil.WriteFile(path.Join(out, "linkIndex.json"), marshalledIndex, 0644)
if writeErr != nil { if writeErr != nil {
return writeErr return writeErr
} }
if toIndex { if toIndex {
marshalledContentIndex, mcErr := yaml.Marshal(&contentIndex) marshalledContentIndex, mcErr := json.MarshalIndent(&contentIndex, "", " ")
if mcErr != nil { if mcErr != nil {
return mcErr return mcErr
} }
writeErr = ioutil.WriteFile(path.Join(out, "contentIndex.yaml"), append([]byte(message), marshalledContentIndex...), 0644) writeErr = ioutil.WriteFile(path.Join(out, "contentIndex.json"), marshalledContentIndex, 0644)
if writeErr != nil { if writeErr != nil {
return writeErr return writeErr
} }