86 lines
2.0 KiB
Go
86 lines
2.0 KiB
Go
package main
|
|
|
|
import(
|
|
"os"
|
|
"io"
|
|
"net/http"
|
|
"fmt"
|
|
|
|
"gorm.io/gorm"
|
|
"gorm.io/driver/sqlite"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/joho/godotenv"
|
|
|
|
"example.com/gin/test/controllers"
|
|
"example.com/gin/test/models"
|
|
"example.com/gin/test/middlewares"
|
|
)
|
|
|
|
var(
|
|
roomController controllers.RoomController
|
|
userController controllers.UserController
|
|
autoValidator middlewares.AuthValidator
|
|
)
|
|
|
|
func LoadEnvVariables() {
|
|
err := godotenv.Load(".env")
|
|
|
|
if err != nil {
|
|
fmt.Println("Error loading .env file")
|
|
}
|
|
}
|
|
|
|
func setupLogOutput() {
|
|
f, _ := os.Create("gin.log")
|
|
gin.DefaultWriter = io.MultiWriter(f, os.Stdout)
|
|
}
|
|
|
|
func SetReply(ctx *gin.Context, err error, message any) {
|
|
if err != nil {
|
|
ctx.JSON(http.StatusBadRequest, gin.H{ "error": err.Error() })
|
|
} else {
|
|
ctx.JSON(http.StatusOK, message)
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
LoadEnvVariables()
|
|
|
|
db, err := gorm.Open(sqlite.Open(os.Getenv("SQLITE_DB")), &gorm.Config{})
|
|
if err != nil {
|
|
panic("failed to connect to database")
|
|
}
|
|
|
|
err = db.AutoMigrate(&models.Room{}, &models.User{}, &models.Booking{})
|
|
if err != nil {
|
|
panic("failed to migrate database")
|
|
}
|
|
|
|
roomController = controllers.NewRoomController(db)
|
|
userController = controllers.NewUserController(db)
|
|
authValidator := middlewares.AuthValidator{ DB: db, }
|
|
|
|
server := gin.New()
|
|
server.Use(gin.Recovery())
|
|
server.Use(gin.Logger())
|
|
|
|
server.Static("/static", "./static")
|
|
server.LoadHTMLGlob("views/*.html")
|
|
|
|
apiRoutes := server.Group("/api")
|
|
//apiRoutes.Use(middlewares.BasicAuth())
|
|
{
|
|
apiRoutes.POST("/rooms", roomController.Create)
|
|
apiRoutes.GET("/rooms", roomController.GetAll)
|
|
apiRoutes.GET("/rooms/:id", roomController.GetById)
|
|
apiRoutes.PUT("/rooms/:id", roomController.Update)
|
|
apiRoutes.DELETE("/rooms/:id", roomController.Delete)
|
|
|
|
apiRoutes.POST("/users/signup", userController.Signup)
|
|
apiRoutes.POST("/users/login", userController.Login)
|
|
apiRoutes.GET("/users/validate", authValidator.RequireAuth, userController.Validate)
|
|
}
|
|
|
|
server.Run(":"+os.Getenv("PORT"))
|
|
}
|