package azblob

import (

	"context"

	"io"

	"net/url"

	"github.com/Azure/azure-pipeline-go/pipeline"

)

const (

	// BlockBlobMaxUploadBlobBytes indicates the maximum number of bytes that can be sent in a call to Upload.

	BlockBlobMaxUploadBlobBytes = 256 * 1024 * 1024 // 256MB

	// BlockBlobMaxStageBlockBytes indicates the maximum number of bytes that can be sent in a call to StageBlock.

	BlockBlobMaxStageBlockBytes = 4000 * 1024 * 1024 // 4000MiB

	// BlockBlobMaxBlocks indicates the maximum number of blocks allowed in a block blob.

	BlockBlobMaxBlocks = 50000

)

// BlockBlobURL defines a set of operations applicable to block blobs.

type BlockBlobURL struct {

	BlobURL

	bbClient blockBlobClient

}

// NewBlockBlobURL creates a BlockBlobURL object using the specified URL and request policy pipeline.

func NewBlockBlobURL(url url.URL, p pipeline.Pipeline) BlockBlobURL {

	blobClient := newBlobClient(url, p)

	bbClient := newBlockBlobClient(url, p)

	return BlockBlobURL{BlobURL: BlobURL{blobClient: blobClient}, bbClient: bbClient}

}

// WithPipeline creates a new BlockBlobURL object identical to the source but with the specific request policy pipeline.

func (bb BlockBlobURL) WithPipeline(p pipeline.Pipeline) BlockBlobURL {

	return NewBlockBlobURL(bb.blobClient.URL(), p)

}

// WithSnapshot creates a new BlockBlobURL object identical to the source but with the specified snapshot timestamp.

// Pass "" to remove the snapshot returning a URL to the base blob.

func (bb BlockBlobURL) WithSnapshot(snapshot string) BlockBlobURL {

	p := NewBlobURLParts(bb.URL())

	p.Snapshot = snapshot

	return NewBlockBlobURL(p.URL(), bb.blobClient.Pipeline())

}

// WithVersionID creates a new BlockBlobURRL object identical to the source but with the specified version id.

// Pass "" to remove the snapshot returning a URL to the base blob.

func (bb BlockBlobURL) WithVersionID(versionId string) BlockBlobURL {

	p := NewBlobURLParts(bb.URL())

	p.VersionID = versionId

	return NewBlockBlobURL(p.URL(), bb.blobClient.Pipeline())

}

func (bb BlockBlobURL) GetAccountInfo(ctx context.Context) (*BlobGetAccountInfoResponse, error) {

	return bb.blobClient.GetAccountInfo(ctx)

}

// Upload creates a new block blob or overwrites an existing block blob.

// Updating an existing block blob overwrites any existing metadata on the blob. Partial updates are not

// supported with Upload; the content of the existing blob is overwritten with the new content. To

// perform a partial update of a block blob, use StageBlock and CommitBlockList.

// This method panics if the stream is not at position 0.

// Note that the http client closes the body stream after the request is sent to the service.

// For more information, see https://docs.microsoft.com/rest/api/storageservices/put-blob.

func (bb BlockBlobURL) Upload(ctx context.Context, body io.ReadSeeker, h BlobHTTPHeaders, metadata Metadata, ac BlobAccessConditions, tier AccessTierType, blobTagsMap BlobTagsMap, cpk ClientProvidedKeyOptions) (*BlockBlobUploadResponse, error) {

	ifModifiedSince, ifUnmodifiedSince, ifMatchETag, ifNoneMatchETag := ac.ModifiedAccessConditions.pointers()

	count, err := validateSeekableStreamAt0AndGetCount(body)

	blobTagsString := SerializeBlobTagsHeader(blobTagsMap)

	if err != nil {

		return nil, err

	}

	return bb.bbClient.Upload(ctx, body, count, nil, nil,

		&h.ContentType, &h.ContentEncoding, &h.ContentLanguage, h.ContentMD5,

		&h.CacheControl, metadata, ac.LeaseAccessConditions.pointers(), &h.ContentDisposition,

		cpk.EncryptionKey, cpk.EncryptionKeySha256, cpk.EncryptionAlgorithm, // CPK-V

		cpk.EncryptionScope, // CPK-N

		tier, ifModifiedSince, ifUnmodifiedSince, ifMatchETag, ifNoneMatchETag,

		nil, // Blob ifTags

		nil,

		blobTagsString, // Blob tags

	)

}

// StageBlock uploads the specified block to the block blob's "staging area" to be later committed by a call to CommitBlockList.

// Note that the http client closes the body stream after the request is sent to the service.

// For more information, see https://docs.microsoft.com/rest/api/storageservices/put-block.

func (bb BlockBlobURL) StageBlock(ctx context.Context, base64BlockID string, body io.ReadSeeker, ac LeaseAccessConditions, transactionalMD5 []byte, cpk ClientProvidedKeyOptions) (*BlockBlobStageBlockResponse, error) {

	count, err := validateSeekableStreamAt0AndGetCount(body)

	if err != nil {

		return nil, err

	}

	return bb.bbClient.StageBlock(ctx, base64BlockID, count, body, transactionalMD5, nil, nil, ac.pointers(),

		cpk.EncryptionKey, cpk.EncryptionKeySha256, cpk.EncryptionAlgorithm, // CPK-V

		cpk.EncryptionScope, // CPK-N

		nil)

}

// StageBlockFromURL copies the specified block from a source URL to the block blob's "staging area" to be later committed by a call to CommitBlockList.

// If count is CountToEnd (0), then data is read from specified offset to the end.

// For more information, see https://docs.microsoft.com/en-us/rest/api/storageservices/put-block-from-url.

func (bb BlockBlobURL) StageBlockFromURL(ctx context.Context, base64BlockID string, sourceURL url.URL, offset int64, count int64, destinationAccessConditions LeaseAccessConditions, sourceAccessConditions ModifiedAccessConditions, cpk ClientProvidedKeyOptions) (*BlockBlobStageBlockFromURLResponse, error) {

	sourceIfModifiedSince, sourceIfUnmodifiedSince, sourceIfMatchETag, sourceIfNoneMatchETag := sourceAccessConditions.pointers()

	return bb.bbClient.StageBlockFromURL(ctx, base64BlockID, 0, sourceURL.String(), httpRange{offset: offset, count: count}.pointers(), nil, nil, nil,

		cpk.EncryptionKey, cpk.EncryptionKeySha256, cpk.EncryptionAlgorithm, // CPK

		cpk.EncryptionScope, // CPK-N

		destinationAccessConditions.pointers(), sourceIfModifiedSince, sourceIfUnmodifiedSince, sourceIfMatchETag, sourceIfNoneMatchETag, nil)

}

// CommitBlockList writes a blob by specifying the list of block IDs that make up the blob.

// In order to be written as part of a blob, a block must have been successfully written

// to the server in a prior PutBlock operation. You can call PutBlockList to update a blob

// by uploading only those blocks that have changed, then committing the new and existing

// blocks together. Any blocks not specified in the block list and permanently deleted.

// For more information, see https://docs.microsoft.com/rest/api/storageservices/put-block-list.

func (bb BlockBlobURL) CommitBlockList(ctx context.Context, base64BlockIDs []string, h BlobHTTPHeaders, metadata Metadata, ac BlobAccessConditions, tier AccessTierType, blobTagsMap BlobTagsMap, cpk ClientProvidedKeyOptions) (*BlockBlobCommitBlockListResponse, error) {

	ifModifiedSince, ifUnmodifiedSince, ifMatchETag, ifNoneMatchETag := ac.ModifiedAccessConditions.pointers()

	blobTagsString := SerializeBlobTagsHeader(blobTagsMap)

	return bb.bbClient.CommitBlockList(ctx, BlockLookupList{Latest: base64BlockIDs}, nil,

		&h.CacheControl, &h.ContentType, &h.ContentEncoding, &h.ContentLanguage, h.ContentMD5, nil, nil,

		metadata, ac.LeaseAccessConditions.pointers(), &h.ContentDisposition,

		cpk.EncryptionKey, cpk.EncryptionKeySha256, cpk.EncryptionAlgorithm, // CPK

		cpk.EncryptionScope, // CPK-N

		tier,

		ifModifiedSince, ifUnmodifiedSince, ifMatchETag, ifNoneMatchETag,

		nil, // Blob ifTags

		nil,

		blobTagsString, // Blob tags

	)

}

// GetBlockList returns the list of blocks that have been uploaded as part of a block blob using the specified block list filter.

// For more information, see https://docs.microsoft.com/rest/api/storageservices/get-block-list.

func (bb BlockBlobURL) GetBlockList(ctx context.Context, listType BlockListType, ac LeaseAccessConditions) (*BlockList, error) {

	return bb.bbClient.GetBlockList(ctx, listType, nil, nil, ac.pointers(),

		nil, // Blob ifTags

		nil)

}

// CopyFromURL synchronously copies the data at the source URL to a block blob, with sizes up to 256 MB.

// For more information, see https://docs.microsoft.com/en-us/rest/api/storageservices/copy-blob-from-url.

func (bb BlockBlobURL) CopyFromURL(ctx context.Context, source url.URL, metadata Metadata, srcac ModifiedAccessConditions, dstac BlobAccessConditions, srcContentMD5 []byte, tier AccessTierType, blobTagsMap BlobTagsMap) (*BlobCopyFromURLResponse, error) {

	srcIfModifiedSince, srcIfUnmodifiedSince, srcIfMatchETag, srcIfNoneMatchETag := srcac.pointers()

	dstIfModifiedSince, dstIfUnmodifiedSince, dstIfMatchETag, dstIfNoneMatchETag := dstac.ModifiedAccessConditions.pointers()

	dstLeaseID := dstac.LeaseAccessConditions.pointers()

	blobTagsString := SerializeBlobTagsHeader(blobTagsMap)

	return bb.blobClient.CopyFromURL(ctx, source.String(), nil, metadata, tier,

		srcIfModifiedSince, srcIfUnmodifiedSince,

		srcIfMatchETag, srcIfNoneMatchETag,

		dstIfModifiedSince, dstIfUnmodifiedSince,

		dstIfMatchETag, dstIfNoneMatchETag,

		nil, // Blob ifTags

		dstLeaseID, nil, srcContentMD5,

		blobTagsString, // Blob tags

		nil,            // seal Blob

	)

}