From 82ad52148fa7d587756b0720fd2dcf797977a4de Mon Sep 17 00:00:00 2001
From: Andrew Davidson <andrew@amdavidson.com>
Date: Sat, 24 Aug 2019 14:34:40 -0400
Subject: [PATCH] Basic working github backup

---
 .gitignore    |   2 +-
 Readme.md     |  17 ++++++++
 cmd/github.go | 112 ++++++++++++++++++++++++++------------------------
 cmd/root.go   |   2 +-
 4 files changed, 77 insertions(+), 56 deletions(-)

diff --git a/.gitignore b/.gitignore
index 41da0ad..1269488 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1 +1 @@
-./data
+data
diff --git a/Readme.md b/Readme.md
index 2d42075..6843ff3 100644
--- a/Readme.md
+++ b/Readme.md
@@ -4,5 +4,22 @@ A simple service to archive various web services.
 
 ## Running The Archivist
 
+The Archivist has a fairly typical command structure, the most basic command is in this format
+
+    archivist ServiceName backup
+
 ## Service Instructions
 
+### Github
+    
+To backup Github repositories, pull a [personal access token](https://github.com/settings/tokens) and add it 
+to your `.the-archivist.yaml` config file.
+
+    github: 
+        user: my-username
+        token: 238u2n3f98va8enf238as9d7hoi23dshaf
+
+Then you can run `archivist github backup` and all your personal repositories will be stored in your `data` directory.
+
+
+
diff --git a/cmd/github.go b/cmd/github.go
index 1cf547b..c880ac2 100644
--- a/cmd/github.go
+++ b/cmd/github.go
@@ -8,12 +8,11 @@ import (
 	"os"
 	"strings"
 	"time"
-	"sync"
 
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
+	//bolt "go.etcd.io/bbolt"
 	"gopkg.in/src-d/go-git.v4"
-	bolt "go.etcd.io/bbolt"
 )
 
 // githubCmd represents the github command
@@ -56,7 +55,7 @@ var backupCmd = &cobra.Command{
 }
 
 type repo struct {
-	FullName    string `json:"full_name"`
+	FullName    string    `json:"full_name"`
 	UpdatedDate time.Time `json:"updated_at"`
 	CreatedDate time.Time `json:"created_at"`
 }
@@ -145,68 +144,73 @@ func ghBackup() error {
 		fmt.Println("Backup up", C.Green("Github"), "repos for", C.Magenta(viper.Get("github.user")))
 	}
 
-	repos, err := ghGetRepos() 
+	repos, err := ghGetRepos()
 	if err != nil {
 		return err
 	}
 
-	var ghBackupDate time.Time
-
-	db.View(func (tx *bolt.Tx) error {
-		b := tx.Bucket([]byte("Github"))
-		ghBackupDate := b.Get([]byte("last-backup"))
-		return nil
-	})
-
-	tasks := make(chan repo)
-
-	// ToDo fix below statements completely non-functioning
-
-	for _, r := range repos {
-		if r.UpdatedDate > ghBackupDate {
-			tasks <- r
-		}
+	c := make(chan error)
+	for _, repo := range repos {
+		go ghBackupRepo(repo, c)
 	}
 
-	var wg sync.WaitGroup 
-	for i := 0; i < 5; i++ {
-		wg.Add(1)
-		go func () {
-			for repo := range tasks {
-				ghBackupRepo(repo)
-			}
-		}
-	}
-	wg.Wait()
-
-	// End ToDo
-}
-
-func ghBackupRepo(repo) error {
-	if Verbose {
-		fmt.Println("Backing up ", repo.FullName)
-	}
-
-	cloneURL := "https://" + viper.GetString("github.token") + "@github.com/" + repo.FullName
-
-	if Verbose {
-		_, err := git.PlainClone(DataDir+"github", false, &git.CloneOptions{
-			URL:      cloneURL,
-			Progress: os.Stdout,
-		})
-		if err != nil {
+	for i := 0; i < len(repos); i++ {
+		if <-c != nil {
 			return err
 		}
-	} else {
-		_, err := git.PlainClone(DataDir+"github", false, &git.CloneOptions{
+
+	}
+
+	return nil
+}
+
+func ghBackupRepo(r repo, c chan error) {
+
+	var tempErr error
+	tempErr = nil
+
+	if Verbose {
+		fmt.Println("Backing up", C.Blue(r.FullName))
+	}
+
+	backupPath := DataDir + "/github/" + r.FullName
+
+	if _, err := os.Stat(backupPath); os.IsNotExist(err) {
+		if Verbose {
+			fmt.Println(C.Blue(r.FullName), "does not exist. Cloning.")
+		}
+		cloneURL := "https://" + viper.GetString("github.token") + "@github.com/" + r.FullName
+
+		_, err := git.PlainClone(backupPath, false, &git.CloneOptions{
 			URL: cloneURL,
 		})
 		if err != nil {
-			return err
+			tempErr = err
+		}
+	} else {
+		if Verbose {
+			fmt.Println(C.Blue(r.FullName), "cloned already, running git pull")
+		}
+
+		l, err := git.PlainOpen(backupPath)
+		if err != nil {
+			fmt.Println(err)
+			tempErr = err
+		}
+		w, err := l.Worktree()
+		if err != nil {
+			fmt.Println(err)
+			tempErr = err
+		}
+		err = w.Pull(&git.PullOptions{RemoteName: "origin"})
+		if err != nil && err != git.NoErrAlreadyUpToDate {
+			fmt.Println(err)
+			tempErr = err
+		}
+		if Verbose {
+			ref, _ := l.Head()
+			fmt.Println(C.Blue(r.FullName), "updated to", C.Green(ref))
 		}
 	}
-
-	if Verbose {
-		fmt.Println("Done backing up", repo.FullName)
-	}
+	c <- tempErr
 }
diff --git a/cmd/root.go b/cmd/root.go
index 6d00514..1382466 100644
--- a/cmd/root.go
+++ b/cmd/root.go
@@ -46,7 +46,7 @@ func init() {
 	rootCmd.PersistentFlags().StringVarP(&DataDir, "data", "d", "./data", "Location to store data.")
 	viper.BindPFlag("data", rootCmd.PersistentFlags().Lookup("data"))
 
-	db, err := bolt.Open(data+"/archivist.db", 0600, nil)
+	db, err := bolt.Open(DataDir+"/archivist.db", 0600, nil)
 	if err != nil {
 		fmt.Println("Cannot open directory")
 	}