From f471af5110804aa656f8be1e7ce3445f59c12515 Mon Sep 17 00:00:00 2001
From: Nicolas MASSE <nicolas.masse@itix.fr>
Date: Thu, 14 May 2020 14:19:59 +0200
Subject: [PATCH] config file as first cli argument

---
 README.md |  6 +++---
 go.mod    |  6 +++---
 main.go   | 47 +++++++++++++++++++++++++++++++++--------------
 3 files changed, 39 insertions(+), 20 deletions(-)

diff --git a/README.md b/README.md
index 7aced0d..fc8d622 100644
--- a/README.md
+++ b/README.md
@@ -19,7 +19,7 @@ go build -o mirror-github
 ## Usage
 
 ```sh
-cat > mirror-github.yaml <<EOF
+cat > config.yaml <<EOF
 GitHub:
   PersonalToken: GITHUB_PERSONAL_ACCESS_TOKEN
 Gitea:
@@ -29,9 +29,9 @@ EOF
 ```
 
 ```sh
-./mirror-github
+./mirror-github config.yaml
 ```
 
 ## In a crontab
 
-In a crontab, you can add the `LogFile` directive to `mirror-github.yaml` to collect logs over multiple runs.
+In a crontab, you can add the `LogFile` directive to `config.yaml` to collect logs over multiple runs.
diff --git a/go.mod b/go.mod
index 18799c6..c04d384 100644
--- a/go.mod
+++ b/go.mod
@@ -4,8 +4,8 @@ go 1.14
 
 require (
 	code.gitea.io/sdk/gitea v0.11.1-0.20200511190314-2883376503c8
-	github.com/google/go-github v17.0.0+incompatible // indirect
+	github.com/google/go-github v17.0.0+incompatible
 	github.com/google/go-querystring v1.0.0 // indirect
-	github.com/spf13/viper v1.7.0 // indirect
-	golang.org/x/oauth2 v0.0.0-20200107190931-bf48bf16ab8d // indirect
+	github.com/spf13/viper v1.7.0
+	golang.org/x/oauth2 v0.0.0-20200107190931-bf48bf16ab8d
 )
diff --git a/main.go b/main.go
index 8c4f931..e537a0c 100644
--- a/main.go
+++ b/main.go
@@ -45,18 +45,29 @@ func (list GHRepositoryList) Swap(i, j int) {
 }
 
 func initConfig() {
-	viper.SetConfigName("mirror-github") // name of config file (without extension)
-	viper.AddConfigPath("/etc/")
-	viper.AddConfigPath("$HOME/.mirror-github")
-	viper.AddConfigPath(".") // optionally look for config in the working directory
-	err := viper.ReadInConfig()
+	if len(os.Args) != 2 {
+		fmt.Printf("Usage: %s config.yaml\n", os.Args[0])
+		os.Exit(1)
+	}
+
+	fd, err := os.Open(os.Args[1])
+	if err != nil {
+		fmt.Printf("open: %s: %s\n", os.Args[0], err)
+		os.Exit(1)
+	}
+	defer fd.Close()
+
+	viper.SetConfigType("yaml")
+	err = viper.ReadConfig(fd)
 	if err != nil {
-		panic(fmt.Errorf("Cannot read config file: %s\n", err))
+		fmt.Println(err)
+		os.Exit(1)
 	}
 
 	for _, config := range []string{"GitHub.PersonalToken", "Gitea.ServerURL", "Gitea.PersonalToken"} {
 		if viper.GetString(config) == "" {
-			panic(fmt.Sprintf("key %s is missing from configuration file", config))
+			fmt.Printf("key %s is missing from configuration file\n", config)
+			os.Exit(1)
 		}
 	}
 }
@@ -66,7 +77,8 @@ func initLogFile() {
 	if logFile != "" {
 		logHandle, err := os.OpenFile(logFile, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0600)
 		if err != nil {
-			panic(fmt.Errorf("Cannot open log file '%s': %s\n", logFile, err))
+			fmt.Printf("Cannot open log file '%s': %s\n", logFile, err)
+			os.Exit(1)
 		}
 		log.SetOutput(logHandle)
 	}
@@ -206,7 +218,7 @@ func computeRepositoriesToMigrate(gh GHRepositoryList, gt GTRepositoryList) GHRe
 	return toMigrate
 }
 
-func migrate(ghRepo *github.Repository, gtClient *gitea.Client, gtUser *gitea.User) {
+func migrate(ghRepo *github.Repository, gtClient *gitea.Client, gtUser *gitea.User) error {
 	migrationOptions := gitea.MigrateRepoOption{
 		CloneAddr:    *ghRepo.CloneURL,
 		AuthUsername: *ghRepo.Owner.Login,
@@ -218,9 +230,7 @@ func migrate(ghRepo *github.Repository, gtClient *gitea.Client, gtUser *gitea.Us
 		Description:  *ghRepo.Description,
 	}
 	_, err := gtClient.MigrateRepo(migrationOptions)
-	if err != nil {
-		log.Fatal(fmt.Sprintf("MigrateRepo: %s: %s", ghRepo.Name, err))
-	}
+	return err
 }
 
 func main() {
@@ -249,8 +259,17 @@ func main() {
 	log.Printf("There are %d repositories to migrate from GitHub to Gitea.\n", len(toMigrate))
 
 	// Migrate each repository
+	var ok bool = true
 	for _, repo := range toMigrate {
-		fmt.Printf("Migrating %s...\n", *repo.Name)
-		migrate(repo, gtClient, gtUserInfo)
+		log.Printf("Migrating %s...\n", *repo.Name)
+		err := migrate(repo, gtClient, gtUserInfo)
+		if err != nil {
+			log.Printf("MigrateRepo: %s: %s", *repo.Name, err)
+			ok = false
+		}
+	}
+
+	if !ok {
+		os.Exit(1)
 	}
 }