Limit preview identity results to 250, limit fields to pull to just id,name,email from search

Author: tyler-mairose-sp

assets/vhs/preview.tape

@@ -0,0 +1,27 @@
+Set FontSize 16
+Set Width 1200
+Set Height 1000
+
+Hide Sleep 2s Show
+
+Sleep 5s
+Type "sail transform preview -f transform_files/DeriveFirstInitialLastNameInUpper.json"
+Enter
+Set TypingSpeed 500ms 
+Sleep 5s
+Down
+Sleep 500ms
+Enter
+Sleep 4.5s
+Type "/"
+Sleep 3s
+Type "Brian"
+Sleep 5s
+Backspace 5
+Sleep 5s
+Type "Adam"
+Down
+Sleep 5s
+Enter
+Sleep 15s
+

cmd/transform/preview.go

@@ -122,7 +122,9 @@ func newPreviewCommand() *cobra.Command {
 					return err
 				}
 
-				formattedResponse, err := search.PerformSearch(*apiClient, searchObj)
+				searchObj.QueryResultFilter.Includes = []string{"id", "displayName", "email"}
+
+				formattedResponse, err := search.PerformSearchWithLimit(*apiClient, searchObj, 250)
 				if err != nil {
 					return err
 				}

internal/search/search.go

@@ -120,6 +120,78 @@ func PerformSearch(apiClient sailpoint.APIClient, search sailpointsdk.Search) (S
 	return SearchResults, nil
 }
 
+func PerformSearchWithLimit(apiClient sailpoint.APIClient, search sailpointsdk.Search, limit int32) (SearchResults, error) {
+	var SearchResults SearchResults
+
+	if limit == 0 {
+		limit = 500
+	}
+
+	ctx := context.TODO()
+	resp, r, err := apiClient.V3.SearchAPI.SearchPost(ctx).Search(search).Limit(limit).Execute()
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "Error: %v\n", err)
+		fmt.Fprintf(os.Stderr, "Full HTTP response: %v\n", r)
+	}
+
+	log.Debug("Search complete")
+
+	for i := 0; i < len(resp); i++ {
+		entry := resp[i]
+		switch entry["_type"] {
+		case "accountactivity":
+			var AccountActivity AccountActivity
+			err := mapstructure.Decode(entry, &AccountActivity)
+			if err != nil {
+				return SearchResults, err
+			}
+			SearchResults.AccountActivities = append(SearchResults.AccountActivities, AccountActivity)
+
+		case "accessprofile":
+			var AccessProfile AccessProfile
+			err := mapstructure.Decode(entry, &AccessProfile)
+			if err != nil {
+				return SearchResults, err
+			}
+			SearchResults.AccessProfiles = append(SearchResults.AccessProfiles, AccessProfile)
+
+		case "entitlement":
+			var Entitlement Entitlement
+			err := mapstructure.Decode(entry, &Entitlement)
+			if err != nil {
+				return SearchResults, err
+			}
+			SearchResults.Entitlements = append(SearchResults.Entitlements, Entitlement)
+
+		case "event":
+			var Event Event
+			err := mapstructure.Decode(entry, &Event)
+			if err != nil {
+				return SearchResults, err
+			}
+			SearchResults.Events = append(SearchResults.Events, Event)
+
+		case "identity":
+			var Identity Identity
+			err := mapstructure.Decode(entry, &Identity)
+			if err != nil {
+				return SearchResults, err
+			}
+			SearchResults.Identities = append(SearchResults.Identities, Identity)
+
+		case "role":
+			var Role Role
+			err := mapstructure.Decode(entry, &Role)
+			if err != nil {
+				return SearchResults, err
+			}
+			SearchResults.Roles = append(SearchResults.Roles, Role)
+		}
+	}
+
+	return SearchResults, nil
+}
+
 func IterateIndices(SearchResults SearchResults, searchQuery string, folderPath string, outputTypes []string) error {
 	if len(SearchResults.AccountActivities) > 0 {
 		fileName := "query=" + searchQuery + "&indices=AccountActivities"