git-lfs/lfs/setup.go

package lfs

import (
	"errors"
	"fmt"
	"github.com/github/git-lfs/git"
	"io/ioutil"
	"os"
	"path/filepath"
	"regexp"
)

var (
	valueRegexp           = regexp.MustCompile("\\Agit[\\-\\s]media")
	prePushHook           = []byte("#!/bin/sh\ngit lfs push --stdin \"$@\"\n")
	NotInARepositoryError = errors.New("Not in a repository")
)

type HookExists struct {
	Name string
	Path string
}

func (e *HookExists) Error() string {
	return fmt.Sprintf("Hook already exists: %s", e.Name)
}

func InstallHooks(force bool) error {
	if !InRepo() {
		return NotInARepositoryError
	}

	if err := os.MkdirAll(filepath.Join(LocalGitDir, "hooks"), 0755); err != nil {
		return err
	}

	hookPath := filepath.Join(LocalGitDir, "hooks", "pre-push")
	if _, err := os.Stat(hookPath); err == nil && !force {
		return &HookExists{"pre-push", hookPath}
	} else {
		return ioutil.WriteFile(hookPath, prePushHook, 0755)
	}
}

func InstallFilters() error {
	var err error
	err = setFilter("clean")
	if err == nil {
		err = setFilter("smudge")
	}
	if err == nil {
		err = requireFilters()
	}
	return err
}

func setFilter(filterName string) error {
	key := fmt.Sprintf("filter.lfs.%s", filterName)
	value := fmt.Sprintf("git lfs %s %%f", filterName)

	existing := git.Config.Find(key)
	if shouldReset(existing) {
		git.Config.UnsetGlobal(key)
		git.Config.SetGlobal(key, value)
	} else if existing != value {
		return fmt.Errorf("The %s filter should be \"%s\" but is \"%s\"", filterName, value, existing)
	}

	return nil
}

func requireFilters() error {
	key := "filter.lfs.required"
	value := "true"

	existing := git.Config.Find(key)
	if shouldReset(existing) {
		git.Config.UnsetGlobal(key)
		git.Config.SetGlobal(key, value)
	} else if existing != value {
		return errors.New("Git LFS filters should be required but are not.")
	}

	return nil
}

func shouldReset(value string) bool {
	if len(value) == 0 {
		return true
	}
	return valueRegexp.MatchString(value)
}