aimlapi

type APIError struct {
	Message string
	Status  int
	Body    string
}

type AuthResult struct {
	Token string `json:"token"`
	Exp   int64  `json:"exp"`
}

type BalanceResult struct {
	Balance             float64 `json:"balance"`
	LowBalance          bool    `json:"lowBalance"`
	LowBalanceThreshold float64 `json:"lowBalanceThreshold"`
}

type CheckResult struct {
	// "sign-in" (account exists) or "sign-up" (no account yet).
	Action   string  `json:"action"`
	Provider *string `json:"provider"`
}

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

type CreatedKey struct {
	Key string `json:"key"`
	ID  string `json:"id"`
}

type Endpoints struct {
	AuthBaseURL      string
	AppBaseURL       string
	InferenceBaseURL string
	PayBaseURL       string
	// VerificationBaseURL is the aimlapi web-app frontend base. The browser is sent
	// back here after the partner checkout completes (see BuildPartnerReturnURL), so
	// it must point at the same environment's front as the one the CLI is talking to
	// to land the user there. Empty falls back to DefaultReturnURL.
	VerificationBaseURL string
}

type ExchangeResult struct {
	APIKey   string `json:"apiKey"`
	APIKeyID string `json:"apiKeyId"`
}

type PartnerCheckoutSession struct {
	ID             string        `json:"id"`
	SessionToken   string        `json:"sessionToken"`
	PartnerID      string        `json:"partnerId"`
	PartnerName    *string       `json:"partnerName"`
	UserID         *int64        `json:"userId"`
	AmountUSDMinor *int64        `json:"amountUsdMinor"`
	Status         SessionStatus `json:"status"`
	IssuedKeyID    *string       `json:"issuedKeyId"`
	ReturnURL      *string       `json:"returnUrl"`
}

type PayResult struct {
	Checkout        PaymentSession         `json:"checkout"`
	PartnerCheckout PartnerCheckoutSession `json:"partnerCheckout"`
}

type PaymentMethod string

type PaymentSession struct {
	ProviderSessionID string `json:"providerSessionId"`
	PayURL            string `json:"payUrl"`
}

type ProvisionedKey struct {
	APIKey   string
	APIKeyID string
	BaseURL  string
	Model    string
}

type SessionStatus string

type Status string

type StreamTopUpByKeyOptions struct {
	APIKey           string // raw key that owns the account (required)
	AmountUSD        string // dollars; parsed via ParseAmountUSD (min $20)
	InferenceBaseURL string // validated key endpoint; empty uses ResolveEndpoints
	AutoTopUp        bool   // enroll the account in automatic top-up at checkout time
	PartnerID        string
	PartnerName      string
	NoOpen           bool

	// ResumeSessionToken resumes an existing partner-checkout session instead of
	// opening a new one. PaymentSessionID is the idempotency handle carried across
	// retries so a re-issued pay returns the same checkout, never a second charge.
	ResumeSessionToken string
	PaymentSessionID   string // required; generate once via NewPaymentSessionID, reuse on retry

	HTTPClient  *http.Client
	OpenBrowser func(string) error
	OnStatus    func(Status, string)
	OnSession   func(sessionToken string)
}

type StreamTopUpOptions struct {
	SessionToken string // user session bearer (required)
	AmountUSD    string // dollars; parsed via ParseAmountUSD (min $20)
	// InferenceBaseURL pins key-bound billing and the returned provider profile to
	// the endpoint that validated the credential. Empty uses ResolveEndpoints.
	InferenceBaseURL string
	Method           PaymentMethod
	AutoTopUp        bool // enroll the account in automatic top-up at checkout time
	Exchange         bool // exchange the paid session into a new key (new accounts)
	Model            string
	PartnerID        string
	PartnerName      string
	NoOpen           bool

	// ResumeSessionToken, when set, resumes that partner-checkout session instead
	// of opening a new one, so a retry after an ambiguous failure (a lost Pay/
	// Exchange response) can never double-charge or strand an already-paid session.
	ResumeSessionToken string
	// PaymentSessionID is generated once per amount/auto-top-up intent and reused
	// on retry so an ambiguous Pay response cannot create a second checkout.
	PaymentSessionID string

	HTTPClient  *http.Client
	OpenBrowser func(string) error
	OnStatus    func(Status, string)
	// OnSession reports the partner-checkout session token the moment it exists (a
	// fresh CreateSession, or the resumed one), so the caller can retain it and
	// resume on a later failure. It fires with "" when the session is dead and the
	// retained token should be dropped (a fresh run must start over).
	OnSession func(sessionToken string)
}

type TopUpByKeyRequest struct {
	SessionToken     string
	AmountUSDMinor   int
	PaymentSessionID string // idempotency: same id → same checkout, never a second charge
	AutoTopUp        bool   // enroll the account in automatic top-up at checkout (card, saved off-session)
	SuccessURL       string
	CancelURL        string
}

type TopUpByKeyResult struct {
	Checkout PaymentSession `json:"checkout"`
}