environment.go

package envs

import (
	"fmt"
	"os"
	"strings"

	"github.com/goharbor/harbor/tests/apitests/api-testing/client"
)

//Environment keeps the testing env info
type Environment struct {
	Protocol       string //env var: HTTP_PROTOCOL
	Hostname       string //env var: TESTING_ENV_HOSTNAME
	Account        string //env var: TESTING_ENV_ACCOUNT
	Password       string //env var: TESTING_ENV_PASSWORD
	Admin          string //env var: TESTING_ENV_ADMIN
	AdminPass      string //env var: TESTING_ENV_ADMIN_PASS
	TestingProject string //env var: TESTING_PROJECT_NAME
	ImageName      string //env var: TESTING_IMAGE_NAME
	ImageTag       string //env var: TESTING_IMAGE_TAG
	CAFile         string //env var: CA_FILE_PATH
	CertFile       string //env var: CERT_FILE_PATH
	KeyFile        string //env var: KEY_FILE_PATH
	ProxyURL       string //env var: http_proxy, https_proxy, HTTP_PROXY, HTTPS_PROXY

	//API client
	HTTPClient *client.APIClient

	//Docker client
	DockerClient *client.DockerClient

	//Initialize status
	loaded bool
}

//Load test env info
func (env *Environment) Load() error {
	host := os.Getenv("TESTING_ENV_HOSTNAME")
	if isNotEmpty(host) {
		env.Hostname = host
	}

	account := os.Getenv("TESTING_ENV_ACCOUNT")
	if isNotEmpty(account) {
		env.Account = account
	}

	pwd := os.Getenv("TESTING_ENV_PASSWORD")
	if isNotEmpty(pwd) {
		env.Password = pwd
	}

	admin := os.Getenv("TESTING_ENV_ADMIN")
	if isNotEmpty(admin) {
		env.Admin = admin
	}

	adminPwd := os.Getenv("TESTING_ENV_ADMIN_PASS")
	if isNotEmpty(adminPwd) {
		env.AdminPass = adminPwd
	}

	pro := os.Getenv("TESTING_PROJECT_NAME")
	if isNotEmpty(pro) {
		env.TestingProject = pro
	}

	imgName := os.Getenv("TESTING_IMAGE_NAME")
	if isNotEmpty(imgName) {
		env.ImageName = imgName
	}

	imgTag := os.Getenv("TESTING_IMAGE_TAG")
	if isNotEmpty(imgTag) {
		env.ImageTag = imgTag
	}

	protocol := os.Getenv("HTTP_PROTOCOL")
	if isNotEmpty(protocol) {
		env.Protocol = protocol
	}

	caFile := os.Getenv("CA_FILE_PATH")
	if isNotEmpty(caFile) {
		env.CAFile = caFile
	}

	keyFile := os.Getenv("KEY_FILE_PATH")
	if isNotEmpty(keyFile) {
		env.KeyFile = keyFile
	}

	certFile := os.Getenv("CERT_FILE_PATH")
	if isNotEmpty(certFile) {
		env.CertFile = certFile
	}

	proxyEnvVar := "https_proxy"
	if env.Protocol == "http" {
		proxyEnvVar = "http_proxy"
	}
	proxyURL := os.Getenv(proxyEnvVar)
	if !isNotEmpty(proxyURL) {
		proxyURL = os.Getenv(strings.ToUpper(proxyEnvVar))
	}
	if isNotEmpty(proxyURL) {
		env.ProxyURL = proxyURL
	}

	if !env.loaded {
		cfg := client.APIClientConfig{
			Username: env.Admin,
			Password: env.AdminPass,
			CaFile:   env.CAFile,
			CertFile: env.CertFile,
			KeyFile:  env.KeyFile,
			Proxy:    env.ProxyURL,
		}

		httpClient, err := client.NewAPIClient(cfg)
		if err != nil {
			return err
		}
		env.HTTPClient = httpClient
		env.DockerClient = &client.DockerClient{}

		env.loaded = true
	}

	return nil
}

//RootURI : The root URI like https://<hostname>
func (env *Environment) RootURI() string {
	return fmt.Sprintf("%s://%s", env.Protocol, env.Hostname)
}

func isNotEmpty(str string) bool {
	return len(strings.TrimSpace(str)) > 0
}