types

type BugsDB interface {
	// GetCountsFromDB returns the latest counts data for the client+source+query combination.
	// If client is not specified then latest counts data for all clients is returned.
	// Similarly if source is not specified then latest counts data for all sources for that client are returned.
	// Similarly if query is not specified then latest counts data for all queries for that client+source are returned.
	GetCountsFromDB(ctx context.Context, client RecognizedClient, source IssueSource, query string) (*IssueCountsData, error)

	// GetQueryDataFromDB returns a slice of query data for the client+source+query combination.
	// If client is not specified then query data for all clients is returned.
	// Similarly if source is not specified then query data for all sources for that client are returned.
	// Similarly if query is not specified then query data for all queries for that client+source are returned.
	GetQueryDataFromDB(ctx context.Context, client RecognizedClient, source IssueSource, query string) ([]*QueryData, error)

	// GetClientsFromDB returns a map from clients to sources to queries.
	GetClientsFromDB(ctx context.Context) (map[RecognizedClient]map[IssueSource]map[string]bool, error)

	// PutInDB puts the specified client+source+query counts data into the DB.
	PutInDB(ctx context.Context, client RecognizedClient, source IssueSource, query, runId string, countsData *IssueCountsData) error

	// GenerateRunId creates a run ID from the current timestamp.
	GenerateRunId(ts time.Time) string

	// Returns a map of all recognized run IDs.
	GetAllRecognizedRunIds(ctx context.Context) (map[string]bool, error)

	// Stores the specified run ID in the DB.
	StoreRunId(ctx context.Context, runId string) error
}

type ClientSourceQueryRequest struct {
	Client RecognizedClient `json:"client"`
	Source IssueSource      `json:"source"`
	Query  string           `json:"query"`
}

type GetChartsDataResponse struct {
	OpenData      interface{} `json:"open_data"`
	SloData       interface{} `json:"slo_data"`
	UntriagedData interface{} `json:"untriaged_data"`
}

type GetClientCountsResponse struct {
	ClientsToStatusData map[RecognizedClient]StatusData `json:"clients_to_status_data"`
}

type GetClientsResponse struct {
	Clients map[RecognizedClient]map[IssueSource]map[string]bool `json:"clients"`
}

type Issue struct {
	Id       string               `json:"id"`
	State    string               `json:"state"`
	Priority StandardizedPriority `json:"priority"`
	Owner    string               `json:"owner"`
	Link     string               `json:"link"`

	SLOViolation         bool          `json:"slo_violation"`
	SLOViolationReason   string        `json:"slo_violation_reason"`
	SLOViolationDuration time.Duration `json:"slo_violation_duration"`

	CreatedTime  time.Time `json:"created"`
	ModifiedTime time.Time `json:"modified"`

	Title   string `json:"title"`   // This is not populated in IssueTracker.
	Summary string `json:"summary"` // This is not returned in IssueTracker or Monorail.
}

type IssueCountsData struct {
	OpenCount       int `json:"open_count"`
	UnassignedCount int `json:"unassigned_count"`
	UntriagedCount  int `json:"untriaged_count"`

	// Priority counts.
	P0Count int `json:"p0_count"`
	P1Count int `json:"p1_count"`
	P2Count int `json:"p2_count"`
	P3Count int `json:"p3_count"`
	P4Count int `json:"p4_count"`
	P5Count int `json:"p5_count"`
	P6Count int `json:"p6_count"`

	// SLO violations per priority.
	// We only do SLOs for P0-P3. Listed here: https://docs.google.com/document/d/1OgpX1KDDq3YkHzRJjqRHSPJ9CJ8hH0RTvMAApKVxwm8/edit
	P0SLOViolationCount int `json:"p0_slo_count"`
	P1SLOViolationCount int `json:"p1_slo_count"`
	P2SLOViolationCount int `json:"p2_slo_count"`
	P3SLOViolationCount int `json:"p3_slo_count"`

	// Links to the issue framework.
	QueryLink          string `json:"query_link"`
	UntriagedQueryLink string `json:"untriaged_query_link"`
	P0Link             string `json:"p0_link"`
	P1Link             string `json:"p1_link"`
	P2Link             string `json:"p2_link"`
	P3AndRestLink      string `json:"p3_and_rest_link"`
}

type IssueSource string

type IssuesOutsideSLOResponse struct {
	PriToSLOIssues map[StandardizedPriority][]*Issue `json:"pri_to_slo_issues"`
}

type QueryData struct {
	Created time.Time `json:"created"`
	RunId   string    `json:"run_id"`

	CountsData *IssueCountsData
}

type RecognizedClient string

type StandardizedPriority string

type StatusData struct {
	UntriagedCount int    `json:"untriaged_count"`
	Link           string `json:"link"`
}