From 26fe4681fcfd42ad9d2db5e0b6ea9acdfed15a01 Mon Sep 17 00:00:00 2001
From: eyalbe4 <eyalb@jfrog.com>
Date: Fri, 20 Dec 2024 23:02:31 +0200
Subject: [PATCH] Add missing file

---
 utils/usage/visibility_system_manager.go | 68 ++++++++++++++++++++++++
 1 file changed, 68 insertions(+)
 create mode 100644 utils/usage/visibility_system_manager.go

diff --git a/utils/usage/visibility_system_manager.go b/utils/usage/visibility_system_manager.go
new file mode 100644
index 000000000..4554e21ec
--- /dev/null
+++ b/utils/usage/visibility_system_manager.go
@@ -0,0 +1,68 @@
+package usage
+
+import (
+	"encoding/json"
+	"os"
+
+	"github.com/jfrog/jfrog-cli-core/v2/artifactory/utils"
+	"github.com/jfrog/jfrog-cli-core/v2/utils/config"
+	"github.com/jfrog/jfrog-cli-core/v2/utils/coreutils"
+)
+
+type VisibilitySystemManager struct {
+	serverDetails *config.ServerDetails
+}
+
+func NewVisibilitySystemManager(serverDetails *config.ServerDetails) *VisibilitySystemManager {
+	return &VisibilitySystemManager{
+		serverDetails: serverDetails,
+	}
+}
+
+type labels struct {
+	ProductID                            string `json:"product_id"`
+	FeatureID                            string `json:"feature_id"`
+	OIDCUsed                             string `json:"oidc_used"`
+	JobID                                string `json:"job_id"`
+	RunID                                string `json:"run_id"`
+	GitRepo                              string `json:"git_repo"`
+	GhTokenForCodeScanningAlertsProvided string `json:"gh_token_for_code_scanning_alerts_provided"`
+}
+
+type visibilityMetric struct {
+	Value       int    `json:"value"`
+	MetricsName string `json:"metrics_name"`
+	Labels      labels `json:"labels"`
+}
+
+func createMetric(commandName string) ([]byte, error) {
+	metricLabels := labels{
+		ProductID:                            coreutils.GetCliUserAgentName(),
+		FeatureID:                            commandName,
+		OIDCUsed:                             os.Getenv("JFROG_CLI_USAGE_OIDC_USED"),
+		JobID:                                os.Getenv("JFROG_CLI_USAGE_JOB_ID"),
+		RunID:                                os.Getenv("JFROG_CLI_USAGE_RUN_ID"),
+		GitRepo:                              os.Getenv("JFROG_CLI_USAGE_GIT_REPO"),
+		GhTokenForCodeScanningAlertsProvided: os.Getenv("JFROG_CLI_USAGE_GH_TOKEN_FOR_CODE_SCANNING_ALERTS_PROVIDED"),
+	}
+
+	metric := visibilityMetric{
+		Value:       1,
+		MetricsName: "jfcli_commands_countaa",
+		Labels:      metricLabels,
+	}
+
+	return json.Marshal(metric)
+}
+
+func (vsm *VisibilitySystemManager) SendUsage(commandName string) error {
+	manager, err := utils.CreateJfConnectServiceManager(vsm.serverDetails)
+	if err != nil {
+		return err
+	}
+	metric, err := createMetric(commandName)
+	if err != nil {
+		return err
+	}
+	return manager.LogMetric(metric)
+}