netquery

type ActiveChartHandler struct {
	Database *Database
}

type BandwidthChartHandler struct {
	Database *Database
}

type BandwidthChartRequest struct {
	Interval int      `json:"interval"`
	Query    Query    `json:"query"`
	GroupBy  []string `json:"groupBy"`
}

type BatchExecute struct {
	ID     string
	SQL    string
	Params map[string]any
	Result *[]map[string]any
}

type BatchQueryHandler struct {
	IsDevMode func() bool
	Database  *Database
}

type BatchQueryRequestPayload map[string]QueryRequestPayload

type Conn struct {
	// ID is a device-unique identifier for the connection. It is built
	// from network.Connection by hashing the connection ID and the start
	// time. We cannot just use the network.Connection.ID because it is only unique
	// as long as the connection is still active and might be, although unlikely,
	// reused afterwards.
	ID              string            `sqlite:"id,primary"`
	ProfileID       string            `sqlite:"profile"`
	Path            string            `sqlite:"path"`
	Type            string            `sqlite:"type,varchar(8)"`
	External        bool              `sqlite:"external"`
	IPVersion       packet.IPVersion  `sqlite:"ip_version"`
	IPProtocol      packet.IPProtocol `sqlite:"ip_protocol"`
	LocalIP         string            `sqlite:"local_ip"`
	LocalPort       uint16            `sqlite:"local_port"`
	RemoteIP        string            `sqlite:"remote_ip"`
	RemotePort      uint16            `sqlite:"remote_port"`
	Domain          string            `sqlite:"domain"`
	Country         string            `sqlite:"country,varchar(2)"`
	ASN             uint              `sqlite:"asn"`
	ASOwner         string            `sqlite:"as_owner"`
	Latitude        float64           `sqlite:"latitude"`
	Longitude       float64           `sqlite:"longitude"`
	Scope           netutils.IPScope  `sqlite:"scope"`
	WorstVerdict    network.Verdict   `sqlite:"worst_verdict"`
	ActiveVerdict   network.Verdict   `sqlite:"verdict"`
	FirewallVerdict network.Verdict   `sqlite:"firewall_verdict"`
	Started         time.Time         `sqlite:"started,text,time"`
	Ended           *time.Time        `sqlite:"ended,text,time"`
	Tunneled        bool              `sqlite:"tunneled"`
	Encrypted       bool              `sqlite:"encrypted"`
	Internal        bool              `sqlite:"internal"`
	Direction       string            `sqlite:"direction"`
	ExtraData       json.RawMessage   `sqlite:"extra_data"`
	Allowed         *bool             `sqlite:"allowed"`
	ProfileRevision int               `sqlite:"profile_revision"`
	ExitNode        *string           `sqlite:"exit_node"`
	BytesReceived   uint64            `sqlite:"bytes_received,default=0"`
	BytesSent       uint64            `sqlite:"bytes_sent,default=0"`

	// TODO(ppacher): support "NOT" in search query to get rid of the following helper fields
	Active bool `sqlite:"active"` // could use "ended IS NOT NULL" or "ended IS NULL"

	// TODO(ppacher): we need to profile here for "suggestion" support. It would be better to keep a table of profiles in sqlite and use joins here
	ProfileName string `sqlite:"profile_name"`
}

type ConnectionStore interface {
	// Save is called to perists the new or updated connection. If required,
	// It's up to the implementation to figure out if the operation is an
	// insert or an update.
	// The ID of Conn is unique and can be trusted to never collide with other
	// connections of the save device.
	Save(context.Context, Conn, bool) error

	// MarkAllHistoryConnectionsEnded marks all active connections in the history
	// database as ended NOW.
	MarkAllHistoryConnectionsEnded(context.Context) error

	// RemoveAllHistoryData removes all connections from the history database.
	RemoveAllHistoryData(context.Context) error

	// RemoveHistoryForProfile removes all connections from the history database.
	// for a given profile ID (source/id)
	RemoveHistoryForProfile(context.Context, string) error

	// UpdateBandwidth updates bandwidth data for the connection and optionally also writes
	// the bandwidth data to the history database.
	UpdateBandwidth(ctx context.Context, enableHistory bool, profileKey string, processKey string, connID string, bytesReceived uint64, bytesSent uint64) error

	// CleanupHistory deletes data outside of the retention time frame from the history database.
	CleanupHistory(ctx context.Context) error

	// Close closes the connection store. It must not be used afterwards.
	Close() error
}

type Count struct {
	As       string `json:"as"`
	Field    string `json:"field"`
	Distinct bool   `json:"distinct"`
}

type Database struct {
	Schema *orm.TableSchema
	// contains filtered or unexported fields
}

type DatabaseName string

type Equal interface{}

type FieldSelect struct {
	Field string `json:"field"`
	As    string `json:"as"`
}

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

type MatchType interface {
	Operator() string
}

type Matcher struct {
	Equal          interface{}   `json:"$eq,omitempty"`
	NotEqual       interface{}   `json:"$ne,omitempty"`
	In             []interface{} `json:"$in,omitempty"`
	NotIn          []interface{} `json:"$notIn,omitempty"`
	Like           string        `json:"$like,omitempty"`
	Greater        *float64      `json:"$gt,omitempty"`
	GreaterOrEqual *float64      `json:"$ge,omitempty"`
	Less           *float64      `json:"$lt,omitempty"`
	LessOrEqual    *float64      `json:"$le,omitempty"`
}

type Min struct {
	Condition *Query `json:"condition,omitempty"`
	Field     string `json:"field"`
	As        string `json:"as"`
	Distinct  bool   `json:"distinct"`
}

type OrderBy struct {
	Field string `json:"field"`
	Desc  bool   `json:"desc"`
}

type OrderBys []OrderBy

type Pagination struct {
	PageSize int `json:"pageSize"`
	Page     int `json:"page"`
}

type Query map[string][]Matcher

type QueryActiveConnectionChartPayload struct {
	Query      Query       `json:"query"`
	TextSearch *TextSearch `json:"textSearch"`
}

type QueryHandler struct {
	IsDevMode func() bool
	Database  *Database
}

type QueryRequestPayload struct {
	Select     Selects     `json:"select"`
	Query      Query       `json:"query"`
	OrderBy    OrderBys    `json:"orderBy"`
	GroupBy    []string    `json:"groupBy"`
	TextSearch *TextSearch `json:"textSearch"`
	// A list of databases to query. If left empty,
	// both, the LiveDatabase and the HistoryDatabase are queried
	Databases []DatabaseName `json:"databases"`

	Pagination
	// contains filtered or unexported fields
}

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

type Select struct {
	Field       string       `json:"field"`
	FieldSelect *FieldSelect `json:"$field"`
	Count       *Count       `json:"$count,omitempty"`
	Sum         *Sum         `json:"$sum,omitempty"`
	Min         *Min         `json:"$min,omitempty"`
	Distinct    *string      `json:"$distinct,omitempty"`
}

type Selects []Select

type Sum struct {
	Condition Query  `json:"condition"`
	Field     string `json:"field"`
	As        string `json:"as"`
	Distinct  bool   `json:"distinct"`
}

type TextSearch struct {
	Fields []string `json:"fields"`
	Value  string   `json:"value"`
}