adding user sync flag (#7)

* adding user sync flag * forgot to remove spinner * wrong spinner, putting back
mona-actions · Nov 1, 2023 · 95fcc1c · 95fcc1c
1 parent fac870f
commit 95fcc1c
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -40,6 +40,7 @@ Flags:
   -a, --source-token string          Source Organization GitHub token. Scopes: read:org, read:user, user:email
   -t, --target-organization string   Target Organization to sync teams from
   -b, --target-token string          Target Organization GitHub token. Scopes: admin:org
+  -u, --user-sync string             User sync mode. One of: all, disable (default "none")
 ```
 
 ### Mapping File Example

diff --git a/cmd/sync.go b/cmd/sync.go
@@ -24,6 +24,7 @@ var syncCmd = &cobra.Command{
 		targetToken := cmd.Flag("target-token").Value.String()
 		mappingFile := cmd.Flag("mapping-file").Value.String()
 		ghHostname := cmd.Flag("source-hostname").Value.String()
+		userSync := cmd.Flag("user-sync").Value.String()
 
 		// Set ENV variables
 		os.Setenv("GHMT_SOURCE_ORGANIZATION", sourceOrganization)
@@ -32,6 +33,7 @@ var syncCmd = &cobra.Command{
 		os.Setenv("GHMT_TARGET_TOKEN", targetToken)
 		os.Setenv("GHMT_MAPPING_FILE", mappingFile)
 		os.Setenv("GHMT_SOURCE_HOSTNAME", ghHostname)
+		os.Setenv("GHMT_USER_SYNC", userSync)
 
 		// Bind ENV variables in Viper
 		viper.BindEnv("SOURCE_ORGANIZATION")
@@ -40,6 +42,7 @@ var syncCmd = &cobra.Command{
 		viper.BindEnv("TARGET_TOKEN")
 		viper.BindEnv("MAPPING_FILE")
 		viper.BindEnv("SOURCE_HOSTNAME")
+		viper.BindEnv("USER_SYNC")
 		// Call syncTeams
 		sync.SyncTeams()
 	},
@@ -65,4 +68,6 @@ func init() {
 
 	syncCmd.Flags().StringP("source-hostname", "u", "", "GitHub Enterprise source hostname url (optional) Ex. https://github.example.com")
 
+	syncCmd.Flags().StringP("user-sync", "u", "all", "User sync mode. One of: all, disable (default \"none\")")
+
 }
diff --git a/internal/team/team.go b/internal/team/team.go
@@ -4,6 +4,7 @@ import (
 	"time"
 
 	"github.com/mona-actions/gh-migrate-teams/internal/api"
+	"github.com/spf13/viper"
 )
 
 type Teams []Team
@@ -104,8 +105,13 @@ func (t Team) CreateTeam() {
 		api.AddTeamRepository(t.Slug, repository.Name, repository.Permission)
 	}
 
-	for _, member := range t.Members {
-		api.AddTeamMember(t.Slug, member.Login)
+	// Check to see if user sync has been disabled
+	userSync := viper.GetString("USER_SYNC")
+
+	if userSync != "disable" {
+		for _, member := range t.Members {
+			api.AddTeamMember(t.Slug, member.Login)
+		}
 	}
 }