package handler

import (
	"net/http"

	"github.com/gin-gonic/gin"
	"github.com/genex/translate-service/internal/application/service"
)

type TranslateHandler struct {
	svc *service.TranslateService
}

func NewTranslateHandler(svc *service.TranslateService) *TranslateHandler {
	return &TranslateHandler{svc: svc}
}

type CreateMappingReq struct {
	UserID          string `json:"userId" binding:"required"`
	InternalAddress string `json:"internalAddress" binding:"required"`
	ChainAddress    string `json:"chainAddress" binding:"required"`
	ChainType       string `json:"chainType" binding:"required"`
}

func (h *TranslateHandler) CreateMapping(c *gin.Context) {
	var req CreateMappingReq
	if err := c.ShouldBindJSON(&req); err != nil {
		c.JSON(http.StatusBadRequest, gin.H{"code": -1, "message": err.Error()})
		return
	}
	mapping := h.svc.CreateMapping(req.UserID, req.InternalAddress, req.ChainAddress, req.ChainType)
	c.JSON(http.StatusOK, gin.H{"code": 0, "data": mapping})
}

func (h *TranslateHandler) Resolve(c *gin.Context) {
	address := c.Query("address")
	direction := c.DefaultQuery("direction", "internal_to_chain")

	if direction == "internal_to_chain" {
		if m, ok := h.svc.InternalToChain(address); ok {
			c.JSON(http.StatusOK, gin.H{"code": 0, "data": m})
			return
		}
	} else {
		if m, ok := h.svc.ChainToInternal(address); ok {
			c.JSON(http.StatusOK, gin.H{"code": 0, "data": m})
			return
		}
	}
	c.JSON(http.StatusNotFound, gin.H{"code": -1, "message": "Address not found"})
}

func (h *TranslateHandler) GetByUser(c *gin.Context) {
	userID := c.Param("userId")
	mappings := h.svc.GetByUserID(userID)
	c.JSON(http.StatusOK, gin.H{"code": 0, "data": mappings})
}