From 81a361d87e5d8468867d38e781fd530001b97a8e Mon Sep 17 00:00:00 2001
From: Alessio <me@playfulpachyderm.com>
Date: Sun, 12 Dec 2021 20:10:08 -0800
Subject: [PATCH] Add color to version upgrading messages

---
 persistence/versions.go | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/persistence/versions.go b/persistence/versions.go
index cacec0a..474ff2a 100644
--- a/persistence/versions.go
+++ b/persistence/versions.go
@@ -3,6 +3,8 @@ package persistence
 import (
 	"fmt"
 	"database/sql"
+
+	"offline_twitter/terminal_utils"
 )
 
 
@@ -83,7 +85,10 @@ func (p Profile) check_and_update_version() error {
 	}
 
 	if ENGINE_DATABASE_VERSION > version {
-		fmt.Printf("Upgrading database from version %d to version %d", version, ENGINE_DATABASE_VERSION)
+		fmt.Printf(terminal_utils.COLOR_YELLOW)
+		fmt.Printf("================================================\n")
+		fmt.Printf("Database version is out of date.  Upgrading database from version v%d to version v%d!\n", version, ENGINE_DATABASE_VERSION)
+		fmt.Printf(terminal_utils.COLOR_RESET)
 		return p.UpgradeFromXToY(version, ENGINE_DATABASE_VERSION)
 	}
 
@@ -95,7 +100,10 @@ func (p Profile) check_and_update_version() error {
  */
 func (p Profile) UpgradeFromXToY(x int, y int) error {
 	for i := x; i < y; i++ {
+		fmt.Printf(terminal_utils.COLOR_CYAN)
 		fmt.Println(MIGRATIONS[i])
+		fmt.Printf(terminal_utils.COLOR_RESET)
+
 		_, err := p.DB.Exec(MIGRATIONS[i])
 		if err != nil {
 			return err
@@ -104,7 +112,13 @@ func (p Profile) UpgradeFromXToY(x int, y int) error {
 		if err != nil {
 			return err
 		}
+		fmt.Printf(terminal_utils.COLOR_YELLOW)
 		fmt.Printf("Now at database schema version %d.\n", i + 1)
+		fmt.Printf(terminal_utils.COLOR_RESET)
 	}
+	fmt.Printf(terminal_utils.COLOR_GREEN)
+	fmt.Printf("================================================\n")
+	fmt.Printf("Database version has been upgraded to version %d.\n", y)
+	fmt.Printf(terminal_utils.COLOR_RESET)
 	return nil
 }