model

type Buffer struct {
	Id        string            `json:"id"`
	CreatedAt time.Time         `json:"createdAt"`
	Location  string            `json:"location,omitempty"`
	Tags      map[string]string `json:"tags,omitempty"`
	ETag      string            `json:"eTag,omitempty"`

	ExpiresAt *time.Time `json:"expiresAt,omitempty"`
}

type BufferAccess struct {
	Uri string `json:"uri"`
}

type BufferParameters struct {
	Inputs  []string `json:"inputs,omitempty"`
	Outputs []string `json:"outputs,omitempty"`
}

type Cluster struct {
	Name      string     `json:"name"`
	Location  string     `json:"location"`
	NodePools []NodePool `json:"nodePools"`
}

type Codespec struct {
	Kind             string `json:"kind"`
	CodespecMetadata `json:",inline"`
	Buffers          *BufferParameters  `json:"buffers,omitempty"`
	Sockets          []Socket           `json:"sockets,omitempty"`
	Image            string             `json:"image"`
	Command          []string           `json:"command,omitempty"`
	Args             []string           `json:"args,omitempty"`
	WorkingDir       string             `json:"workingDir,omitempty"`
	Env              map[string]string  `json:"env,omitempty"`
	Identity         string             `json:"identity,omitempty"`
	Resources        *CodespecResources `json:"resources,omitempty"`
	MaxReplicas      *int               `json:"maxReplicas,omitempty"`
	Endpoints        map[string]int     `json:"endpoints,omitempty"`
}

type CodespecMetadata struct {
	Name      *string    `json:"name,omitempty"`
	Version   *int       `json:"version,omitempty"`
	CreatedAt *time.Time `json:"createdAt,omitempty"`
}

type CodespecRef struct {
	Named  *NamedCodespecRef  `json:"-"`
	Inline *InlineCodespecRef `json:"-"`
}

type CodespecResources struct {
	Requests *OvercommittableResources `json:"requests,omitempty"`
	Limits   *OvercommittableResources `json:"limits,omitempty"`
	Gpu      *resource.Quantity        `json:"gpu,omitempty"`
	Shm      *resource.Quantity        `json:"shm,omitempty"`
}

type ErrorInfo struct {
	Code        string   `json:"code"`
	Message     string   `json:"message"`
	ApiVersions []string `json:"apiVersions,omitempty"`
}

type ErrorResponse struct {
	Error ErrorInfo `json:"error"`
}

type ExportBuffersRequest struct {
	SourceStorageAccountName   string            `json:"sourceStorageAccountName"`
	DestinationStorageEndpoint string            `json:"destinationStorageEndpoint"`
	Filters                    map[string]string `json:"filters,omitempty"`
	HashIds                    bool              `json:"hashIds,omitempty"`
}

type ImportBuffersRequest struct {
	StorageAccountName string `json:"storageAccountName"`
}

type InlineCodespecRef Codespec

type NamedCodespecRef string

type NodePool struct {
	Name   string `json:"name"`
	VmSize string `json:"vmSize"`
}

type OvercommittableResources struct {
	Cpu    *resource.Quantity `json:"cpu,omitempty"`
	Memory *resource.Quantity `json:"memory,omitempty"`
}

type Page[T any] struct {
	Items    []T    `json:"items"`
	NextLink string `json:"nextLink,omitempty"`
}

type Run struct {
	RunMetadata
	Kind            string            `json:"kind,omitempty"`
	Job             RunCodeTarget     `json:"job,omitempty"`
	Worker          *RunCodeTarget    `json:"worker,omitempty"`
	Cluster         string            `json:"cluster,omitempty"`
	TimeoutSeconds  *int              `json:"timeoutSeconds,omitempty"`
	Tags            map[string]string `json:"tags,omitempty"`
	ETag            string            `json:"eTag,omitempty"`
	BufferAccessTtl string            `json:"bufferAccessTtl,omitempty"`
}

type RunCodeTarget struct {
	Codespec  CodespecRef       `json:"codespec"`
	Buffers   map[string]string `json:"buffers,omitempty"`
	Tags      map[string]string `json:"tags,omitempty"`
	BufferTtl string            `json:"bufferTtl,omitempty"`
	NodePool  string            `json:"nodePool,omitempty"`
	Replicas  int               `json:"replicas,omitempty"`
}

type RunMetadata struct {
	Id           int64             `json:"id,omitempty"`
	Status       *RunStatus        `json:"status,omitempty"`
	StatusReason string            `json:"statusReason,omitempty"`
	RunningCount *int              `json:"runningCount,omitempty"`
	CreatedAt    time.Time         `json:"createdAt,omitempty"`
	StartedAt    *time.Time        `json:"startedAt,omitempty"`
	FinishedAt   *time.Time        `json:"finishedAt,omitempty"`
	Tags         map[string]string `json:"tags,omitempty"`
}

type RunStatus uint

type ServiceMetadata struct {
	RbacEnabled       bool     `json:"rbacEnabled,omitempty"`
	Authority         string   `json:"authority,omitempty"`
	Audience          string   `json:"audience,omitempty"`
	ApiAppUri         string   `json:"serverAppUri,omitempty"`
	ApiAppId          string   `json:"serverAppId,omitempty"`
	CliAppUri         string   `json:"cliAppUri,omitempty"`
	CliAppId          string   `json:"cliAppId,omitempty"`
	Capabilities      []string `json:"capabilities,omitempty"`
	ApiVersions       []string `json:"apiVersions,omitempty"`
	DataPlaneProxy    string   `json:"dataPlaneProxy,omitempty"`    // Only used by tyger-proxy
	TlsCaCertificates string   `json:"tlsCaCertificates,omitempty"` // Only used by tyger-proxy
	StorageEndpoints  []string `json:"storageEndpoints,omitempty"`
}

type Socket struct {
	Port         int    `json:"port"`
	InputBuffer  string `json:"inputBuffer,omitempty"`
	OutputBuffer string `json:"outputBuffer,omitempty"`
}

type StorageAccount struct {
	Name     string `json:"name"`
	Location string `json:"location"`
	Endpoint string `json:"endpoint"`
}