51 lines
1.0 KiB
Go
51 lines
1.0 KiB
Go
|
package handler
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net/http"
|
||
|
|
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
"github.com/muyuqingfeng/iloom/shared/pkg/response"
|
||
|
|
"github.com/muyuqingfeng/iloom/textile-service/internal/service"
|
||
|
|
)
|
||
|
|
|
||
|
|
type PlanHandler struct {
|
||
|
|
svc *service.PlanService
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewPlanHandler(svc *service.PlanService) *PlanHandler {
|
||
|
|
return &PlanHandler{svc: svc}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (h *PlanHandler) List(c *gin.Context) {
|
||
|
|
companyID := c.GetHeader("X-Company-ID")
|
||
|
|
if companyID == "" {
|
||
|
|
response.BadRequest(c, "missing X-Company-ID header")
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
plans, err := h.svc.List(c.Request.Context(), companyID)
|
||
|
|
if err != nil {
|
||
|
|
response.Error(c, http.StatusInternalServerError, 50000, err.Error())
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
response.OK(c, plans)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (h *PlanHandler) Get(c *gin.Context) {
|
||
|
|
companyID := c.GetHeader("X-Company-ID")
|
||
|
|
if companyID == "" {
|
||
|
|
response.BadRequest(c, "missing X-Company-ID header")
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
planID := c.Param("id")
|
||
|
|
plan, err := h.svc.GetByID(c.Request.Context(), planID, companyID)
|
||
|
|
if err != nil {
|
||
|
|
response.NotFound(c, "plan not found")
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
response.OK(c, plan)
|
||
|
|
}
|