Initial import

main.go

@@ -0,0 +1,181 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"log"
+	"os"
+	"time"
+
+	"gopkg.in/src-d/go-git.v4"
+	"gopkg.in/src-d/go-git.v4/plumbing/object"
+	"gopkg.in/src-d/go-git.v4/plumbing/transport"
+	"gopkg.in/src-d/go-git.v4/plumbing/transport/ssh"
+)
+
+var (
+	author   = flag.String("author", "gitomatic", "author name for git commits")
+	email    = flag.String("email", "gitomatic@fribbledom.com", "email address for git commits")
+	interval = flag.String("interval", "1m", "how often to check for changes")
+	privkey  = flag.String("privkey", "", "location of private key used for auth")
+)
+
+func gitAdd(w *git.Worktree, path string) error {
+	log.Printf("Adding file to work-tree: %s\n", path)
+	_, err := w.Add(path)
+	return err
+}
+
+func gitRemove(w *git.Worktree, path string) error {
+	log.Printf("Removing file from work-tree: %s\n", path)
+	_, err := w.Remove(path)
+	return err
+}
+
+func gitCommit(w *git.Worktree, message string) error {
+	log.Printf("Creating commit: %s", message)
+	_, err := w.Commit(message, &git.CommitOptions{
+		Author: &object.Signature{
+			Name:  *author,
+			Email: *email,
+			When:  time.Now(),
+		},
+	})
+	return err
+}
+
+func gitPush(r *git.Repository, auth transport.AuthMethod) error {
+	if !gitHasRemote(r) {
+		log.Println("Not pushing: no remotes configured.")
+		return nil
+	}
+
+	log.Println("Pushing changes...")
+	return r.Push(&git.PushOptions{
+		Auth: auth,
+	})
+}
+
+func gitPull(r *git.Repository, w *git.Worktree, auth transport.AuthMethod) error {
+	if !gitHasRemote(r) {
+		log.Println("Not pulling: no remotes configured.")
+		return nil
+	}
+
+	log.Println("Pulling changes...")
+	err := w.Pull(&git.PullOptions{
+		RemoteName: "origin",
+		Auth:       auth,
+	})
+	if err == transport.ErrEmptyRemoteRepository {
+		return nil
+	}
+	if err == git.NoErrAlreadyUpToDate {
+		return nil
+	}
+	return err
+}
+
+func gitHasRemote(r *git.Repository) bool {
+	remotes, _ := r.Remotes()
+	return len(remotes) > 0
+}
+
+func main() {
+	fmt.Println("git-o-matic")
+	flag.Parse()
+
+	if len(flag.Args()) < 1 {
+		fmt.Println("usage: gitomatic <path>")
+		os.Exit(1)
+	}
+
+	timeout, err := time.ParseDuration(*interval)
+	if err != nil {
+		panic(err)
+	}
+
+	var auth transport.AuthMethod
+	if len(*privkey) > 0 {
+		var err error
+		auth, err = ssh.NewPublicKeysFromFile("git", *privkey, "")
+		if err != nil {
+			panic(err)
+		}
+	}
+
+	path := flag.Args()[0]
+	r, err := git.PlainOpen(path)
+	if err != nil {
+		panic(err)
+	}
+	w, err := r.Worktree()
+	if err != nil {
+		panic(err)
+	}
+
+	for {
+		log.Println("Checking repository:", path)
+		err = gitPull(r, w, auth)
+		if err != nil {
+			panic(err)
+		}
+		status, err := w.Status()
+		if err != nil {
+			panic(err)
+		}
+
+		changes := 0
+		msg := ""
+		for path, s := range status {
+			log.Printf("%s %s %s\n", string(s.Worktree), string(s.Staging), path)
+			switch s.Worktree {
+			case git.Untracked:
+				log.Printf("New file detected: %s\n", path)
+				err := gitAdd(w, path)
+				if err != nil {
+					panic(err)
+				}
+
+				msg += fmt.Sprintf("Added %s.\n", path)
+				changes++
+
+			case git.Modified:
+				log.Printf("Modified file detected: %s\n", path)
+				err := gitAdd(w, path)
+				if err != nil {
+					panic(err)
+				}
+
+				msg += fmt.Sprintf("Updated %s.\n", path)
+				changes++
+
+			case git.Deleted:
+				log.Printf("Deleted file detected: %s\n", path)
+				err := gitRemove(w, path)
+				if err != nil {
+					panic(err)
+				}
+
+				msg += fmt.Sprintf("Removed %s.\n", path)
+				changes++
+			}
+		}
+
+		if changes == 0 {
+			log.Println("No changes detected.")
+		} else {
+			err = gitCommit(w, msg)
+			if err != nil {
+				panic(err)
+			}
+			err = gitPush(r, auth)
+			if err != nil {
+				panic(err)
+			}
+		}
+
+		log.Printf("Sleeping until next check in %s...\n", timeout)
+		time.Sleep(timeout)
+	}
+}