ForFarm/backend/internal/cmd/api.go

package cmd

import (
	"context"
	"os"

	"log/slog"
	"net/http"

	"github.com/spf13/cobra"

	"github.com/forfarm/backend/internal/api"
	"github.com/forfarm/backend/internal/cmdutil"
	"github.com/forfarm/backend/internal/config"
	"github.com/forfarm/backend/internal/event"
	"github.com/forfarm/backend/internal/repository"
)

func APICmd(ctx context.Context) *cobra.Command {
	var port int = config.PORT

	cmd := &cobra.Command{
		Use:   "api",
		Args:  cobra.ExactArgs(0),
		Short: "Run RESTful API",
		RunE: func(cmd *cobra.Command, args []string) error {
			logger := slog.New(slog.NewTextHandler(os.Stdout, nil))

			pool, err := cmdutil.NewDatabasePool(ctx, 16)
			if err != nil {
				return err
			}
			defer pool.Close()

			logger.Info("connected to database")

			// Events

			eventBus, err := event.NewRabbitMQEventBus(config.RABBITMQ_URL, logger)
			if err != nil {
				logger.Error("failed to connect to event bus", "err", err)
				os.Exit(1)
			}
			logger.Info("connecting to event bus", "url", config.RABBITMQ_URL)

			aggregator := event.NewEventAggregator(eventBus, eventBus, logger)
			if err := aggregator.Start(ctx); err != nil {
				logger.Error("failed to start event aggregator", "err", err)
				os.Exit(1)
			}
			logger.Info("Start event aggregator")

			analyticRepo := repository.NewPostgresAnalyticsRepository(pool)
			projection := event.NewAnalyticsProjection(eventBus, analyticRepo, logger)
			if err := projection.Start(ctx); err != nil {
				logger.Error("failed to start analytics projection", "err", err)
				os.Exit(1)
			}
			logger.Info("Start analytics projection")
			//

			api := api.NewAPI(ctx, logger, pool, eventBus)
			server := api.Server(port)

			go func() {
				if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
					logger.Error("failed to start server", "err", err)
				}
			}()

			logger.Info("started API", "port", port)

			<-ctx.Done()

			if err := server.Shutdown(ctx); err != nil {
				logger.Error("failed to gracefully shutdown server", "err", err)
			}

			return nil
		},
	}

	return cmd
}