refactor: migrate storage backend from SQLite to PostgreSQL

Author: CodeMonkeyCybersecurity

go.mod

@@ -18,7 +18,6 @@ require (
 	github.com/lib/pq v1.10.9
 	github.com/likexian/whois v1.15.6
 	github.com/likexian/whois-parser v1.24.20
-	github.com/mattn/go-sqlite3 v1.14.22
 	github.com/miekg/dns v1.1.67
 	github.com/redis/go-redis/v9 v9.11.0
 	github.com/spf13/cobra v1.9.1

internal/api/hera.go

@@ -164,8 +164,7 @@ import (
 	"github.com/CodeMonkeyCybersecurity/shells/internal/logger"
 	"github.com/gin-gonic/gin"
 	"github.com/jmoiron/sqlx"
-	_ "github.com/lib/pq"           // PostgreSQL driver
-	_ "github.com/mattn/go-sqlite3" // SQLite driver
+	_ "github.com/lib/pq" // PostgreSQL driver
 )
 
 // Hera-specific API types

pkg/monitoring/continuous.go

@@ -1140,16 +1140,18 @@ func sortStrings(items []string) {
 func createStorage(backend string) (MonitoringStorage, error) {
 	// Create appropriate storage backend
 	switch backend {
-	case "sqlite", "sqlite3":
-		// Use the same database as the main application
-		return NewSQLiteStorage("shells.db")
+	case "postgres", "postgresql":
+		// Use PostgreSQL connection string from environment or default
+		dsn := "postgres://shells:shells_password@localhost:5432/shells?sslmode=disable"
+		return NewSQLiteStorage(dsn)
 	case "memory":
 		return &InMemoryStorage{
 			data: make(map[string]interface{}),
 		}, nil
 	default:
-		// Default to SQLite for persistence
-		return NewSQLiteStorage("shells.db")
+		// Default to PostgreSQL for persistence
+		dsn := "postgres://shells:shells_password@localhost:5432/shells?sslmode=disable"
+		return NewSQLiteStorage(dsn)
 	}
 }
 

pkg/monitoring/storage.go

@@ -9,19 +9,21 @@ import (
 	"time"
 
 	"github.com/jmoiron/sqlx"
-	_ "github.com/mattn/go-sqlite3"
+	_ "github.com/lib/pq" // PostgreSQL driver
 
 	"github.com/CodeMonkeyCybersecurity/shells/internal/config"
 	"github.com/CodeMonkeyCybersecurity/shells/internal/logger"
 )
 
-// SQLiteStorage implements MonitoringStorage using SQLite3
+// SQLiteStorage implements MonitoringStorage using PostgreSQL
+// Note: Name kept for backward compatibility but now uses PostgreSQL
 type SQLiteStorage struct {
 	db     *sqlx.DB
 	logger *logger.Logger
 }
 
-// NewSQLiteStorage creates a new SQLite storage backend for monitoring
+// NewSQLiteStorage creates a new PostgreSQL storage backend for monitoring
+// Note: Name kept for backward compatibility but now uses PostgreSQL
 func NewSQLiteStorage(dsn string) (*SQLiteStorage, error) {
 	// Initialize logger
 	log, err := logger.New(config.LoggerConfig{Level: "debug", Format: "json"})
@@ -30,10 +32,10 @@ func NewSQLiteStorage(dsn string) (*SQLiteStorage, error) {
 	}
 	log = log.WithComponent("monitoring-storage")
 
-	// Connect to SQLite database
-	db, err := sqlx.Connect("sqlite3", dsn)
+	// Connect to PostgreSQL database
+	db, err := sqlx.Connect("postgres", dsn)
 	if err != nil {
-		return nil, fmt.Errorf("failed to connect to SQLite database: %w", err)
+		return nil, fmt.Errorf("failed to connect to PostgreSQL database: %w", err)
 	}
 
 	storage := &SQLiteStorage{