144 lines
3.0 KiB
Go
144 lines
3.0 KiB
Go
package handlers
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"errors"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/hacdias/caddy-filemanager/config"
|
|
"github.com/hacdias/caddy-filemanager/frontmatter"
|
|
)
|
|
|
|
// PreProccessPUT is used to update a file that was edited
|
|
func PreProccessPUT(
|
|
w http.ResponseWriter,
|
|
r *http.Request,
|
|
c *config.Config,
|
|
u *config.User,
|
|
) (err error) {
|
|
var (
|
|
data = map[string]interface{}{}
|
|
file []byte
|
|
kind string
|
|
rawBuffer = new(bytes.Buffer)
|
|
)
|
|
|
|
kind = r.Header.Get("kind")
|
|
rawBuffer.ReadFrom(r.Body)
|
|
|
|
if kind != "" {
|
|
err = json.Unmarshal(rawBuffer.Bytes(), &data)
|
|
|
|
if err != nil {
|
|
return
|
|
}
|
|
}
|
|
|
|
switch kind {
|
|
case "frontmatter-only":
|
|
if file, err = ParseFrontMatterOnlyFile(data, r.URL.Path); err != nil {
|
|
return
|
|
}
|
|
case "content-only":
|
|
mainContent := data["content"].(string)
|
|
mainContent = strings.TrimSpace(mainContent)
|
|
file = []byte(mainContent)
|
|
case "complete":
|
|
var mark rune
|
|
|
|
if v := r.Header.Get("Rune"); v != "" {
|
|
n, err := strconv.Atoi(v)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
mark = rune(n)
|
|
}
|
|
|
|
if file, err = ParseCompleteFile(data, r.URL.Path, mark); err != nil {
|
|
return
|
|
}
|
|
default:
|
|
file = rawBuffer.Bytes()
|
|
}
|
|
|
|
// Overwrite the request Body
|
|
r.Body = ioutil.NopCloser(bytes.NewReader(file))
|
|
return
|
|
}
|
|
|
|
// ParseFrontMatterOnlyFile parses a frontmatter only file
|
|
func ParseFrontMatterOnlyFile(data interface{}, filename string) ([]byte, error) {
|
|
frontmatter := strings.TrimPrefix(filepath.Ext(filename), ".")
|
|
f, err := ParseFrontMatter(data, frontmatter)
|
|
fString := string(f)
|
|
|
|
// If it's toml or yaml, strip frontmatter identifier
|
|
if frontmatter == "toml" {
|
|
fString = strings.TrimSuffix(fString, "+++\n")
|
|
fString = strings.TrimPrefix(fString, "+++\n")
|
|
}
|
|
|
|
if frontmatter == "yaml" {
|
|
fString = strings.TrimSuffix(fString, "---\n")
|
|
fString = strings.TrimPrefix(fString, "---\n")
|
|
}
|
|
|
|
f = []byte(fString)
|
|
return f, err
|
|
}
|
|
|
|
// ParseFrontMatter is the frontmatter parser
|
|
func ParseFrontMatter(data interface{}, front string) ([]byte, error) {
|
|
var mark rune
|
|
|
|
switch front {
|
|
case "toml":
|
|
mark = '+'
|
|
case "json":
|
|
mark = '{'
|
|
case "yaml":
|
|
mark = '-'
|
|
default:
|
|
return nil, errors.New("Unsupported Format provided")
|
|
}
|
|
|
|
return frontmatter.Marshal(data, mark)
|
|
}
|
|
|
|
// ParseCompleteFile parses a complete file
|
|
func ParseCompleteFile(data map[string]interface{}, filename string, mark rune) ([]byte, error) {
|
|
mainContent := ""
|
|
|
|
if _, ok := data["content"]; ok {
|
|
// The main content of the file
|
|
mainContent = data["content"].(string)
|
|
mainContent = "\n\n" + strings.TrimSpace(mainContent) + "\n"
|
|
|
|
// Removes the main content from the rest of the frontmatter
|
|
delete(data, "content")
|
|
}
|
|
|
|
if _, ok := data["date"]; ok {
|
|
data["date"] = data["date"].(string) + ":00"
|
|
}
|
|
|
|
front, err := frontmatter.Marshal(data, mark)
|
|
if err != nil {
|
|
return []byte{}, err
|
|
}
|
|
|
|
front = frontmatter.AppendRune(front, mark)
|
|
|
|
// Generates the final file
|
|
f := new(bytes.Buffer)
|
|
f.Write(front)
|
|
f.Write([]byte(mainContent))
|
|
return f.Bytes(), nil
|
|
}
|