core

type ActivityState int

type AdminConfig struct {
	// Enabled controls whether admin endpoints are active.
	// When false, all /admin/* routes return 404.
	Enabled bool `yaml:"enabled"`

	// User is the admin username for /admin/login authentication.
	User string `yaml:"user"`

	// Password is the admin password for /admin/login authentication.
	// WARNING: Change the default before deploying to production.
	Password string `yaml:"password"`
}

type BrainState struct {
	IndexID        IndexID       `msgpack:"index_id"`
	State          ActivityState `msgpack:"state"`
	LastInvoke     time.Time     `msgpack:"last_invoke"`
	InvokeCount    uint64        `msgpack:"invoke_count"`
	SessionStart   time.Time     `msgpack:"session_start"`
	IdleThreshold  time.Duration `msgpack:"idle_threshold"`
	SleepThreshold time.Duration `msgpack:"sleep_threshold"`
}

type CLIOverrides struct {
	ConfigPath             *string
	HTTPAddr               *string
	DataPath               *string
	Compress               *bool
	MinDimension           *int
	MaxDimension           *int
	MaxNeurons             *int
	IdleThreshold          *time.Duration
	SleepThreshold         *time.Duration
	DormantThreshold       *time.Duration
	DecayInterval          *time.Duration
	ConsolidateInt         *time.Duration
	PruneInterval          *time.Duration
	PersistInterval        *time.Duration
	ReorgInterval          *time.Duration
	MaxIdleTime            *time.Duration
	RegistryEnabled        *bool
	VectorEnabled          *bool
	VectorModelPath        *string
	VectorGPULayers        *int
	VectorAlpha            *float64
	VectorQueryRepeat      *int
	VectorEmbedContextSize *uint32
	AdminEnabled           *bool
	AdminUser              *string
	AdminPassword          *string
	AllowedOrigins         *string
	MaxRequestBody         *int64
	MaxNeuronContentBytes  *int64
	TLSCert                *string
	TLSKey                 *string
}

type Config struct {
	Server    ServerConfig    `yaml:"server"`
	Storage   StorageConfig   `yaml:"storage"`
	Matrix    MatrixConfig    `yaml:"matrix"`
	Lifecycle LifecycleConfig `yaml:"lifecycle"`
	Daemons   DaemonConfig    `yaml:"daemons"`
	Worker    WorkerConfig    `yaml:"worker"`
	Registry  RegistryConfig  `yaml:"registry"`
	Vector    VectorConfig    `yaml:"vector"`
	Admin     AdminConfig     `yaml:"admin"`
	MCP       MCPConfig       `yaml:"mcp"`
	Security  SecurityConfig  `yaml:"security"`
}

type ConnInfo struct {
	// Scheme is the protocol scheme ("qubicdb" or "qubicdb+tls").
	Scheme string

	// User is the authentication username (empty if not provided).
	User string

	// Password is the authentication password (empty if not provided).
	Password string

	// Hosts is a list of host:port pairs. At least one is always present.
	Hosts []string

	// IndexID is the optional default index (path segment after the first slash).
	IndexID string

	// TLS is true when the scheme is "qubicdb+tls".
	TLS bool
}

type DaemonConfig struct {
	// DecayInterval controls how often the decay daemon runs.
	// Decay reduces energy of unused neurons over time.
	DecayInterval time.Duration `yaml:"decayInterval"`

	// ConsolidateInterval controls how often the consolidation daemon runs.
	// Consolidation moves important neurons to deeper, permanent layers.
	ConsolidateInterval time.Duration `yaml:"consolidateInterval"`

	// PruneInterval controls how often the pruning daemon runs.
	// Pruning removes dead neurons and weak synapses.
	PruneInterval time.Duration `yaml:"pruneInterval"`

	// PersistInterval controls how often in-memory state is flushed to disk.
	PersistInterval time.Duration `yaml:"persistInterval"`

	// ReorgInterval controls how often the matrix reorganisation daemon runs.
	// Reorg optimises spatial locality for frequently co-accessed neurons.
	ReorgInterval time.Duration `yaml:"reorgInterval"`
}

type IndexID string

type LifecycleConfig struct {
	// IdleThreshold is how long a brain must be inactive before
	// transitioning from Active → Idle.
	IdleThreshold time.Duration `yaml:"idleThreshold"`

	// SleepThreshold is how long a brain must be idle before
	// transitioning from Idle → Sleeping (consolidation begins).
	SleepThreshold time.Duration `yaml:"sleepThreshold"`

	// DormantThreshold is how long a brain must be sleeping before
	// transitioning from Sleeping → Dormant (eligible for eviction).
	DormantThreshold time.Duration `yaml:"dormantThreshold"`
}

type MCPConfig struct {
	// Enabled controls whether /mcp endpoint is exposed.
	Enabled bool `yaml:"enabled"`

	// Path is the HTTP route for MCP transport.
	Path string `yaml:"path"`

	// APIKey is optional shared secret validated from X-API-Key or Bearer token.
	APIKey string `yaml:"apiKey"`

	// Stateless enables stateless session-id handling for streamable HTTP.
	Stateless bool `yaml:"stateless"`

	// RateLimitRPS controls per-client rate limiting in requests/second.
	// Set to 0 to disable MCP-specific rate limiting.
	RateLimitRPS float64 `yaml:"rateLimitRPS"`

	// RateLimitBurst controls burst capacity for MCP-specific rate limiting.
	RateLimitBurst int `yaml:"rateLimitBurst"`

	// EnablePrompts toggles MCP prompt registration.
	EnablePrompts bool `yaml:"enablePrompts"`

	// AllowedTools is an optional allowlist; empty means all built-in MCP tools.
	AllowedTools []string `yaml:"allowedTools"`
}

type Matrix struct {
	IndexID    IndexID      `msgpack:"index_id"`
	Bounds     MatrixBounds `msgpack:"bounds"`
	CurrentDim int          `msgpack:"current_dim"`

	// Neurons indexed by ID
	Neurons map[NeuronID]*Neuron `msgpack:"neurons"`

	// Synapses indexed by ID
	Synapses map[SynapseID]*Synapse `msgpack:"synapses"`

	// Adjacency list for fast traversal (neuron -> connected neurons)
	Adjacency map[NeuronID][]NeuronID `msgpack:"adjacency"`

	// Learned parameters (self-tuning)
	DecayRate       float64       `msgpack:"decay_rate"`
	LinkThreshold   float64       `msgpack:"link_threshold"`
	ConsolFrequency time.Duration `msgpack:"consol_freq"`

	// Statistics
	TotalActivations  uint64    `msgpack:"total_activations"`
	LastActivity      time.Time `msgpack:"last_activity"`
	LastConsolidation time.Time `msgpack:"last_consolidation"`

	// Version for persistence
	Version    uint64    `msgpack:"version"`
	CreatedAt  time.Time `msgpack:"created_at"`
	ModifiedAt time.Time `msgpack:"modified_at"`
	// contains filtered or unexported fields
}

type MatrixBounds struct {
	MinDimension int `msgpack:"min_dim"`
	MaxDimension int `msgpack:"max_dim"`
	MinNeurons   int `msgpack:"min_neurons"`
	MaxNeurons   int `msgpack:"max_neurons"`
}

type MatrixConfig struct {
	// MinDimension is the initial dimensionality of a new brain matrix.
	MinDimension int `yaml:"minDimension"`

	// MaxDimension is the upper limit the matrix can grow to.
	MaxDimension int `yaml:"maxDimension"`

	// MaxNeurons is the hard cap on the number of neurons per brain.
	MaxNeurons int `yaml:"maxNeurons"`
}

type Neuron struct {
	ID          NeuronID `msgpack:"id"`
	Content     string   `msgpack:"content"`
	ContentHash string   `msgpack:"content_hash"`

	// Spatial position in N-dimensional matrix (organic, grows/shrinks)
	Position []float64 `msgpack:"position"`

	// Activation dynamics
	Energy     float64 `msgpack:"energy"`      // Current activation level (0.0 - 1.0)
	BaseEnergy float64 `msgpack:"base_energy"` // Resting energy level

	// Depth in the memory hierarchy (0 = surface/hot, higher = deeper/consolidated)
	Depth int `msgpack:"depth"`

	// Temporal markers
	CreatedAt   time.Time `msgpack:"created_at"`
	LastFiredAt time.Time `msgpack:"last_fired_at"`
	LastDecayAt time.Time `msgpack:"last_decay_at"`
	AccessCount uint64    `msgpack:"access_count"`

	// Classification tags (emergent, not predefined)
	Tags []string `msgpack:"tags"`

	// Sentiment classification (set on creation, updated on content change)
	SentimentLabel string  `msgpack:"sentiment_label,omitempty"` // happiness|sadness|fear|anger|disgust|surprise|neutral
	SentimentScore float64 `msgpack:"sentiment_score,omitempty"` // VADER compound [-1, 1]

	// Vector embedding for semantic search (set once on creation, nil if vector layer disabled)
	Embedding []float32 `msgpack:"embedding,omitempty"`

	// Metadata
	Metadata map[string]any `msgpack:"metadata"`
	// contains filtered or unexported fields
}

type NeuronID string

type RegistryConfig struct {
	// Enabled controls whether the UUID registry guard is active.
	// When true, only registered UUIDs may access brain operations.
	Enabled bool `yaml:"enabled"`
}

type SecurityConfig struct {
	// AllowedOrigins controls the CORS Access-Control-Allow-Origin header.
	// Use "*" to allow all origins (development only) or a comma-separated
	// list of allowed origins for production.
	AllowedOrigins string `yaml:"allowedOrigins"`

	// MaxRequestBody is the maximum allowed HTTP request body size in bytes.
	// Requests exceeding this limit are rejected with 413 Payload Too Large.
	// Default: 1048576 (1 MB). Set to 0 to disable the limit (not recommended).
	MaxRequestBody int64 `yaml:"maxRequestBody"`

	// MaxNeuronContentBytes is the maximum allowed neuron content payload size in bytes.
	// Requests that try to write larger neuron content are rejected.
	// Default: 65536 (64 KB).
	MaxNeuronContentBytes int64 `yaml:"maxNeuronContentBytes"`

	// TLSCert is the path to a TLS certificate file for HTTPS.
	// Leave empty to disable TLS (plain HTTP). Requires TLSKey.
	TLSCert string `yaml:"tlsCert"`

	// TLSKey is the path to the TLS private key file.
	// Leave empty to disable TLS. Requires TLSCert.
	TLSKey string `yaml:"tlsKey"`

	// ReadTimeout is the maximum duration for reading the entire request.
	ReadTimeout time.Duration `yaml:"readTimeout"`

	// WriteTimeout is the maximum duration before timing out writes of the response.
	WriteTimeout time.Duration `yaml:"writeTimeout"`
}

type ServerConfig struct {
	// HTTPAddr is the TCP address the HTTP/REST API binds to.
	HTTPAddr string `yaml:"httpAddr"`
}

type StorageConfig struct {
	// DataPath is the directory where .nrdb brain files are stored.
	DataPath string `yaml:"dataPath"`

	// Compress enables msgpack-level compression for persistence.
	Compress bool `yaml:"compress"`

	// WALEnabled controls write-ahead logging for crash recovery.
	WALEnabled bool `yaml:"walEnabled"`

	// FsyncPolicy controls persistence fsync behavior: always | interval | off.
	FsyncPolicy string `yaml:"fsyncPolicy"`

	// FsyncInterval controls fsync cadence when fsyncPolicy is interval.
	FsyncInterval time.Duration `yaml:"fsyncInterval"`

	// ChecksumValidationInterval controls periodic on-disk .nrdb checksum scans.
	// 0 disables periodic background validation.
	ChecksumValidationInterval time.Duration `yaml:"checksumValidationInterval"`

	// StartupRepair enables startup integrity repair for corrupt/missing persisted data files.
	StartupRepair bool `yaml:"startupRepair"`
}

type Synapse struct {
	ID     SynapseID `msgpack:"id"`
	FromID NeuronID  `msgpack:"from_id"`
	ToID   NeuronID  `msgpack:"to_id"`

	// Connection strength (0.0 - 1.0)
	Weight float64 `msgpack:"weight"`

	// Hebbian learning metrics
	CoFireCount uint64    `msgpack:"co_fire_count"` // Times fired together
	LastCoFire  time.Time `msgpack:"last_co_fire"`

	// Bidirectional flag
	Bidirectional bool `msgpack:"bidirectional"`

	CreatedAt time.Time `msgpack:"created_at"`
	// contains filtered or unexported fields
}

type SynapseID string

type VectorConfig struct {
	// Enabled activates the vector embedding layer.
	// When false, search falls back to pure lexical matching.
	Enabled bool `yaml:"enabled"`

	// ModelPath is the path to a GGUF BERT embedding model file.
	ModelPath string `yaml:"modelPath"`

	// GPULayers controls how many model layers are offloaded to GPU.
	// 0 = CPU only.
	GPULayers int `yaml:"gpuLayers"`

	// Alpha is the weight of the vector score in hybrid search.
	// finalScore = Alpha*vectorScore + (1-Alpha)*stringScore
	// Range: 0.0 (pure string) to 1.0 (pure vector). Default: 0.6
	Alpha float64 `yaml:"alpha"`

	// QueryRepeat controls how many times the query is repeated before embedding.
	// Repeating the query improves embedding quality for short queries by allowing
	// the model to attend to all query tokens bidirectionally (Springer et al. 2024).
	// 1 = no repeat (baseline), 2 = repeat once (recommended), 3 = repeat twice.
	// Default: 2
	QueryRepeat int `yaml:"queryRepeat"`

	// EmbedContextSize is the llama.cpp context window size used for embedding.
	// Must be >= 512 for MiniLM. Increase if QueryRepeat×queryTokens > 512.
	// Default: 512
	EmbedContextSize uint32 `yaml:"embedContextSize"`
}

type WorkerConfig struct {
	// MaxIdleTime is the maximum duration a brain worker may remain idle
	// before being evicted from the in-memory pool.
	MaxIdleTime time.Duration `yaml:"maxIdleTime"`
}