gitpod/components/supervisor/cmd/send-analytics.go
Pudong Zheng 352393456b gitpod-cli: add analytics
Co-authored-by: Andrea Falzetti <andrea@gitpod.io>
2023-02-08 17:43:46 +01:00

77 lines
2.0 KiB
Go

// Copyright (c) 2023 Gitpod GmbH. All rights reserved.
// Licensed under the GNU Affero General Public License (AGPL).
// See License.AGPL.txt in the project root for license information.
package cmd
import (
"context"
"encoding/json"
"os"
"time"
"github.com/gitpod-io/gitpod/common-go/analytics"
"github.com/gitpod-io/gitpod/supervisor/api"
log "github.com/sirupsen/logrus"
"github.com/spf13/cobra"
)
var sendAnalyticsCmdOpts struct {
data string
event string
}
// sendAnalyticsCmd represents the send-analytics command
var sendAnalyticsCmd = &cobra.Command{
Use: "send-analytics",
Long: "Sending anonymous statistics",
Hidden: true,
Args: cobra.ExactArgs(0),
Run: func(cmd *cobra.Command, args []string) {
file, err := os.OpenFile(os.TempDir()+"/supervisor-errors.log", os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)
if err == nil {
log.SetOutput(file)
defer file.Close()
} else {
log.SetLevel(log.FatalLevel)
}
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
data := make(map[string]interface{})
err = json.Unmarshal([]byte(sendAnalyticsCmdOpts.data), &data)
if err != nil {
log.Fatal(err)
}
conn := dialSupervisor()
defer conn.Close()
wsInfo, err := api.NewInfoServiceClient(conn).WorkspaceInfo(ctx, &api.WorkspaceInfoRequest{})
if err != nil {
log.Fatal(err)
}
data["instanceId"] = wsInfo.InstanceId
data["workspaceId"] = wsInfo.WorkspaceId
w := analytics.NewFromEnvironment()
w.Track(analytics.TrackMessage{
Identity: analytics.Identity{UserID: wsInfo.OwnerId},
Event: sendAnalyticsCmdOpts.event,
Properties: data,
})
w.Close()
},
}
func init() {
rootCmd.AddCommand(sendAnalyticsCmd)
sendAnalyticsCmd.Flags().StringVarP(&sendAnalyticsCmdOpts.event, "event", "", "", "event name")
sendAnalyticsCmd.Flags().StringVarP(&sendAnalyticsCmdOpts.data, "data", "", "", "json data")
_ = sendAnalyticsCmd.MarkFlagRequired("event")
_ = sendAnalyticsCmd.MarkFlagRequired("data")
}