local-server/pkg/clustersconfig/defaults.go

package clustersconfig

import (
	"fmt"
	"io"
	"io/ioutil"
	"log"
	"path"
	"path/filepath"
	"strings"

	billy "gopkg.in/src-d/go-billy.v4"
	git "gopkg.in/src-d/go-git.v4"
	"gopkg.in/src-d/go-git.v4/plumbing"
	"gopkg.in/src-d/go-git.v4/plumbing/object"
	yaml "gopkg.in/yaml.v2"
)

type Defaults struct {
	repo *git.Repository
	fs   billy.Filesystem
}

type defaultRef struct {
	From string
}

func NewDefaults(path string) (d *Defaults, err error) {
	repo, err := git.PlainOpen(path)
	if err != nil {
		return
	}

	d = &Defaults{
		repo: repo,
	}

	return
}

func (d *Defaults) Load(dir, suffix string, value Rev, data []byte) (err error) {
	ref := defaultRef{}

	if err = yaml.Unmarshal(data, &ref); err != nil {
		return
	}

	if len(ref.From) != 0 {
		if Debug {
			log.Printf("loading defaults %q", ref.From)
		}

		parts := strings.SplitN(ref.From, ":", 2)
		if len(parts) != 2 {
			err = fmt.Errorf("bad default reference: %q", ref.From)
			return
		}
		rev, fileName := parts[0], parts[1]

		if err = d.decodeDefault(rev, path.Join(dir, fileName+suffix), value); err != nil {
			return
		}

		value.SetRev(rev)
	}

	err = yaml.Unmarshal(data, value)

	return
}

func (d *Defaults) Open(rev, filePath string) (rd io.Reader, err error) {
	tree, err := d.treeAt(rev)
	if err != nil {
		return
	}

	file, err := tree.File(filePath)
	if err == object.ErrFileNotFound {
		return nil, nil
	} else if err != nil {
		return
	}

	return file.Reader()
}

func (d *Defaults) ReadAll(rev, filePath string) (ba []byte, err error) {
	rd, err := d.Open(rev, filePath)
	if err != nil || rd == nil {
		return
	}
	return ioutil.ReadAll(rd)
}

func (d *Defaults) List(rev, dir string) (names []string, err error) {
	tree, err := d.treeAt(rev)
	if err != nil {
		return
	}

	dirPrefix := dir + "/"
	err = tree.Files().ForEach(func(f *object.File) (err error) {
		if !strings.HasPrefix(f.Name, dirPrefix) {
			return
		}
		if !strings.HasSuffix(f.Name, ".yaml") {
			return
		}
		names = append(names, strings.TrimSuffix(filepath.Base(f.Name), ".yaml"))
		return
	})
	return
}

func (d *Defaults) treeAt(rev string) (tree *object.Tree, err error) {
	h, err := d.repo.ResolveRevision(plumbing.Revision(rev))
	if err != nil {
		return
	}

	obj, err := d.repo.Object(plumbing.AnyObject, *h)
	if err != nil {
		return
	}

	for {
		switch o := obj.(type) {
		case *object.Tag: // tag -> commit
			obj, err = o.Object()

		case *object.Commit: // commit -> tree
			return o.Tree()

		default:
			err = object.ErrUnsupportedObject
		}

		if err != nil {
			return
		}
	}
}

func (d *Defaults) decodeDefault(rev, filePath string, value Rev) (err error) {
	ba, err := d.ReadAll(rev, filePath)

	if err != nil {
		return
	}

	return yaml.Unmarshal(ba, value)
}