From fbd799c8ed9c065bf4b8872f21d2634bedd9868a Mon Sep 17 00:00:00 2001
From: James Tombleson <luther38@gmail.com>
Date: Fri, 5 Apr 2024 17:53:28 -0700
Subject: [PATCH] Added migrations on startup

---
 main.go | 26 +++++++++++++++++++++-----
 1 file changed, 21 insertions(+), 5 deletions(-)

diff --git a/main.go b/main.go
index 411f3f4..63fb65d 100644
--- a/main.go
+++ b/main.go
@@ -2,15 +2,17 @@ package main
 
 import (
 	"database/sql"
-	v1 "go-cook/api/handlers/v1"
-	"go-cook/api/services"
 	"log"
 	"net/http"
 
+	"git.jamestombleson.com/jtom38/go-cook/api/services"
+	v1 "git.jamestombleson.com/jtom38/go-cook/api/handlers/v1"
+
 	_ "github.com/glebarez/go-sqlite"
 	"github.com/go-playground/validator"
 	"github.com/labstack/echo/v4"
 	"github.com/labstack/echo/v4/middleware"
+	"github.com/pressly/goose/v3"
 )
 
 // @title Swagger Example API
@@ -24,6 +26,20 @@ func main() {
 
 	cfg := services.NewEnvConfig()
 
+	// Check if we should apply migrations on startup
+	// You can disable this in the ENV configuration of your application.
+	if !cfg.DisableMigrationsOnStartUp {
+		err = goose.SetDialect("sqlite3")
+		if err != nil {
+			panic(err)
+		}
+
+		err = goose.Up(db, "api/migrations")
+		if err != nil {
+			panic(err)
+		}
+	}
+
 	e := echo.New()
 	e.Validator = &CustomValidator{
 		validator: validator.New(),
@@ -45,8 +61,8 @@ type CustomValidator struct {
 
 func (cv *CustomValidator) Validate(i interface{}) error {
 	if err := cv.validator.Struct(i); err != nil {
-	  // Optionally, you could return the error to give each route more control over the status code
-	  return echo.NewHTTPError(http.StatusBadRequest, err.Error())
+		// Optionally, you could return the error to give each route more control over the status code
+		return echo.NewHTTPError(http.StatusBadRequest, err.Error())
 	}
 	return nil
-  }
+}