about summary refs log tree commit diff
diff options
context:
space:
mode:
-rw-r--r--services/sync-gcsr/default.nix10
-rw-r--r--services/sync-gcsr/main.go80
2 files changed, 90 insertions, 0 deletions
diff --git a/services/sync-gcsr/default.nix b/services/sync-gcsr/default.nix
new file mode 100644
index 000000000000..114ff221bed1
--- /dev/null
+++ b/services/sync-gcsr/default.nix
@@ -0,0 +1,10 @@
+{ pkgs, ... }:
+
+pkgs.buildGo.program {
+  name = "sync-gcsr";
+  srcs = [ ./main.go ];
+
+  deps = with pkgs.third_party; map (p: p.gopkg) [
+    gopkgs."gopkg.in".src-d.go-git
+  ];
+}
diff --git a/services/sync-gcsr/main.go b/services/sync-gcsr/main.go
new file mode 100644
index 000000000000..0b295a2c817e
--- /dev/null
+++ b/services/sync-gcsr/main.go
@@ -0,0 +1,80 @@
+// 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 main() {
+	var dest = EnvOr("SYNC_DEST", "/git/depot")
+	var project = EnvOr("SYNC_PROJECT", "tazjins-infrastructure")
+	var repo = EnvOr("SYNC_REPO", "depot")
+	var user = os.Getenv("SYNC_USER")
+	var pass = os.Getenv("SYNC_PASS")
+
+	log.Printf("Syncing repository '%s/%s' to destination '%s'", project, repo, dest)
+
+	var cloneOpts = git.CloneOptions{
+		URL: fmt.Sprintf("https://source.developers.google.com/p/%s/r/%s", project, repo),
+	}
+
+	if user != "" && pass != "" {
+		cloneOpts.Auth = &http.BasicAuth{
+			Username: user,
+			Password: pass,
+		}
+		log.Println("Enabling basic authentication as user", user)
+	}
+
+	action := "clone"
+	handle, err := git.PlainClone(dest, true, &cloneOpts)
+
+	if err == git.ErrRepositoryAlreadyExists {
+		log.Println("Repository has already been cloned!")
+		handle, err = git.PlainOpen(dest)
+		action = "open"
+	}
+
+	if err != nil {
+		log.Fatalln("Failed to %s repository:", action, err)
+	} else {
+		log.Println("Initiating update loop")
+	}
+
+	fetchOpts := git.FetchOptions{
+		Auth: cloneOpts.Auth,
+	}
+
+	for {
+		time.Sleep(30 * time.Second) //  TODO(tazjin): Config option for fetch interval?
+		err = handle.Fetch(&fetchOpts)
+
+		if err == git.NoErrAlreadyUpToDate {
+			// no-op ...
+			err = nil
+		} else if err != nil {
+			log.Fatalf("Failed to fetch updated repository: %s", err)
+		} else {
+			log.Println("Fetched new updates from remote repository")
+		}
+	}
+}