main.go

package main

import (
	"context"
	"fmt"
	"log"
	"os"
	"strings"

	"github.com/Soup666/modelmaker/controller"
	db "github.com/Soup666/modelmaker/database"
	repositories "github.com/Soup666/modelmaker/repository"
	"github.com/Soup666/modelmaker/router"
	"github.com/Soup666/modelmaker/services"

	firebase "firebase.google.com/go/v4"
	"google.golang.org/api/option"

	_ "github.com/joho/godotenv/autoload"
)

func printEnvironmentBanner() {
	appEnv := os.Getenv("APP_ENV")
	if appEnv == "" {
		appEnv = "production"
	}

	// ANSI color codes
	const (
		colorRed     = "\033[1;31m" // Bold Red
		colorGreen   = "\033[1;32m" // Bold Green
		colorYellow  = "\033[1;33m" // Bold Yellow
		colorBlue    = "\033[1;34m" // Bold Blue
		colorMagenta = "\033[1;35m" // Bold Magenta
		colorCyan    = "\033[1;36m" // Bold Cyan
		colorReset   = "\033[0m"
	)

	var bannerColor, envText string
	switch appEnv {
	case "dev":
		bannerColor = colorYellow
		envText = "DEVELOPMENT"
	case "staging":
		bannerColor = colorCyan
		envText = "STAGING"
	case "test":
		bannerColor = colorMagenta
		envText = "TEST"
	case "production":
		bannerColor = colorGreen
		envText = "PRODUCTION"
	default:
		bannerColor = colorRed
		envText = strings.ToUpper(appEnv)
	}

	banner := []string{
		"",
		"╔════════════════════════════════════════╗",
		"║             MODEL MAKER                ║",
		fmt.Sprintf("║         Environment: %-16s  ║", envText),
		"╚════════════════════════════════════════╝",
		"",
	}

	for _, line := range banner {
		fmt.Printf("%s%s%s\n", bannerColor, line, colorReset)
	}
}

func main() {
	// Print environment banner
	printEnvironmentBanner()

	// Set up the database connection
	log.Println("Connecting to database...")

	db.ConnectDatabase()

	// Create a Firebase app instance
	opt := option.WithCredentialsFile(os.Getenv("GOOGLE_CREDENTIALS_FILE"))
	app, err := firebase.NewApp(context.Background(), nil, opt)
	if err != nil {
		log.Fatalf("Failed to create Firebase app: %v", err)
	}

	// Create a Firebase auth client instance
	authClient, err := app.Auth(context.Background())
	if err != nil {
		log.Fatalf("Failed to create Firebase auth client: %v", err)
	}

	// Set up the repositories
	userRepo := repositories.NewUserRepository(db.DB)
	taskRepo := repositories.NewTaskRepository(db.DB)
	appFileRepo := repositories.NewAppFileRepository(db.DB)
	reportsRepo := repositories.NewReportsRepository(db.DB)
	collectionsRepo := repositories.NewCollectionsRepository(db.DB)
	chatRepo := repositories.NewChatRepository(db.DB)
	userAnalyticsRepo := repositories.NewUserAnalyticsRepository(db.DB)

	// Set up the services
	authService := services.NewAuthService(authClient, db.DB, userRepo)
	userService := services.NewUserService(userRepo)
	notificationService := services.NewNotificationService()
	appFileService := services.NewAppFileServiceFile(appFileRepo)
	storageService := services.NewKatapultStorageService()
	taskService := services.NewTaskService(taskRepo, appFileService, chatRepo, notificationService, storageService)
	visionService := services.NewVisionService()
	reportsService := services.NewReportsService(reportsRepo)
	collectionsService := services.NewCollectionsService(collectionsRepo)
	userAnalyticsService := services.NewUserAnalyticsService(userAnalyticsRepo)

	// Initialize Job Queue
	taskService.StartWorker()

	// Set up the controllers
	authController := controller.NewAuthController(authService, userService)
	uploadController := controller.NewUploadController(storageService)
	objectController := controller.NewObjectController(storageService)
	taskController := controller.NewTaskController(&taskService, appFileService, visionService, storageService)
	visionController := controller.NewVisionController(visionService, taskRepo, &taskService)
	reportsController := controller.NewReportsController(reportsService)
	collectionsController := controller.NewCollectionsController(collectionsService)
	userAnalyticsController := controller.NewUserAnalyticsController(userAnalyticsService)
	notificationController := controller.NewNotificationController(notificationService)

	// Set up the HTTP router
	r := router.NewRouter(
		authController,
		taskController,
		uploadController,
		objectController,
		visionController,
		authService,
		reportsController,
		collectionsController,
		userAnalyticsController,
		notificationController,
	)

	// Start the server
	port := os.Getenv("PORT")
	if port == "" {
		port = "3333"
	}

	log.Printf("Starting server on port %s...", port)
	if err := r.Run(":" + port); err != nil {
		log.Fatalf("Failed to start server: %v", err)
	}
}