97 lines
2.1 KiB
Go
97 lines
2.1 KiB
Go
// Copyright 2019 Google LLC.
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
// sync-gcsr implements a small utility that periodically mirrors a
|
|
// remote Google Cloud Source Repository to a local file path.
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"time"
|
|
|
|
git "gopkg.in/src-d/go-git.v4"
|
|
"gopkg.in/src-d/go-git.v4/plumbing/transport/http"
|
|
)
|
|
|
|
func EnvOr(key, def string) string {
|
|
v := os.Getenv(key)
|
|
if v == "" {
|
|
return def
|
|
}
|
|
|
|
return v
|
|
}
|
|
|
|
func updateRepo(repo *git.Repository, tree *git.Worktree, opts *git.PullOptions) error {
|
|
err := tree.Pull(opts)
|
|
if err == git.NoErrAlreadyUpToDate {
|
|
// nothing to do ...
|
|
return nil
|
|
} else if err != nil {
|
|
return err
|
|
}
|
|
|
|
log.Println("Updated local repository mirror")
|
|
return nil
|
|
}
|
|
|
|
func cloneRepo(dest, project, repo string, auth *http.BasicAuth) (*git.Repository, error) {
|
|
var cloneOpts = git.CloneOptions{
|
|
Auth: auth,
|
|
URL: fmt.Sprintf("https://source.developers.google.com/p/%s/r/%s", project, repo),
|
|
}
|
|
|
|
handle, err := git.PlainClone(dest, false, &cloneOpts)
|
|
|
|
if err == git.ErrRepositoryAlreadyExists {
|
|
handle, err = git.PlainOpen(dest)
|
|
}
|
|
|
|
return handle, err
|
|
}
|
|
|
|
func main() {
|
|
dest := EnvOr("SYNC_DEST", "/git/depot")
|
|
project := EnvOr("SYNC_PROJECT", "tazjins-infrastructure")
|
|
repo := EnvOr("SYNC_REPO", "depot")
|
|
user := os.Getenv("SYNC_USER")
|
|
pass := os.Getenv("SYNC_PASS")
|
|
|
|
log.Printf("Syncing repository '%s/%s' to destination '%s'", project, repo, dest)
|
|
|
|
var auth *http.BasicAuth
|
|
if user != "" && pass != "" {
|
|
auth = &http.BasicAuth{
|
|
Username: user,
|
|
Password: pass,
|
|
}
|
|
log.Println("Enabling basic authentication as user", user)
|
|
}
|
|
|
|
handle, err := cloneRepo(dest, project, repo, auth)
|
|
|
|
if err != nil {
|
|
log.Fatalf("Failed to clone repository: %s", err)
|
|
} else {
|
|
log.Println("Initiating update loop")
|
|
}
|
|
|
|
tree, err := handle.Worktree()
|
|
if err != nil {
|
|
log.Fatalln("Failed to open repository worktree:", err)
|
|
}
|
|
|
|
pullOpts := git.PullOptions{
|
|
Auth: auth,
|
|
Force: true,
|
|
}
|
|
|
|
for {
|
|
if err = updateRepo(handle, tree, &pullOpts); err != nil {
|
|
log.Fatalf("Failed to pull updated repository: %s", err)
|
|
}
|
|
time.Sleep(30 * time.Second) // TODO(tazjin): Config option for pull interval?
|
|
}
|
|
}
|