s3_output.go

package awos

import (
	"strconv"

	"github.com/aws/aws-sdk-go/service/s3"
)

type HeadGetObjectOutputWrapper struct {
	getObjectOutput  *s3.GetObjectOutput
	headObjectOutput *s3.HeadObjectOutput
}

func (h *HeadGetObjectOutputWrapper) getContentType() *string {
	if h.getObjectOutput != nil {
		return h.getObjectOutput.ContentType
	}
	return h.headObjectOutput.ContentType
}

func (h *HeadGetObjectOutputWrapper) getContentEncoding() *string {
	if h.getObjectOutput != nil {
		return h.getObjectOutput.ContentEncoding
	}
	return h.headObjectOutput.ContentEncoding
}

func (h *HeadGetObjectOutputWrapper) getContentLength() *string {
	if h.getObjectOutput != nil && h.getObjectOutput.ContentLength != nil {
		clStr := strconv.FormatInt(*h.getObjectOutput.ContentLength, 10)
		return &clStr
	}
	if h.headObjectOutput != nil && h.headObjectOutput.ContentLength != nil {
		clStr := strconv.FormatInt(*h.headObjectOutput.ContentLength, 10)
		return &clStr
	}
	clStr := "0"
	return &clStr
}

func (h *HeadGetObjectOutputWrapper) getContentDisposition() *string {
	if h.getObjectOutput != nil {
		return h.getObjectOutput.ContentDisposition
	}
	return h.headObjectOutput.ContentDisposition
}

func (h *HeadGetObjectOutputWrapper) metaData() map[string]*string {
	if h.getObjectOutput != nil {
		return h.getObjectOutput.Metadata
	}
	return h.headObjectOutput.Metadata
}

func mergeHttpStandardHeaders(output *HeadGetObjectOutputWrapper) map[string]*string {
	res := make(map[string]*string)
	for k, v := range output.metaData() {
		res[k] = v
	}

	res["Content-Length"] = output.getContentLength()
	res["Content-Encoding"] = output.getContentEncoding()
	res["Content-Type"] = output.getContentType()
	res["Content-Disposition"] = output.getContentDisposition()

	return res
}