models

type BlueTeamStore struct {
	Name       string `json:"name"`        // name
	Hash       []byte `json:"-"`           // hash
	BlueteamIP int16  `json:"blueteam_ip"` // blueteam_ip
}

type BlueTeamStoreSlice []BlueTeamStore

type BlueteamView struct {
	ID         int    `json:"id"`          // id
	Name       string `json:"name"`        // name
	BlueteamIP int16  `json:"blueteam_ip"` // blueteam_ip
}

type CTFProgress struct {
	Category string `json:"category"`
	Amount   int    `json:"count"`
	Max      int    `json:"max"`
}

type CapturedChallenge struct {
	Designer  string    `json:"designer"` // challenge.designer
	Category  string    `json:"category"` // challenge.category
	Name      string    `json:"name"`     // challenge.name
	Timestamp time.Time `json:"time"`     // ctf_solve.created_at
}

type Challenge struct {
	ID       int     `json:"id"`       // id
	Name     string  `json:"name"`     // name
	Category string  `json:"category"` // category
	Designer string  `json:"designer"` // designer
	Flag     string  `json:"flag"`     // flag
	Total    float32 `json:"total"`    // total
	Body     string  `json:"body"`     // body
	Hidden   bool    `json:"hidden"`   // hidden

	CreatedAt  time.Time `json:"created_at"`  // created_at
	ModifiedAt time.Time `json:"modified_at"` // modified_at
}

type ChallengeCaptureCount struct {
	Designer  string     `json:"designer"`             // challenge.designer
	Category  string     `json:"category"`             // challenge.category
	Name      string     `json:"name"`                 // challenge.name
	Count     int        `json:"count"`                // --- (calculated column)
	FirstTeam *string    `json:"first_team,omitempty"` // team.name
	Timestamp *time.Time `json:"timestamp,omitempty"`  // ctf_solve.created_at
}

type ChallengeGuess struct {
	Name     string  `json:"name"`     // name
	Category string  `json:"category"` // category
	Flag     string  `json:"flag"`     // flag
	Points   float32 `json:"points"`   // total
}

type ChallengeSlice []Challenge

type ChallengeView struct {
	ID     int    `json:"id"`     // id
	Name   string `json:"name"`   // name
	Points int    `json:"points"` // points (rounded down to nearest int)

	Captured bool `json:"captured"` // Whether the viewing team has already got this flag
}

type ChallengeViewGroup struct {
	Category   string          `json:"category"`
	Challenges []ChallengeView `json:"challenges"`
}

type CtfSolve struct {
	CreatedAt   time.Time `json:"created_at"`   // created_at
	TeamID      int       `json:"team_id"`      // team_id
	ChallengeID int       `json:"challenge_id"` // challenge_id
}

type CtfSolveResult struct {
	Timestamp     time.Time `json:"timestamp"`      // ctf_solve.created_at
	TeamID        int       `json:"team_id"`        // team.id
	TeamName      string    `json:"team_name"`      // team.name
	ChallengeID   int       `json:"challenge_id"`   // challenge.id
	ChallengeName string    `json:"challenge_name"` // challenge.name
	Category      string    `json:"category"`       // challenge.category
	Points        float32   `json:"points"`         // challenge.total
}

type DB interface {
	Exec(string, ...interface{}) (pgx.CommandTag, error)
	Query(string, ...interface{}) (*pgx.Rows, error)
	QueryRow(string, ...interface{}) *pgx.Row
	CopyFrom(tableName pgx.Identifier, columnNames []string, rowSrc pgx.CopyFromSource) (int, error)
}

type DBClient interface {
	DB
	TXer
}

type Deleter interface {
	Delete(DB) error
}

type ExitStatus uint16

type FlagState int

type Inserter interface {
	Insert(DB) error
}

type ManyInserter interface {
	Insert(TXer) error
}

type MonitorTeamService struct {
	Team struct {
		ID   int    // id
		Name string // name
		IP   int16  // blueteam_ip
	}
	Service struct {
		ID       int       // id
		Name     string    // name
		Script   string    // script
		Args     []string  // args
		StartsAt time.Time // starts_at
	}
}

type OtherPoints struct {
	CreatedAt time.Time `json:"created_at"` // created_at
	TeamID    int       `json:"team_id"`    // team_id
	Points    float32   `json:"points"`     // points
	Reason    string    `json:"reason"`     // reason
}

type OtherPointsSlice []OtherPoints

type OtherPointsView struct {
	CreatedAt time.Time `json:"created_at"` // created_at
	Teams     []string  `json:"teams"`      // array of team names
	Points    float32   `json:"points"`     // points
	Reason    string    `json:"reason"`     // reason
}

type Service struct {
	ID          int      `json:"id"`           // id
	Name        string   `json:"name"`         // name
	Category    string   `json:"category"`     // category
	Description string   `json:"description"`  // description
	TotalPoints float32  `json:"total_points"` // total_points
	Points      *float32 `json:"points"`       // points
	Script      string   `json:"script"`       // script
	Args        []string `json:"args"`         // args
	Disabled    bool     `json:"disabled"`     // disabled

	StartsAt   time.Time `json:"starts_at"`   // starts_at
	CreatedAt  time.Time `json:"created_at"`  // created_at
	ModifiedAt time.Time `json:"modified_at"` // modified_at
}

type ServiceCheck struct {
	CreatedAt time.Time  `json:"created_at"` // created_at
	TeamID    int        `json:"team_id"`    // team_id
	ServiceID int        `json:"service_id"` // service_id
	Status    ExitStatus `json:"status"`     // status
	ExitCode  int16      `json:"exit_code"`  // exit_code
}

type ServiceCheckSlice []ServiceCheck

type ServiceSlice []Service

type TXer interface {
	Begin() (Tx, error)
	BeginEx(context.Context, *pgx.TxOptions) (Tx, error)
}

type Team struct {
	ID         int      `json:"id"`          // id
	Name       string   `json:"name"`        // name
	RoleName   TeamRole `json:"role_name"`   // role_name
	Hash       []byte   `json:"-"`           // hash
	Disabled   bool     `json:"disabled"`    // disabled
	BlueteamIP *int16   `json:"blueteam_ip"` // blueteam_ip
}

type TeamCapturedChallenges struct {
	Team       string              `json:"team"` // team.name
	Challenges []CapturedChallenge `json:"challenges"`
}

type TeamRole uint16

type TeamServiceStatusesView struct {
	ServiceID   int          `json:"service_id"`
	ServiceName string       `json:"service_name"`
	Statuses    []ExitStatus `json:"statuses"`
}

type TeamsScoresResponse struct {
	TeamID  int    `json:"team_id"`
	Name    string `json:"name"`
	Score   int    `json:"score"`
	Service int    `json:"service"`
	Ctf     int    `json:"ctf"`
	Other   int    `json:"other"`
}

type Tx = *pgx.Tx

type Updater interface {
	Update(DB) error
}