Set analytics clientID

hack/gendocs/main.go

@@ -55,7 +55,7 @@ aliases:
 
 // ref: https://github.com/spf13/cobra/blob/master/doc/md_docs.md
 func main() {
-	rootCmd := cmds.NewCmdStash("")
+	rootCmd := cmds.NewRootCmd()
 	dir := runtime.GOPath() + "/src/github.com/appscode/stash/docs/reference"
 	fmt.Printf("Generating cli markdown tree in: %v\n", dir)
 	err := os.RemoveAll(dir)

main.go

@@ -16,7 +16,7 @@ func main() {
 	logs.InitLogs()
 	defer logs.FlushLogs()
 
-	if err := cmds.NewCmdStash(Version).Execute(); err != nil {
+	if err := cmds.NewRootCmd().Execute(); err != nil {
 		log.Fatalln("Error in Stash Main:", err)
 	}
 	log.Infoln("Exiting Stash Main")

pkg/cmds/backup.go

@@ -1,12 +1,14 @@
 package cmds
 
 import (
+	"fmt"
 	"os"
 	"strings"
 	"time"
 
 	"github.com/appscode/go/log"
 	"github.com/appscode/kutil/meta"
+	"github.com/appscode/kutil/tools/analytics"
 	cs "github.com/appscode/stash/client/typed/stash/v1alpha1"
 	"github.com/appscode/stash/pkg/backup"
 	"github.com/appscode/stash/pkg/util"
@@ -22,12 +24,12 @@ func NewCmdBackup() *cobra.Command {
 		opt            = backup.Options{
 			Namespace:      meta.Namespace(),
 			ScratchDir:     "/tmp",
-			PushgatewayURL: "http://stash-operator.kube-system.svc:56789",
 			PodLabelsPath:  "/etc/stash/labels",
 			ResyncPeriod:   5 * time.Minute,
 			MaxNumRequeues: 5,
 		}
 	)
+	opt.PushgatewayURL = fmt.Sprintf("http://stash-operator.%s.svc:56789", opt.Namespace)
 
 	cmd := &cobra.Command{
 		Use:               "backup",
@@ -38,8 +40,12 @@ func NewCmdBackup() *cobra.Command {
 			if err != nil {
 				log.Fatalf("Could not get Kubernetes config: %s", err)
 			}
-			kubeClient = kubernetes.NewForConfigOrDie(config)
-			stashClient = cs.NewForConfigOrDie(config)
+			kubeClient := kubernetes.NewForConfigOrDie(config)
+			stashClient := cs.NewForConfigOrDie(config)
+
+			if meta.PossiblyInCluster() {
+				sendAnalytics(cmd, analytics.ClientID(kubeClient.CoreV1().Nodes()))
+			}
 
 			opt.NodeName = os.Getenv("NODE_NAME")
 			if opt.NodeName == "" {

pkg/cmds/check.go

@@ -3,7 +3,8 @@ package cmds
 import (
 	"github.com/appscode/go/log"
 	"github.com/appscode/kutil/meta"
-	"github.com/appscode/stash/client/typed/stash/v1alpha1"
+	"github.com/appscode/kutil/tools/analytics"
+	cs "github.com/appscode/stash/client/typed/stash/v1alpha1"
 	"github.com/appscode/stash/pkg/check"
 	"github.com/spf13/cobra"
 	"k8s.io/client-go/kubernetes"
@@ -28,11 +29,14 @@ func NewCmdCheck() *cobra.Command {
 			if err != nil {
 				log.Fatalln(err)
 			}
-			c := check.New(
-				kubernetes.NewForConfigOrDie(config),
-				v1alpha1.NewForConfigOrDie(config),
-				opt,
-			)
+			kubeClient := kubernetes.NewForConfigOrDie(config)
+			stashClient := cs.NewForConfigOrDie(config)
+
+			if meta.PossiblyInCluster() {
+				sendAnalytics(cmd, analytics.ClientID(kubeClient.CoreV1().Nodes()))
+			}
+
+			c := check.New(kubeClient, stashClient, opt)
 			if err = c.Run(); err != nil {
 				log.Fatal(err)
 			}

pkg/cmds/recover.go

@@ -3,7 +3,8 @@ package cmds
 import (
 	"github.com/appscode/go/log"
 	"github.com/appscode/kutil/meta"
-	"github.com/appscode/stash/client/typed/stash/v1alpha1"
+	"github.com/appscode/kutil/tools/analytics"
+	cs "github.com/appscode/stash/client/typed/stash/v1alpha1"
 	"github.com/appscode/stash/pkg/recovery"
 	"github.com/spf13/cobra"
 	"k8s.io/client-go/kubernetes"
@@ -26,12 +27,14 @@ func NewCmdRecover() *cobra.Command {
 			if err != nil {
 				log.Fatalln(err)
 			}
-			c := recovery.New(
-				kubernetes.NewForConfigOrDie(config),
-				v1alpha1.NewForConfigOrDie(config),
-				meta.Namespace(),
-				recoveryName,
-			)
+			kubeClient := kubernetes.NewForConfigOrDie(config)
+			stashClient := cs.NewForConfigOrDie(config)
+
+			if meta.PossiblyInCluster() {
+				sendAnalytics(cmd, analytics.ClientID(kubeClient.CoreV1().Nodes()))
+			}
+
+			c := recovery.New(kubeClient, stashClient, meta.Namespace(), recoveryName)
 			c.Run()
 		},
 	}

pkg/cmds/root.go

@@ -5,7 +5,9 @@ import (
 	"log"
 	"strings"
 
+	"github.com/appscode/go/analytics"
 	v "github.com/appscode/go/version"
+	"github.com/appscode/kutil/meta"
 	"github.com/appscode/stash/client/scheme"
 	"github.com/jpillora/go-ogle-analytics"
 	"github.com/spf13/cobra"
@@ -17,10 +19,7 @@ const (
 	gaTrackingCode = "UA-62096468-20"
 )
 
-func NewCmdStash(version string) *cobra.Command {
-	var (
-		enableAnalytics = true
-	)
+func NewRootCmd() *cobra.Command {
 	var rootCmd = &cobra.Command{
 		Use:               "stash",
 		Short:             `Stash by AppsCode - Backup your Kubernetes Volumes`,
@@ -30,11 +29,8 @@ func NewCmdStash(version string) *cobra.Command {
 			c.Flags().VisitAll(func(flag *pflag.Flag) {
 				log.Printf("FLAG: --%s=%q", flag.Name, flag.Value)
 			})
-			if enableAnalytics && gaTrackingCode != "" {
-				if client, err := ga.NewClient(gaTrackingCode); err == nil {
-					parts := strings.Split(c.CommandPath(), " ")
-					client.Send(ga.NewEvent(parts[0], strings.Join(parts[1:], "/")).Label(version))
-				}
+			if !meta.PossiblyInCluster() {
+				sendAnalytics(c, analytics.ClientID())
 			}
 			scheme.AddToScheme(clientsetscheme.Scheme)
 		},
@@ -45,9 +41,23 @@ func NewCmdStash(version string) *cobra.Command {
 	rootCmd.PersistentFlags().BoolVar(&enableAnalytics, "analytics", enableAnalytics, "Send analytical events to Google Analytics")
 
 	rootCmd.AddCommand(v.NewCmdVersion())
-	rootCmd.AddCommand(NewCmdRun(version))
+	rootCmd.AddCommand(NewCmdRun())
 	rootCmd.AddCommand(NewCmdBackup())
 	rootCmd.AddCommand(NewCmdRecover())
 	rootCmd.AddCommand(NewCmdCheck())
 	return rootCmd
 }
+
+var (
+	enableAnalytics = true
+)
+
+func sendAnalytics(c *cobra.Command, clientID string) {
+	if enableAnalytics && gaTrackingCode != "" {
+		if client, err := ga.NewClient(gaTrackingCode); err == nil {
+			client.ClientID(clientID)
+			parts := strings.Split(c.CommandPath(), " ")
+			client.Send(ga.NewEvent(parts[0], strings.Join(parts[1:], "/")).Label(v.Version.Version))
+		}
+	}
+}

pkg/cmds/run.go

@@ -7,6 +7,9 @@ import (
 
 	"github.com/appscode/go/log"
 	stringz "github.com/appscode/go/strings"
+	v "github.com/appscode/go/version"
+	"github.com/appscode/kutil/meta"
+	"github.com/appscode/kutil/tools/analytics"
 	"github.com/appscode/pat"
 	api "github.com/appscode/stash/apis/stash"
 	cs "github.com/appscode/stash/client/typed/stash/v1alpha1"
@@ -20,23 +23,17 @@ import (
 	"k8s.io/client-go/tools/clientcmd"
 )
 
-var (
-	kubeClient  kubernetes.Interface
-	stashClient cs.StashV1alpha1Interface
-
-	scratchDir string = "/tmp"
-)
-
-func NewCmdRun(version string) *cobra.Command {
+func NewCmdRun() *cobra.Command {
 	var (
 		masterURL      string
 		kubeconfigPath string
 		address        string = ":56790"
 		opts                  = controller.Options{
-			SidecarImageTag: stringz.Val(version, "canary"),
+			SidecarImageTag: stringz.Val(v.Version.Version, "canary"),
 			ResyncPeriod:    5 * time.Minute,
 			MaxNumRequeues:  5,
 		}
+		scratchDir = "/tmp"
 	)
 
 	cmd := &cobra.Command{
@@ -52,10 +49,14 @@ func NewCmdRun(version string) *cobra.Command {
 			if err != nil {
 				log.Fatalln(err)
 			}
-			kubeClient = kubernetes.NewForConfigOrDie(config)
-			stashClient = cs.NewForConfigOrDie(config)
+			kubeClient := kubernetes.NewForConfigOrDie(config)
+			stashClient := cs.NewForConfigOrDie(config)
 			crdClient := crd_cs.NewForConfigOrDie(config)
 
+			if meta.PossiblyInCluster() {
+				sendAnalytics(cmd, analytics.ClientID(kubeClient.CoreV1().Nodes()))
+			}
+
 			// get kube api server version
 			version, err := kubeClient.Discovery().ServerVersion()
 			if err != nil {
@@ -89,7 +90,12 @@ func NewCmdRun(version string) *cobra.Command {
 
 			pattern := fmt.Sprintf("/%s/v1beta1/namespaces/%s/restics/%s/metrics", api.GroupName, PathParamNamespace, PathParamName)
 			log.Infof("URL pattern: %s", pattern)
-			m.Get(pattern, http.HandlerFunc(ExportSnapshots))
+			exporter := &PrometheusExporter{
+				kubeClient:  kubeClient,
+				stashClient: stashClient,
+				scratchDir:  scratchDir,
+			}
+			m.Get(pattern, exporter)
 
 			http.Handle("/", m)
 			log.Infoln("Listening on", address)