uploadflow

type ConflictError struct {
	SessionID string
	State     string
	Stale     bool
	Err       error
}

type Events interface {
	// Progress reports one file's transfer progress. committed grows with
	// server-acknowledged bytes; committed == total is emitted exactly once
	// per file and signals that file's completion.
	Progress(file string, committed, total int64)
	// Note reports one fully formatted, human-readable flow message
	// (without a trailing newline).
	Note(msg string)
}

type ManifestInputs struct {
	ModelFile     string
	Inputs        []string
	External      []string
	InputManifest string
	// Buckets are the already-parsed .pt2 shape buckets (the CLI's --bucket
	// flags); only valid together with ModelFile.
	Buckets []upload.BucketSpec
}

type NopEvents struct{}

type Orchestrator struct {
	// Gen is the generated API client over the authenticated transport.
	Gen *gen.ClientWithResponses
	// Events receives progress and notes; nil discards them.
	Events Events
	// Bare is the HTTP client used against signed GCS URLs. It must carry
	// NO API transport chain (no Authorization header, no debug logging):
	// signed URLs and resumable session URIs are credentials.
	Bare *http.Client
	// StallTimeout is the per-chunk inactivity budget during transfers.
	StallTimeout time.Duration

	// Poll seams for completion polling: nil selects the real
	// jitter/sleeper/clock in internal/wait.
	Jitter func(time.Duration) time.Duration
	Sleep  func(context.Context, time.Duration) error
	Now    func() time.Time
	// TransferSleep, when non-nil, replaces the GCS uploader's retry
	// backoff sleep (tests inject it; the CLI leaves it nil).
	TransferSleep func(context.Context, time.Duration) error
}

type Phase int

type Request struct {
	Account string
	Name    string
	// Repo is ACCOUNT/NAME as displayed and persisted in resume state.
	Repo    string
	Specs   []upload.FileSpec
	Buckets []upload.BucketSpec
	// Wait polls completion (with deliberate replays) until a model
	// reference or terminal state is observable, within Timeout.
	Wait    bool
	Timeout time.Duration
}

type Result struct {
	SessionID string
	Repo      string
	// Response is the typed final completion response (nil on partial
	// results).
	Response *gen.CompleteModelUploadResponse
	// Completion is Response's raw body, byte-exact.
	Completion json.RawMessage
	// Model is the raw "model" object from Completion; nil when absent.
	Model json.RawMessage
	// WaitStarted is the completion clock start, for callers sharing the
	// Wait budget with follow-up polling.
	WaitStarted time.Time
	Lease       *upload.SessionLease
}

type ResumeOptions struct {
	Account string
	Name    string
	Repo    string
	// BuildSpecs lazily digests the local files for a state rebuild; it is
	// invoked only when the local state file is missing or corrupt. nil
	// means the caller provided no local file arguments.
	BuildSpecs func() ([]upload.FileSpec, error)
	Wait       bool
	Timeout    time.Duration
}

type SessionError struct {
	Phase     Phase
	SessionID string
	Repo      string
	Err       error
}

type TerminalStateError struct {
	SessionID string
	State     string
}

type UsageError struct{ Err error }