ez-api/internal/api/admin_handler.go

package api

import (
	"net/http"

	"github.com/ez-api/ez-api/internal/service"
	"github.com/gin-gonic/gin"
)

type AdminHandler struct {
	masterService *service.MasterService
	syncService   *service.SyncService
}

func NewAdminHandler(masterService *service.MasterService, syncService *service.SyncService) *AdminHandler {
	return &AdminHandler{masterService: masterService, syncService: syncService}
}

type CreateMasterRequest struct {
	Name         string `json:"name" binding:"required"`
	Group        string `json:"group" binding:"required"`
	MaxChildKeys int    `json:"max_child_keys"`
	GlobalQPS    int    `json:"global_qps"`
}

func (h *AdminHandler) CreateMaster(c *gin.Context) {
	var req CreateMasterRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
		return
	}

	// Use defaults if not provided
	if req.MaxChildKeys == 0 {
		req.MaxChildKeys = 5
	}
	if req.GlobalQPS == 0 {
		req.GlobalQPS = 3
	}

	master, rawMasterKey, err := h.masterService.CreateMaster(req.Name, req.Group, req.MaxChildKeys, req.GlobalQPS)
	if err != nil {
		c.JSON(http.StatusInternalServerError, gin.H{"error": "failed to create master key", "details": err.Error()})
		return
	}

	if err := h.syncService.SyncMaster(master); err != nil {
		c.JSON(http.StatusInternalServerError, gin.H{"error": "failed to sync master key", "details": err.Error()})
		return
	}

	c.JSON(http.StatusCreated, gin.H{
		"id":             master.ID,
		"name":           master.Name,
		"group":          master.Group,
		"master_key":     rawMasterKey, // Only show this on creation
		"max_child_keys": master.MaxChildKeys,
		"global_qps":     master.GlobalQPS,
	})
}