telemetry.go

package cmd

import (
	"fmt"

	"github.com/pingcap/tiup/pkg/environment"
	"github.com/pingcap/tiup/pkg/telemetry"
	"github.com/spf13/cobra"
)

func newTelemetryCmd() *cobra.Command {
	cmd := &cobra.Command{
		Use:   "telemetry",
		Short: "Controls things about telemetry",
	}

	cmd.AddCommand(&cobra.Command{
		Use:   "reset",
		Short: "Reset the uuid used for telemetry",
		RunE: func(cmd *cobra.Command, args []string) error {
			teleCommand = cmd.CommandPath()
			env := environment.GlobalEnv()
			teleMeta, fname, err := telemetry.GetMeta(env)
			if err != nil {
				return err
			}

			teleMeta.UUID = telemetry.NewUUID()
			err = teleMeta.SaveTo(fname)
			if err != nil {
				return err
			}

			fmt.Printf("Reset uuid as: %s success\n", teleMeta.UUID)
			return nil
		},
	})

	cmd.AddCommand(&cobra.Command{
		Use:   "enable",
		Short: "Enable telemetry of tiup",
		RunE: func(cmd *cobra.Command, args []string) error {
			teleCommand = cmd.CommandPath()
			env := environment.GlobalEnv()
			teleMeta, fname, err := telemetry.GetMeta(env)
			if err != nil {
				return err
			}

			teleMeta.Status = telemetry.EnableStatus
			err = teleMeta.SaveTo(fname)
			if err != nil {
				return err
			}

			fmt.Printf("Enable telemetry success\n")
			return nil
		},
	})

	cmd.AddCommand(&cobra.Command{
		Use:   "disable",
		Short: "Disable telemetry of tiup",
		RunE: func(cmd *cobra.Command, args []string) error {
			teleCommand = cmd.CommandPath()
			env := environment.GlobalEnv()
			teleMeta, fname, err := telemetry.GetMeta(env)
			if err != nil {
				return err
			}

			teleMeta.Status = telemetry.DisableStatus
			err = teleMeta.SaveTo(fname)
			if err != nil {
				return err
			}

			fmt.Printf("Disable telemetry success\n")
			return nil
		},
	})

	cmd.AddCommand(&cobra.Command{
		Use:   "status",
		Short: "Display the current status of tiup telemetry",
		RunE: func(cmd *cobra.Command, args []string) error {
			teleCommand = cmd.CommandPath()
			env := environment.GlobalEnv()
			teleMeta, _, err := telemetry.GetMeta(env)
			if err != nil {
				return err
			}

			fmt.Printf("status: %s\n", teleMeta.Status)
			fmt.Printf("uuid: %s\n", teleMeta.UUID)
			return nil
		},
	})

	return cmd
}