Bugfix: Gitops commit state

pkg/dashboard/server/flux.go

@@ -13,11 +13,15 @@ import (
 	"github.com/gimlet-io/gimlet-cli/pkg/dashboard/notifications"
 	"github.com/gimlet-io/gimlet-cli/pkg/dashboard/server/streaming"
 	"github.com/gimlet-io/gimlet-cli/pkg/dashboard/store"
+	"github.com/gimlet-io/gimlet-cli/pkg/dx"
+	"github.com/gimlet-io/gimlet-cli/pkg/git/nativeGit"
+	"github.com/go-git/go-git/v5"
+	"github.com/go-git/go-git/v5/plumbing"
+	"github.com/go-git/go-git/v5/plumbing/object"
 	log "github.com/sirupsen/logrus"
 )
 
 func fluxEvent(w http.ResponseWriter, r *http.Request) {
-
 	buf, _ := ioutil.ReadAll(r.Body)
 	rdr1 := ioutil.NopCloser(bytes.NewBuffer(buf))
 	rdr2 := ioutil.NopCloser(bytes.NewBuffer(buf))
@@ -54,6 +58,15 @@ func fluxEvent(w http.ResponseWriter, r *http.Request) {
 		log.Errorf("could not save or update gitops commit: %s", err)
 	}
 
+	gitopsRepoCache := ctx.Value("gitRepoCache").(*nativeGit.RepoCache)
+	clientHub, _ := ctx.Value("clientHub").(*streaming.ClientHub)
+	err = updateGitopsCommitStatuses(clientHub, gitopsRepoCache, store, event, repoName, env)
+	if err != nil {
+		log.Errorf("cannot update releases: %s", err)
+		http.Error(w, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
+		return
+	}
+
 	if !stateUpdated {
 		w.WriteHeader(http.StatusOK)
 		w.Write([]byte(""))
@@ -63,7 +76,6 @@ func fluxEvent(w http.ResponseWriter, r *http.Request) {
 	notificationsManager := ctx.Value("notificationsManager").(notifications.Manager)
 	notificationsManager.Broadcast(notifications.NewMessage(repoName, gitopsCommit, env))
 
-	clientHub, _ := ctx.Value("clientHub").(*streaming.ClientHub)
 	streaming.BroadcastGitopsCommitEvent(clientHub, *gitopsCommit)
 
 	w.WriteHeader(http.StatusOK)
@@ -101,3 +113,76 @@ func parseRev(rev string) (string, error) {
 
 	return parts[1], nil
 }
+
+func updateGitopsCommitStatuses(
+	clientHub *streaming.ClientHub,
+	gitopsRepoCache *nativeGit.RepoCache,
+	store *store.Store,
+	event fluxEvents.Event,
+	repoName, env string,
+) error {
+	if _, ok := event.Metadata["revision"]; !ok {
+		return fmt.Errorf("could not extract gitops sha from Flux message: %s", event)
+	}
+	eventHash, err := parseRev(event.Metadata["revision"])
+	if err != nil {
+		return err
+	}
+
+	repo, err := gitopsRepoCache.InstanceForRead(repoName)
+	if err != nil {
+		return err
+	}
+
+	hash := plumbing.NewHash(eventHash)
+	commitWalker, err := repo.Log(&git.LogOptions{
+		From: hash,
+	})
+	if err != nil {
+		return err
+	}
+
+	err = commitWalker.ForEach(func(c *object.Commit) error {
+		hashString := c.Hash.String()
+		if hashString == eventHash {
+			return nil
+		}
+
+		gitopsCommitFromDb, err := store.GitopsCommit(hashString)
+		if err != nil {
+			log.Warnf("cannot get gitops commit: %s", err)
+			return nil
+		}
+
+		if gitopsCommitFromDb != nil &&
+			(gitopsCommitFromDb.Status == dx.ValidationFailed ||
+				gitopsCommitFromDb.Status == dx.ReconciliationFailed ||
+				gitopsCommitFromDb.Status == dx.HealthCheckFailed ||
+				gitopsCommitFromDb.Status == dx.ReconciliationSucceeded) {
+			return fmt.Errorf("%s", "EOF")
+		}
+
+		gitopsCommitToSave := model.GitopsCommit{
+			Sha:        hashString,
+			Status:     event.Reason,
+			StatusDesc: event.Message,
+			Created:    event.Timestamp.Unix(),
+			Env:        env,
+		}
+
+		_, err = store.SaveOrUpdateGitopsCommit(&gitopsCommitToSave)
+		if err != nil {
+			log.Warnf("could not save or update gitops commit: %s", err)
+			return nil
+		}
+		streaming.BroadcastGitopsCommitEvent(clientHub, gitopsCommitToSave)
+
+		return nil
+	})
+	if err != nil &&
+		err.Error() != "EOF" {
+		return err
+	}
+
+	return nil
+}