cloud

type BurstCredentials struct {
	// GitHubToken is a fine-grained token for the repository, with contents
	// and pull requests write.
	GitHubToken string `json:"github_token"`
	// HuggingFaceToken writes the run logs to the corpus dataset.
	HuggingFaceToken string `json:"hf_token"`
	// CorpusDataset is the private dataset the run logs go to, owner/name.
	CorpusDataset string `json:"corpus_dataset"`
	// ExecutorEnvironment is the executor's model credential, by the
	// environment variable it reads, such as ANTHROPIC_API_KEY.
	ExecutorEnvironment map[string]string `json:"executor_environment"`
}

type CloudJob struct {
	ID       string `json:"id"`
	Provider string `json:"provider"`
	Flavor   string `json:"flavor"`
	// Purpose is what the job is for: the burst and its slices.
	Purpose string `json:"purpose"`
	// URL is the job's own page at the provider.
	URL        string       `json:"url,omitempty"`
	CreatedAt  time.Time    `json:"created_at"`
	FinishedAt *time.Time   `json:"finished_at,omitempty"`
	Stage      hfjobs.Stage `json:"stage"`
	CapUSD     float64      `json:"cap_usd"`
	// UnitCostMicroUSD is the flavor's price per minute when it launched.
	UnitCostMicroUSD int64 `json:"unit_cost_micro_usd"`
	// TimeoutSeconds is the provider's own stop: the minutes the cap buys.
	TimeoutSeconds int64 `json:"timeout_seconds"`
	// SpendUSD is the spend so far: the minutes since the job was created,
	// until it finished, at the flavor's price. Billing from creation rather
	// than from the start of running reads high, never low.
	SpendUSD   float64    `json:"spend_usd"`
	StopReason StopReason `json:"stop_reason,omitempty"`
	// Slices are the dispatcher's slices the job runs, held on this host.
	Slices []string `json:"slices,omitempty"`
	// Branches are the work branches its sessions push.
	Branches []string `json:"branches,omitempty"`
	// PullRequests is how many of those branches' pull requests merged,
	// counted once the job ended; -1 until then.
	PullRequests int `json:"pull_requests"`
}

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

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

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

type IJobProvider interface {
	Run(ctx context.Context, spec hfjobs.Spec) (hfjobs.Job, error)
	List(ctx context.Context) ([]hfjobs.Job, error)
	Cancel(ctx context.Context, id string) error
	Hardware(ctx context.Context) ([]hfjobs.Hardware, error)
	JobURL(id string) string
}

type INotifier interface {
	Notify(ctx context.Context, notice Notice) error
}

type LaunchBurstInput struct {
	Sessions int     `json:"sessions" jsonschema:"how many fixer sessions the job runs, one ready slice each"`
	Flavor   string  `json:"flavor" jsonschema:"the provider's hardware flavor, such as cpu-xl"`
	CapUSD   float64 `` /* 128-byte string literal not displayed */
	DryRun   bool    `json:"dry_run,omitempty" jsonschema:"build and show the job spec, secrets redacted, and launch nothing"`
}

type LaunchBurstOutput struct {
	DryRun  bool        `json:"dry_run"`
	Spec    hfjobs.Spec `json:"spec"`
	Slices  []string    `json:"slices"`
	Missing []string    `json:"missing"`
	Job     *CloudJob   `json:"job,omitempty"`
}

type LaunchProbeInput struct {
	Flavor  string  `json:"flavor" jsonschema:"the provider's hardware flavor, such as cpu-basic"`
	Purpose string  `json:"purpose" jsonschema:"what the probe is for, as the Cloud panel shows it"`
	CapUSD  float64 `json:"cap_usd" jsonschema:"its spend cap in dollars; the provider's timeout is the minutes it buys"`
}

type LaunchSuiteShardInput struct {
	Build   string          `json:"build"`
	Suite   json.RawMessage `json:"suite"`
	Jobs    json.RawMessage `json:"jobs"`
	Tickets []string        `json:"tickets"`
	Flavor  string          `json:"flavor"`
	CapUSD  float64         `json:"cap_usd"`
	DryRun  bool            `json:"dry_run,omitempty"`
}

type Ledger struct {
	DailyCapUSD float64    `json:"daily_cap_usd"`
	Jobs        []CloudJob `json:"jobs"`
	Notices     []Notice   `json:"notices"`
	UpdatedAt   time.Time  `json:"updated_at"`
}

type ListCloudJobsInput struct{}

type Notice struct {
	At    time.Time  `json:"at"`
	Kind  NoticeKind `json:"kind"`
	JobID string     `json:"job_id"`
	Text  string     `json:"text"`
}

type NoticeKind string

type Option func(jobs *CloudJobs) error

type StopCloudJobInput struct {
	ID string `json:"id" jsonschema:"the job's identifier, as the record and the Cloud panel show it"`
}

type StopReason string