71 lines
1.5 KiB
Go
71 lines
1.5 KiB
Go
|
package goorgeous
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bufio"
|
||
|
|
"bytes"
|
||
|
|
"regexp"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
// ExtractOrgHeaders finds and returns all of the headers
|
||
|
|
// from a bufio.Reader and returns them as their own byte slice
|
||
|
|
func ExtractOrgHeaders(r *bufio.Reader) (fm []byte, err error) {
|
||
|
|
var out bytes.Buffer
|
||
|
|
endOfHeaders := true
|
||
|
|
for endOfHeaders {
|
||
|
|
p, err := r.Peek(2)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
if !charMatches(p[0], '#') && !charMatches(p[1], '+') {
|
||
|
|
endOfHeaders = false
|
||
|
|
break
|
||
|
|
}
|
||
|
|
line, _, err := r.ReadLine()
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
out.Write(line)
|
||
|
|
out.WriteByte('\n')
|
||
|
|
}
|
||
|
|
return out.Bytes(), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
var reHeader = regexp.MustCompile(`^#\+(\w+?): (.*)`)
|
||
|
|
|
||
|
|
// OrgHeaders find all of the headers from a byte slice and returns
|
||
|
|
// them as a map of string interface
|
||
|
|
func OrgHeaders(input []byte) (map[string]interface{}, error) {
|
||
|
|
out := make(map[string]interface{})
|
||
|
|
scanner := bufio.NewScanner(bytes.NewReader(input))
|
||
|
|
|
||
|
|
for scanner.Scan() {
|
||
|
|
data := scanner.Bytes()
|
||
|
|
if !charMatches(data[0], '#') && !charMatches(data[1], '+') {
|
||
|
|
return out, nil
|
||
|
|
}
|
||
|
|
matches := reHeader.FindSubmatch(data)
|
||
|
|
|
||
|
|
if len(matches) < 3 {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
key := string(matches[1])
|
||
|
|
val := matches[2]
|
||
|
|
switch {
|
||
|
|
case strings.ToLower(key) == "tags" || strings.ToLower(key) == "categories" || strings.ToLower(key) == "aliases":
|
||
|
|
bTags := bytes.Split(val, []byte(" "))
|
||
|
|
tags := make([]string, len(bTags))
|
||
|
|
for idx, tag := range bTags {
|
||
|
|
tags[idx] = string(tag)
|
||
|
|
}
|
||
|
|
out[key] = tags
|
||
|
|
default:
|
||
|
|
out[key] = string(val)
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
return out, nil
|
||
|
|
|
||
|
|
}
|