2015-03-19 19:30:55 +00:00
|
|
|
package lfs
|
2013-11-05 17:07:03 +00:00
|
|
|
|
2013-11-05 17:45:01 +00:00
|
|
|
import (
|
2014-05-22 23:02:56 +00:00
|
|
|
"fmt"
|
2015-01-12 00:53:10 +00:00
|
|
|
"net/http"
|
2015-02-12 23:25:18 +00:00
|
|
|
"net/url"
|
2014-03-12 15:31:41 +00:00
|
|
|
"os"
|
2015-06-01 18:53:08 +00:00
|
|
|
"path/filepath"
|
2015-04-23 14:48:44 +00:00
|
|
|
"strconv"
|
2014-02-01 20:38:29 +00:00
|
|
|
"strings"
|
2015-04-23 01:07:52 +00:00
|
|
|
"sync"
|
2015-05-13 19:43:41 +00:00
|
|
|
|
|
|
|
"github.com/github/git-lfs/git"
|
2013-11-05 17:45:01 +00:00
|
|
|
)
|
|
|
|
|
2015-06-17 20:38:09 +00:00
|
|
|
var (
|
|
|
|
Config = NewConfig()
|
|
|
|
defaultRemote = "origin"
|
|
|
|
)
|
|
|
|
|
2013-11-05 17:07:03 +00:00
|
|
|
type Configuration struct {
|
2015-02-17 16:39:15 +00:00
|
|
|
CurrentRemote string
|
2015-06-03 21:33:00 +00:00
|
|
|
httpClient *HttpClient
|
2015-02-13 23:51:01 +00:00
|
|
|
redirectingHttpClient *http.Client
|
2015-06-13 23:13:58 +00:00
|
|
|
envVars map[string]string
|
2015-03-05 19:49:15 +00:00
|
|
|
isTracingHttp bool
|
2015-06-04 18:30:47 +00:00
|
|
|
isLoggingStats bool
|
2015-06-13 23:00:36 +00:00
|
|
|
|
|
|
|
loading sync.Mutex // guards initialization of gitConfig and remotes
|
|
|
|
gitConfig map[string]string
|
|
|
|
remotes []string
|
2013-11-05 17:07:03 +00:00
|
|
|
}
|
|
|
|
|
2015-03-05 19:59:52 +00:00
|
|
|
func NewConfig() *Configuration {
|
2015-03-05 20:04:00 +00:00
|
|
|
c := &Configuration{
|
|
|
|
CurrentRemote: defaultRemote,
|
2015-06-13 23:13:58 +00:00
|
|
|
envVars: make(map[string]string),
|
2015-03-05 19:59:52 +00:00
|
|
|
}
|
2015-06-16 23:30:16 +00:00
|
|
|
c.isTracingHttp = c.GetenvBool("GIT_CURL_VERBOSE", false)
|
|
|
|
c.isLoggingStats = c.GetenvBool("GIT_LOG_STATS", false)
|
2015-03-05 19:59:52 +00:00
|
|
|
return c
|
|
|
|
}
|
|
|
|
|
2015-06-13 23:13:58 +00:00
|
|
|
func (c *Configuration) Getenv(key string) string {
|
|
|
|
if i, ok := c.envVars[key]; ok {
|
|
|
|
return i
|
2015-03-29 19:16:28 +00:00
|
|
|
}
|
|
|
|
|
2015-06-13 23:13:58 +00:00
|
|
|
v := os.Getenv(key)
|
|
|
|
c.envVars[key] = v
|
|
|
|
return v
|
2015-03-29 19:16:28 +00:00
|
|
|
}
|
|
|
|
|
2015-06-16 23:30:16 +00:00
|
|
|
// GetenvBool parses a boolean environment variable and returns the result as a bool.
|
|
|
|
// If the environment variable is unset, empty, or if the parsing fails,
|
|
|
|
// the value of def (default) is returned instead.
|
|
|
|
func (c *Configuration) GetenvBool(key string, def bool) bool {
|
|
|
|
s := c.Getenv(key)
|
|
|
|
if len(s) == 0 {
|
|
|
|
return def
|
|
|
|
}
|
|
|
|
|
|
|
|
b, err := strconv.ParseBool(s)
|
|
|
|
if err != nil {
|
|
|
|
return def
|
|
|
|
}
|
|
|
|
return b
|
|
|
|
}
|
|
|
|
|
2015-03-27 22:36:33 +00:00
|
|
|
func (c *Configuration) Endpoint() Endpoint {
|
2015-03-19 19:30:55 +00:00
|
|
|
if url, ok := c.GitConfig("lfs.url"); ok {
|
2015-06-17 10:04:22 +00:00
|
|
|
return NewEndpoint(url)
|
2013-11-05 17:07:03 +00:00
|
|
|
}
|
|
|
|
|
2015-02-02 22:39:03 +00:00
|
|
|
if len(c.CurrentRemote) > 0 && c.CurrentRemote != defaultRemote {
|
2015-03-27 22:36:33 +00:00
|
|
|
if endpoint := c.RemoteEndpoint(c.CurrentRemote); len(endpoint.Url) > 0 {
|
2015-02-02 22:39:03 +00:00
|
|
|
return endpoint
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return c.RemoteEndpoint(defaultRemote)
|
2013-11-05 17:07:03 +00:00
|
|
|
}
|
2013-11-05 17:45:01 +00:00
|
|
|
|
2015-05-13 15:27:06 +00:00
|
|
|
func (c *Configuration) ConcurrentTransfers() int {
|
2015-04-23 14:48:44 +00:00
|
|
|
uploads := 3
|
|
|
|
|
2015-05-13 15:27:06 +00:00
|
|
|
if v, ok := c.GitConfig("lfs.concurrenttransfers"); ok {
|
2015-04-23 14:48:44 +00:00
|
|
|
n, err := strconv.Atoi(v)
|
2015-05-26 19:42:37 +00:00
|
|
|
if err == nil && n > 0 {
|
2015-04-23 14:48:44 +00:00
|
|
|
uploads = n
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return uploads
|
|
|
|
}
|
|
|
|
|
2015-05-07 18:37:30 +00:00
|
|
|
func (c *Configuration) BatchTransfer() bool {
|
|
|
|
if v, ok := c.GitConfig("lfs.batch"); ok {
|
2015-05-26 19:42:37 +00:00
|
|
|
if v == "true" || v == "" {
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
|
|
|
|
// Any numeric value except 0 is considered true
|
|
|
|
if n, err := strconv.Atoi(v); err == nil && n != 0 {
|
|
|
|
return true
|
|
|
|
}
|
2015-05-07 18:37:30 +00:00
|
|
|
}
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
2015-03-27 22:36:33 +00:00
|
|
|
func (c *Configuration) RemoteEndpoint(remote string) Endpoint {
|
2015-02-02 22:39:03 +00:00
|
|
|
if len(remote) == 0 {
|
|
|
|
remote = defaultRemote
|
|
|
|
}
|
2015-04-15 19:24:44 +00:00
|
|
|
if url, ok := c.GitConfig("remote." + remote + ".lfsurl"); ok {
|
2015-06-17 10:04:22 +00:00
|
|
|
return NewEndpoint(url)
|
2014-02-01 20:38:29 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
if url, ok := c.GitConfig("remote." + remote + ".url"); ok {
|
2015-06-17 10:04:22 +00:00
|
|
|
return NewEndpointFromCloneURL(url)
|
2015-06-13 22:55:23 +00:00
|
|
|
}
|
2014-03-19 21:21:52 +00:00
|
|
|
|
2015-06-13 22:55:23 +00:00
|
|
|
return Endpoint{}
|
|
|
|
}
|
2014-03-19 21:21:52 +00:00
|
|
|
|
2014-02-01 20:38:29 +00:00
|
|
|
func (c *Configuration) Remotes() []string {
|
2015-03-05 19:49:15 +00:00
|
|
|
c.loadGitConfig()
|
2014-02-01 20:38:29 +00:00
|
|
|
return c.remotes
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *Configuration) GitConfig(key string) (string, bool) {
|
2015-03-05 19:49:15 +00:00
|
|
|
c.loadGitConfig()
|
2015-01-12 00:53:10 +00:00
|
|
|
value, ok := c.gitConfig[strings.ToLower(key)]
|
2014-02-01 20:38:29 +00:00
|
|
|
return value, ok
|
|
|
|
}
|
|
|
|
|
2014-02-01 20:46:58 +00:00
|
|
|
func (c *Configuration) SetConfig(key, value string) {
|
2015-03-05 19:49:15 +00:00
|
|
|
c.loadGitConfig()
|
2014-02-01 20:46:58 +00:00
|
|
|
c.gitConfig[key] = value
|
|
|
|
}
|
|
|
|
|
2015-03-19 20:39:44 +00:00
|
|
|
func (c *Configuration) ObjectUrl(oid string) (*url.URL, error) {
|
2015-03-29 19:16:28 +00:00
|
|
|
return ObjectUrl(c.Endpoint(), oid)
|
2015-02-12 23:25:18 +00:00
|
|
|
}
|
|
|
|
|
2014-02-01 20:38:29 +00:00
|
|
|
func (c *Configuration) loadGitConfig() {
|
2015-04-23 01:07:52 +00:00
|
|
|
c.loading.Lock()
|
|
|
|
defer c.loading.Unlock()
|
|
|
|
|
2015-03-05 19:49:15 +00:00
|
|
|
if c.gitConfig != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2014-02-01 20:38:29 +00:00
|
|
|
uniqRemotes := make(map[string]bool)
|
|
|
|
|
|
|
|
c.gitConfig = make(map[string]string)
|
2014-03-12 15:31:41 +00:00
|
|
|
|
|
|
|
var output string
|
2014-09-23 15:40:23 +00:00
|
|
|
listOutput, err := git.Config.List()
|
2014-06-03 22:03:43 +00:00
|
|
|
if err != nil {
|
2014-06-05 18:48:23 +00:00
|
|
|
panic(fmt.Errorf("Error listing git config: %s", err))
|
2014-06-03 22:03:43 +00:00
|
|
|
}
|
|
|
|
|
2015-06-01 18:53:08 +00:00
|
|
|
configFile := filepath.Join(LocalWorkingDir, ".gitconfig")
|
|
|
|
fileOutput, err := git.Config.ListFromFile(configFile)
|
2014-06-03 22:03:43 +00:00
|
|
|
if err != nil {
|
2014-06-05 18:48:23 +00:00
|
|
|
panic(fmt.Errorf("Error listing git config from file: %s", err))
|
2014-06-03 22:03:43 +00:00
|
|
|
}
|
|
|
|
|
2015-04-20 16:38:17 +00:00
|
|
|
output = fileOutput + "\n" + listOutput
|
2014-03-12 15:31:41 +00:00
|
|
|
|
|
|
|
lines := strings.Split(output, "\n")
|
2014-02-01 20:38:29 +00:00
|
|
|
for _, line := range lines {
|
|
|
|
pieces := strings.SplitN(line, "=", 2)
|
2014-03-12 15:31:41 +00:00
|
|
|
if len(pieces) < 2 {
|
|
|
|
continue
|
|
|
|
}
|
2015-01-12 00:53:10 +00:00
|
|
|
key := strings.ToLower(pieces[0])
|
2014-02-01 20:38:29 +00:00
|
|
|
c.gitConfig[key] = pieces[1]
|
|
|
|
|
|
|
|
keyParts := strings.Split(key, ".")
|
|
|
|
if len(keyParts) > 1 && keyParts[0] == "remote" {
|
2014-02-01 21:04:40 +00:00
|
|
|
remote := keyParts[1]
|
|
|
|
uniqRemotes[remote] = remote == "origin"
|
2014-02-01 20:38:29 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-02-01 21:04:40 +00:00
|
|
|
c.remotes = make([]string, 0, len(uniqRemotes))
|
|
|
|
for remote, isOrigin := range uniqRemotes {
|
|
|
|
if isOrigin {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
c.remotes = append(c.remotes, remote)
|
2014-02-01 20:38:29 +00:00
|
|
|
}
|
|
|
|
}
|