sessiontools

type CommandACLDecision struct {
	Action    string   `json:"action"`
	ACLID     string   `json:"acl_id,omitempty"`
	ItemID    string   `json:"item_id,omitempty"`
	Name      string   `json:"name,omitempty"`
	Matched   string   `json:"matched,omitempty"`
	DetailURL string   `json:"detail_url,omitempty"`
	Reviewers []string `json:"reviewers,omitempty"`
	Processor string   `json:"processor,omitempty"`
	Reviewed  bool     `json:"reviewed,omitempty"`
}

type CommandConstraints struct {
	BackgroundEligible  bool
	MaxExecutionSeconds int
}

type CommandExecutor interface {
	Execute(context.Context, string, func(string)) (string, *int, error)
	Close() error
}

type CommandValidator func(string) (CommandConstraints, error)

type ContextSnapshot struct {
	SessionKind      string `json:"session_kind,omitempty"`
	InteractionMode  string `json:"interaction_mode,omitempty"`
	CommandLanguage  string `json:"command_language,omitempty"`
	Dialect          string `json:"dialect,omitempty"`
	Protocol         string `json:"protocol"`
	ConnectionMethod string `json:"connection_method,omitempty"`
	AssetID          string `json:"asset_id,omitempty"`
	AssetName        string `json:"asset_name,omitempty"`
	AssetAddress     string `json:"asset_address,omitempty"`
	PlatformID       int    `json:"platform_id,omitempty"`
	PlatformCategory string `json:"platform_category,omitempty"`
	PlatformType     string `json:"platform_type,omitempty"`
	PlatformName     string `json:"platform_name,omitempty"`
	BaseOS           string `json:"base_os,omitempty"`
	Charset          string `json:"charset,omitempty"`
	Database         string `json:"database,omitempty"`
	Schema           string `json:"schema,omitempty"`
}

type DatabaseConfig struct {
	Protocol         string
	Host             string
	Port             int
	ServerName       string
	Username         string
	Password         string
	Database         string
	UseSSL           bool
	PGSSLMode        string
	CACert           string
	ClientCert       string
	ClientKey        string
	AllowInvalidCert bool
	Encrypt          bool
	DisableEncrypt   bool
	ClusterMode      bool
	AuthSource       string
	ConnectionOpts   string
	ProxyURL         string
	DataMaskingRules []model.DataMaskingRule
}

type DirectoryResult struct {
	Path      string      `json:"path"`
	Entries   []FileEntry `json:"entries"`
	Truncated bool        `json:"truncated,omitempty"`
}

type ExecutionUnavailableError struct {
	Cause error
}

type FileEntry struct {
	Name    string `json:"name"`
	Path    string `json:"path"`
	Exists  bool   `json:"exists"`
	Size    int64  `json:"size"`
	Perm    string `json:"perm"`
	ModTime int64  `json:"modTime"`
	IsDir   bool   `json:"isDir"`
	Version string `json:"version"`
}

type FileExecutor interface {
	ListDirectory(context.Context, string, int) (DirectoryResult, error)
	Stat(context.Context, string) (FileEntry, error)
	ReadText(context.Context, string, int64) (TextResult, error)
	SaveText(context.Context, string, string, string) (FileEntry, error)
	Mkdir(context.Context, string) error
	Rename(context.Context, string, string, string) error
	Delete(context.Context, string, string, bool) error
}

type FileToolCapabilities struct {
	ReadText bool
	SaveText bool
	Mkdir    bool
	Rename   bool
	Delete   bool
}

type MCPCallToolResult struct {
	Content           []MCPTextContent `json:"content"`
	StructuredContent any              `json:"structuredContent,omitempty"`
	IsError           bool             `json:"isError,omitempty"`
	Meta              map[string]any   `json:"_meta"`
}

type MCPCommandHooks struct {
	CommandACLCheck  func(string) CommandACLDecision
	CommandACLReview func(
		context.Context,
		CommandACLDecision,
		string,
	) (CommandACLDecision, error)
	BackgroundRecord    func(string, string, *int, *CommandACLDecision)
	ExecutionGuard      func() error
	BackgroundGuard     func() error
	BackgroundAvailable func() bool
	PTYExecute          func(
		context.Context,
		string,
		*CommandACLDecision,
	) (string, *int, error)
}

type MCPCommandResult struct {
	Command         string              `json:"command"`
	Execution       string              `json:"execution"`
	Output          string              `json:"output"`
	OutputTruncated bool                `json:"output_truncated,omitempty"`
	ExitCode        *int                `json:"exit_code,omitempty"`
	CommandACL      *CommandACLDecision `json:"command_acl,omitempty"`
}

type MCPCommandToolOptions struct {
	Executor CommandExecutor
	Validate CommandValidator
	Protocol string
	Hooks    MCPCommandHooks
}

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

type MCPDispatcherOptions struct {
	ResourceSessionID string
	Profile           string
	Context           ContextSnapshot
	Handlers          []MCPToolHandler
	Emit              func(MCPOutbound)
}

type MCPManifest struct {
	Version           int                 `json:"version"`
	ResourceSessionID string              `json:"resource_session_id"`
	Profile           string              `json:"profile"`
	Context           ContextSnapshot     `json:"context,omitempty"`
	Revision          int                 `json:"revision"`
	Tools             []MCPToolDefinition `json:"tools"`
}

type MCPOutbound struct {
	Type string
	Data []byte
}

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

type MCPResponse struct {
	JSONRPC string       `json:"jsonrpc"`
	ID      string       `json:"id"`
	Result  any          `json:"result,omitempty"`
	Error   *MCPRPCError `json:"error,omitempty"`
}

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

type MCPToolDefinition struct {
	Name         string         `json:"name"`
	Title        string         `json:"title,omitempty"`
	Description  string         `json:"description,omitempty"`
	Icons        []ToolIcon     `json:"icons,omitempty"`
	InputSchema  map[string]any `json:"inputSchema"`
	OutputSchema map[string]any `json:"outputSchema,omitempty"`
	Annotations  map[string]any `json:"annotations,omitempty"`
	Meta         map[string]any `json:"_meta,omitempty"`
}

type MCPToolHandler interface {
	Definition() MCPToolDefinition
	Call(context.Context, json.RawMessage) (any, error)
}

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

type MySQLConfig = DatabaseConfig

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

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

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

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

type SQLMetadataProvider interface {
	SQLMetadataScope() string
	LookupSQLSchema(context.Context, SQLSchemaLookupRequest) (SQLSchemaLookupResult, error)
	InvalidateSQLMetadata()
}

type SQLSchemaColumn struct {
	Name     string  `json:"name"`
	Type     string  `json:"type"`
	Nullable bool    `json:"nullable"`
	Default  *string `json:"default"`
	Ordinal  int     `json:"-"`
}

type SQLSchemaLookupRequest struct {
	Tables []string `json:"tables"`
	Query  string   `json:"query"`
}

type SQLSchemaLookupResult struct {
	Database  string           `json:"database"`
	Matches   []string         `json:"matches,omitempty"`
	Tables    []SQLTableSchema `json:"tables"`
	Truncated bool             `json:"truncated,omitempty"`
}

type SQLTableSchema struct {
	Database string            `json:"database"`
	Schema   string            `json:"schema,omitempty"`
	Table    string            `json:"table"`
	Columns  []SQLSchemaColumn `json:"columns"`
}

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

type TerminalCommandResult struct {
	Command string
	Output  string
}

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

type TextResult struct {
	Path      string `json:"path"`
	Exists    bool   `json:"exists"`
	Content   string `json:"content"`
	Version   string `json:"version"`
	Truncated bool   `json:"truncated,omitempty"`
}

type ToolIcon struct {
	Source   string   `json:"src"`
	MIMEType string   `json:"mimeType,omitempty"`
	Sizes    []string `json:"sizes,omitempty"`
	Theme    string   `json:"theme,omitempty"`
}

type ValueSanitizer interface {
	Sanitize(column, value string) string
}