diff --git a/internal/commands/telemetry.go b/internal/commands/telemetry.go index 414db7c80..3b5bbefe0 100644 --- a/internal/commands/telemetry.go +++ b/internal/commands/telemetry.go @@ -40,6 +40,9 @@ func telemetryAISubCommand(telemetryAIWrapper wrappers.TelemetryWrapper) *cobra. telemetryAICmd.PersistentFlags().String(params.SubTypeFlag, "", "Sub Type") telemetryAICmd.PersistentFlags().String(params.EngineFlag, "", "Engine") telemetryAICmd.PersistentFlags().String(params.AgentFlag, "", "Agent") + telemetryAICmd.PersistentFlags().String(params.ScanTypeFlag, "", "Scan Type") + telemetryAICmd.PersistentFlags().String(params.StatusFlag, "", "Status") + telemetryAICmd.PersistentFlags().Int(params.TotalCountFlag, 0, "Total Count") return telemetryAICmd } @@ -52,6 +55,9 @@ func runTelemetryAI(telemetryWrapper wrappers.TelemetryWrapper) func(*cobra.Comm subType, _ := cmd.Flags().GetString("sub-type") agent, _ := cmd.Flags().GetString("agent") engine, _ := cmd.Flags().GetString("engine") + scanType, _ := cmd.Flags().GetString("scan-type") + status, _ := cmd.Flags().GetString("status") + totalCount, _ := cmd.Flags().GetInt("total-count") err := telemetryWrapper.SendAIDataToLog(&wrappers.DataForAITelemetry{ AIProvider: aiProvider, @@ -60,6 +66,9 @@ func runTelemetryAI(telemetryWrapper wrappers.TelemetryWrapper) func(*cobra.Comm SubType: subType, Agent: agent, Engine: engine, + ScanType: scanType, + Status: status, + TotalCount: totalCount, }) if err != nil { diff --git a/internal/params/flags.go b/internal/params/flags.go index 7541e6d5f..85bfb4576 100644 --- a/internal/params/flags.go +++ b/internal/params/flags.go @@ -9,6 +9,9 @@ const ( TypeFlag = "type" SubTypeFlag = "sub-type" EngineFlag = "engine" + StatusFlag = "status" + scanTypeFlag = "scan-type" + TotalCountFlag = "total-count" OriginFlag = "origin" AgentFlagUsage = "Scan origin name" ApplicationName = "application-name" diff --git a/internal/wrappers/realtime-scanner-http.go b/internal/wrappers/realtime-scanner-http.go index be06b7379..68d0f3763 100644 --- a/internal/wrappers/realtime-scanner-http.go +++ b/internal/wrappers/realtime-scanner-http.go @@ -34,7 +34,7 @@ func (r RealtimeScannerHTTPWrapper) ScanPackages(packages *RealtimeScannerPackag } fn := func() (*http.Response, error) { - return SendHTTPRequest(http.MethodPost, fmt.Sprint(r.path, "/analyze-manifest"), bytes.NewBuffer(jsonBytes), true, clientTimeout) + return SendHTTPRequest(http.MethodPost, fmt.Sprint(r.path, "/scan/packages"), bytes.NewBuffer(jsonBytes), true, clientTimeout) } resp, err := retryHTTPRequest(fn, retryAttempts, retryDelay*time.Millisecond) if err != nil { diff --git a/internal/wrappers/telemetry.go b/internal/wrappers/telemetry.go index 30d872e93..58e8a5b73 100644 --- a/internal/wrappers/telemetry.go +++ b/internal/wrappers/telemetry.go @@ -7,6 +7,9 @@ type DataForAITelemetry struct { SubType string `json:"subType"` Agent string `json:"agent"` Engine string `json:"engine"` + ScanType string `json:"scanType"` + Status string `json:"status"` + TotalCount int `json:"totalCount"` } type TelemetryWrapper interface {