mcp

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

type CompletionItem struct {
	Name             string   `json:"name"`
	Type             string   `json:"type"`
	Documentation    string   `json:"documentation"`
	InsertText       string   `json:"insertText"`
	Parameters       []string `json:"parameters,omitempty"`
	ReturnType       string   `json:"returnType,omitempty"`
	Category         string   `json:"category,omitempty"`
	SortText         string   `json:"sortText,omitempty"`
	FilterText       string   `json:"filterText,omitempty"`
	CommitCharacters []string `json:"commitCharacters,omitempty"`
}

type CompressionConfig struct {
	SegmentBy string `json:"segmentBy,omitempty"`
	OrderBy   string `json:"orderBy,omitempty"`
	Interval  string `json:"interval,omitempty"`
}

type ExecuteTool struct {
	BaseToolType
}

type HypertableColumnInfo struct {
	Name        string `json:"name"`
	Type        string `json:"type"`
	Nullable    bool   `json:"nullable"`
	PrimaryKey  bool   `json:"primaryKey"`
	Indexed     bool   `json:"indexed"`
	Description string `json:"description,omitempty"`
}

type HypertableSchemaInfo struct {
	TableName          string                 `json:"tableName"`
	SchemaName         string                 `json:"schemaName"`
	TimeColumn         string                 `json:"timeColumn"`
	ChunkTimeInterval  string                 `json:"chunkTimeInterval"`
	Size               string                 `json:"size"`
	ChunkCount         int                    `json:"chunkCount"`
	RowCount           int64                  `json:"rowCount"`
	SpacePartitioning  []string               `json:"spacePartitioning,omitempty"`
	CompressionEnabled bool                   `json:"compressionEnabled"`
	CompressionConfig  CompressionConfig      `json:"compressionConfig,omitempty"`
	RetentionEnabled   bool                   `json:"retentionEnabled"`
	RetentionInterval  string                 `json:"retentionInterval,omitempty"`
	Columns            []HypertableColumnInfo `json:"columns"`
}

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

type ListDatabasesTool struct {
	BaseToolType
}

type PerformanceTool struct {
	BaseToolType
}

type QuerySuggestion struct {
	Title       string `json:"title"`
	Description string `json:"description"`
	Query       string `json:"query"`
	Category    string `json:"category"`
}

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 TimescaleDBCompletionProvider struct {
	// contains filtered or unexported fields
}

type TimescaleDBContextInfo struct {
	IsTimescaleDB bool                        `json:"isTimescaleDB"`
	Version       string                      `json:"version,omitempty"`
	Hypertables   []TimescaleDBHypertableInfo `json:"hypertables,omitempty"`
}

type TimescaleDBContextProvider struct{}

type TimescaleDBHypertableInfo struct {
	TableName     string `json:"tableName"`
	TimeColumn    string `json:"timeColumn"`
	ChunkInterval string `json:"chunkInterval"`
}

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

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)
}