internal/adminapi/tool_inventory_handler.go

Rama main · réplica de lectura del repositorio original.

package adminapi

import (
	"net/http"

	"github.com/yeomyeonggeori/blueclaw/internal/agentruntime"
)

type QuarantinedToolProvider struct {
	ProviderID string `json:"providerID"`
	Reason     string `json:"reason"`
}

type ToolInventoryDocument struct {
	Tools                []string                  `json:"tools"`
	ProviderByTool       map[string]string         `json:"providerByTool"`
	QuarantinedProviders []QuarantinedToolProvider `json:"quarantinedProviders"`
}

type ToolInventoryHandler struct {
	ToolCatalogBuilder *agentruntime.ToolCatalogBuilder
}

func (handler ToolInventoryHandler) HandleListTools(responseWriter http.ResponseWriter, request *http.Request) {
	if handler.ToolCatalogBuilder == nil {
		http.Error(responseWriter, "the tool catalog is unavailable", http.StatusServiceUnavailable)
		return
	}
	toolSet := handler.ToolCatalogBuilder.BuildToolSet(agentruntime.ToolCatalogRequest{
		ProfileName:    "default",
		RequesterEmail: request.URL.Query().Get("requester"),
	})
	document := ToolInventoryDocument{
		Tools:                toolSet.ListRegisteredToolNames(),
		ProviderByTool:       map[string]string{},
		QuarantinedProviders: []QuarantinedToolProvider{},
	}
	for _, definition := range toolSet.ListRegisteredToolDefinitions() {
		document.ProviderByTool[definition.Name] = definition.ProviderID
	}
	for _, quarantinedProvider := range toolSet.QuarantinedProviders() {
		document.QuarantinedProviders = append(document.QuarantinedProviders, QuarantinedToolProvider{
			ProviderID: quarantinedProvider.ProviderID,
			Reason:     quarantinedProvider.Reason,
		})
	}
	writeJSON(responseWriter, http.StatusOK, document)
}