fix: improve sqlite support

Author: lucasnevespereira

assets/demo.gif

@@ -58,7 +58,7 @@ var appAddCmd = &cobra.Command{
 	Short: "Add a new app to monitor",
 	Long: `Add a new application to monitor with its database connection.
 
-Supported database types: postgres, mysql, mongodb
+Supported database types: postgres, mysql, mongodb, sqlite
 
 Examples:
   dashmin app add myapp postgres "postgres://readonly:password@localhost:5432/myapp?sslmode=disable"
@@ -74,10 +74,11 @@ Examples:
 			"postgres": true,
 			"mysql":    true,
 			"mongodb":  true,
+			"sqlite":   true,
 		}
 
 		if !validTypes[dbType] {
-			return fmt.Errorf("invalid database type '%s'. Supported: postgres, mysql, mongodb", dbType)
+			return fmt.Errorf("invalid database type '%s'. Supported: postgres, mysql, mongodb, sqlite", dbType)
 		}
 
 		cfg, err := config.Load()
@@ -265,6 +266,8 @@ Examples:
 				fmt.Printf("    user:password@tcp(host:port)/database\n")
 			case "mongodb":
 				fmt.Printf("    mongodb://user:password@host:port/database\n")
+			case "sqlite":
+				fmt.Printf("    sqlite:///path/to/database.db\n")
 			}
 			fmt.Printf("  - Check network connectivity\n")
 			fmt.Printf("  - Verify credentials are correct\n")
@@ -277,7 +280,7 @@ Examples:
 		fmt.Printf("Testing basic query...\n")
 		var testQuery string
 		switch app.Type {
-		case "postgres", "mysql":
+		case "postgres", "mysql", "sqlite":
 			testQuery = "SELECT 1 as test"
 		case "mongodb":
 			testQuery = "test.count({})"

cmd/app.go

@@ -7,6 +7,7 @@ require (
 	github.com/charmbracelet/lipgloss v1.1.0
 	github.com/go-sql-driver/mysql v1.9.3
 	github.com/jackc/pgx/v5 v5.7.5
+	github.com/mattn/go-sqlite3 v1.14.33
 	github.com/spf13/cobra v1.10.1
 	go.mongodb.org/mongo-driver v1.17.4
 	golang.org/x/sync v0.19.0

go.mod

@@ -51,6 +51,8 @@ github.com/mattn/go-localereader v0.0.1 h1:ygSAOl7ZXTx4RdPYinUpg6W99U8jWvWi9Ye2J
 github.com/mattn/go-localereader v0.0.1/go.mod h1:8fBrzywKY7BI3czFoHkuzRoWE9C+EiG4R1k4Cjx5p88=
 github.com/mattn/go-runewidth v0.0.16 h1:E5ScNMtiwvlvB5paMFdw9p4kSQzbXFikJ5SQO6TULQc=
 github.com/mattn/go-runewidth v0.0.16/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w=
+github.com/mattn/go-sqlite3 v1.14.33 h1:A5blZ5ulQo2AtayQ9/limgHEkFreKj1Dv226a1K73s0=
+github.com/mattn/go-sqlite3 v1.14.33/go.mod h1:Uh1q+B4BYcTPb+yiD3kU8Ct7aC0hY9fxUwlHK0RXw+Y=
 github.com/montanaflynn/stats v0.7.1 h1:etflOAAHORrCC44V+aR6Ftzort912ZU+YLiSTuV8eaE=
 github.com/montanaflynn/stats v0.7.1/go.mod h1:etXPPgVO6n31NxCd9KQUMvCM+ve0ruNzt6R8Bnaayow=
 github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6 h1:ZK8zHtRHOkbHy6Mmr5D264iyp3TiX5OmNcI5cIARiQI=

go.sum

@@ -10,6 +10,7 @@ import (
 
 	_ "github.com/go-sql-driver/mysql"
 	_ "github.com/jackc/pgx/v5/stdlib"
+	_ "github.com/mattn/go-sqlite3"
 	"go.mongodb.org/mongo-driver/bson"
 	"go.mongodb.org/mongo-driver/mongo"
 	"go.mongodb.org/mongo-driver/mongo/options"
@@ -196,6 +197,24 @@ func ConnectMongoDB(connectionString string) (Connection, error) {
 	return &MongoConnection{client: client, dbName: dbName}, nil
 }
 
+// ConnectSQLite connects to SQLite using a file path
+// Connection string format: sqlite:///path/to/database.db
+func ConnectSQLite(connectionString string) (Connection, error) {
+	// Remove the sqlite:// prefix if present
+	dbPath := strings.TrimPrefix(connectionString, "sqlite://")
+
+	db, err := sql.Open("sqlite3", dbPath)
+	if err != nil {
+		return nil, fmt.Errorf("failed to connect to sqlite: %w", err)
+	}
+
+	if err := db.Ping(); err != nil {
+		return nil, fmt.Errorf("failed to ping sqlite: %w", err)
+	}
+
+	return &SQLConnection{db: db}, nil
+}
+
 // ConnectByType connects to a database given its type and connection string
 func ConnectByType(dbType, connectionString string) (Connection, error) {
 	switch dbType {
@@ -205,6 +224,8 @@ func ConnectByType(dbType, connectionString string) (Connection, error) {
 		return ConnectMySQL(connectionString)
 	case "mongodb":
 		return ConnectMongoDB(connectionString)
+	case "sqlite":
+		return ConnectSQLite(connectionString)
 	default:
 		return nil, fmt.Errorf("unsupported database type: %s", dbType)
 	}

internal/db/connections.go

@@ -11,6 +11,8 @@ func GetDatabaseSchema(conn Connection, dbType string) (string, error) {
 		return getPostgresSchema(conn)
 	case "mysql":
 		return getMySQLSchema(conn)
+	case "sqlite":
+		return getSQLiteSchema(conn)
 	case "mongodb":
 		return getMongoDBSchema(conn)
 	default:
@@ -66,6 +68,31 @@ func getMySQLSchema(conn Connection) (string, error) {
 	return formatSchemaResult(result, "MySQL"), nil
 }
 
+func getSQLiteSchema(conn Connection) (string, error) {
+	query := `
+		SELECT 
+			m.name as table_name,
+			p.name as column_name,
+			p.type as data_type,
+			CASE WHEN p."notnull" = 0 THEN 'YES' ELSE 'NO' END as is_nullable
+		FROM sqlite_master m
+		JOIN pragma_table_info(m.name) p
+		WHERE m.type = 'table' AND m.name NOT LIKE 'sqlite_%'
+		ORDER BY m.name, p.cid
+	`
+
+	result, err := conn.Query(query)
+	if err != nil {
+		return "", fmt.Errorf("failed to get SQLite schema: %w", err)
+	}
+
+	if result.Error != nil {
+		return "", fmt.Errorf("schema query error: %w", result.Error)
+	}
+
+	return formatSchemaResult(result, "SQLite"), nil
+}
+
 func getMongoDBSchema(conn Connection) (string, error) {
 	return `MongoDB collections (common field patterns):
 - users: {_id, email, name, status, created_at/createdAt, updated_at/updatedAt}
@@ -112,4 +139,4 @@ func formatSchemaResult(result *Result, dbType string) string {
 	}
 
 	return schema.String()
-}
+}