gochat/backend/internal/handler/api/v1/upload_handler.go

package v1

import (
	"errors"
	"net/http"
	"strings"

	"github.com/gin-gonic/gin"
	"gorm.io/gorm"

	"github.com/gochat/gochat/internal/service"
	"github.com/gochat/gochat/pkg/response"
)

// UploadHandler handles file upload endpoints (account-level + widget direct + account direct).
type UploadHandler struct {
	svc *service.UploadService
}

// NewUploadHandler creates a new UploadHandler.
func NewUploadHandler(svc *service.UploadService) *UploadHandler {
	return &UploadHandler{svc: svc}
}

// Upload handles POST /api/v1/accounts/:id/upload — account-level file upload.
// Reference: Chatwoot api/v1/accounts/:account_id/upload
func (h *UploadHandler) Upload(c *gin.Context) {
	accountID := getAccountID(c)
	if accountID == 0 {
		response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrValidation, "account_id is required")
		return
	}

	var result *service.UploadResponse
	var svcErr error
	if strings.Contains(c.GetHeader("Content-Type"), "application/json") {
		var req struct {
			ExternalURL string `json:"external_url"`
		}
		if err := c.ShouldBindJSON(&req); err != nil || strings.TrimSpace(req.ExternalURL) == "" {
			c.JSON(http.StatusUnprocessableEntity, gin.H{"error": "missing input"})
			return
		}
		result, svcErr = h.svc.AccountUploadFromURL(c.Request.Context(), accountID, strings.TrimSpace(req.ExternalURL))
	} else {
		fileHeader, err := c.FormFile("attachment")
		if err != nil {
			fileHeader, err = c.FormFile("file")
		}
		if err != nil {
			c.JSON(http.StatusUnprocessableEntity, gin.H{"error": "missing input"})
			return
		}
		result, svcErr = h.svc.AccountUpload(c.Request.Context(), accountID, service.AccountUploadRequest{
			FileHeader: fileHeader,
		})
	}
	if svcErr != nil {
		status := http.StatusUnprocessableEntity
		if errors.Is(svcErr, gorm.ErrRecordNotFound) {
			status = http.StatusNotFound
		}
		c.JSON(status, gin.H{"error": svcErr.Error()})
		return
	}

	c.JSON(http.StatusOK, gin.H{
		"file_url": result.FileURL,
		"blob_id":  result.UploadUUID,
		"blob_key": result.UploadUUID,
	})
}

// DirectUpload handles POST /api/v1/widget/direct_uploads — widget direct file upload.
// Reference: Chatwoot POST /widget/direct_uploads
func (h *UploadHandler) DirectUpload(c *gin.Context) {
	if strings.Contains(c.GetHeader("Content-Type"), "application/json") {
		var req service.ActiveStorageDirectUploadRequest
		if err := c.ShouldBindJSON(&req); err != nil {
			response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrValidation, "invalid direct upload metadata")
			return
		}
		req.WebsiteToken = c.Query("website_token")
		req.AuthToken = c.GetHeader("X-Auth-Token")
		result, svcErr := h.svc.CreateWidgetDirectUpload(c.Request.Context(), req)
		if svcErr != nil {
			handleServiceError(c, svcErr)
			return
		}
		c.JSON(http.StatusOK, result)
		return
	}

	fileHeader, err := c.FormFile("file")
	if err != nil {
		response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrValidation, "file is required")
		return
	}

	result, svcErr := h.svc.WidgetDirectUpload(c.Request.Context(), service.WidgetDirectUploadRequest{
		FileHeader: fileHeader,
	})
	if svcErr != nil {
		handleServiceError(c, svcErr)
		return
	}

	response.OK(c, result)
}

func (h *UploadHandler) CompleteWidgetDirectUpload(c *gin.Context) {
	result, svcErr := h.svc.CompleteWidgetDirectUpload(c.Request.Context(), c.Param("upload_uuid"), c.Request.Body)
	if svcErr != nil {
		handleServiceError(c, svcErr)
		return
	}
	response.OK(c, result)
}

// AccountDirectUpload handles POST /api/v1/accounts/:id/direct_uploads — account-level staged upload.
// Returns a blob/UUID for later attachment to messages.
// Reference: Chatwoot POST /api/v1/accounts/:account_id/direct_uploads
func (h *UploadHandler) AccountDirectUpload(c *gin.Context) {
	accountID := getAccountID(c)
	if accountID == 0 {
		response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrValidation, "account_id is required")
		return
	}

	fileHeader, err := c.FormFile("file")
	if err != nil {
		response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrValidation, "file is required")
		return
	}

	result, svcErr := h.svc.AccountDirectUpload(c.Request.Context(), accountID, service.AccountDirectUploadRequest{
		FileHeader: fileHeader,
	})
	if svcErr != nil {
		handleServiceError(c, svcErr)
		return
	}

	response.OK(c, result)
}

// ConversationDirectUpload handles Chatwoot's nested conversation direct upload
// endpoint used by the reused dashboard message composer.
// Reference: POST /api/v1/accounts/:account_id/conversations/:conversation_id/direct_uploads
func (h *UploadHandler) ConversationDirectUpload(c *gin.Context) {
	accountID := getAccountID(c)
	if accountID == 0 {
		response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrValidation, "account_id is required")
		return
	}
	conversationID, err := parseUintParam(c, "conversation_id")
	if err != nil || conversationID == 0 {
		response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrBadRequest, "invalid conversation_id")
		return
	}

	if !strings.Contains(c.GetHeader("Content-Type"), "application/json") {
		response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrValidation, "invalid direct upload metadata")
		return
	}

	var req service.ActiveStorageDirectUploadRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrValidation, "invalid direct upload metadata")
		return
	}
	result, svcErr := h.svc.CreateConversationDirectUpload(c.Request.Context(), accountID, conversationID, req)
	if svcErr != nil {
		handleServiceError(c, svcErr)
		return
	}
	c.JSON(http.StatusOK, result)
}

func (h *UploadHandler) CompleteConversationDirectUpload(c *gin.Context) {
	accountID := getAccountID(c)
	if accountID == 0 {
		response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrValidation, "account_id is required")
		return
	}
	conversationID, err := parseUintParam(c, "conversation_id")
	if err != nil || conversationID == 0 {
		response.AbortWithStatusError(c, http.StatusBadRequest, response.ErrBadRequest, "invalid conversation_id")
		return
	}

	result, svcErr := h.svc.CompleteConversationDirectUpload(c.Request.Context(), accountID, conversationID, c.Param("upload_uuid"), c.Request.Body)
	if svcErr != nil {
		handleServiceError(c, svcErr)
		return
	}
	response.OK(c, result)
}