add more telemetry (#102)

* add more telemetry

Author: Spartakovic

cmd/digger/main.go

@@ -11,6 +11,7 @@ import (
 )
 
 func main() {
+	utils.SendUsageRecord("", "log", "initialize")
 	args := os.Args[1:]
 	if len(args) > 0 && args[0] == "version" {
 		fmt.Println(utils.GetVersion())
@@ -21,39 +22,34 @@ func main() {
 		os.Exit(0)
 	}
 
-	diggerConfig, err := digger.NewDiggerConfig("")
-	if err != nil {
-		fmt.Printf("Failed to read digger config. %s\n", err)
-		os.Exit(1)
-	}
-	println("Digger config read successfully")
-
-	lock, err := utils.GetLock()
-	if err != nil {
-		fmt.Printf("Failed to create lock provider. %s\n", err)
-		os.Exit(2)
-	}
-	println("Lock provider has been created successfully")
-
 	ghToken := os.Getenv("GITHUB_TOKEN")
 	if ghToken == "" {
-		println("GITHUB_TOKEN is not defined")
-		os.Exit(3)
+		reportErrorAndExit("", "GITHUB_TOKEN is not defined", 1)
 	}
 
 	ghContext := os.Getenv("GITHUB_CONTEXT")
 	if ghContext == "" {
-		fmt.Printf("GITHUB_CONTEXT is not defined. %s\n", err)
-		os.Exit(4)
+		reportErrorAndExit("", "GITHUB_CONTEXT is not defined", 2)
 	}
 
 	parsedGhContext, err := models.GetGitHubContext(ghContext)
 	if err != nil {
-		fmt.Printf("failed to parse GitHub context. %s\n", err.Error())
-		os.Exit(5)
+		reportErrorAndExit("", fmt.Sprintf("Failed to parse GitHub context. %s", err), 3)
 	}
 	println("GitHub context parsed successfully")
 
+	diggerConfig, err := digger.NewDiggerConfig("")
+	if err != nil {
+		reportErrorAndExit(parsedGhContext.RepositoryOwner, fmt.Sprintf("Failed to read Digger config. %s", err), 4)
+	}
+	println("Digger config read successfully")
+
+	lock, err := utils.GetLock()
+	if err != nil {
+		reportErrorAndExit(parsedGhContext.RepositoryOwner, fmt.Sprintf("Failed to create lock provider. %s", err), 5)
+	}
+	println("Lock provider has been created successfully")
+
 	ghEvent := parsedGhContext.Event
 	eventName := parsedGhContext.EventName
 	splitRepositoryName := strings.Split(parsedGhContext.Repository, "/")
@@ -62,8 +58,7 @@ func main() {
 
 	impactedProjects, prNumber, err := digger.ProcessGitHubEvent(ghEvent, diggerConfig, githubPrService)
 	if err != nil {
-		fmt.Printf("failed to process GitHub event, %v", err)
-		os.Exit(6)
+		reportErrorAndExit(repoOwner, fmt.Sprintf("Failed to process GitHub event. %s", err), 6)
 	}
 	println("GitHub event processed successfully")
 
@@ -74,15 +69,31 @@ func main() {
 
 	commandsToRunPerProject, err := digger.ConvertGithubEventToCommands(ghEvent, impactedProjects)
 	if err != nil {
-		fmt.Printf("failed to convert event to command, %v", err)
-		os.Exit(7)
+		reportErrorAndExit(repoOwner, fmt.Sprintf("Failed to convert GitHub event to commands. %s", err), 7)
 	}
 	println("GitHub event converted to commands successfully")
 
 	err = digger.RunCommandsPerProject(commandsToRunPerProject, repoOwner, repositoryName, eventName, prNumber, diggerConfig, githubPrService, lock, "")
 	if err != nil {
-		fmt.Printf("failed to execute command, %v", err)
-		os.Exit(8)
+		reportErrorAndExit(repoOwner, fmt.Sprintf("Failed to run commands. %s", err), 8)
 	}
 	println("Commands executed successfully")
+
+	reportErrorAndExit(repoOwner, "Digger finished successfully", 0)
+
+	defer func() {
+		if r := recover(); r != nil {
+			reportErrorAndExit(repoOwner, fmt.Sprintf("Panic occurred. %s", r), 1)
+		}
+	}()
+
+}
+
+func reportErrorAndExit(repoOwner string, message string, exitCode int) {
+	fmt.Printf(message)
+	err := utils.SendLogRecord(repoOwner, message)
+	if err != nil {
+		fmt.Printf("Failed to send log record. %s\n", err)
+	}
+	os.Exit(exitCode)
 }

pkg/utils/usage.go

@@ -27,6 +27,25 @@ func SendUsageRecord(repoOwner string, eventName string, action string) error {
 		Action:    action,
 		Token:     "diggerABC@@1998fE",
 	}
+	return sendPayload(payload)
+}
+
+func SendLogRecord(repoOwner string, message string) error {
+	h := sha256.New()
+	h.Write([]byte(repoOwner))
+	sha := h.Sum(nil)
+	shaStr := hex.EncodeToString(sha)
+	payload := UsageRecord{
+		UserId:    shaStr,
+		EventName: "log",
+		Action:    message,
+		Token:     "diggerABC@@1998fE",
+	}
+	return sendPayload(payload)
+}
+
+func sendPayload(payload interface{}) error {
+
 	jsonData, err := json.Marshal(payload)
 	if err != nil {
 		log.Printf("Error marshalling usage record: %v", err)