Init

cmd/server/main.go

@@ -0,0 +1,114 @@
+package main
+
+import (
+	"ResendIt/internal/auth"
+	"ResendIt/internal/db"
+	"ResendIt/internal/file"
+	"ResendIt/internal/user"
+	"ResendIt/internal/util"
+	"ResendIt/internal/web"
+	"errors"
+	"fmt"
+	"html/template"
+	"net/http"
+	"os"
+
+	"github.com/gin-gonic/gin"
+	"github.com/joho/godotenv"
+)
+
+func main() {
+	err := godotenv.Load()
+	if err != nil {
+		fmt.Printf("Error loading .env file")
+		return
+	}
+
+	dbCon, err := db.Connect()
+	if err != nil {
+		panic(fmt.Errorf("failed to connect database: %w", err))
+	}
+
+	err = dbCon.AutoMigrate(&user.User{}, &file.FileRecord{})
+	if err != nil {
+		fmt.Printf("Error migrating database: %v\n", err)
+		return
+	}
+
+	r := gin.Default()
+
+	r.MaxMultipartMemory = 10 << 30
+	r.SetFuncMap(template.FuncMap{
+		"add":       func(a, b int) int { return a + b },
+		"sub":       func(a, b int) int { return a - b },
+		"humanSize": util.HumanSize,
+	})
+
+	r.LoadHTMLGlob("internal/templates/*.html")
+	//r.LoadHTMLGlob("internal/templates/new/*.html")
+	r.Static("/static", "./internal/static")
+
+	r.GET("/ping", func(c *gin.Context) {
+		c.JSON(http.StatusOK, gin.H{
+			"message": "hello",
+		})
+	})
+
+	r.NoRoute(func(c *gin.Context) {
+		c.HTML(404, "error.html", nil)
+	})
+
+	authRepo := auth.NewRepository(dbCon)
+	authService := auth.NewService(authRepo)
+	authHandler := auth.NewHandler(authService)
+
+	userRepo := user.NewRepository(dbCon)
+	userService := user.NewService(userRepo)
+	userHandler := user.NewHandler(userService)
+
+	fileRepo := file.NewRepository(dbCon)
+	fileService := file.NewService(fileRepo, "./uploads")
+	fileHandler := file.NewHandler(fileService)
+
+	createAdminUser(userService)
+
+	apiRoute := r.Group("/api")
+
+	auth.RegisterRoutes(apiRoute, authHandler)
+	user.RegisterRoutes(apiRoute, userHandler)
+	file.RegisterRoutes(apiRoute, fileHandler)
+
+	webHandler := web.NewHandler(fileService)
+	web.RegisterRoutes(r, webHandler)
+
+	err = r.Run(":" + os.Getenv("PORT"))
+	if err != nil {
+		return
+	}
+}
+
+func createAdminUser(service *user.Service) {
+	//Check if admin user already exists
+	_, err := service.FindByUsername("admin")
+	if err == nil {
+		fmt.Println("Admin user already exists, skipping creation")
+		return
+	} else if !errors.Is(err, user.ErrUserNotFound) {
+		fmt.Printf("Error checking for admin user: %v\n", err)
+		return
+	}
+
+	adminPassword, exists := os.LookupEnv("ADMIN_PASSWORD")
+	if !exists || adminPassword == "" {
+		fmt.Println("ADMIN_PASSWORD not set in environment variables")
+		fmt.Println("NO ADMIN ACCOUNT WILL BE CREATED")
+		return
+	}
+
+	_, err = service.CreateUser("admin", adminPassword, "admin")
+	if err != nil {
+		fmt.Printf("Error creating admin user: %v\n", err)
+	} else {
+		fmt.Println("Admin user created successfully")
+	}
+}