From 321ed4f6f793564a8ffe21a4fd7dcab0ef6181f4 Mon Sep 17 00:00:00 2001
From: Ronen Lubin <63970571+ronenlu@users.noreply.github.com>
Date: Sun, 16 Jun 2024 15:33:49 +0300
Subject: [PATCH] atlasexec: add summary method on migration apply (#73)

* atlasexec: add dirName, url and drvier to migrate apply output

* atlasexec: add summary method on migration apply

* fix lint

* update format
---
 atlasexec/atlas_models.go | 101 +++++++++++++++++++++++++++++++++++++-
 1 file changed, 99 insertions(+), 2 deletions(-)

diff --git a/atlasexec/atlas_models.go b/atlasexec/atlas_models.go
index 5fee952..2af22c7 100644
--- a/atlasexec/atlas_models.go
+++ b/atlasexec/atlas_models.go
@@ -1,6 +1,8 @@
 package atlasexec
 
 import (
+	"fmt"
+	"strings"
 	"time"
 
 	"ariga.io/atlas/sql/sqlcheck"
@@ -19,8 +21,9 @@ type (
 		File
 		Start   time.Time
 		End     time.Time
-		Skipped int      // Amount of skipped SQL statements in a partially applied file.
-		Applied []string // SQL statements applied with success
+		Skipped int           // Amount of skipped SQL statements in a partially applied file.
+		Applied []string      // SQL statements applied with success
+		Checks  []*FileChecks // Assertion checks
 		Error   *struct {
 			SQL   string // SQL statement that failed.
 			Error string // Error returned by the database.
@@ -120,6 +123,19 @@ type (
 		Error   string            `json:"Error,omitempty"`   // File specific error.
 	}
 
+	// FileChecks represents a set of checks to run before applying a file.
+	FileChecks struct {
+		Name  string     `json:"Name,omitempty"`  // File/group name.
+		Stmts []*Check   `json:"Stmts,omitempty"` // Checks statements executed.
+		Error *StmtError `json:"Error,omitempty"` // Assertion error.
+		Start time.Time  `json:"Start,omitempty"` // Start assertion time.
+		End   time.Time  `json:"End,omitempty"`   // End assertion time.
+	}
+	// Check represents an assertion and its status.
+	Check struct {
+		Stmt  string  `json:"Stmt,omitempty"`  // Assertion statement.
+		Error *string `json:"Error,omitempty"` // Assertion error, if any.
+	}
 	// StmtError groups a statement with its execution error.
 	StmtError struct {
 		Stmt string `json:"Stmt,omitempty"` // SQL statement that failed.
@@ -179,6 +195,87 @@ type (
 	}
 )
 
+// Summary of the migration attempt.
+func (a *MigrateApply) Summary(ident string) string {
+	var (
+		passedC, failedC int
+		passedS, failedS int
+		passedF, failedF int
+		lines            = make([]string, 0, 3)
+	)
+	for _, f := range a.Applied {
+		// For each check file, count the
+		// number of failed assertions.
+		for _, cf := range f.Checks {
+			for _, s := range cf.Stmts {
+				if s.Error != nil {
+					failedC++
+				} else {
+					passedC++
+				}
+			}
+		}
+		passedS += len(f.Applied)
+		if f.Error != nil {
+			failedF++
+			// Last statement failed (not an assertion).
+			if len(f.Checks) == 0 || f.Checks[len(f.Checks)-1].Error == nil {
+				passedS--
+				failedS++
+			}
+		} else {
+			passedF++
+		}
+	}
+	// Execution time.
+	lines = append(lines, a.End.Sub(a.Start).String())
+	// Executed files.
+	switch {
+	case passedF > 0 && failedF > 0:
+		lines = append(lines, fmt.Sprintf("%d migration%s ok, %d with errors", passedF, plural(passedF), failedF))
+	case passedF > 0:
+		lines = append(lines, fmt.Sprintf("%d migration%s", passedF, plural(passedF)))
+	case failedF > 0:
+		lines = append(lines, fmt.Sprintf("%d migration%s with errors", failedF, plural(failedF)))
+	}
+	// Executed checks.
+	switch {
+	case passedC > 0 && failedC > 0:
+		lines = append(lines, fmt.Sprintf("%d check%s ok, %d failure%s", passedC, plural(passedC), failedC, plural(failedC)))
+	case passedC > 0:
+		lines = append(lines, fmt.Sprintf("%d check%s", passedC, plural(passedC)))
+	case failedC > 0:
+		lines = append(lines, fmt.Sprintf("%d check error%s", failedC, plural(failedC)))
+	}
+	// Executed statements.
+	switch {
+	case passedS > 0 && failedS > 0:
+		lines = append(lines, fmt.Sprintf("%d sql statement%s ok, %d with errors", passedS, plural(passedS), failedS))
+	case passedS > 0:
+		lines = append(lines, fmt.Sprintf("%d sql statement%s", passedS, plural(passedS)))
+	case failedS > 0:
+		lines = append(lines, fmt.Sprintf("%d sql statement%s with errors", failedS, plural(failedS)))
+	}
+	var b strings.Builder
+	for i, l := range lines {
+		b.WriteString("-")
+		b.WriteByte(' ')
+		b.WriteString(fmt.Sprintf("**%s**", l))
+		if i < len(lines)-1 {
+			b.WriteByte('\n')
+			b.WriteString(ident)
+		}
+	}
+	return b.String()
+}
+
+func plural(n int) (s string) {
+	if n > 1 {
+		s += "s"
+	}
+	return
+}
+
 // Error implements the error interface.
 func (e *MigrateApplyError) Error() string { return last(e.Result).Error }