mcp

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

type AuditSink interface {
	Log(event MCPAuditEvent) error
}

type AuthConfig struct {
	// RequireAuth enables authentication (default: true)
	RequireAuth bool
	// AllowAnonymous allows unauthenticated read-only access
	AllowAnonymous bool
	// SecurityEnabled enables/disables security (false = development mode)
	SecurityEnabled bool
	// AuditEnabled enables audit logging
	AuditEnabled bool
	// RateLimitEnabled enables rate limiting
	RateLimitEnabled bool
}

type AuthContext struct {
	UserID    string
	Username  string
	Email     string
	Roles     []MCPRole
	OrgID     string
	Workspace string
	Claims    *auth.JWTClaims // Uses existing auth package claims
	Timestamp time.Time
}

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

type CallToolRequest struct {
	Name      string                 `json:"name"`
	Arguments map[string]interface{} `json:"arguments"`
}

type CallToolResponse struct {
	Content []Content `json:"content"`
	IsError bool      `json:"isError,omitempty"`
}

type ClientInfo struct {
	Name    string `json:"name"`
	Version string `json:"version"`
}

type ConsoleSink struct{}

type Content struct {
	Type string `json:"type"` // "text" or "resource"
	Text string `json:"text,omitempty"`
}

type Dependency struct {
	From string `json:"from"`
	To   string `json:"to"`
	Type string `json:"type"`
}

type DiscoverParams struct {
	Query         string   `json:"query"`                    // Required
	Type          []string `json:"type,omitempty"`           // Optional, filter by types
	Limit         int      `json:"limit,omitempty"`          // Optional, default: 10
	MinSimilarity float64  `json:"min_similarity,omitempty"` // Optional, normalized relevance threshold (default: 0)
	Depth         int      `json:"depth,omitempty"`          // Optional, default: 1, range: 1-3
	Database      string   `json:"database,omitempty"`       // Optional, default: configured default database
}

type DiscoverResult struct {
	Results     []SearchResult `json:"results"`
	Method      string         `json:"method"` // "vector" or "keyword"
	Total       int            `json:"total"`
	Suggestions []string       `json:"suggestions,omitempty"`
}

type Edge struct {
	ID         string                 `json:"id"`
	From       string                 `json:"from"`
	To         string                 `json:"to"`
	Type       string                 `json:"type"`
	Strength   float64                `json:"strength,omitempty"`
	Properties map[string]interface{} `json:"properties,omitempty"`
	Created    time.Time              `json:"created,omitempty"`
}

type Embedder interface {
	Embed(ctx context.Context, text string) ([]float32, error)
	EmbedBatch(ctx context.Context, texts []string) ([][]float32, error)
	ChunkText(text string, maxTokens, overlap int) ([]string, error)
	Model() string
	Dimensions() int
}

type InitRequest struct {
	ProtocolVersion string                 `json:"protocolVersion"`
	Capabilities    map[string]interface{} `json:"capabilities"`
	ClientInfo      ClientInfo             `json:"clientInfo"`
}

type InitResponse struct {
	ProtocolVersion string                 `json:"protocolVersion"`
	Capabilities    map[string]interface{} `json:"capabilities"`
	ServerInfo      ServerInfo             `json:"serverInfo"`
}

type LinkParams struct {
	From     string                 `json:"from"`               // Required
	To       string                 `json:"to"`                 // Required
	Relation string                 `json:"relation"`           // Required
	Strength float64                `json:"strength,omitempty"` // Optional, default: 1.0
	Metadata map[string]interface{} `json:"metadata,omitempty"` // Optional
	Database string                 `json:"database,omitempty"` // Optional, default: configured default database
}

type LinkResult struct {
	EdgeID    string      `json:"edge_id"`
	From      Node        `json:"from"`
	To        Node        `json:"to"`
	Suggested []Edge      `json:"suggested,omitempty"`
	Receipt   interface{} `json:"receipt,omitempty"`
}

type ListToolsRequest struct{}

type ListToolsResponse struct {
	Tools []Tool `json:"tools"`
}

type MCPAuditEvent struct {
	Timestamp    time.Time              `json:"timestamp"`
	RequestID    string                 `json:"request_id"`
	UserID       string                 `json:"user_id"`
	Username     string                 `json:"username,omitempty"`
	Role         string                 `json:"role"`
	OrgID        string                 `json:"org_id,omitempty"`
	Tool         string                 `json:"tool"`
	Operation    string                 `json:"operation"`
	ResourceType string                 `json:"resource_type,omitempty"`
	ResourceID   string                 `json:"resource_id,omitempty"`
	IPAddress    string                 `json:"ip_address,omitempty"`
	UserAgent    string                 `json:"user_agent,omitempty"`
	Success      bool                   `json:"success"`
	ErrorMessage string                 `json:"error_message,omitempty"`
	Duration     time.Duration          `json:"duration"`
	Metadata     map[string]interface{} `json:"metadata,omitempty"`
}

type MCPPermission string

type MCPRole string

type Node struct {
	ID         string                 `json:"id"`
	Type       string                 `json:"type"`
	Title      string                 `json:"title,omitempty"`
	Content    string                 `json:"content,omitempty"`
	Tags       []string               `json:"tags,omitempty"`
	Properties map[string]interface{} `json:"properties,omitempty"`
	Created    time.Time              `json:"created,omitempty"`
	Updated    time.Time              `json:"updated,omitempty"`
}

type RateLimit struct {
	RequestsPerMinute int
	RequestsPerHour   int
	BurstSize         int
}

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

type RecallParams struct {
	ID       string    `json:"id,omitempty"`       // Optional, if provided ignores other filters
	Type     []string  `json:"type,omitempty"`     // Optional, filter by types
	Tags     []string  `json:"tags,omitempty"`     // Optional, filter by tags
	Since    time.Time `json:"since,omitempty"`    // Optional, filter by creation time
	Limit    int       `json:"limit,omitempty"`    // Optional, default: 10
	Database string    `json:"database,omitempty"` // Optional, default: configured default database
}

type RecallResult struct {
	Nodes   []Node `json:"nodes"`
	Count   int    `json:"count"`
	Related []Node `json:"related,omitempty"`
}

type RelatedNode struct {
	ID           string   `json:"id"`
	Type         string   `json:"type"`
	Title        string   `json:"title,omitempty"`
	Distance     int      `json:"distance"`            // Hops from the source node (1 = direct, 2 = two hops, etc.)
	Relationship string   `json:"relationship"`        // Relationship type that connects to this node
	Direction    string   `json:"direction,omitempty"` // "outgoing", "incoming", or "both"
	Path         []string `json:"path,omitempty"`      // Node IDs in the path (for depth > 1)
}

type SearchResult struct {
	ID             string                 `json:"id"`
	Type           string                 `json:"type"`
	Title          string                 `json:"title"`
	ContentPreview string                 `json:"content_preview,omitempty"`
	Similarity     float64                `json:"similarity"`
	Properties     map[string]interface{} `json:"properties,omitempty"`
	// Related nodes discovered via graph traversal (only populated when depth > 1)
	Related []RelatedNode `json:"related,omitempty"`
}

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

type ServerConfig struct {
	// Address to bind to (default: "localhost")
	Address string `yaml:"address"`
	// Port to listen on (default: 7474, same as NornicDB HTTP API)
	Port int `yaml:"port"`
	// ReadTimeout for requests
	ReadTimeout time.Duration `yaml:"read_timeout"`
	// WriteTimeout for responses
	WriteTimeout time.Duration `yaml:"write_timeout"`
	// MaxRequestSize in bytes (default: 10MB)
	MaxRequestSize int64 `yaml:"max_request_size"`
	// EnableCORS for cross-origin requests
	EnableCORS bool `yaml:"enable_cors"`
	// EmbeddingEnabled controls whether embeddings are generated
	EmbeddingEnabled bool `yaml:"embedding_enabled"`
	// EmbeddingModel is the model name (for error messages)
	EmbeddingModel string `yaml:"embedding_model"`
	// EmbeddingDimensions is the expected vector dimensions (for validation)
	EmbeddingDimensions int `yaml:"embedding_dimensions"`
	// Embedder is the embedding service (set externally if needed)
	Embedder Embedder `yaml:"-"`

	// DatabaseScopedExecutor returns an executor and node getter for the given database name.
	// When set, MCP tool calls that include a database in context (e.g. from the agentic loop)
	// use this to run store/recall/link/task against the request's database instead of the default.
	// If nil or the context has no database, the server uses its single db.
	DatabaseScopedExecutor func(dbName string) (exec *cypher.StorageExecutor, getNode func(context.Context, string) (*nornicdb.Node, error), err error)

	// DatabaseScopedStorage returns a storage engine scoped to dbName.
	// When set, tools that need direct storage access (e.g. discover/search) can operate
	// on the request's database without relying on a single default DB instance.
	DatabaseScopedStorage func(dbName string) (storage.Engine, error)

	// DefaultNodeLabel is the label used when the store tool is called without
	// explicit labels or type. Defaults to "Memory" for backward compatibility.
	// Configured via NORNICDB_DEFAULT_NODE_LABEL env var or config.Memory.DefaultNodeLabel.
	DefaultNodeLabel string
}

type ServerInfo struct {
	Name    string `json:"name"`
	Version string `json:"version"`
}

type StoreParams struct {
	Content  string                 `json:"content"`            // Required
	Type     string                 `json:"type,omitempty"`     // Optional, default: "memory"
	Title    string                 `json:"title,omitempty"`    // Optional, auto-generated if empty
	Tags     []string               `json:"tags,omitempty"`     // Optional
	Metadata map[string]interface{} `json:"metadata,omitempty"` // Optional, will be flattened
}

type StoreResult struct {
	ID          string       `json:"id"`
	Title       string       `json:"title"`
	Embedded    bool         `json:"embedded"`
	Suggestions []Suggestion `json:"suggestions,omitempty"`
	Receipt     interface{}  `json:"receipt,omitempty"`
}

type Suggestion struct {
	ID         string  `json:"id"`
	Title      string  `json:"title"`
	Type       string  `json:"type"`
	Similarity float64 `json:"similarity"`
}

type TaskParams struct {
	ID          string   `json:"id,omitempty"`          // Optional, for update/complete
	Title       string   `json:"title,omitempty"`       // Required for create
	Description string   `json:"description,omitempty"` // Optional
	Status      string   `json:"status,omitempty"`      // Optional: pending|active|completed|blocked
	Priority    string   `json:"priority,omitempty"`    // Optional: low|medium|high|critical
	DependsOn   []string `json:"depends_on,omitempty"`  // Optional, task IDs
	Assign      string   `json:"assign,omitempty"`      // Optional, agent/person
	Database    string   `json:"database,omitempty"`    // Optional, default: configured default database
}

type TaskResult struct {
	Task       Node        `json:"task"`
	Blockers   []Node      `json:"blockers,omitempty"`
	Subtasks   []Node      `json:"subtasks,omitempty"`
	NextAction string      `json:"next_action,omitempty"`
	Receipt    interface{} `json:"receipt,omitempty"`
}

type TaskRow struct {
	ID          string `cypher:"id" json:"id"`
	Title       string `cypher:"title" json:"title"`
	Description string `cypher:"description" json:"description"`
	Status      string `cypher:"status" json:"status"`
	Priority    string `cypher:"priority" json:"priority"`
	AssignedTo  string `cypher:"assigned_to" json:"assigned_to"`
}

type TaskStatRow struct {
	Status   string `cypher:"status" json:"status"`
	Priority string `cypher:"priority" json:"priority"`
	Count    int64  `cypher:"count" json:"count"`
}

type TaskStats struct {
	Total      int            `json:"total"`
	ByStatus   map[string]int `json:"by_status"`
	ByPriority map[string]int `json:"by_priority"`
}

type TasksParams struct {
	Status        []string `json:"status,omitempty"`         // Optional, filter by status
	Priority      []string `json:"priority,omitempty"`       // Optional, filter by priority
	AssignedTo    string   `json:"assigned_to,omitempty"`    // Optional, filter by assignee
	UnblockedOnly bool     `json:"unblocked_only,omitempty"` // Optional, default: false
	Limit         int      `json:"limit,omitempty"`          // Optional, default: 20
	Database      string   `json:"database,omitempty"`       // Optional, default: configured default database
}

type TasksResult struct {
	Tasks           []Node       `json:"tasks"`
	Stats           TaskStats    `json:"stats"`
	DependencyGraph []Dependency `json:"dependency_graph,omitempty"`
	Recommended     []Node       `json:"recommended,omitempty"`
}

type Tool struct {
	Name        string          `json:"name"`
	Description string          `json:"description"`
	InputSchema json.RawMessage `json:"inputSchema"`
}

type ToolHandler func(ctx context.Context, args map[string]interface{}) (interface{}, error)