[add] Using default git data if not provided via flags

Author: filipecosta90

Makefile

@@ -21,6 +21,9 @@ all: dist hash
 gofmt:
 	@test -z $(shell gofmt -l -s $(SOURCE_DIRS) ./ | tee /dev/stderr) || (echo "[WARN] Fix formatting issues with 'make fmt'" && exit 1)
 
+build:
+	$(GOCMD) build .
+
 fmt:
 	$(GOFMT) ./...
 
@@ -32,11 +35,11 @@ lint:
 dist:
 	mkdir -p bin/
 	rm -rf bin/pprof-exporter*
-	CGO_ENABLED=0 GOARCH=amd64 GOOS=linux go build -mod=vendor -a -ldflags $(LDFLAGS) -installsuffix cgo -o bin/pprof-exporter
-	CGO_ENABLED=0 GOARCH=amd64 GOOS=darwin go build -mod=vendor -a -ldflags $(LDFLAGS) -installsuffix cgo -o bin/pprof-exporter-darwin
-	GOARM=6 GOARCH=arm CGO_ENABLED=0 GOOS=linux go build -mod=vendor -a -ldflags $(LDFLAGS) -installsuffix cgo -o bin/pprof-exporter-armhf
-	GOARCH=arm64 CGO_ENABLED=0 GOOS=linux go build -mod=vendor -a -ldflags $(LDFLAGS) -installsuffix cgo -o bin/pprof-exporter-arm64
-	GOOS=windows GOARCH=amd64 CGO_ENABLED=0 go build -mod=vendor -a -ldflags $(LDFLAGS) -installsuffix cgo -o bin/pprof-exporter.exe
+	CGO_ENABLED=0 GOARCH=amd64 GOOS=linux go build -a -ldflags $(LDFLAGS) -installsuffix cgo -o bin/pprof-exporter
+	CGO_ENABLED=0 GOARCH=amd64 GOOS=darwin go build -a -ldflags $(LDFLAGS) -installsuffix cgo -o bin/pprof-exporter-darwin
+	GOARM=6 GOARCH=arm CGO_ENABLED=0 GOOS=linux go build -a -ldflags $(LDFLAGS) -installsuffix cgo -o bin/pprof-exporter-armhf
+	GOARCH=arm64 CGO_ENABLED=0 GOOS=linux go build -a -ldflags $(LDFLAGS) -installsuffix cgo -o bin/pprof-exporter-arm64
+	GOOS=windows GOARCH=amd64 CGO_ENABLED=0 go build -a -ldflags $(LDFLAGS) -installsuffix cgo -o bin/pprof-exporter.exe
 
 .PHONY: hash
 hash:

cmd/export.go

@@ -59,7 +59,7 @@ func exportLogic() func(cmd *cobra.Command, args []string) {
 					log.Fatal(err)
 				}
 			}
-			log.Printf("Successfully published profile data. Check it at: %s/#/gh/%s/%s/commit/%s/bench/%s/cpu", codeperfUrl, gitOrg, gitRepo, gitCommit, bench)
+			log.Printf("Successfully published profile data. Check it at: %s/gh/%s/%s/commit/%s/bench/%s/cpu", codeperfUrl, gitOrg, gitRepo, gitCommit, bench)
 		}
 
 	}

cmd/root.go

@@ -17,6 +17,9 @@ package cmd
 
 import (
 	"fmt"
+	"github.com/go-git/go-git/v5"
+	"log"
+	"strings"
 
 	"github.com/spf13/cobra"
 	"os"
@@ -57,6 +60,27 @@ func Execute() {
 
 func init() {
 	cobra.OnInitialize(initConfig)
+	defaultGitOrg := ""
+	defaultGitRepo := ""
+	defaultGitCommit := ""
+
+	r, err := git.PlainOpen(".")
+	if err != nil {
+		log.Println("Unable to retrieve current repo git info. Use the --git-org, --git-repo, and --git-hash to properly fill the git info.")
+	} else {
+		ref, _ := r.Head()
+		refHash := ref.Hash().String()
+		remotes, _ := r.Remotes()
+		remoteUsed := remotes[0].Config().URLs[0]
+		fmt.Println(remoteUsed)
+		toOrg := remoteUsed[:strings.LastIndex(remoteUsed, "/")]
+		fmt.Println(toOrg)
+		defaultGitOrg = toOrg[strings.LastIndexAny(toOrg, "/:")+1:]
+		repoStartPos := strings.LastIndex(remoteUsed, "/") + 1
+		defaultGitRepo = remoteUsed[repoStartPos : len(remoteUsed)-4]
+		defaultGitCommit = refHash
+		log.Printf("Using the following default git vars org=%s repo=%s hash=%s", defaultGitOrg, defaultGitRepo, defaultGitCommit)
+	}
 
 	// Here you will define your flags and configuration settings.
 	// Cobra supports persistent flags, which, if defined here,
@@ -65,9 +89,9 @@ func init() {
 	rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.pprof-exporter.yaml)")
 	rootCmd.PersistentFlags().BoolVar(&local, "local", false, "don't push the data to https://codeperf.io")
 	rootCmd.PersistentFlags().StringVar(&bench, "bench", "", "Benchmark name")
-	rootCmd.PersistentFlags().StringVar(&gitOrg, "git-org", "", "git org")
-	rootCmd.PersistentFlags().StringVar(&gitRepo, "git-repo", "", "git repo")
-	rootCmd.PersistentFlags().StringVar(&gitCommit, "git-commit", "", "git commit hash")
+	rootCmd.PersistentFlags().StringVar(&gitOrg, "git-org", defaultGitOrg, "git org")
+	rootCmd.PersistentFlags().StringVar(&gitRepo, "git-repo", defaultGitRepo, "git repo")
+	rootCmd.PersistentFlags().StringVar(&gitCommit, "git-hash", defaultGitCommit, "git commit hash")
 	rootCmd.PersistentFlags().StringVar(&localFilename, "local-filename", "profile.json", "Local file to export the json to. Only used when the --local flag is set")
 	rootCmd.PersistentFlags().StringVar(&codeperfUrl, "codeperf-url", "https://codeperf.io", "codeperf URL")
 	rootCmd.MarkPersistentFlagRequired("bench")

go.mod

@@ -3,7 +3,9 @@ module github.com/codeperfio/pprof-exporter
 go 1.16
 
 require (
+	github.com/go-git/go-git/v5 v5.4.2
 	github.com/google/pprof v0.0.0-20210720184732-4bb14d4b1be1
+	github.com/sergi/go-diff v1.2.0 // indirect
 	github.com/spf13/cobra v1.2.1
 	github.com/spf13/viper v1.9.0
 )