postgres and such

main.go

@@ -1,20 +1,121 @@
 package main
 
 import (
+	"context"
 	"log"
 	"net/http"
+	"os"
+	"path/filepath"
 
-	"michaelthomson.dev/mthomson/go-todos-app/db"
+	pgxuuid "github.com/jackc/pgx-gofrs-uuid"
+	"github.com/jackc/pgx/v5"
+	"github.com/jackc/pgx/v5/pgxpool"
+	"github.com/joho/godotenv"
 	"michaelthomson.dev/mthomson/go-todos-app/handlers"
+	"michaelthomson.dev/mthomson/go-todos-app/repositories"
 	"michaelthomson.dev/mthomson/go-todos-app/services"
 )
 
 func main() {
-  todosStore := db.NewTodoStore()
-  ts := services.NewTodoService(&todosStore)
+  // load environment
+  err := godotenv.Load()
 
-  homeHandler := handlers.NewHomeHandler(ts)
-  todoHandler := handlers.NewTodoHandler(ts)
+  if err != nil {
+    log.Fatalf("Error loading .env file: %v", err)
+    os.Exit(1)
+  }
+
+  databaseUrl := os.Getenv("DATABASE_URL")
+
+  // connect to Db
+  dbconfig, err := pgxpool.ParseConfig(databaseUrl)
+  if err != nil {
+    log.Fatalf("Unable to parse db config: %v", err)
+    os.Exit(1)
+  }
+
+  dbconfig.AfterConnect = func(ctx context.Context, conn *pgx.Conn) error {
+    pgxuuid.Register(conn.TypeMap())
+    return nil
+  }
+
+  dbpool, err := pgxpool.New(context.Background(), databaseUrl)
+  if err != nil {
+    log.Fatalf("Unable to create connection pool: %v", err)
+    os.Exit(1)
+  }
+  defer dbpool.Close()
+
+  // apply initial sql
+  log.Print("Applying migrations table")
+  content, err := os.ReadFile(filepath.Join("sql", "init.sql"))
+
+  if err != nil {
+    log.Fatal(err)
+    os.Exit(1)
+  }
+
+  _, err = dbpool.Exec(context.Background(), string(content))
+
+  if err != nil {
+    log.Fatal(err)
+    os.Exit(1)
+  }
+
+  //apply migrations
+  log.Print("Starting migrations...")
+  files, err := os.ReadDir("./sql/migrations")
+  if err != nil {
+    log.Fatal(err)
+    os.Exit(1)
+  }
+
+  for _, file := range files {
+    path := filepath.Join("sql", "migrations", file.Name())
+    content, err := os.ReadFile(path)
+
+    if err != nil {
+      log.Fatal(err)
+      os.Exit(1)
+    }
+
+    var count int
+    err = dbpool.QueryRow(context.Background(), "SELECT COUNT(1) FROM migration WHERE name=$1", file.Name()).Scan(&count)
+
+    if err != nil {
+      log.Fatal(err)
+      os.Exit(1)
+    }
+
+    if count == 0 {
+      log.Printf("Running migration: %s", file.Name())
+      _, err = dbpool.Exec(context.Background(), string(content))
+
+      if err != nil {
+	log.Fatal(err)
+	os.Exit(1)
+      }
+
+      _, err = dbpool.Exec(context.Background(), "INSERT INTO migration(name) values($1)", file.Name())
+
+      if err != nil {
+	log.Fatal(err)
+	os.Exit(1)
+      }
+    } else {
+      log.Printf("Migration already exists: %s", file.Name())
+    }
+  }
+
+  // set up repositories
+  todoRepository := repositories.NewPostgresTodoRepository(dbpool)
+
+  // set up services
+  todoService := services.NewTodoService(&todoRepository)
+
+  // set up handlers
+  homeHandler := handlers.NewHomeHandler(todoService)
+  todoHandler := handlers.NewTodoHandler(todoService)
 
   router := http.NewServeMux()
 
@@ -25,6 +126,8 @@ func main() {
   router.HandleFunc("GET /{$}", homeHandler.Home)
   router.HandleFunc("POST /todos", todoHandler.Create)
   router.HandleFunc("DELETE /todos/{id}", todoHandler.Delete)
+  router.HandleFunc("PATCH /todos/{id}/done", todoHandler.Done)
+  router.HandleFunc("PATCH /todos/{id}/undone", todoHandler.Undone)
 
   server := http.Server{
     Addr: "localhost:3000",