offline-labs/gas-stack
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

textutils: move colors to textutils

Browse Source
This commit is contained in:
~wispem-wantex
2026-02-14 18:33:41 -08:00
parent ba89f0b818
commit d2b3a8173f
2 changed files with 25 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
pkg/db/connect.go
View File

@@ -8,6 +8,8 @@ import (
"github.com/jmoiron/sqlx" "github.com/jmoiron/sqlx"
_ "github.com/mattn/go-sqlite3" _ "github.com/mattn/go-sqlite3"
"git.offline-twitter.com/offline-labs/gas-stack/pkg/textutils"
) )
type DBConfig struct { type DBConfig struct {
@@ -22,20 +24,6 @@ type DBConfig struct {
version_number uint version_number uint
} }
// Colors for terminal output
const (
ColorReset = "\033[0m"
ColorBlack = "\033[30m"
ColorRed = "\033[31m"
ColorGreen = "\033[32m"
ColorYellow = "\033[33m"
ColorBlue = "\033[34m"
ColorPurple = "\033[35m"
ColorCyan = "\033[36m"
ColorGray = "\033[37m"
ColorWhite = "\033[97m"
)
func Init(schema *string, migrationsList *[]string) DBConfig { func Init(schema *string, migrationsList *[]string) DBConfig {
return DBConfig{ return DBConfig{
sql_schema: schema, sql_schema: schema,
@@ -96,11 +84,11 @@ func (c DBConfig) CheckAndUpdateVersion(db *sqlx.DB) error {
} }
if c.version_number > version { if c.version_number > version {
fmt.Print(ColorYellow) fmt.Print(textutils.ColorYellow)
fmt.Printf("================================================\n") fmt.Printf("================================================\n")
fmt.Printf("Database version is out of date. Upgrading database from version %d to version %d!\n", version, fmt.Printf("Database version is out of date. Upgrading database from version %d to version %d!\n", version,
c.version_number) c.version_number)
fmt.Print(ColorReset) fmt.Print(textutils.ColorReset)
c.UpgradeFromXToY(db, version, c.version_number) c.UpgradeFromXToY(db, version, c.version_number)
} }
@@ -110,9 +98,9 @@ func (c DBConfig) CheckAndUpdateVersion(db *sqlx.DB) error {
// UpgradeFromXToY runs all the migrations from version X to version Y, and update the `database_version` table's `version_number` // UpgradeFromXToY runs all the migrations from version X to version Y, and update the `database_version` table's `version_number`
func (c DBConfig) UpgradeFromXToY(db *sqlx.DB, x uint, y uint) { func (c DBConfig) UpgradeFromXToY(db *sqlx.DB, x uint, y uint) {
for i := x; i < y; i++ { for i := x; i < y; i++ {
fmt.Print(ColorCyan) fmt.Print(textutils.ColorCyan)
fmt.Println((*c.migrations)[i]) fmt.Println((*c.migrations)[i])
fmt.Print(ColorReset) fmt.Print(textutils.ColorReset)
// Execute the migration in a transaction // Execute the migration in a transaction
tx, err := db.Beginx() tx, err := db.Beginx()
@@ -125,14 +113,14 @@ func (c DBConfig) UpgradeFromXToY(db *sqlx.DB, x uint, y uint) {
panic(err) panic(err)
} }
fmt.Print(ColorYellow) fmt.Print(textutils.ColorYellow)
fmt.Printf("Now at database schema version %d.\n", i+1) fmt.Printf("Now at database schema version %d.\n", i+1)
fmt.Print(ColorReset) fmt.Print(textutils.ColorReset)
} }
fmt.Print(ColorGreen) fmt.Print(textutils.ColorGreen)
fmt.Printf("================================================\n") fmt.Printf("================================================\n")
fmt.Printf("Database version has been upgraded to version %d.\n", y) fmt.Printf("Database version has been upgraded to version %d.\n", y)
fmt.Print(ColorReset) fmt.Print(textutils.ColorReset)
} }
type VersionMismatchError struct { type VersionMismatchError struct {

15
pkg/textutils/terminal_colors.go Normal file
View File

@@ -0,0 +1,15 @@
package textutils
// Colors for terminal output
const (
ColorReset = "\033[0m"
ColorBlack = "\033[30m"
ColorRed = "\033[31m"
ColorGreen = "\033[32m"
ColorYellow = "\033[33m"
ColorBlue = "\033[34m"
ColorPurple = "\033[35m"
ColorCyan = "\033[36m"
ColorGray = "\033[37m"
ColorWhite = "\033[97m"
)