1 files changed, 131 insertions, 0 deletions

diff --git a/vendor/github.com/hashicorp/go-getter/get_hg.go b/vendor/github.com/hashicorp/go-getter/get_hg.go
new file mode 100644
index 0000000..820bdd4
--- /dev/null
+++ b/vendor/github.com/hashicorp/go-getter/get_hg.go
@@ -0,0 +1,131 @@
+package getter
+
+import (
+        "fmt"
+        "io/ioutil"
+        "net/url"
+        "os"
+        "os/exec"
+        "path/filepath"
+        "runtime"
+
+        urlhelper "github.com/hashicorp/go-getter/helper/url"
+)
+
+// HgGetter is a Getter implementation that will download a module from
+// a Mercurial repository.
+type HgGetter struct{}
+
+func (g *HgGetter) ClientMode(_ *url.URL) (ClientMode, error) {
+        return ClientModeDir, nil
+}
+
+func (g *HgGetter) Get(dst string, u *url.URL) error {
+        if _, err := exec.LookPath("hg"); err != nil {
+                return fmt.Errorf("hg must be available and on the PATH")
+        }
+
+        newURL, err := urlhelper.Parse(u.String())
+        if err != nil {
+                return err
+        }
+        if fixWindowsDrivePath(newURL) {
+                // See valid file path form on http://www.selenic.com/hg/help/urls
+                newURL.Path = fmt.Sprintf("/%s", newURL.Path)
+        }
+
+        // Extract some query parameters we use
+        var rev string
+        q := newURL.Query()
+        if len(q) > 0 {
+                rev = q.Get("rev")
+                q.Del("rev")
+
+                newURL.RawQuery = q.Encode()
+        }
+
+        _, err = os.Stat(dst)
+        if err != nil && !os.IsNotExist(err) {
+                return err
+        }
+        if err != nil {
+                if err := g.clone(dst, newURL); err != nil {
+                        return err
+                }
+        }
+
+        if err := g.pull(dst, newURL); err != nil {
+                return err
+        }
+
+        return g.update(dst, newURL, rev)
+}
+
+// GetFile for Hg doesn't support updating at this time. It will download
+// the file every time.
+func (g *HgGetter) GetFile(dst string, u *url.URL) error {
+        td, err := ioutil.TempDir("", "getter-hg")
+        if err != nil {
+                return err
+        }
+        if err := os.RemoveAll(td); err != nil {
+                return err
+        }
+
+        // Get the filename, and strip the filename from the URL so we can
+        // just get the repository directly.
+        filename := filepath.Base(u.Path)
+        u.Path = filepath.ToSlash(filepath.Dir(u.Path))
+
+        // If we're on Windows, we need to set the host to "localhost" for hg
+        if runtime.GOOS == "windows" {
+                u.Host = "localhost"
+        }
+
+        // Get the full repository
+        if err := g.Get(td, u); err != nil {
+                return err
+        }
+
+        // Copy the single file
+        u, err = urlhelper.Parse(fmtFileURL(filepath.Join(td, filename)))
+        if err != nil {
+                return err
+        }
+
+        fg := &FileGetter{Copy: true}
+        return fg.GetFile(dst, u)
+}
+
+func (g *HgGetter) clone(dst string, u *url.URL) error {
+        cmd := exec.Command("hg", "clone", "-U", u.String(), dst)
+        return getRunCommand(cmd)
+}
+
+func (g *HgGetter) pull(dst string, u *url.URL) error {
+        cmd := exec.Command("hg", "pull")
+        cmd.Dir = dst
+        return getRunCommand(cmd)
+}
+
+func (g *HgGetter) update(dst string, u *url.URL, rev string) error {
+        args := []string{"update"}
+        if rev != "" {
+                args = append(args, rev)
+        }
+
+        cmd := exec.Command("hg", args...)
+        cmd.Dir = dst
+        return getRunCommand(cmd)
+}
+
+func fixWindowsDrivePath(u *url.URL) bool {
+        // hg assumes a file:/// prefix for Windows drive letter file paths.
+        // (e.g. file:///c:/foo/bar)
+        // If the URL Path does not begin with a '/' character, the resulting URL
+        // path will have a file:// prefix. (e.g. file://c:/foo/bar)
+        // See http://www.selenic.com/hg/help/urls and the examples listed in
+        // http://selenic.com/repo/hg-stable/file/1265a3a71d75/mercurial/util.py#l1936
+        return runtime.GOOS == "windows" && u.Scheme == "file" &&
+                len(u.Path) > 1 && u.Path[0] != '/' && u.Path[1] == ':'
+}