rename pkg/ to internal/

1 files changed, 138 insertions, 0 deletions

diff --git a/internal/prompt/git.go b/internal/prompt/git.go
new file mode 100644
index 0000000..8393c3e
--- /dev/null
+++ b/internal/prompt/git.go
@@ -0,0 +1,138 @@
+/*******************************************************************************
+*
+* Copyright 2017 Stefan Majewsky <majewsky@gmx.net>
+*
+* This program is free software: you can redistribute it and/or modify it under
+* the terms of the GNU General Public License as published by the Free Software
+* Foundation, either version 3 of the License, or (at your option) any later
+* version.
+*
+* This program is distributed in the hope that it will be useful, but WITHOUT ANY
+* WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
+* A PARTICULAR PURPOSE. See the GNU General Public License for more details.
+*
+* You should have received a copy of the GNU General Public License along with
+* this program. If not, see <http://www.gnu.org/licenses/>.
+*
+*******************************************************************************/
+
+package prompt
+
+import (
+	"fmt"
+	"io/ioutil"
+	"os"
+	"path/filepath"
+	"strings"
+)
+
+type gitRepo struct {
+	RootPath string
+	GitDir   string
+}
+
+//Returns two empty strings if `path` is not inside a Git repo.
+func findRepo(path string) (*gitRepo, error) {
+	//find .git directory or file
+	gitEntry := filepath.Join(path, ".git")
+	fi, err := os.Stat(gitEntry)
+	switch {
+	case err == nil:
+		//found - continue below with further checks
+	case !os.IsNotExist(err):
+		return nil, err
+	case path == "/":
+		return nil, nil
+	default:
+		return findRepo(filepath.Dir(path))
+	}
+
+	//found .git - what is it?
+	if fi.Mode().IsDir() {
+		//normal case - .git is a directory
+		return &gitRepo{RootPath: path, GitDir: gitEntry}, nil
+	}
+
+	//.git is a file (e.g. for submodules) - it contains a line like "gitdir: path/to/gitdir"
+	bytes, err := ioutil.ReadFile(gitEntry)
+	if err != nil {
+		return nil, err
+	}
+	for _, line := range strings.Split(string(bytes), "\n") {
+		line = strings.TrimSpace(line)
+		if strings.HasPrefix(line, "gitdir:") {
+			return &gitRepo{
+				RootPath: path,
+				GitDir:   filepath.Join(path, strings.TrimSpace(strings.TrimPrefix(line, "gitdir:"))),
+			}, nil
+		}
+	}
+
+	return nil, fmt.Errorf("read %s: missing gitdir directive", gitEntry)
+}
+
+func getRepoStatusField(repo *gitRepo) string {
+	if repo == nil {
+		return ""
+	}
+
+	bytes, err := ioutil.ReadFile(filepath.Join(repo.GitDir, "HEAD"))
+	if err != nil {
+		handleError(err)
+		return withType("git", withColor("1;41", "unknown"))
+	}
+	refSpec := strings.TrimSpace(string(bytes))
+
+	//is current HEAD detached?
+	if !strings.HasPrefix(refSpec, "ref: refs/") {
+		return formatRepoStatusField(withColor("1;41", "detached"), refSpec)
+	}
+
+	//current HEAD is a ref
+	refSpec = strings.TrimPrefix(refSpec, "ref: ")
+	refSpecDisplay := strings.TrimPrefix(refSpec, "refs/")
+	refSpecDisplay = strings.TrimPrefix(refSpecDisplay, "heads/")
+
+	//read file corresponding to refspec to find commit ID
+	bytes, err = ioutil.ReadFile(filepath.Join(repo.GitDir, refSpec))
+	commitID := strings.TrimSpace(string(bytes))
+	if err != nil {
+		if os.IsNotExist(err) {
+			commitID = tryReadFromPackedRefs(repo, refSpec)
+			if commitID == "" {
+				commitID = withColor("37", "blank")
+			}
+		} else {
+			handleError(err)
+			commitID = withColor("1;41", "unknown")
+		}
+	}
+
+	return formatRepoStatusField(refSpecDisplay, commitID)
+}
+
+func tryReadFromPackedRefs(repo *gitRepo, refSpec string) string {
+	bytes, err := ioutil.ReadFile(filepath.Join(repo.GitDir, "packed-refs"))
+	if err != nil {
+		return ""
+	}
+	for _, line := range strings.Split(string(bytes), "\n") {
+		line = strings.TrimSpace(line)
+		if line == "" || strings.HasPrefix(line, "#") {
+			continue
+		}
+		fields := strings.Fields(line)
+		if len(fields) == 2 && fields[1] == refSpec {
+			return fields[0]
+		}
+	}
+	return ""
+}
+
+func formatRepoStatusField(refSpec, commitID string) string {
+	//shorten plain commit IDs from 40 to 10 bytes
+	if len(commitID) == 40 && !strings.Contains(commitID, "\x1B") {
+		commitID = commitID[0:10]
+	}
+	return withType("git", refSpec+"/"+commitID)
+}