database setup and getcurrent

cmd/internal/db/connection.go

@@ -7,11 +7,29 @@ import (
 	"github.com/siestaw/laterna/server/cmd/internal/logger"
 )
 
+var DB *sql.DB
+
 func ConnectDB() {
-	db, err := sql.Open("sqlite3", "./db.sql")
+	var err error
+	DB, err = sql.Open("sqlite3", "./db.sql")
 	if err != nil {
 		logger.DBLogger.Printf("An Error occured while connecting to the database: %v", err)
+		return
+	}
+	logger.DBLogger.Printf("Successfully connected to the database!")
+	InitDB()
+}
+
+func InitDB() {
+	_, err := DB.Exec(`
+		CREATE TABLE IF NOT EXISTS lamp_state (
+		id TEXT PRIMARY KEY,
+		color TEXT NOT NULL,
+		updated_at DATETIME NOT NULL
+		);
+	`)
+
+	if err != nil {
+		logger.DBLogger.Fatalf("An error occured while initializing the database: %v", err)
 	}
-	logger.DBLogger.Printf("Connected")
-	defer db.Close()
 }

cmd/internal/db/queries.go

@@ -0,0 +1,19 @@
+package db
+
+import (
+	"github.com/siestaw/laterna/server/cmd/internal/logger"
+	"github.com/siestaw/laterna/server/cmd/internal/models"
+)
+
+func ViewColor(id int) (*models.LampState, error) {
+	row := DB.QueryRow("SELECT * FROM lamp_state WHERE id = ?", id)
+
+	var state models.LampState
+	err := row.Scan(&state.ID, &state.Color, &state.UpdatedAt)
+	if err != nil {
+		logger.DBLogger.Print(err)
+		return nil, err
+	}
+
+	return &state, nil
+}

cmd/internal/db/utils.go

@@ -1 +0,0 @@
-package db