114 lines
2.6 KiB
Go
114 lines
2.6 KiB
Go
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")
|
|
}
|
|
|
|
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("templates/*.html")
|
|
//r.LoadHTMLGlob("internal/templates/new/*.html")
|
|
r.Static("/static", "/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")
|
|
}
|
|
}
|