88 lines
2.6 KiB
Go
88 lines
2.6 KiB
Go
package infrastructure
|
|
|
|
import (
|
|
"bill-server/internal/bill"
|
|
"bill-server/internal/label"
|
|
"bill-server/internal/misc"
|
|
"fmt"
|
|
"github.com/gofiber/fiber/v2"
|
|
"log"
|
|
|
|
_ "github.com/go-sql-driver/mysql"
|
|
"github.com/gofiber/fiber/v2/middleware/compress"
|
|
"github.com/gofiber/fiber/v2/middleware/cors"
|
|
"github.com/gofiber/fiber/v2/middleware/etag"
|
|
"github.com/gofiber/fiber/v2/middleware/favicon"
|
|
"github.com/gofiber/fiber/v2/middleware/limiter"
|
|
"github.com/gofiber/fiber/v2/middleware/logger"
|
|
"github.com/gofiber/fiber/v2/middleware/recover"
|
|
"github.com/gofiber/fiber/v2/middleware/requestid"
|
|
)
|
|
|
|
// Run our Fiber webserver.
|
|
func Run() {
|
|
// Try to connect to our database as the initial part.
|
|
//mariadb, err := ConnectToMariaDB()
|
|
sqlite, err := ConnectToSqlite()
|
|
if err != nil {
|
|
log.Fatal("Database connection error: $s", err)
|
|
}
|
|
|
|
// Creates a new Fiber instance.
|
|
app := fiber.New(fiber.Config{
|
|
AppName: "Bill Server",
|
|
ServerHeader: "Fiber",
|
|
})
|
|
|
|
// Use global middlewares.
|
|
app.Use(cors.New())
|
|
app.Use(compress.New())
|
|
app.Use(etag.New())
|
|
app.Use(favicon.New())
|
|
app.Use(limiter.New(limiter.Config{
|
|
Max: 100,
|
|
LimitReached: func(c *fiber.Ctx) error {
|
|
return c.Status(fiber.StatusTooManyRequests).JSON(&fiber.Map{
|
|
"status": "fail",
|
|
"message": "You have requested too many in a single time-frame! Please wait another minute!",
|
|
})
|
|
},
|
|
}))
|
|
app.Use(logger.New())
|
|
app.Use(recover.New())
|
|
app.Use(requestid.New())
|
|
|
|
// Create repositories.
|
|
//cityRepository := city.NewCityRepository(mariadb)
|
|
//userRepository := user.NewUserRepository(mariadb)
|
|
billRepository := bill.NewBillRepository(sqlite)
|
|
labelRepository := label.NewLabelRepository(sqlite)
|
|
|
|
// Create all of our services.
|
|
//cityService := city.NewCityService(cityRepository)
|
|
//userService := user.NewUserService(userRepository)
|
|
billService := bill.NewBillService(billRepository)
|
|
labelService := label.NewLabelService(labelRepository)
|
|
|
|
// Prepare our endpoints for the API.
|
|
misc.NewMiscHandler(app.Group("/api/v1"))
|
|
//auth.NewAuthHandler(app.Group("/api/v1/auth"))
|
|
//city.NewCityHandler(app.Group("/api/v1/cities"), cityService)
|
|
//user.NewUserHandler(app.Group("/api/v1/users"), userService)
|
|
bill.NewBillHandler(app.Group("/api/v1/bills"), billService)
|
|
label.NewLabelHandler(app.Group("/api/v1/labels"), labelService)
|
|
|
|
// Prepare an endpoint for 'Not Found'.
|
|
app.All("*", func(c *fiber.Ctx) error {
|
|
errorMessage := fmt.Sprintf("Route '%s' does not exist in this API!", c.OriginalURL())
|
|
|
|
return c.Status(fiber.StatusNotFound).JSON(&fiber.Map{
|
|
"status": "fail",
|
|
"message": errorMessage,
|
|
})
|
|
})
|
|
|
|
// Listen to port 8080.
|
|
log.Fatal(app.Listen(":8080"))
|
|
}
|