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    []StoryAction `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"`

	// Result is a buffered channel to be closed by the receiver
	Result 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
	// Buttons are a list of buttons to be displayed in the UI.
	Buttons      []StoryAction `log:"buttons"`
	Stories      []StoryInfo   `log:"stories"`
	StatesList   []string      `log:"states_list"`
	ActivateList []bool        `log:"activate_list"`
	Result       am.Result
}

type Config struct {
	AI    ConfigAI
	Agent ConfigAgent
	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
}

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

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

type ConfigAgentLog struct {
	// duplicate to state-machine log
	Machine bool
	// path to the log file
	File string
	// log level 0-5
	Level   am.LogLevel
	Prompts bool
}

type ConfigDebug struct {
	// display extra info about these stories in the machine log
	Story []string `kdl:",multiple"`
	// Run the mock scenarios (if any)
	Mock bool
	// Enable misc debugging modes (eg SQL history)
	Misc bool
	// Enable REPL for each machine
	REPL bool
	// Enable debugging in am-dbg ("1" expands to default). am-dbg has to be started BEFORE the bot.
	DBGAddr string
}

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

type ConfigTUI struct {
	// SSH port
	Port int
	// SSH host
	Host string
	// Number of transitions to show on the clock
	ClockRange int
}

type ConfigTools struct {
	SearXNG ConfigSearXNG
}

type From enum.Member[string]

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

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

type Story[G any] struct {
	StoryInfo

	// 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(s *G) bool
}

type StoryAction struct {

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

	Label string
	Desc  string

	StateAdd    string
	StateRemove string

	VisibleCook amhelp.Cond
	VisibleMem  amhelp.Cond
	Action      func()
	IsDisabled  func() bool
	LabelEnd    string

	Pos         int
	PosInferred bool
}

type StoryActionsByIdx []StoryAction

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 StoryImpl[G any] interface {
	Clone() *G
}

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
}