main.go

package main

import (
	"log"
	"net/http"

	"github.com/gin-gonic/gin"
	swaggerFiles "github.com/swaggo/files"
	ginSwagger "github.com/swaggo/gin-swagger"
	"github.com/vhybZApp/api/agent"
	"github.com/vhybZApp/api/azure"
	"github.com/vhybZApp/api/config"
	"github.com/vhybZApp/api/database"
	_ "github.com/vhybZApp/api/docs"
)

// @title           Vhybz API
// @version         1.0.0
// @description     API for Vhybz application with JWT authentication and Azure OpenAI integration
// @termsOfService  http://swagger.io/terms/

// @contact.name   API Support
// @contact.url    https://github.com/vhybZApp/api
// @contact.email  support@vhybz.com

// @license.name  Apache 2.0
// @license.url   http://www.apache.org/licenses/LICENSE-2.0.html

// @host      localhost:8080
// @BasePath  /
// @schemes   http https

// @securityDefinitions.apikey BearerAuth
// @in header
// @name Authorization
// @description JWT Authorization header using the Bearer scheme. Example: "Bearer {token}"

func main() {
	// Load configuration
	config.LoadConfig()

	// Initialize database
	if err := database.Initialize(); err != nil {
		log.Fatalf("Error initializing database: %v", err)
	}

	// Create Gin router
	r := gin.Default()

	// Health check endpoint
	r.GET("/health", func(c *gin.Context) {
		c.JSON(http.StatusOK, gin.H{
			"status": "healthy",
		})
	})

	// Swagger documentation

	r.GET("/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler))

	// Auth routes
	auth := r.Group("/auth")
	{
		auth.POST("/register", register)
		auth.POST("/login", login)
		auth.POST("/refresh", refresh)
		auth.GET("/profile", authMiddleware(), getProfile)
	}

	// Azure OpenAI routes
	azureGroup := r.Group("/azure")
	{
		azureGroup.POST("/chat/completions", authMiddleware(), azure.ChatCompletion)
	}

	//  Agent routes
	agentGroup := r.Group("/agent")
	{
		agentGroup.POST("/agent/make-html", authMiddleware(), agent.MakeHTML)
	}

	// Start server
	if err := r.Run(":8080"); err != nil {
		log.Fatalf("Error starting server: %v", err)
	}
}