regclient/types/blob/blob.go

// Package blob is the underlying type for pushing and pulling blobs
package blob

import (
	"io"
	"net/http"

	"github.com/regclient/regclient/types"
	v1 "github.com/regclient/regclient/types/oci/v1"
	"github.com/regclient/regclient/types/ref"
)

// Blob interface is used for returning blobs
type Blob interface {
	Common
	RawBody() ([]byte, error)
}

type blobConfig struct {
	desc    types.Descriptor
	header  http.Header
	image   *v1.Image
	r       ref.Ref
	rdr     io.Reader
	resp    *http.Response
	rawBody []byte
}

type Opts func(*blobConfig)

// WithDesc specifies the descriptor associated with the blob
func WithDesc(d types.Descriptor) Opts {
	return func(bc *blobConfig) {
		bc.desc = d
	}
}

// WithHeader defines the headers received when pulling a blob
func WithHeader(header http.Header) Opts {
	return func(bc *blobConfig) {
		bc.header = header
	}
}

// WithImage provides the OCI Image config needed for config blobs
func WithImage(image v1.Image) Opts {
	return func(bc *blobConfig) {
		bc.image = &image
	}
}

// WithRawBody defines the raw blob contents for OCIConfig
func WithRawBody(raw []byte) Opts {
	return func(bc *blobConfig) {
		bc.rawBody = raw
	}
}

// WithReader defines the reader for a new blob
func WithReader(rc io.Reader) Opts {
	return func(bc *blobConfig) {
		bc.rdr = rc
	}
}

// WithRef specifies the reference where the blob was pulled from
func WithRef(r ref.Ref) Opts {
	return func(bc *blobConfig) {
		bc.r = r
	}
}

// WithResp includes the http response, which is used to extract the headers and reader
func WithResp(resp *http.Response) Opts {
	return func(bc *blobConfig) {
		bc.resp = resp
		if bc.header == nil && resp != nil {
			bc.header = resp.Header
		}
	}
}