package handler

import (
	"github.com/gin-gonic/gin"
	"github.com/muyuqingfeng/iloom/auth-service/internal/service"
	"github.com/muyuqingfeng/iloom/shared/pkg/response"
)

type CompanyHandler struct {
	svc *service.CompanyService
}

func NewCompanyHandler(svc *service.CompanyService) *CompanyHandler {
	return &CompanyHandler{svc: svc}
}

func (h *CompanyHandler) GetCompany(c *gin.Context) {
	id := c.Param("id")
	company, err := h.svc.GetByID(c.Request.Context(), id)
	if err != nil {
		response.NotFound(c, err.Error())
		return
	}
	response.OK(c, company)
}

func (h *CompanyHandler) ListRelationships(c *gin.Context) {
	companyID := c.GetHeader("X-Company-ID")
	if companyID == "" {
		response.BadRequest(c, "company id not found")
		return
	}

	rels, err := h.svc.ListRelationships(c.Request.Context(), companyID)
	if err != nil {
		response.InternalError(c, err.Error())
		return
	}

	response.OK(c, rels)
}