checks

type Custom struct {
	Service
	Command string
	Regex   string
}

type Dns struct {
	Service
	Record []DnsRecord
}

type DnsRecord struct {
	Kind   string
	Domain string
	Answer []string
}

type Ftp struct {
	Service
	File []FtpFile
}

type FtpFile struct {
	Name  string
	Hash  string
	Regex string
}

type Imap struct {
	Service
	Domain    string
	Encrypted bool
}

type Ldap struct {
	Service
	Domain    string
	Encrypted bool
}

type Ping struct {
	Service
	Count           int
	AllowPacketLoss bool
	Percent         int
}

type Pop3 struct {
	Service
	Domain    string
	Encrypted bool
}

type Rdp struct {
	Service
}

type Result struct {
	ServiceName string `json:"name,omitempty"`
	Target      string `json:"target,omitempty"`
	TeamID      uint   `json:"team_id,omitempty"`
	Status      bool   `json:"status,omitempty"`
	Debug       string `json:"debug,omitempty"`
	Error       string `json:"error,omitempty"`
	Points      int    `json:"points,omitempty"`
	ServiceType string `json:"service_type,omitempty"`
	RoundID     uint   `json:"round_id"`

	// Added for runner visualization
	RunnerID   string `json:"runner_id,omitempty"`
	StartTime  string `json:"start_time,omitempty"`
	EndTime    string `json:"end_time,omitempty"`
	StatusText string `json:"status_text,omitempty"` // "running", "success", or "failed"
}

type Runner interface {
	Run(teamID uint, identifier string, roundID uint, resultsChan chan Result)
	Runnable() bool
	Verify(box string, ip string, points int, timeout int, slapenalty int, slathreshold int) error
	GetType() string
	GetName() string
	GetAttempts() int
	GetCredlists() []string
	SetTaskCredentials(creds []TaskCredential)
}

type Service struct {
	Name            string           `toml:"-"`          // Name is the box name plus the service (ex. lunar-dns)
	Display         string           `toml:",omitempty"` // Display is the name of the service (ex. dns)
	CredLists       []string         `toml:",omitempty"`
	Port            int              `toml:",omitzero"` // omitzero because custom checks might not specify port, and shouldn't be assigned 0
	Points          int              `toml:",omitempty"`
	Timeout         int              `toml:",omitempty"`
	SlaPenalty      int              `toml:",omitempty"`
	SlaThreshold    int              `toml:",omitempty"`
	LaunchTime      time.Time        `toml:",omitempty"`
	StopTime        time.Time        `toml:",omitempty"`
	Disabled        bool             `toml:",omitempty"`
	Target          string           `toml:",omitempty"` // Target is the IP address or hostname for the box
	ServiceType     string           `toml:",omitempty"` // ServiceType is the name of the Runner that checks the service
	Attempts        int              `toml:",omitempty"` // Attempts is the number of times the service has been checked
	TaskCredentials []TaskCredential `toml:"-"`          // Credentials from task payload (set per-task, not from config)
}

type Smb struct {
	Service
	Domain string
	Share  string
	File   []smbFile
}

type Smtp struct {
	Service
	Encrypted   bool
	Domain      string
	RequireAuth bool
}

type Sql struct {
	Service
	Kind  string
	Query []queryData
}

type Ssh struct {
	Service
	PrivKey     string `toml:",omitempty"`
	BadAttempts int    `toml:",omitzero"`
	Command     []commandData
}

type TaskCredential struct {
	Username string
	Password string
}

type Tcp struct {
	Service
}

type Vnc struct {
	Service
}

type Web struct {
	Service
	Url    []urlData
	Scheme string
}

type WinRM struct {
	Service
	Encrypted   bool
	BadAttempts int
	Command     []winCommandData
}