bill-server-go/internal/infrastructure/fiber.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"))
}