Make /git a parameter rather than const

This makes easier testing.
This commit is contained in:
Tim Hockin 2016-08-20 20:52:06 -07:00
parent 0a403496bf
commit c39a9369b9
1 changed files with 20 additions and 21 deletions

41
main.go
View File

@ -35,11 +35,10 @@ import (
"time" "time"
) )
const volMount = "/git"
var flRepo = flag.String("repo", envString("GIT_SYNC_REPO", ""), "git repo url") var flRepo = flag.String("repo", envString("GIT_SYNC_REPO", ""), "git repo url")
var flBranch = flag.String("branch", envString("GIT_SYNC_BRANCH", "master"), "git branch") var flBranch = flag.String("branch", envString("GIT_SYNC_BRANCH", "master"), "git branch")
var flRev = flag.String("rev", envString("GIT_SYNC_REV", "HEAD"), "git rev") var flRev = flag.String("rev", envString("GIT_SYNC_REV", "HEAD"), "git rev")
var flRoot = flag.String("root", envString("GIT_SYNC_ROOT", "/git"), "root directory for git operations")
var flDest = flag.String("dest", envString("GIT_SYNC_DEST", ""), "destination subdirectory path within volume") var flDest = flag.String("dest", envString("GIT_SYNC_DEST", ""), "destination subdirectory path within volume")
var flWait = flag.Int("wait", envInt("GIT_SYNC_WAIT", 0), "number of seconds to wait before next sync") var flWait = flag.Int("wait", envInt("GIT_SYNC_WAIT", 0), "number of seconds to wait before next sync")
var flOneTime = flag.Bool("one-time", envBool("GIT_SYNC_ONE_TIME", false), "exit after the initial checkout") var flOneTime = flag.Bool("one-time", envBool("GIT_SYNC_ONE_TIME", false), "exit after the initial checkout")
@ -114,7 +113,7 @@ func main() {
initialSync := true initialSync := true
failCount := 0 failCount := 0
for { for {
if err := syncRepo(*flRepo, *flDest, *flBranch, *flRev, *flDepth); err != nil { if err := syncRepo(*flRepo, *flRoot, *flDest, *flBranch, *flRev, *flDepth); err != nil {
if initialSync || failCount >= *flMaxSyncFailures { if initialSync || failCount >= *flMaxSyncFailures {
log.Fatalf("error syncing repo: %v", err) log.Fatalf("error syncing repo: %v", err)
} }
@ -140,27 +139,27 @@ func main() {
} }
// updateSymlink atomically swaps the symlink to point at the specified directory and cleans up the previous worktree. // updateSymlink atomically swaps the symlink to point at the specified directory and cleans up the previous worktree.
func updateSymlink(link, newDir string) error { func updateSymlink(gitRoot, link, newDir string) error {
// Get currently-linked repo directory (to be removed), unless it doesn't exist // Get currently-linked repo directory (to be removed), unless it doesn't exist
currentDir, err := filepath.EvalSymlinks(path.Join(volMount, link)) currentDir, err := filepath.EvalSymlinks(path.Join(gitRoot, link))
if err != nil && !os.IsNotExist(err) { if err != nil && !os.IsNotExist(err) {
return fmt.Errorf("error accessing symlink: %v", err) return fmt.Errorf("error accessing symlink: %v", err)
} }
// newDir is /git/rev-..., we need to change it to relative path. // newDir is /git/rev-..., we need to change it to relative path.
// Volume in other container may not be mounted at /git, so the symlink can't point to /git. // Volume in other container may not be mounted at /git, so the symlink can't point to /git.
newDirRelative, err := filepath.Rel(volMount, newDir) newDirRelative, err := filepath.Rel(gitRoot, newDir)
if err != nil { if err != nil {
return fmt.Errorf("error converting to relative path: %v", err) return fmt.Errorf("error converting to relative path: %v", err)
} }
if _, err := runCommand("ln", volMount, []string{"-snf", newDirRelative, "tmp-link"}); err != nil { if _, err := runCommand("ln", gitRoot, []string{"-snf", newDirRelative, "tmp-link"}); err != nil {
return fmt.Errorf("error creating symlink: %v", err) return fmt.Errorf("error creating symlink: %v", err)
} }
log.Printf("create symlink %v->%v", "tmp-link", newDirRelative) log.Printf("create symlink %v->%v", "tmp-link", newDirRelative)
if _, err := runCommand("mv", volMount, []string{"-T", "tmp-link", link}); err != nil { if _, err := runCommand("mv", gitRoot, []string{"-T", "tmp-link", link}); err != nil {
return fmt.Errorf("error replacing symlink: %v", err) return fmt.Errorf("error replacing symlink: %v", err)
} }
@ -174,7 +173,7 @@ func updateSymlink(link, newDir string) error {
log.Printf("remove %v", currentDir) log.Printf("remove %v", currentDir)
output, err := runCommand("git", volMount, []string{"worktree", "prune"}) output, err := runCommand("git", gitRoot, []string{"worktree", "prune"})
if err != nil { if err != nil {
return err return err
} }
@ -186,9 +185,9 @@ func updateSymlink(link, newDir string) error {
} }
// addWorktreeAndSwap creates a new worktree and calls updateSymlink to swap the symlink to point to the new worktree // addWorktreeAndSwap creates a new worktree and calls updateSymlink to swap the symlink to point to the new worktree
func addWorktreeAndSwap(dest, branch, rev string) error { func addWorktreeAndSwap(gitRoot, dest, branch, rev string) error {
// fetch branch // fetch branch
output, err := runCommand("git", volMount, []string{"fetch", "origin", branch}) output, err := runCommand("git", gitRoot, []string{"fetch", "origin", branch})
if err != nil { if err != nil {
return err return err
} }
@ -197,8 +196,8 @@ func addWorktreeAndSwap(dest, branch, rev string) error {
// add worktree in subdir // add worktree in subdir
rand.Seed(time.Now().UnixNano()) rand.Seed(time.Now().UnixNano())
worktreePath := path.Join(volMount, "rev-"+strconv.Itoa(rand.Int())) worktreePath := path.Join(gitRoot, "rev-"+strconv.Itoa(rand.Int()))
output, err = runCommand("git", volMount, []string{"worktree", "add", worktreePath, "origin/" + branch}) output, err = runCommand("git", gitRoot, []string{"worktree", "add", worktreePath, "origin/" + branch})
if err != nil { if err != nil {
return err return err
} }
@ -207,7 +206,7 @@ func addWorktreeAndSwap(dest, branch, rev string) error {
// .git file in worktree directory holds a reference to /git/.git/worktrees/<worktree-dir-name> // .git file in worktree directory holds a reference to /git/.git/worktrees/<worktree-dir-name>
// Replace it with a reference using relative paths, so that other containers can use a different volume mount name // Replace it with a reference using relative paths, so that other containers can use a different volume mount name
worktreePathRelative, err := filepath.Rel(volMount, worktreePath) worktreePathRelative, err := filepath.Rel(gitRoot, worktreePath)
if err != nil { if err != nil {
return err return err
} }
@ -232,10 +231,10 @@ func addWorktreeAndSwap(dest, branch, rev string) error {
} }
} }
return updateSymlink(dest, worktreePath) return updateSymlink(gitRoot, dest, worktreePath)
} }
func initRepo(repo, dest, branch, rev string, depth int) error { func initRepo(repo, dest, branch, rev string, depth int, gitRoot string) error {
// clone repo // clone repo
args := []string{"clone", "--no-checkout", "-b", branch} args := []string{"clone", "--no-checkout", "-b", branch}
if depth != 0 { if depth != 0 {
@ -243,7 +242,7 @@ func initRepo(repo, dest, branch, rev string, depth int) error {
args = append(args, string(depth)) args = append(args, string(depth))
} }
args = append(args, repo) args = append(args, repo)
args = append(args, volMount) args = append(args, gitRoot)
output, err := runCommand("git", "", args) output, err := runCommand("git", "", args)
if err != nil { if err != nil {
return err return err
@ -255,13 +254,13 @@ func initRepo(repo, dest, branch, rev string, depth int) error {
} }
// syncRepo syncs the branch of a given repository to the destination at the given rev. // syncRepo syncs the branch of a given repository to the destination at the given rev.
func syncRepo(repo, dest, branch, rev string, depth int) error { func syncRepo(repo, gitRoot, dest, branch, rev string, depth int) error {
target := path.Join(volMount, dest) target := path.Join(gitRoot, dest)
gitRepoPath := path.Join(target, ".git") gitRepoPath := path.Join(target, ".git")
_, err := os.Stat(gitRepoPath) _, err := os.Stat(gitRepoPath)
switch { switch {
case os.IsNotExist(err): case os.IsNotExist(err):
err = initRepo(repo, target, branch, rev, depth) err = initRepo(repo, target, branch, rev, depth, gitRoot)
if err != nil { if err != nil {
return err return err
} }
@ -278,7 +277,7 @@ func syncRepo(repo, dest, branch, rev string, depth int) error {
} }
} }
return addWorktreeAndSwap(dest, branch, rev) return addWorktreeAndSwap(gitRoot, dest, branch, rev)
} }
// gitRemoteChanged returns true if the remote HEAD is different from the local HEAD, false otherwise // gitRemoteChanged returns true if the remote HEAD is different from the local HEAD, false otherwise