1 files changed, 12 insertions, 1 deletions
diff --git a/main.go b/main.go
index c763c5a..247d751 100644
--- a/main.go
+++ b/main.go
@@ -2,19 +2,30 @@ package main
 
 import (
 	"embed"
+	"flag"
 	"fmt"
 
 	"git.sr.ht/~gabrielgio/midr/db"
 	"git.sr.ht/~gabrielgio/midr/routes"
+	"github.com/gin-gonic/gin"
 )
 
 var Version = "development"
+var dataDir = flag.String("d", ".", "Folder where midr will store the data")
+var production = flag.Bool("p", false, "Set production")
 
 //go:embed assets/* templates/*
 var f embed.FS
 
 func main() {
+	flag.Parse()
 	fmt.Println("Version:\t", Version)
-	db.ConnectDb()
+	fmt.Println("Datadir:\t", *dataDir)
+
+	if *production {
+		gin.SetMode(gin.ReleaseMode)
+	}
+
+	db.ConnectDb(*dataDir)
 	routes.HandleRequests(f)
 }