libhoclient

type AccessTokenBuildAPICreds struct {
	AccessToken   string
	UserProjectID string
}

type ApiCallError struct {
	opapi.ErrorMsg

	HTTPStatusCode int
}

type BuildAPICreds interface {
	ApplyToHTTPRequest(rb *HTTPRequestBuilder)
}

type ConnectWebRTCOpts struct {
	LocalICEConfig *wclient.ICEConfig
}

type CreateBugReportOpts struct {
	IncludeADBBugReport bool
}

type ExpBackOffOptions struct {
	InitialDuration     time.Duration
	RandomizationFactor float64
	Multiplier          float64
	MaxElapsedTime      time.Duration
}

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

type FilesUploader struct {
	HTTPHelper     HTTPHelper
	UploadEndpoint string
	DumpOut        io.Writer
	UploadOptions
}

type HTTPHelper struct {
	Client            *http.Client
	RootEndpoint      string
	Dumpster          io.Writer
	AccessToken       string
	HTTPBasicUsername string
}

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

type HostOrchestratorClient interface {
	InstancesClient
	UserArtifactsClient
	ImageDirectoriesClient
	InstanceOperationsClient
	InstanceConnectionsClient
	OperationsClient
	SnapshotsClient
}

type HostOrchestratorClientImpl struct {
	HTTPHelper HTTPHelper
	ProxyURL   string
}

type ImageDirectoriesClient interface {
	// Create an empty image directory.
	CreateImageDirectory() (*hoapi.Operation, error)
	// List all image directories.
	ListImageDirectories() (*hoapi.ListImageDirectoriesResponse, error)
	// Update image directory to include uploaded or extracted user artifact at
	// Host Orchestrator.
	UpdateImageDirectoryWithUserArtifact(id, filename string) (*hoapi.Operation, error)
	// Delete the specified image directory.
	DeleteImageDirectory(id string) (*hoapi.Operation, error)
}

type InstanceConnectionsClient interface {
	// Creates a webRTC connection to a device running in this host.
	ConnectWebRTC(device string, observer wclient.Observer, logger io.Writer, opts ConnectWebRTCOpts) (*wclient.Connection, error)
	// Connect to ADB WebSocket endpoint.
	ConnectADBWebSocket(device string) (*websocket.Conn, error)
}

type InstanceOperationsClient interface {
	// Create cvd bugreport.
	CreateBugReport(group string, opts CreateBugReportOpts, dst io.Writer) error
	// Powerwash the device.
	Powerwash(groupName, instanceName string) error
	// Stop the device.
	Stop(groupName, instanceName string) error
	// Press power button.
	Powerbtn(groupName, instanceName string) error
	// Start the device.
	Start(groupName, instanceName string, req *hoapi.StartCVDRequest) error
	// Create device snapshot.
	CreateSnapshot(groupName, instanceName string, req *hoapi.CreateSnapshotRequest) (*hoapi.CreateSnapshotResponse, error)
}

type InstancesClient interface {
	// Lists currently running devices.
	ListCVDs() ([]*hoapi.CVD, error)
	// Calls cvd fetch in the remote host, the downloaded artifacts can be used to create a CVD later.
	// If not empty, the provided credentials will be used by the host orchestrator to access the build api.
	FetchArtifacts(req *hoapi.FetchArtifactsRequest, creds BuildAPICreds) (*hoapi.FetchArtifactsResponse, error)
	// Create a new device with artifacts from the build server or previously uploaded by the user.
	// If not empty, the provided credentials will be used to download necessary artifacts from the build api.
	CreateCVD(req *hoapi.CreateCVDRequest, creds BuildAPICreds) (*hoapi.CreateCVDResponse, error)
	CreateCVDOp(req *hoapi.CreateCVDRequest, creds BuildAPICreds) (*hoapi.Operation, error)
	// Deletes an existing cvd instance.
	DeleteCVD(id string) error
	// Forcefully deletes all existing cvd instances and cleans up the host's state.
	Reset() error
}

type JournalGatewaydClient struct {
	HTTPHelper HTTPHelper
	ProxyURL   string
}

type OperationsClient interface {
	// Wait for an operation, `result` will be populated with the relevant operation's result object.
	WaitForOperation(name string, result any) error
}

type RetryOptions struct {
	StatusCodes []int
	RetryDelay  time.Duration
	// Keep retrying until the MaxWait threshold is reached out
	MaxWait time.Duration
}

type ServiceLogsClient interface {
	PullHOLogs(dst io.Writer) error
}

type SnapshotsClient interface {
	// Create device snapshot.
	DeleteSnapshot(id string) error
}

type StatusCodeError struct {
	Code int
}

type UploadOptions struct {
	BackOffOpts    ExpBackOffOptions
	ChunkSizeBytes int64
	NumWorkers     int
}

type UserArtifactsClient interface {
	// Upload artifact into the artifacts repository.
	// Artifacts are identified by their SHA256 checksum in the artifacts repository
	UploadArtifact(filename string) error
	// Extract artifact into the artifacts repository.
	// Artifacts are identified by their SHA256 checksum in the artifacts repository
	ExtractArtifact(filename string) (*hoapi.Operation, error)
}