server.go

package main

import (
	"context"
	"time"

	"github.com/jatindotdev/tinybits/api/db"
	"github.com/jatindotdev/tinybits/api/handlers"
	"github.com/jatindotdev/tinybits/api/lib"
	"github.com/jatindotdev/tinybits/api/middlewares"
	"github.com/jatindotdev/tinybits/api/routes"
	"github.com/jatindotdev/tinybits/api/storage"
	"github.com/jmoiron/sqlx"
	"github.com/labstack/echo/v4"
	"github.com/redis/go-redis/v9"
	"go.uber.org/fx"
)

func createServer(linkHandler *handlers.LinkHandler, db *sqlx.DB, redis *redis.Client) *echo.Echo {
	app := echo.New()
	app.Validator = lib.NewValidator()

	middlewares.SetupMiddlewares(app)
	routes.SetupHelperRoutes(app, db, redis)

	api := app.Group("/api")
	routes.SetupRoutes(api, linkHandler)

	return app
}

func initServer(lifecycle fx.Lifecycle, app *echo.Echo) {
	lifecycle.Append(fx.Hook{
		OnStart: func(context.Context) error {
			routes.StartTime = time.Now()
			address := lib.GetServerAddress()
			go app.Start(address)
			return nil
		},
		OnStop: func(ctx context.Context) error {
			return app.Shutdown(ctx)
		},
	})
}

func main() {
	fx.New(
		fx.Provide(
			handlers.NewLinkHandler,
			storage.NewLinkStorage,
			db.CreatePostgresConnection,
			db.CreateRedisClient,
			createServer,
		),
		fx.Invoke(
			lib.LoadEnv,
			initServer,
		),
	).Run()
}