feat: migrate team member role too (#15)

mona-actions · May 13, 2024 · a674e43 · a674e43
1 parent f89b6e3
commit a674e43
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 8 deletions.
diff --git a/internal/api/api.go b/internal/api/api.go
@@ -125,6 +125,7 @@ func GetTeamMemberships(team string) []map[string]string {
 							Login string
 							Email string
 						}
+						Role string
 					}
 				} `graphql:"members(first: $first, after: $after)"`
 			} `graphql:"team(slug: $slug)"`
@@ -146,7 +147,7 @@ func GetTeamMemberships(team string) []map[string]string {
 		}
 
 		for _, member := range query.Organization.Team.Members.Edges {
-			members = append(members, map[string]string{"Login": member.Node.Login, "Email": member.Node.Email})
+			members = append(members, map[string]string{"Login": member.Node.Login, "Email": member.Node.Email, "Role": member.Role})
 		}
 
 		if !query.Organization.Team.Members.PageInfo.HasNextPage {
@@ -312,8 +313,9 @@ func CreateTeam(name string, description string, privacy string, parentTeamName
 		parentTeamID, err := GetTeamId(parentTeamName)
 		if err != nil {
 			fmt.Println(err)
+		} else {
+			t.ParentTeamID = &parentTeamID
 		}
-		t.ParentTeamID = &parentTeamID
 	}
 
 	_, _, err := client.Teams.CreateTeam(context.Background(), viper.Get("TARGET_ORGANIZATION").(string), t)
@@ -345,21 +347,22 @@ func AddTeamRepository(slug string, repo string, permission string) {
 	}
 }
 
-func AddTeamMember(slug string, member string) {
+func AddTeamMember(slug string, member string, role string) {
 	client := newGHRestClient(viper.GetString("TARGET_TOKEN"))
 
-	fmt.Println("Adding member to team: ", slug, member)
-	_, _, err := client.Teams.AddTeamMembershipBySlug(context.Background(), viper.Get("TARGET_ORGANIZATION").(string), slug, member, &github.TeamAddTeamMembershipOptions{Role: "member"})
+	role = strings.ToLower(role) // lowercase to match github api
+	fmt.Println("Adding member to team: ", slug, member, role)
+	_, _, err := client.Teams.AddTeamMembershipBySlug(context.Background(), viper.Get("TARGET_ORGANIZATION").(string), slug, member, &github.TeamAddTeamMembershipOptions{Role: role})
 	if err != nil {
-		fmt.Println("Error adding member to team: ", slug, member)
+		fmt.Println("Error adding member to team: ", slug, member, err)
 	}
 }
 
 func GetTeamId(TeamName string) (int64, error) {
 	client := newGHRestClient(viper.GetString("TARGET_TOKEN"))
 	team, _, err := client.Teams.GetTeamBySlug(context.Background(), viper.Get("TARGET_ORGANIZATION").(string), TeamName)
 	if err != nil {
-		fmt.Println("Error getting team ID: ", TeamName)
+		fmt.Println("Error getting parent team ID: ", TeamName)
 		return 0, err
 	}
 	return *team.ID, nil

diff --git a/internal/team/team.go b/internal/team/team.go
@@ -25,6 +25,7 @@ type Team struct {
 type Member struct {
 	Login string
 	Email string
+	Role  string
 }
 
 type Repository struct {
@@ -67,6 +68,7 @@ func getTeamMemberships(team string) []Member {
 		members = append(members, Member{
 			Login: member["Login"],
 			Email: member["Email"],
+			Role:  member["Role"],
 		})
 	}
 
@@ -113,7 +115,7 @@ func (t Team) CreateTeam() {
 
 	if userSync != "disable" {
 		for _, member := range t.Members {
-			api.AddTeamMember(t.Slug, member.Login)
+			api.AddTeamMember(t.Slug, member.Login, member.Role)
 		}
 	}
 }