mcp

type BaseToolType struct {
	// contains filtered or unexported fields
}

type ExecuteTool struct {
	BaseToolType
}

type ListDatabasesTool struct {
	BaseToolType
}

type PerformanceTool struct {
	BaseToolType
}

type QueryTool struct {
	BaseToolType
}

type Response struct {
	Content  []TextContent          `json:"content"`
	Metadata map[string]interface{} `json:"metadata,omitempty"`
}

type SchemaTool struct {
	BaseToolType
}

type ServerWrapper struct {
	// contains filtered or unexported fields
}

type TextContent struct {
	Type string `json:"type"`
	Text string `json:"text"`
}

type ToolRegistry struct {
	// contains filtered or unexported fields
}

type ToolType interface {
	// GetName returns the base name of the tool type (e.g., "query", "execute")
	GetName() string

	// GetDescription returns a description for this tool type
	GetDescription(dbID string) string

	// CreateTool creates a tool with the specified name
	// The returned tool must be compatible with server.MCPServer.AddTool's first parameter
	CreateTool(name string, dbID string) interface{}

	// HandleRequest handles tool requests for this tool type
	HandleRequest(ctx context.Context, request server.ToolCallRequest, dbID string, useCase UseCaseProvider) (interface{}, error)
}

type ToolTypeFactory struct {
	// contains filtered or unexported fields
}

type TransactionTool struct {
	BaseToolType
}

type UseCaseProvider interface {
	ExecuteQuery(ctx context.Context, dbID, query string, params []interface{}) (string, error)
	ExecuteStatement(ctx context.Context, dbID, statement string, params []interface{}) (string, error)
	ExecuteTransaction(ctx context.Context, dbID, action string, txID string, statement string, params []interface{}, readOnly bool) (string, map[string]interface{}, error)
	GetDatabaseInfo(dbID string) (map[string]interface{}, error)
	ListDatabases() []string
	GetDatabaseType(dbID string) (string, error)
}