domain

type ClientSession struct {
	ID        string
	UserAgent string
	Connected bool
}

type ConnectionManager interface {
	// Add adds a session to the manager.
	AddSession(session SSESession)

	// Remove removes a session from the manager.
	RemoveSession(sessionID string)

	// Get retrieves a session by ID.
	GetSession(sessionID string) (SSESession, bool)

	// Broadcast sends an event to all sessions.
	Broadcast(event interface{}) error

	// CloseAll closes all active sessions.
	CloseAll()

	// Count returns the number of active sessions.
	Count() int
}

type Error struct {
	Message string
	Code    int
}

type JSONRPCError struct {
	Code    int         `json:"code"`
	Message string      `json:"message"`
	Data    interface{} `json:"data,omitempty"`
}

type JSONRPCNotification struct {
	JSONRPC string                 `json:"jsonrpc"`
	Method  string                 `json:"method"`
	Params  map[string]interface{} `json:"params,omitempty"`
}

type JSONRPCRequest struct {
	JSONRPC string      `json:"jsonrpc"`
	ID      interface{} `json:"id"`
	Method  string      `json:"method"`
	Params  interface{} `json:"params,omitempty"`
}

type JSONRPCResponse struct {
	JSONRPC string        `json:"jsonrpc"`
	ID      interface{}   `json:"id"`
	Result  interface{}   `json:"result,omitempty"`
	Error   *JSONRPCError `json:"error,omitempty"`
}

type MessageHandler interface {
	// HandleMessage processes a raw JSON message and returns a response.
	HandleMessage(ctx context.Context, rawMessage json.RawMessage) interface{}
}

type Notification struct {
	Method string
	Params map[string]interface{}
}

type NotificationSender interface {
	// SendNotification sends a notification to a specific client.
	SendNotification(ctx context.Context, sessionID string, notification *Notification) error

	// BroadcastNotification sends a notification to all connected clients.
	BroadcastNotification(ctx context.Context, notification *Notification) error
}

type Prompt struct {
	Name        string
	Description string
	Template    string
	Parameters  []PromptParameter
}

type PromptNotFoundError struct {
	Name string
	Err  *Error
}

type PromptParameter struct {
	Name        string
	Description string
	Type        string
	Required    bool
}

type PromptRepository interface {
	// GetPrompt retrieves a prompt by its name.
	GetPrompt(ctx context.Context, name string) (*Prompt, error)

	// ListPrompts returns all available prompts.
	ListPrompts(ctx context.Context) ([]*Prompt, error)

	// AddPrompt adds a new prompt to the repository.
	AddPrompt(ctx context.Context, prompt *Prompt) error

	// DeletePrompt removes a prompt from the repository.
	DeletePrompt(ctx context.Context, name string) error
}

type PromptRequest struct {
	Name       string
	Parameters map[string]interface{}
	Session    *ClientSession
}

type PromptResult struct {
	Text  string
	Error error
}

type Resource struct {
	URI         string
	Name        string
	Description string
	MIMEType    string
}

type ResourceContents struct {
	URI      string
	MIMEType string
	Content  []byte
	Text     string
}

type ResourceNotFoundError struct {
	URI string
	Err *Error
}

type ResourceRepository interface {
	// GetResource retrieves a resource by its URI.
	GetResource(ctx context.Context, uri string) (*Resource, error)

	// ListResources returns all available resources.
	ListResources(ctx context.Context) ([]*Resource, error)

	// AddResource adds a new resource to the repository.
	AddResource(ctx context.Context, resource *Resource) error

	// DeleteResource removes a resource from the repository.
	DeleteResource(ctx context.Context, uri string) error
}

type SSEHandler interface {
	// ServeHTTP handles HTTP requests for SSE events.
	ServeHTTP(w http.ResponseWriter, r *http.Request)

	// Start starts the SSE server.
	Start(addr string) error

	// Shutdown gracefully stops the SSE server.
	Shutdown(ctx context.Context) error

	// BroadcastEvent sends an event to all connected clients.
	BroadcastEvent(event interface{}) error

	// SendEventToSession sends an event to a specific client session.
	SendEventToSession(sessionID string, event interface{}) error
}

type SSESession interface {
	// ID returns the session identifier.
	ID() string

	// Close closes the session.
	Close()

	// NotificationChannel returns the channel used to send notifications to this session.
	NotificationChannel() chan<- string

	// Start begins processing events for this session.
	Start()

	// Context returns the session's context.
	Context() context.Context
}

type SessionNotFoundError struct {
	ID  string
	Err *Error
}

type SessionRepository interface {
	// GetSession retrieves a session by its ID.
	GetSession(ctx context.Context, id string) (*ClientSession, error)

	// ListSessions returns all active sessions.
	ListSessions(ctx context.Context) ([]*ClientSession, error)

	// AddSession adds a new session to the repository.
	AddSession(ctx context.Context, session *ClientSession) error

	// DeleteSession removes a session from the repository.
	DeleteSession(ctx context.Context, id string) error
}

type Tool struct {
	Name        string
	Description string
	Parameters  []ToolParameter
}

type ToolCall struct {
	Name       string
	Parameters map[string]interface{}
	Session    *ClientSession
}

type ToolNotFoundError struct {
	Name string
	Err  *Error
}

type ToolParameter struct {
	Name        string
	Description string
	Type        string
	Required    bool
	Items       map[string]interface{}
}

type ToolRepository interface {
	// GetTool retrieves a tool by its name.
	GetTool(ctx context.Context, name string) (*Tool, error)

	// ListTools returns all available tools.
	ListTools(ctx context.Context) ([]*Tool, error)

	// AddTool adds a new tool to the repository.
	AddTool(ctx context.Context, tool *Tool) error

	// DeleteTool removes a tool from the repository.
	DeleteTool(ctx context.Context, name string) error
}

type ToolResult struct {
	Data  interface{}
	Error error
}

type ValidationError struct {
	Field   string
	Message string
	Err     *Error
}