2017-06-24 11:12:15 +00:00
|
|
|
package frontmatter
|
|
|
|
|
|
|
|
import (
|
|
|
|
"strings"
|
|
|
|
)
|
|
|
|
|
|
|
|
// HasRune checks if the file has the frontmatter rune
|
2017-06-29 09:17:35 +00:00
|
|
|
func HasRune(file string) bool {
|
|
|
|
return strings.HasPrefix(file, "---") ||
|
|
|
|
strings.HasPrefix(file, "+++") ||
|
|
|
|
strings.HasPrefix(file, "{")
|
2017-06-24 11:12:15 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// AppendRune appends the frontmatter rune to a file
|
2017-06-29 09:17:35 +00:00
|
|
|
func AppendRune(frontmatter string, mark rune) string {
|
|
|
|
frontmatter = strings.TrimSpace(frontmatter)
|
2017-06-24 11:12:15 +00:00
|
|
|
|
|
|
|
switch mark {
|
|
|
|
case '-':
|
2017-06-29 09:17:35 +00:00
|
|
|
return "---\n" + frontmatter + "\n---"
|
2017-06-24 11:12:15 +00:00
|
|
|
case '+':
|
2017-06-29 09:17:35 +00:00
|
|
|
return "+++\n" + frontmatter + "\n+++"
|
2017-06-24 11:12:15 +00:00
|
|
|
case '{':
|
2017-06-29 09:17:35 +00:00
|
|
|
return "{\n" + frontmatter + "\n}"
|
2017-06-24 11:12:15 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
return frontmatter
|
|
|
|
}
|