// RAINBOND, Application Management Platform
// Copyright (C) 2014-2017 Goodrain Co., Ltd.

// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version. For any non-GPL usage of Rainbond,
// one or multiple Commercial Licenses authorized by Goodrain Co., Ltd.
// must be obtained first.

// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.

package builder

import (
	"fmt"
	"github.com/goodrain/rainbond/util/constants"
	"github.com/sirupsen/logrus"
	"os"
	"path"
	"runtime"
	"strings"
)

func init() {
	if os.Getenv("BUILD_IMAGE_REPOSTORY_DOMAIN") != "" {
		REGISTRYDOMAIN = os.Getenv("BUILD_IMAGE_REPOSTORY_DOMAIN")
	}
	if os.Getenv("BUILD_IMAGE_REPOSTORY_USER") != "" {
		REGISTRYUSER = os.Getenv("BUILD_IMAGE_REPOSTORY_USER")
	}
	if os.Getenv("BUILD_IMAGE_REPOSTORY_PASS") != "" {
		REGISTRYPASS = os.Getenv("BUILD_IMAGE_REPOSTORY_PASS")
	}
	if os.Getenv("REGISTRY_MIRRORS") != "" {
		SetRegistryMirrors(os.Getenv("REGISTRY_MIRRORS"))
	}
	arch := runtime.GOARCH
	RUNNERIMAGENAME = fmt.Sprintf("%s:latest-%s", "/runner", arch)
	if os.Getenv("RUNNER_IMAGE_NAME") != "" {
		RUNNERIMAGENAME = os.Getenv("RUNNER_IMAGE_NAME")
	}
	RUNNERIMAGENAME = path.Join(REGISTRYDOMAIN, RUNNERIMAGENAME)
	BUILDERIMAGENAME = fmt.Sprintf("%s:latest-%s", "/builder", arch)
	if os.Getenv("BUILDER_IMAGE_NAME") != "" {
		BUILDERIMAGENAME = os.Getenv("BUILDER_IMAGE_NAME")
	}

	BUILDERIMAGENAME = path.Join(REGISTRYDOMAIN, BUILDERIMAGENAME)
	if os.Getenv("ABROAD") != "" {
		ONLINEREGISTRYDOMAIN = "docker.io/rainbond"
	}
	if release := os.Getenv("RELEASE_DESC"); release != "" {
		releaseList := strings.Split(release, "-")
		if len(releaseList) > 0 {
			CIVERSION = fmt.Sprintf("%v-%v", releaseList[0], releaseList[1])
		}
	}

}

// GetImageUserInfoV2 -
func GetImageUserInfoV2(domain, user, pass string) (string, string) {
	logrus.Debugf("domain: %s, user: %s, pass: %s", domain, user, pass)
	if user != "" && pass != "" {
		return user, pass
	}
	if strings.HasPrefix(domain, REGISTRYDOMAIN) {
		return REGISTRYUSER, REGISTRYPASS
	}
	return "", ""
}

// GetImageRepo -
func GetImageRepo(imageRepo string) string {
	if imageRepo == "" {
		return REGISTRYDOMAIN
	}
	return imageRepo
}

// REGISTRYDOMAIN REGISTRY_DOMAIN
var REGISTRYDOMAIN = constants.DefImageRepository

// REGISTRYUSER REGISTRY USER NAME
var REGISTRYUSER = ""

// REGISTRYPASS REGISTRY PASSWORD
var REGISTRYPASS = ""

// REGISTRYMIRRORS registry mirrors used when pulling docker.io base images in
// Dockerfile builds. Empty (the default) means no mirror is configured and the
// generated buildkitd.toml is identical to the historical content.
var REGISTRYMIRRORS []string

// SetRegistryMirrors parses a comma-separated mirror list into REGISTRYMIRRORS.
// Empty / whitespace-only entries are dropped. An empty input resets the value
// to nil so the default (no-mirror) behaviour is preserved.
func SetRegistryMirrors(raw string) {
	mirrors := make([]string, 0)
	for _, m := range strings.Split(raw, ",") {
		m = strings.TrimSpace(m)
		if m == "" {
			continue
		}
		mirrors = append(mirrors, m)
	}
	if len(mirrors) == 0 {
		REGISTRYMIRRORS = nil
		return
	}
	REGISTRYMIRRORS = mirrors
}

// RUNNERIMAGENAME runner image name
var RUNNERIMAGENAME string

// BUILDERIMAGENAME builder image name
var BUILDERIMAGENAME string

// ONLINEREGISTRYDOMAIN online REGISTRY_DOMAIN
var ONLINEREGISTRYDOMAIN = constants.DefOnlineImageRepository

// GetBuilderImage GetBuilderImage
func GetBuilderImage(brVersion string) string {
	arch := runtime.GOARCH
	if brVersion == "" {
		brVersion = CIVERSION
	}

	return fmt.Sprintf("%s:%s-%s", path.Join(ONLINEREGISTRYDOMAIN, "builder"), brVersion, arch)
}

// GetRunnerImage GetRunnerImage
func GetRunnerImage(brVersion string) string {
	arch := runtime.GOARCH
	if brVersion == "" {
		brVersion = CIVERSION
	}

	return fmt.Sprintf("%s:%s-%s", path.Join(ONLINEREGISTRYDOMAIN, "runner"), brVersion, arch)
}

// CIVERSION -
var CIVERSION = "stable"