wiki/storage.go

package main

import (
	"fmt"
	"io/ioutil"
	"os"
	"path"
	"strings"
	"time"

	"github.com/pkg/errors"
	"gopkg.in/src-d/go-git.v4"
	"gopkg.in/src-d/go-git.v4/config"
	"gopkg.in/src-d/go-git.v4/plumbing/object"
	"gopkg.in/yaml.v2"
)

const yamlDelimiter = `---`

var errFileNotFound = errors.New("Specified file was not found")

type storedFile struct {
	Meta    map[string]interface{} `json:"meta"`
	Content string                 `json:"content"`
}

func loadStoredFile(filename string) (*storedFile, error) {
	if _, err := os.Stat(path.Join(cfg.DataDir, filename)); err != nil {
		return nil, errFileNotFound
	}

	content, err := ioutil.ReadFile(path.Join(cfg.DataDir, filename))
	if err != nil {
		return nil, errors.Wrap(err, "Unable to read file")
	}

	return storedFileFromString(string(content))
}

func storedFileFromString(content string) (*storedFile, error) {
	// Look at first line and see whether this file has a metadata part
	lines := strings.Split(strings.TrimSpace(content), "\n")
	if len(lines) == 0 {
		// Empty file
		return &storedFile{}, nil
	}

	var (
		metadata     []string
		contentStart int
	)

	if lines[0] == yamlDelimiter {
		// This file has a metadata part
		for i := 1; i < len(lines); i++ {
			if lines[i] == yamlDelimiter {
				contentStart = i + 1
				break
			}

			metadata = append(metadata, lines[i])
		}
	}

	file := &storedFile{
		Content: strings.TrimSpace(strings.Join(lines[contentStart:], "\n")),
		Meta:    map[string]interface{}{},
	}

	if len(metadata) > 0 {
		if err := yaml.NewDecoder(strings.NewReader(strings.Join(metadata, "\n"))).Decode(&file.Meta); err != nil {
			return nil, errors.Wrap(err, "Unable to parse metadata part")
		}
	}

	return file, nil
}

func (s storedFile) GetMetaString(key string) string {
	if v, ok := s.Meta[key].(string); ok {
		return v
	}
	return ""
}

func (s storedFile) Save(filename string) error {
	repo, err := git.PlainOpen(cfg.DataDir)
	if err != nil {
		if err != git.ErrRepositoryNotExists {
			return errors.Wrap(err, "Unable to open repository")
		}

		if repo, err = s.initRepo(); err != nil {
			return errors.Wrap(err, "Unable to init repo")
		}
	}

	wt, err := repo.Worktree()
	if err != nil {
		return errors.Wrap(err, "Unable to get worktree")
	}

	f, err := os.Create(path.Join(cfg.DataDir, filename))
	if err != nil {
		return errors.Wrap(err, "Unable to create file")
	}
	defer f.Close()

	if len(s.Meta) > 0 {
		fmt.Fprintln(f, yamlDelimiter)
		if err := yaml.NewEncoder(f).Encode(s.Meta); err != nil {
			return errors.Wrap(err, "Unable to write metadata")
		}
		fmt.Fprintln(f, yamlDelimiter)
	}

	fmt.Fprintln(f, s.Content)

	if _, err := wt.Add(filename); err != nil {
		return errors.Wrap(err, "Unable to add file to index")
	}

	_, err = wt.Commit("Web-Update of "+filename, &git.CommitOptions{Author: s.authorSignature()})
	return errors.Wrap(err, "Unable to commit file change")
}

func (s storedFile) authorSignature() *object.Signature {
	return &object.Signature{Name: "wiki " + version, Email: "wiki@luzifer.io", When: time.Now()}
}

func (s storedFile) initRepo() (*git.Repository, error) {
	repo, err := git.PlainInit(cfg.DataDir, false)
	if err != nil {
		return nil, errors.Wrap(err, "Unable to initialize repo")
	}

	if _, err := repo.Branch("master"); err == git.ErrBranchNotFound {
		if err := repo.CreateBranch(&config.Branch{Name: "master"}); err != nil {
			return nil, errors.Wrap(err, "Unable to create master branch")
		}
	}

	wt, err := repo.Worktree()
	if err != nil {
		return nil, errors.Wrap(err, "Unable to get worktree")
	}

	_, err = wt.Commit("Initial commit", &git.CommitOptions{Author: s.authorSignature()})

	return repo, errors.Wrap(err, "Unable to create initial commit")
}