package main import ( "bytes" "errors" "gopkg.in/yaml.v2" "strings" ) type HugoFrontmatter struct { Title string `yaml:"title,omitempty"` Date string `yaml:"date,omitempty"` Lastmod string `yaml:"lastmod,omitempty"` Tags []string `yaml:"tags,omitempty"` ExternalURL string `yaml:"externalURL,omitempty"` Indieweb HugoFrontmatterIndieweb `yaml:"indieweb,omitempty"` Syndicate []string `yaml:"syndicate,omitempty"` TranslationKey string `yaml:"translationKey,omitempty"` Images []string `yaml:"images,omitempty"` Audio string `yaml:"audio,omitempty"` } type HugoFrontmatterIndieweb struct { Reply HugoFrontmatterReply `yaml:"reply,omitempty"` Like HugoFrontmatterLike `yaml:"like,omitempty"` } type HugoFrontmatterReply struct { Link string `yaml:"link,omitempty"` Title string `yaml:"title,omitempty"` } type HugoFrontmatterLike struct { Link string `yaml:"link,omitempty"` Title string `yaml:"title,omitempty"` } func writeFrontMatter(entry *Entry) (frontmatter string, err error) { var buff bytes.Buffer writeFrontmatter := &HugoFrontmatter{ Title: entry.title, Date: entry.date, Lastmod: entry.lastmod, Tags: entry.tags, ExternalURL: entry.link, Indieweb: HugoFrontmatterIndieweb{ Reply: HugoFrontmatterReply{ Link: entry.replyLink, Title: entry.replyTitle, }, Like: HugoFrontmatterLike{ Link: entry.likeLink, Title: entry.likeTitle, }, }, Syndicate: entry.syndicate, TranslationKey: entry.translationKey, Images: entry.images, Audio: entry.audio, } yamlBytes, err := yaml.Marshal(&writeFrontmatter) if err != nil { err = errors.New("failed marshaling frontmatter") return } buff.WriteString("---\n") buff.Write(yamlBytes) buff.WriteString("---\n") frontmatter = buff.String() return } func WriteHugoPost(entry *Entry) (string, error) { var buff bytes.Buffer frontmatter, err := writeFrontMatter(entry) if err != nil { return "", err } buff.WriteString(frontmatter) if len(entry.content) > 0 { buff.WriteString(entry.content) } return buff.String(), nil } func readFrontMatter(frontmatter string, entry *Entry) (err error) { parsedFrontmatter := &HugoFrontmatter{} err = yaml.Unmarshal([]byte(frontmatter), &parsedFrontmatter) if err != nil { err = errors.New("failed parsing frontmatter") } if len(parsedFrontmatter.Title) > 0 { entry.title = parsedFrontmatter.Title } if len(parsedFrontmatter.Date) > 0 { entry.date = parsedFrontmatter.Date } if len(parsedFrontmatter.Lastmod) > 0 { entry.lastmod = parsedFrontmatter.Lastmod } if len(parsedFrontmatter.Tags) > 0 { entry.tags = parsedFrontmatter.Tags } if len(parsedFrontmatter.ExternalURL) > 0 { entry.link = parsedFrontmatter.ExternalURL } if len(parsedFrontmatter.Indieweb.Reply.Link) > 0 { entry.replyLink = parsedFrontmatter.Indieweb.Reply.Link } if len(parsedFrontmatter.Indieweb.Reply.Title) > 0 { entry.replyTitle = parsedFrontmatter.Indieweb.Reply.Title } if len(parsedFrontmatter.Indieweb.Like.Link) > 0 { entry.replyLink = parsedFrontmatter.Indieweb.Like.Link } if len(parsedFrontmatter.Indieweb.Like.Title) > 0 { entry.replyTitle = parsedFrontmatter.Indieweb.Like.Title } if len(parsedFrontmatter.Syndicate) > 0 { entry.syndicate = parsedFrontmatter.Syndicate } if len(parsedFrontmatter.TranslationKey) > 0 { entry.translationKey = parsedFrontmatter.TranslationKey } if len(parsedFrontmatter.Images) > 0 { entry.images = parsedFrontmatter.Images } if len(parsedFrontmatter.Audio) > 0 { entry.audio = parsedFrontmatter.Audio } return } func ReadHugoPost(fileContent string) (entry *Entry, err error) { parts := strings.Split(fileContent, "---\n") if len(parts) != 3 { err = errors.New("empty frontmatter or content") return } entry = new(Entry) err = readFrontMatter(parts[1], entry) if err != nil { return } entry.content = strings.TrimSuffix(parts[2], "\n") return }