kubeshark/cli/telemetry/telemetry.go

package telemetry

import (
	"bytes"
	"encoding/json"
	"fmt"
	"net/http"
	"os"
	"time"

	"github.com/denisbrodbeck/machineid"
	"github.com/up9inc/mizu/cli/apiserver"
	"github.com/up9inc/mizu/cli/config"
	"github.com/up9inc/mizu/cli/mizu"
	"github.com/up9inc/mizu/logger"
)

const telemetryUrl = "https://us-east4-up9-prod.cloudfunctions.net/mizu-telemetry"

func ReportRun(cmd string, args interface{}) {
	if !shouldRunTelemetry() {
		logger.Log.Debug("not reporting telemetry")
		return
	}

	argsBytes, _ := json.Marshal(args)
	argsMap := map[string]interface{}{
		"cmd":  cmd,
		"args": string(argsBytes),
	}

	if err := sendTelemetry(argsMap); err != nil {
		logger.Log.Debug(err)
		return
	}

	logger.Log.Debugf("successfully reported telemetry for cmd %v", cmd)
}

func ReportTapTelemetry(apiProvider *apiserver.Provider, args interface{}, startTime time.Time) {
	if !shouldRunTelemetry() {
		logger.Log.Debug("not reporting telemetry")
		return
	}

	generalStats, err := apiProvider.GetGeneralStats()
	if err != nil {
		logger.Log.Debugf("[ERROR] failed to get general stats from api server %v", err)
		return
	}
	argsBytes, _ := json.Marshal(args)
	argsMap := map[string]interface{}{
		"cmd":                    "tap",
		"args":                   string(argsBytes),
		"executionTimeInSeconds": int(time.Since(startTime).Seconds()),
		"apiCallsCount":          generalStats["EntriesCount"],
		"trafficVolumeInGB":      generalStats["EntriesVolumeInGB"],
	}

	if err := sendTelemetry(argsMap); err != nil {
		logger.Log.Debug(err)
		return
	}

	logger.Log.Debug("successfully reported telemetry of tap command")
}

func shouldRunTelemetry() bool {
	if _, present := os.LookupEnv(mizu.DEVENVVAR); present {
		return false
	}
	if !config.Config.Telemetry {
		return false
	}

	return mizu.Branch == "main" || mizu.Branch == "develop"
}

func sendTelemetry(argsMap map[string]interface{}) error {
	argsMap["component"] = "mizu_cli"
	argsMap["buildTimestamp"] = mizu.BuildTimestamp
	argsMap["branch"] = mizu.Branch
	argsMap["version"] = mizu.Ver
	argsMap["platform"] = mizu.Platform

	if machineId, err := machineid.ProtectedID("mizu"); err == nil {
		argsMap["machineId"] = machineId
	}

	jsonValue, _ := json.Marshal(argsMap)

	if resp, err := http.Post(telemetryUrl, "application/json", bytes.NewBuffer(jsonValue)); err != nil {
		return fmt.Errorf("ERROR: failed sending telemetry, err: %v, response %v", err, resp)
	}

	return nil
}