git-lfs/gitmedia.go

package gitmedia

import (
	"fmt"
	"io/ioutil"
	"os"
	"os/exec"
	"path/filepath"
	"strings"
)

const Version = "0.0.1"

var (
	LargeSizeThreshold = 5 * 1024 * 1024
	TempDir            = filepath.Join(os.TempDir(), "git-media")
	LocalWorkingDir    string
	LocalGitDir        string
	LocalMediaDir      string
	LocalLogDir        string
)

func TempFile() (*os.File, error) {
	return ioutil.TempFile(TempDir, "")
}

func SimpleExec(name string, arg ...string) string {
	output, err := exec.Command(name, arg...).Output()
	if _, ok := err.(*exec.ExitError); ok {
		return ""
	} else if err != nil {
		Panic(err, "Error running %s %s", name, arg)
	}

	return strings.Trim(string(output), " \n")
}

func LocalMediaPath(sha string) string {
	path := filepath.Join(LocalMediaDir, sha[0:2], sha[2:4])
	if err := os.MkdirAll(path, 0744); err != nil {
		Panic(err, "Error trying to create local media directory: %s", path)
	}

	return filepath.Join(path, sha)
}

func Environ() []string {
	osEnviron := os.Environ()
	env := make([]string, 4, len(osEnviron)+4)
	env[0] = fmt.Sprintf("LocalWorkingDir=%s", LocalWorkingDir)
	env[1] = fmt.Sprintf("LocalGitDir=%s", LocalGitDir)
	env[2] = fmt.Sprintf("LocalMediaDir=%s", LocalMediaDir)
	env[3] = fmt.Sprintf("TempDir=%s", TempDir)

	for _, e := range os.Environ() {
		if !strings.Contains(e, "GIT_") {
			continue
		}
		env = append(env, e)
	}

	return env
}

func init() {
	LocalWorkingDir, LocalGitDir = resolveGitDir()
	LocalMediaDir = filepath.Join(LocalGitDir, "media")
	LocalLogDir = filepath.Join(LocalMediaDir, "logs")
	queueDir = setupQueueDir()

	if err := os.MkdirAll(TempDir, 0744); err != nil {
		Panic(err, "Error trying to create temp directory: %s", TempDir)
	}
}

func resolveGitDir() (string, string) {
	wd, err := os.Getwd()
	if err != nil {
		Panic(err, "Error reading working directory")
	}

	return recursiveResolveGitDir(wd)
}

func recursiveResolveGitDir(dir string) (string, string) {
	if filepath.Base(dir) == gitExt {
		return filepath.Dir(dir), dir
	}

	gitDir := filepath.Join(dir, gitExt)
	if info, err := os.Stat(gitDir); err == nil {
		if info.IsDir() {
			return dir, gitDir
		}
	}

	return recursiveResolveGitDir(filepath.Dir(dir))
}

const gitExt = ".git"