shared

type A struct {
	// ID is a general string ID param.
	ID string `log:"id"`
	// Addr is a network address.
	Addr string `log:"addr"`
	// Timeout is a generic timeout.
	Timeout time.Duration `log:"timeout"`
	// Prompt is a prompt to be sent to LLM.
	Prompt string `log:"prompt"`
	// IntByTimeout means the interruption was caused by timeout.
	IntByTimeout bool `log:"int_by_timeout"`
	// Msg is a single message with an author and text.
	Msg *Msg `log:"msg"`
	// Perform additional checks via LLM
	CheckLLM bool `log:"check_llm"`
	// List of choices
	Choices []string
	// Actions are a list of buttons to be displayed in the UI.
	Actions    []ActionInfo `log:"actions"`
	Stories    []StoryInfo  `log:"stories"`
	StatesList []string     `log:"states_list"`
	// ActivateList is a list of booleans for StatesList, indicating an active state at the given index.
	ActivateList []bool `log:"activate_list"`
	ConfigAI     *ConfigAI
	ClockDiff    [][]int

	// Result is a buffered channel to be closed by the receiver
	ResultCh chan<- am.Result
	// DBQuery is a function that executes a query on the database.
	DBQuery func(ctx context.Context) error
	// RetStr returns dereferenced user prompts based on the list of offer.
	RetOfferRef chan<- *OfferRef
	SSHServer   *ssh.Server
	SSHSess     ssh.Session
	// Done is a buffered channel to be closed by the receiver
	Done chan<- struct{}
}

type ARPC struct {
	// ID is a general string ID param.
	ID string `log:"id"`
	// Addr is a network address.
	Addr string `log:"addr"`
	// Timeout is a generic timeout.
	Timeout time.Duration `log:"timeout"`
	// Prompt is a prompt to be sent to LLM.
	Prompt string `log:"prompt"`
	// IntByTimeout means the interruption was caused by timeout.
	IntByTimeout bool `log:"int_by_timeout"`
	// Msg is a single message with an author and text.
	Msg *Msg `log:"msg"`
	// Perform additional checks via LLM
	CheckLLM bool `log:"check_llm"`
	// List of choices
	Choices []string
	// Actions are a list of buttons to be displayed in the UI.
	Actions      []ActionInfo `log:"actions"`
	Stories      []StoryInfo  `log:"stories"`
	StatesList   []string     `log:"states_list"`
	ActivateList []bool       `log:"activate_list"`
	Result       am.Result
	ConfigAI     *ConfigAI
	ClockDiff    [][]int
}

type Action struct {
	// random ID
	ID string

	// Current value.
	Value func() int
	// Maximum value.
	ValueEnd func() int

	Label  string
	Desc   string
	Action func()

	StateAdd    string
	StateRemove string

	VisibleAgent amhelp.Cond
	VisibleMem   amhelp.Cond
	IsDisabled   func() bool
	LabelEnd     string

	Pos         int
	PosInferred bool
}

type ActionInfo struct {
	// random ID
	ID string

	// Current value.
	Value int
	// Maximum value.
	ValueEnd int

	Label  string
	Desc   string
	Action bool

	StateAdd    string
	StateRemove string

	VisibleAgent bool
	VisibleMem   bool
	IsDisabled   bool
	LabelEnd     string

	Pos         int
	PosInferred bool
}

type AgentAPI interface {
	AgentBaseAPI

	Msgs() []*Msg
	Splash() string
	MachSchema() (am.Schema, am.S)
	Actions() []ActionInfo
	Stories() []StoryInfo
	Story(state string) *Story
	DBAgent() *sql.DB
	MachMem() *am.Machine
	// Moves return a list of moves and their descriptions for orienting.
	OrientingMoves() map[string]string
	// HistoryStates returns a list of states to track in the history.
	HistoryStates() am.S
}

type AgentBaseAPI interface {
	// AgentImpl returns the top-level agent implementation.
	AgentImpl() AgentAPI
	// Output outputs text to the user.
	Output(txt string, from From) am.Result

	Mach() *am.Machine
	DBG() *debugger.Debugger
	Hist() (history.MemoryApi, error)

	ConfigBase() *Config
	OpenAI() []*OpenAIClient
	Gemini() []*GeminiClient

	Start() am.Result
	Stop(disposeCtx context.Context) am.Result
	Log(msg string, args ...any)
	LogErr(msg string, err error, args ...any)
	Logger() *slog.Logger
	Store() *AgentStore

	QueriesBase() *sqlc.Queries

	DBBase() *sql.DB
	DBHistory() *sql.DB
}

type AgentInit interface {
	Init(
		agentImpl AgentAPI, cfg *Config, logArgs am.LogArgsMapperFn, groups any, states am.States, args any,
	) error
}

type AgentQueries[TQueries any] interface {
	Queries() *TQueries
}

type AgentStore struct {
	M         map[string]any
	ClockDiff [][]int
	Web       fs.FS
}

type Config struct {
	// File is the path to the loaded config file.
	File  string `kdl:"-"`
	AI    ConfigAI
	Agent ConfigAgent
	Web   ConfigWeb
	TUI   ConfigTUI
	Tools ConfigTools
	Debug ConfigDebug
}

type ConfigAI struct {
	OpenAI []ConfigAIOpenAI `kdl:"OpenAI,multiple"`
	Gemini []ConfigAIGemini `kdl:"Gemini,multiple"`
	// Max LLM requests per session.
	ReqLimit int
}

type ConfigAIGemini struct {
	Key      string
	Disabled bool
	Model    string
	Tags     []string
	Retries  int
}

type ConfigAIOpenAI struct {
	Key      string
	Disabled bool
	URL      string
	Model    string
	Tags     []string
	Retries  int `kdl:",omitempty"`
}

type ConfigAgent struct {
	// bot ID
	ID    string
	Label string
	// // dir for tmp files, defaults to CWD
	Dir       string
	Intro     string
	IntroDash string
	Footer    string
	Log       ConfigAgentLog
	History   ConfigAgentHistory
}

type ConfigAgentHistory struct {
	Backend string
	// TODO BackendParsed enum
	Max int
}

type ConfigAgentLog struct {
	// path to the log file
	File string
	// log prompts
	Prompts bool
	// duplicate log to state-machine log
	MachFwd bool
	// mach log level 0-5
	MachLevel am.LogLevel
	// print machine log
	MachPrint bool
}

type ConfigDebug struct {
	// Display extra info about these stories in the machine log
	Story []string
	// Run the mock scenario
	Mock bool
	// Start pprof on addr
	ProfilerAddr string
	// Enable misc debugging modes (SQL history, am-relay, browser RPC)
	// TODO extract SQL
	Verbose bool
	// Create value files for inspection
	ValFiles bool
	// Enable REPL for agent, mem, and tools
	REPL bool
	// Connect and send dbg info to am-dbg
	DBGAddr string

	// Start an embedded debugger on localhost:{DBGEmbed}
	DBGEmbed bool
	// Expose the embedded debugger on http://localhost:{DBGEmbedWeb}
	DBGEmbedWeb int
	// Start a web REPL on http://localhost:{REPLWeb}
	REPLWeb int
}

type ConfigSearXNG struct {
	// Port to start a local instance on
	Port string
	// URL of an existing instance (disables Port).
	URL string
}

type ConfigTUI struct {
	// TODO Addr
	// TODO WebAddr
	// SSH host
	Host string
	// SSH port
	PortSSH int
	// Web host
	WebHost string
	// Web port
	PortWeb int
	// Number of transitions to show on the clock
	ClockRange int
}

type ConfigTools struct {
	SearXNG ConfigSearXNG
}

type ConfigWeb struct {
	// Base address for HTTP services:
	// - main HTTP server
	// - +1 WebSocket addr of the agent's RPC server (dashboard)
	// - +2 WebSocket addr of the agent's RPC server (agent UI)
	// - +3 TCP addr of the dashboard REPL server (WS tunnel)
	// - +4 TCP addr of the agent UI REPL server (WS tunnel)
	Addr string
	// Start a DBPort web UI on http://localhost:{DBPort[0-2]}
	DBPort int
	// Start a log web UI on http://localhost:{LogPort}
	LogPort int
}

type From enum.Member[string]

type GeminiClient struct {
	Cfg *ConfigAIGemini
	C   *instructor.InstructorGoogle
}

type Msg struct {
	From      From
	Text      string
	CreatedAt time.Time
}

type OfferRef struct {
	// Index from 0
	Index int
	Text  string
}

type OpenAIClient struct {
	Cfg *ConfigAIOpenAI
	C   *instructor.InstructorOpenAI
}

type Story struct {
	StoryInfo

	Agent  StoryActor
	Memory StoryActor

	// If is an optional function used to confirm that this story can activate. It has access to the whole story struct,
	// so all the involved state machines and their historical snapshots (relative to activation and deactivation of
	// this story).
	CanActivate func(instance *Story) bool
	Actions     []Action
}

type StoryActionsByIdx []Action

type StoryActor struct {
	Mach *am.Machine

	// actor's time when the story activated
	TimeActivated   am.Time
	TimeDeactivated am.Time

	// When these conditions are met, the story will activate itself.
	Trigger amhelp.Cond
}

type StoryInfo struct {
	// Name of the bound state (eg StoryFoo).
	State string
	// Tick is the current tick of the bound state.
	Tick uint64
	// Epoch is the sum of all previous memories, before a replacement.
	Epoch uint64

	// Title of this story.
	Title string
	// Description of this story.
	Desc string

	// The story was last deactivated at this human time.
	DeactivatedAt time.Time
	// The story was last active for this many ticks of the AgentLLM machine.
	LastActiveTicks uint64
}