agentdb

type AgentInfo struct {
	AgentID      string
	AgentName    string
	AgentNetwork string
	Version      string
	OS           string
	Arch         string
	NumCPU       int
	Hostname     string
	PID          int
	NetworkInfo  NetInfo
	StartupArgs  string // masked os.Args as JSON array
	StartupEnv   string // masked key env vars as JSON object
	StartedAt    time.Time
	UpdatedAt    time.Time
}

type InterfaceInfo struct {
	Name  string   `json:"name"`
	Addrs []string `json:"addrs"` // CIDR notation
}

type LogEntry struct {
	ID        int64
	Timestamp time.Time
	Line      string // full slog-formatted line
}

type LogFilter struct {
	Since  time.Time // zero = no lower bound
	Until  time.Time // zero = no upper bound
	Offset int
	Limit  int // 0 = default 500
}

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

type MetricSample struct {
	ID               int64
	Timestamp        time.Time
	CPUPercent       float64
	RSSMB            uint64
	HeapAllocMB      uint64
	HeapSysMB        uint64
	FDUsed           int
	FDLimit          int
	MemTotalMB       uint64
	MemAvailMB       uint64
	Goroutines       int
	ActiveWorkers    int32
	ChunkParallelism int
}

type NetInfo struct {
	Interfaces   []InterfaceInfo `json:"interfaces"`
	Gateway      string          `json:"gateway"`
	DNSResolvers []string        `json:"dns_resolvers"`
	PublicIPs    []string        `json:"public_ips"`
	PrivateIPs   []string        `json:"private_ips"`
	NetworkType  string          `json:"network_type"` // "direct_public", "nat_likely", "no_external"
}

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

type SizeCaps struct {
	LogCapBytes    int64
	MetricCapBytes int64
}

type Store interface {
	// UpsertAgentInfo inserts or replaces the single agent_info row.
	UpsertAgentInfo(ctx context.Context, info *AgentInfo) error

	// GetAgentInfo returns the current agent info, or nil if not yet stored.
	GetAgentInfo(ctx context.Context) (*AgentInfo, error)

	// InsertLog appends a structured log entry.
	InsertLog(ctx context.Context, entry *LogEntry) error

	// InsertMetric appends a resource metrics sample.
	InsertMetric(ctx context.Context, sample *MetricSample) error

	// QueryLogs returns log entries matching the filter, ordered oldest-first.
	QueryLogs(ctx context.Context, filter LogFilter) ([]LogEntry, error)

	// QueryMetrics returns metric samples within the time range, ordered oldest-first.
	QueryMetrics(ctx context.Context, since, until time.Time, limit int) ([]MetricSample, error)

	// InsertTask records a new scan/enumeration task as running.
	InsertTask(ctx context.Context, task *Task) error

	// FinishTask updates a task's status and sets finished_at.
	FinishTask(ctx context.Context, taskID, status string) error

	// ActiveTasks returns tasks with status "running", newest first.
	ActiveTasks(ctx context.Context) ([]Task, error)

	// RecentTasks returns the most recent tasks (any status), newest first.
	RecentTasks(ctx context.Context, limit int) ([]Task, error)

	// DBSizeBytes returns the database file size in bytes.
	DBSizeBytes() (int64, error)

	// Close closes the database connection.
	Close() error
}

type Task struct {
	ID         int64
	Type       string // "scan" or "enumeration"
	TaskID     string // scan_id or enumeration_id
	Status     string // "running", "completed", "failed"
	StartedAt  time.Time
	FinishedAt time.Time // zero if still running
}

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