config

type AgentOptions struct {
	Prompt         string   `short:"p" long:"prompt" description:"Natural language task for the agent"`
	Inputs         []string `short:"i" long:"input" description:"Target input: IP, URL, IP:port, or CIDR. Can specify multiple"`
	Skills         []string `short:"s" long:"skill" description:"Skill to apply (name or file path). Can specify multiple"`
	Tools          []string `` /* 126-byte string literal not displayed */
	TaskFile       string   `long:"task-file" description:"File containing task description"`
	Heartbeat      int      `` /* 132-byte string literal not displayed */
	Timeout        int      `long:"timeout" config:"timeout" description:"Overall timeout in seconds" default:"3600"`
	EvalCriteria   string   `` /* 146-byte string literal not displayed */
	EvalModel      string   `long:"eval-model" config:"eval_model" description:"Model for goal evaluation (defaults to main model)"`
	EvalMaxRetries int      `long:"eval-retries" config:"eval_retries" description:"Max goal evaluation retry rounds" default:"3"`
	WebURL         string   `long:"web-url" config:"web_url" description:"AIScan web server URL for remote REPL and PTY access"`
	Resume         string   `` /* 148-byte string literal not displayed */
	SaveSession    bool     `` /* 135-byte string literal not displayed */
}

type IOAClientArgs struct {
	Space     string
	MessageID string
}

type IOAConfig struct {
	URL           string
	NodeID        string
	NodeName      string
	Space         string
	RegisterTools bool
	AutoRegister  bool
	NodeMeta      map[string]any
}

type IOAOptions struct {
	IOAURL      string `long:"ioa-url" config:"url" description:"IOA server URL (supports http://token@host:port for auth)"`
	IOAToken    string `long:"ioa-token" config:"token" description:"IOA server access key (for 'ioa serve'; auto-generated if empty)"`
	IOANodeID   string `long:"ioa-node-id" description:"Existing IOA node id for agent tools"`
	IOANodeName string `long:"ioa-node-name" config:"node_name" description:"IOA node name when auto-registering"`
	Space       string `long:"space" config:"space" description:"IOA space name" default:"default"`
	IOAJSON     bool   `long:"json" description:"Output IOA query results in JSON format"`
}

type LLMOptions struct {
	Provider  string             `` /* 135-byte string literal not displayed */
	BaseURL   string             `long:"base-url" config:"base_url" description:"LLM API base URL (leave empty to use provider default)"`
	APIKey    string             `long:"api-key" config:"api_key" description:"LLM API key (or env: OPENAI_API_KEY, ANTHROPIC_API_KEY, AISCAN_API_KEY)"`
	Model     string             `long:"model" config:"model" description:"LLM model name"`
	LLMProxy  string             `long:"llm-proxy" config:"proxy" description:"Proxy for LLM API requests"`
	Providers []LLMProviderEntry `no-flag:"true" config:"providers" description:"Additional LLM providers for fallback or multi-model routing"`
	AI        bool               `long:"ai" description:"Analyze direct scanner output with an LLM"`
}

type LLMProviderEntry struct {
	Provider string `config:"provider" yaml:"provider"`
	BaseURL  string `config:"base_url" yaml:"base_url"`
	APIKey   string `config:"api_key" yaml:"api_key"`
	Model    string `config:"model" yaml:"model"`
	Proxy    string `config:"proxy" yaml:"proxy"`
	Timeout  int    `config:"timeout" yaml:"timeout"`
	Images   *bool  `config:"images" yaml:"images,omitempty"`
}

type MiscOptions struct {
	ConfigFile string `short:"c" long:"config" description:"Path to config file (default: ./aiscan.yaml, <binary_dir>/aiscan.yaml)"`
	DataDir    string `long:"data-dir" config:"data_dir" description:"Data directory for cache, arsenal, history (default: <binary_dir>/.aiscan)"`
	InitConfig bool   `long:"init" description:"Generate default aiscan.yaml and exit"`
	ViewFile   string `short:"F" long:"view" description:"View a scan record JSONL file"`
	ViewFormat string `short:"o" long:"output" description:"Output format for -F: terminal (default), markdown" default:"terminal"`
	ViewOutput string `short:"f" long:"file" description:"Write -F output to file instead of stdout"`
	Debug      bool   `long:"debug" config:"debug" description:"Enable debug logging"`
	Verbose    []bool `short:"v" long:"verbose" description:"Increase verbosity (-v tools, -vv thinking)"`
	Quiet      bool   `short:"q" long:"quiet" config:"quiet" description:"Quiet mode — only show final result"`
	NoColor    bool   `long:"no-color" config:"no_color" description:"Disable ANSI colors in scanner output"`
	Version    bool   `long:"version" description:"Print version and exit"`
}

type Option struct {
	LLMOptions     `group:"LLM Options" config:"llm"`
	ScannerOptions `group:"Scanner Options" config:"cyberhub"`
	AgentOptions   `group:"Agent Options" config:"agent"`
	IOAOptions     `group:"IOA Options" config:"ioa"`
	ReconOptions   `group:"Recon Options" config:"recon"`
	MiscOptions    `group:"Miscellaneous Options" config:"misc"`
	ScanConfig     ScanConfigOptions `no-flag:"true" config:"scan"`
}

type ReconOptions struct {
	FofaEmail    string `long:"fofa-email" config:"fofa_email" hidden:"true"`
	FofaKey      string `long:"fofa-key" config:"fofa_key" hidden:"true"`
	HunterToken  string `long:"hunter-token" config:"hunter_token" hidden:"true"`
	HunterAPIKey string `long:"hunter-api-key" config:"hunter_api_key" hidden:"true"`
	ReconProxy   string `long:"recon-proxy" config:"proxy" hidden:"true"`
	ReconLimit   *int   `long:"recon-limit" config:"limit" hidden:"true"`
}

type RunMode string

type RuntimeConfig struct {
	Provider      RuntimeProviderConfig
	Scanner       ScannerConfig
	Tools         ToolConfig
	IOA           *IOAConfig
	Logger        telemetry.Logger
	CLISkillPaths []string
}

type RuntimeFeatures struct {
	ProviderEnabled  bool
	ProviderOptional bool
	ToolsEnabled     bool
	AIEnabled        bool
	ScannerAI        bool
	Warning          string
}

type RuntimeProviderConfig struct {
	Enabled   bool
	Config    agent.ProviderConfig
	Fallbacks []agent.ProviderConfig
	Optional  bool
}

type ScanConfigOptions struct {
	Verify        string `config:"verify"`
	VerifyTimeout int    `config:"verify_timeout"`
}

type ScannerCommands struct {
	Scan    struct{} `command:"scan" description:"Run the scan pipeline"`
	Gogo    struct{} `command:"gogo" description:"Run gogo scanner"`
	Spray   struct{} `command:"spray" description:"Run spray scanner"`
	Katana  struct{} `command:"katana" description:"Run katana web crawler"`
	Zombie  struct{} `command:"zombie" description:"Run zombie weakpass scanner"`
	Neutron struct{} `command:"neutron" description:"Run neutron POC scanner"`
	Passive struct{} `command:"passive" description:"Run passive cyberspace recon"`
}

type ScannerConfig struct {
	CyberhubURL       string
	CyberhubKey       string
	CyberhubMode      string
	AIEnabled         bool
	EnableAllAISkills bool
	AITimeout         int
	VerifyMode        string
	Proxy             string
	FofaEmail         string
	FofaKey           string
	HunterToken       string
	HunterAPIKey      string
	ReconProxy        string
	ReconLimit        int
}

type ScannerOptions struct {
	CyberhubURL  string `long:"cyberhub-url" config:"url" description:"Cyberhub server URL for loading fingers/templates"`
	CyberhubKey  string `long:"cyberhub-key" config:"key" description:"Cyberhub API key"`
	CyberhubMode string `long:"cyberhub-mode" config:"mode" description:"Cyberhub resource mode: merge or override"`
	Proxy        string `` /* 151-byte string literal not displayed */
}

type ToolConfig struct {
	Enabled       bool
	BashTimeout   int
	TavilyKeys    string
	OptionalTools []string // optional tool groups to enable (e.g. "search", "browser")
}