From 1ee9b72fd97d0a0e49ec2c0bad3fe892fa9d70a3 Mon Sep 17 00:00:00 2001
From: Thales Costa <thalescostaa@gmail.com>
Date: Wed, 22 Feb 2023 14:41:26 +0000
Subject: [PATCH] fix(accounts): avoiding break if have more than 20 accounts

---
 pkg/aws/sso.go | 20 +++++++++++++++-----
 1 file changed, 15 insertions(+), 5 deletions(-)

diff --git a/pkg/aws/sso.go b/pkg/aws/sso.go
index 59cb897..45f3737 100644
--- a/pkg/aws/sso.go
+++ b/pkg/aws/sso.go
@@ -171,12 +171,22 @@ func newOIDCToken(ctx context.Context, startUrl, region string) (*ssooidc.Create
 }
 
 func ListAccounts(ctx context.Context, client *sso.Client, token *string) []types.AccountInfo {
-	acc, err := client.ListAccounts(ctx, &sso.ListAccountsInput{AccessToken: token})
-	if err != nil {
-		log.Fatalf("unable to list accounts, %v", err)
+	var (
+		nextToken *string
+		accounts  []types.AccountInfo
+	)
+	for {
+		acc, err := client.ListAccounts(ctx, &sso.ListAccountsInput{AccessToken: token, NextToken: nextToken, MaxResults: aws.Int32(50)})
+		if err != nil {
+			log.Fatalf("unable to list accounts, %v", err)
+		}
+		accounts = append(accounts, acc.AccountList...)
+		nextToken = acc.NextToken
+		if nextToken == nil {
+			break
+		}
 	}
-
-	return acc.AccountList
+	return accounts
 }
 
 func GetRolesByAccount(ctx context.Context, client *sso.Client, account types.AccountInfo, token *string) []types.RoleInfo {