utils

type APICount struct {
	Root string
	Cnt  int
}

type APIPathRow struct {
	RootURL   string `json:"root_url"`
	SourceURL string `json:"source_url"`
	Path      string `json:"path"`
	SaveDir   string `json:"save_dir"`
}

type CDNHostRow struct {
	Root string
	Host string
}

type Config struct {
	Rarity   int
	Routines int

	SendTimeout time.Duration
	ReadTimeout time.Duration

	NULLProbeOnly bool
	UseAllProbes  bool
	SSLAlwaysTry  bool
}

type Details struct {
	ProbeName     string `json:"probe_name"`
	ProbeData     string `json:"probe_data"`
	MatchMatched  string `json:"match_matched"`
	IsSoftMatched bool   `json:"soft_matched"`
}

type Directive struct {
	DirectiveName string
	Flag          string
	Delimiter     string
	DirectiveStr  string
}

type EntropyHit struct {
	SourceURL string  `json:"source_url"`
	Category  string  `json:"category"`
	Content   string  `json:"content"`
	Entropy   float64 `json:"entropy"`
	SaveDir   string  `json:"save_dir"`
}

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

type Extras struct {
	VendorProduct   string `json:"vendor_product,omitempty"`
	Version         string `json:"version,omitempty"`
	Info            string `json:"info,omitempty"`
	Hostname        string `json:"hostname,omitempty"`
	OperatingSystem string `json:"operating_system,omitempty"`
	DeviceType      string `json:"device_type,omitempty"`
	CPE             string `json:"cpe,omitempty"`
}

type FingerResult struct {
	Finger      string
	Server      string
	Title       string
	ContentType string
	Location    string
	HeadersJSON string
	Body        []byte
	Status      int
	Err         error
}

type Fingerprint struct {
	Cms      string
	Method   string
	Location string
	Keyword  []string
}

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

type GraphEdge struct {
	Root  string `json:"root"`
	From  string `json:"from"`
	To    string `json:"to"`
	Depth int    `json:"depth"`
}

type HTMLReportData struct {
	GeneratedAt   string
	Summary       []SpiderRecord
	APIs          []APIPathRow
	Sensitive     []SensitiveHit
	SourceMaps    []SourceMapHit
	Pages         []PageSnapshotMeta
	PageBodies    []PageSnapshotLite
	Scores        []ScoredRow
	CDNHosts      []CDNHostRow
	Graph         []GraphEdge
	ImportantAPIs []string
	LLMSummary    *LLMSummary
}

type HostErrorStat struct {
	Count  int
	Sample string
}

type IpHash struct {
	Ip   string
	Hash uint64
}

type LLMConfig struct {
	Provider      string
	Model         string
	Prompt        string
	APIKey        string
	BaseURL       string
	MaxInputChars int
	MaxOutputTok  int
	DryRun        bool
}

type LLMProfile struct {
	Name     string `json:"name"`
	Provider string `json:"provider"`
	Model    string `json:"model"`
	APIKey   string `json:"api_key"`
	BaseURL  string `json:"base_url,omitempty"`
}

type LLMSummary struct {
	Enabled      bool   `json:"enabled"`
	Provider     string `json:"provider"`
	Model        string `json:"model"`
	Prompt       string `json:"prompt"`
	Output       string `json:"output"`
	Error        string `json:"error,omitempty"`
	InputPreview string `json:"input_preview,omitempty"`
	InputTotal   int    `json:"input_total"`
	InputLimit   int    `json:"input_limit,omitempty"`
	OutputLimit  int    `json:"output_limit,omitempty"`
	GeneratedAt  string `json:"generated_at,omitempty"`
	DryRun       bool   `json:"dry_run,omitempty"`
}

type LogLevel int

type Logger interface {
	Debug(msg string, args ...interface{})
	Info(msg string, args ...interface{})
	Warning(msg string, args ...interface{})
	Error(msg string, args ...interface{})
	Fatal(msg string, args ...interface{})

	SetLevel(level LogLevel)
	SetOutput(w io.Writer)
	WithField(key string, value interface{}) Logger
	WithFields(fields map[string]interface{}) Logger
}

type Match struct {
	IsSoft bool

	Service     string
	Pattern     string
	VersionInfo string

	PatternCompiled *regexp2.Regexp
}

type Packjson struct {
	Fingerprint []Fingerprint
}

type PageSnapshot struct {
	RootURL     string `json:"root_url"`
	URL         string `json:"url"`
	Status      int    `json:"status"`
	ContentType string `json:"content_type"`
	Headers     string `json:"headers"`
	Body        string `json:"body"`
	Length      int    `json:"length"`
}

type PageSnapshotLite struct {
	RootURL     string `json:"root_url"`
	URL         string `json:"url"`
	Status      int    `json:"status"`
	ContentType string `json:"content_type"`
	Length      int    `json:"length"`
	Headers     string `json:"headers"`
	Snippet     string `json:"snippet"`
}

type PageSnapshotMeta struct {
	RootURL     string `json:"root_url"`
	URL         string `json:"url"`
	Status      int    `json:"status"`
	ContentType string `json:"content_type"`
	Length      int    `json:"length"`
}

type Probe struct {
	Name        string
	Data        string
	DecodedData []byte
	Protocol    string

	Ports    string
	SSLPorts string

	TotalWaitMS  int
	TCPWrappedMS int
	Rarity       int
	Fallback     string

	Matchs *[]Match
}

type ProbesRarity []Probe

type ProtocolInfo struct {
	Ip   string
	Port int
}

type Result struct {
	Target
	Service `json:"service"`

	Timestamp int32  `json:"timestamp"`
	Error     string `json:"error"`
}

type ScoredRow struct {
	RootURL   string         `json:"root_url"`
	Score     int            `json:"score"`
	Status    int            `json:"status"`
	ApiCount  int            `json:"api_count"`
	Important int            `json:"important_api"`
	UrlCount  int            `json:"url_count"`
	CDNCount  int            `json:"cdn_count"`
	SaveDir   string         `json:"save_dir"`
	Reasons   []string       `json:"reasons"`
	RiskFlags []string       `json:"risk_flags"`
	DebugMeta map[string]any `json:"debug_meta"`
}

type SelfUpdateOptions struct {
	Owner          string
	Repo           string
	CurrentVersion string
	TargetVersion  string
	DownloadURL    string
	OS             string
	Arch           string
	DryRun         bool
	Force          bool
	SkipChecksum   bool
	HTTPClient     *http.Client
	UserAgent      string
	Token          string
}

type SensitiveCount struct {
	Category string `json:"category"`
	Count    int    `json:"count"`
}

type SensitiveData struct {
	Content string
	Entropy float64
}

type SensitiveHit struct {
	SourceURL string  `json:"source_url"`
	Category  string  `json:"category"`
	Content   string  `json:"content"`
	Entropy   float64 `json:"entropy"`
	SaveDir   string  `json:"save_dir"`
}

type Service struct {
	Target

	Name        string `json:"name"`
	Protocol    string `json:"protocol"`
	Banner      string `json:"banner"`
	BannerBytes []byte `json:"banner_bytes"`

	Extras  `json:"extras"`
	Details `json:"details"`
}

type SourceMapHit struct {
	RootURL string `json:"root_url"`
	JSURL   string `json:"js_url"`
	MapURL  string `json:"map_url"`
	Status  int    `json:"status"`
	Length  int    `json:"length"`
}

type SpiderRecord struct {
	Url        string `json:"Url"`
	IconHash   string `json:"IconHash"`
	IconBase64 string `json:"IconBase64"`
	ApiCount   int    `json:"ApiCount"`
	UrlCount   int    `json:"UrlCount"`
	CDNCount   int    `json:"CDNCount"`
	CDNHosts   string `json:"CDNHosts"`
	SaveDir    string `json:"SaveDir"`
	Status     int    `json:"Status"`
}

type SpiderSummary struct {
	URL         string
	Title       string
	Finger      string
	ContentType string
	Status      int
	Length      int
	Keyword     string
	SimHash     string
	IconHash    string
	IconBase64  string
	ApiCount    int
	UrlCount    int
	CDNCount    int
	CDNHosts    string
	SaveDir     string
	Err         error
}

type Target struct {
	IP       string `json:"ip"`
	Port     int    `json:"port"`
	Protocol string `json:"protocol"`
}

type VScan struct {
	Exclude string

	Probes []Probe

	ProbesMapKName map[string]Probe
}

type Worker struct {
	In     chan Target
	Out    chan Result
	Config *Config
}

type Worksheet struct {
	Name string
	Rows [][]string
}