package api

import (
	"context"
	"errors"
	"net/http"

	"github.com/gin-gonic/gin"

	"monica-go-completion-backend/internal/completion"
)

type CompletionService interface {
	Complete(ctx context.Context, req completion.Request) (completion.Response, error)
}

func RegisterRoutes(router *gin.Engine, service CompletionService) {
	router.GET("/health", func(c *gin.Context) {
		c.JSON(http.StatusOK, gin.H{"status": "ok"})
	})

	registerWSRoutes(router, service)

	router.POST("/api/v1/completions/go", func(c *gin.Context) {
		var req completion.Request
		if err := c.ShouldBindJSON(&req); err != nil {
			c.JSON(http.StatusBadRequest, gin.H{"error": "invalid JSON payload"})
			return
		}

		resp, err := service.Complete(c.Request.Context(), req)
		if err != nil {
			if errors.Is(err, completion.ErrInvalidRequest) {
				c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
				return
			}
			c.JSON(http.StatusInternalServerError, gin.H{"error": "completion failed"})
			return
		}

		c.JSON(http.StatusOK, resp)
	})
}