calling-telemetry-api-post-valid-login

Author: cx-hitesh-madgulkar

internal/commands/auth.go

@@ -39,7 +39,7 @@ type ClientCreated struct {
 	Secret string `json:"secret"`
 }
 
-func NewAuthCommand(authWrapper wrappers.AuthWrapper) *cobra.Command {
+func NewAuthCommand(authWrapper wrappers.AuthWrapper, telemetryWrapper wrappers.TelemetryWrapper) *cobra.Command {
 	authCmd := &cobra.Command{
 		Use:   "auth",
 		Short: "Validate authentication and create OAuth2 credentials",
@@ -111,19 +111,25 @@ func NewAuthCommand(authWrapper wrappers.AuthWrapper) *cobra.Command {
 			`,
 			),
 		},
-		RunE: validLogin(),
+		RunE: validLogin(telemetryWrapper),
 	}
 	authCmd.AddCommand(createClientCmd, validLoginCmd)
 	return authCmd
 }
 
-func validLogin() func(cmd *cobra.Command, args []string) error {
+func validLogin(telemetryWrapper wrappers.TelemetryWrapper) func(cmd *cobra.Command, args []string) error {
 	return func(cmd *cobra.Command, args []string) error {
 		defer func() {
 			logger.PrintIfVerbose("Calling GetUniqueId func")
 			uniqueID := wrappers.GetUniqueID()
 			if uniqueID != "" {
 				logger.PrintIfVerbose("Set unique id: " + uniqueID)
+				err := telemetryWrapper.SendAIDataToLog(&wrappers.DataForAITelemetry{
+					UniqueID: uniqueID,
+				})
+				if err != nil {
+					logger.PrintIfVerbose("Failed to send telemetry data: " + err.Error())
+				}
 			}
 		}()
 		clientID := viper.GetString(params.AccessKeyIDConfigKey)

internal/commands/root.go

@@ -205,7 +205,7 @@ func NewAstCLI(
 	)
 
 	versionCmd := util.NewVersionCommand()
-	authCmd := NewAuthCommand(authWrapper)
+	authCmd := NewAuthCommand(authWrapper, telemetryWrapper)
 	utilsCmd := util.NewUtilsCommand(
 		gitHubWrapper,
 		azureWrapper,

internal/wrappers/client.go

@@ -1028,7 +1028,7 @@ func GetUniqueID() string {
 		if !ok {
 			continue
 		}
-		if strings.EqualFold(engineStr, "Checkmarx Developer Assist") {
+		if strings.EqualFold(engineStr, "Checkmarx One Assist") {
 			isAllowed = true
 			break
 		}