endpoint.go

package storage

import (
	"fmt"
	"strings"

	"github.com/deis/builder/pkg/sys"
)

const (
	minioHostEnvVar        = "DEIS_MINIO_SERVICE_HOST"
	minioPortEnvVar        = "DEIS_MINIO_SERVICE_PORT"
	outsideStorageEndpoint = "DEIS_OUTSIDE_STORAGE"
)

var (
	errNoStorageConfig = fmt.Errorf(
		"no storage config variables found (%s:%s or %s)",
		minioHostEnvVar,
		minioPortEnvVar,
		outsideStorageEndpoint,
	)
)

func stripScheme(str string) string {
	schemes := []string{"http://", "https://"}
	for _, scheme := range schemes {
		if strings.HasPrefix(str, scheme) {
			str = str[len(scheme):]
		}
	}
	return str
}

// Endpoint represents all the details about a storage endpoint.
type Endpoint struct {
	// URLStr is the url string, stripped of its scheme.
	URLStr string
	// Secure determines if TLS should be used (e.g. a "https://" scheme).
	Secure bool
}

// FullURL returns the full URL string of an endpoint, including its URL scheme, based on e.Secure.
func (e *Endpoint) FullURL() string {
	if e.Secure {
		return fmt.Sprintf("https://%s", e.URLStr)
	}
	return fmt.Sprintf("http://%s", e.URLStr)
}

func getEndpoint(env sys.Env) (*Endpoint, error) {
	mHost := env.Get(minioHostEnvVar)
	mPort := env.Get(minioPortEnvVar)
	S3EP := env.Get(outsideStorageEndpoint)
	if S3EP != "" {
		return &Endpoint{URLStr: stripScheme(S3EP), Secure: true}, nil
	} else if mHost != "" && mPort != "" {
		return &Endpoint{URLStr: fmt.Sprintf("%s:%s", mHost, mPort), Secure: false}, nil
	} else {
		return nil, errNoStorageConfig
	}
}