telemetry for customaimodes and secrets

Author: sawka

cmd/server/main-server.go

@@ -22,6 +22,7 @@ import (
 	"github.com/wavetermdev/waveterm/pkg/panichandler"
 	"github.com/wavetermdev/waveterm/pkg/remote/conncontroller"
 	"github.com/wavetermdev/waveterm/pkg/remote/fileshare/wshfs"
+	"github.com/wavetermdev/waveterm/pkg/secretstore"
 	"github.com/wavetermdev/waveterm/pkg/service"
 	"github.com/wavetermdev/waveterm/pkg/telemetry"
 	"github.com/wavetermdev/waveterm/pkg/telemetry/telemetrydata"
@@ -224,18 +225,25 @@ func updateTelemetryCounts(lastCounts telemetrydata.TEventProps) telemetrydata.T
 	customWidgets := fullConfig.CountCustomWidgets()
 	customAIPresets := fullConfig.CountCustomAIPresets()
 	customSettings := wconfig.CountCustomSettings()
+	customAIModes := fullConfig.CountCustomAIModes()
 
 	props.UserSet = &telemetrydata.TEventUserProps{
 		SettingsCustomWidgets:   customWidgets,
 		SettingsCustomAIPresets: customAIPresets,
 		SettingsCustomSettings:  customSettings,
+		SettingsCustomAIModes:   customAIModes,
+	}
+	
+	secretsCount, err := secretstore.CountSecrets()
+	if err == nil {
+		props.UserSet.SettingsSecretsCount = secretsCount
 	}
 
 	if utilfn.CompareAsMarshaledJson(props, lastCounts) {
 		return lastCounts
 	}
 	tevent := telemetrydata.MakeTEvent("app:counts", props)
-	err := telemetry.RecordTEvent(ctx, tevent)
+	err = telemetry.RecordTEvent(ctx, tevent)
 	if err != nil {
 		log.Printf("error recording counts tevent: %v\n", err)
 	}

pkg/secretstore/secretstore.go

@@ -279,6 +279,24 @@ func GetSecretNames() ([]string, error) {
 	return names, nil
 }
 
+func CountSecrets() (int, error) {
+	lock.Lock()
+	defer lock.Unlock()
+	
+	if !initialized {
+		return 0, fmt.Errorf("secret store not initialized")
+	}
+
+	count := 0
+	for name := range secrets {
+		if name == WriteTsKey {
+			continue
+		}
+		count++
+	}
+	return count, nil
+}
+
 func GetLinuxStorageBackend() (string, error) {
 	if runtime.GOOS != "linux" {
 		return "", nil

pkg/telemetry/telemetrydata/telemetrydata.go

@@ -84,6 +84,8 @@ type TEventUserProps struct {
 	SettingsCustomWidgets   int `json:"settings:customwidgets,omitempty"`
 	SettingsCustomAIPresets int `json:"settings:customaipresets,omitempty"`
 	SettingsCustomSettings  int `json:"settings:customsettings,omitempty"`
+	SettingsCustomAIModes   int `json:"settings:customaimodes,omitempty"`
+	SettingsSecretsCount    int `json:"settings:secretscount,omitempty"`
 }
 
 type TEventProps struct {

pkg/wconfig/settingsconfig.go

@@ -887,6 +887,18 @@ func (fc *FullConfigType) CountCustomAIPresets() int {
 	return count
 }
 
+// CountCustomAIModes returns the number of custom AI modes the user has defined.
+// Custom AI modes are identified as modes that don't start with "waveai@".
+func (fc *FullConfigType) CountCustomAIModes() int {
+	count := 0
+	for modeID := range fc.WaveAIModes {
+		if !strings.HasPrefix(modeID, "waveai@") {
+			count++
+		}
+	}
+	return count
+}
+
 // CountCustomSettings returns the number of settings in the user's settings file.
 // This excludes telemetry:enabled and autoupdate:channel which don't count as customizations.
 func CountCustomSettings() int {