diagram

type HTTPClient interface {
	Do(req *http.Request) (*http.Response, error)
}

type HTTPHandler func(ctx context.Context, input Input) (Output, error)

type Input interface {
	Validate() error
	GetUser() *User
	GetPrompt() string
	GetRequestID() string
}

type MockHTTPClient struct {
	V   *http.Response
	Err error
}

type MockInput struct {
	Err       error
	Prompt    string
	RequestID string
	User      *User
}

type MockModelInference struct {
	V               []byte
	UsagePrompt     uint16
	UsageCompletion uint16
	Err             error
}

type MockOutput struct {
	V   []byte
	Err error
}

type MockRepositoryPrediction struct {
	Timestamps []time.Time
	Err        error
}

type MockRepositorySecretsVault struct {
	V   []byte
	Err error
}

type MockRepositoryToken struct {
	V   string
	Err error
}

type ModelInference interface {
	Do(ctx context.Context, userPrompt string, systemContent string, model string) (
		predictionRaw string, prediction []byte, usageTokensPrompt uint16, usageTokensCompletions uint16, err error,
	)
}

type Output interface {
	Serialize() ([]byte, error)
}

type QuotaRequestsConsumption struct {
	Limit int   `json:"limit"`
	Used  int   `json:"used"`
	Reset int64 `json:"reset"`
}

type QuotasUsage struct {
	PromptLengthMax int                      `json:"prompt_length_max"`
	RateMinute      QuotaRequestsConsumption `json:"rate_minute"`
	RateDay         QuotaRequestsConsumption `json:"rate_day"`
}

type RepositoryPrediction interface {
	// WriteInputPrompt records user's input prompt.
	WriteInputPrompt(ctx context.Context, requestID, userID, prompt string) error

	// WriteModelResult records the model's prediction result and the associated costs in tokens.
	WriteModelResult(
		ctx context.Context, requestID, userID, predictionRaw, prediction, model string,
		usageTokensPrompt, usageTokensCompletions uint16,
	) error

	// WriteSuccessFlag records the instance of a successful diagram generation
	// based on the model's prediction result.
	WriteSuccessFlag(ctx context.Context, requestID, userID, token string) error

	// GetDailySuccessfulResultsTimestampsByUserID reads the timestamps of all user's successful requests
	// which led to successful diagrams generation over the last 24 hours / day.
	GetDailySuccessfulResultsTimestampsByUserID(ctx context.Context, userID string) ([]time.Time, error)

	// Close closes connection to persistence service.
	Close(ctx context.Context) error
}

type RepositorySecretsVault interface {
	ReadLastVersion(ctx context.Context, uri string, output interface{}) error
}

type RepositoryToken interface {
	// GetActiveUserIDByActiveTokenID reads userID from the repository given the tokenID.
	// It returns a non-empty value if and only if the token and user are active.
	GetActiveUserIDByActiveTokenID(ctx context.Context, id string) (string, error)
}

type User struct {
	ID           string
	APIToken     string
	IsRegistered bool
}