package cnb

import (
	"os"
	"path"

	"github.com/goodrain/rainbond/builder"
	"github.com/goodrain/rainbond/builder/parser/code"
	"github.com/goodrain/rainbond/util/constants"
	"github.com/sirupsen/logrus"
)

const (
	// DefaultCNBBuilder is the default online CNB builder image
	DefaultCNBBuilder = constants.DefOnlineImageRepository + "/" + constants.CNBBuilderImageName
	// DefaultCNBRunImage is the default online CNB run image
	DefaultCNBRunImage = constants.DefOnlineImageRepository + "/" + constants.CNBRunImageName
	// DefaultPHPCNBBuilder is the default Jammy Full builder image for PHP CNB builds.
	DefaultPHPCNBBuilder = constants.DefOnlineImageRepository + "/" + constants.PHPCNBBuilderImageName
	// DefaultPHPCNBRunImage is the default Jammy Full run image for PHP CNB builds.
	DefaultPHPCNBRunImage = constants.DefOnlineImageRepository + "/" + constants.PHPCNBRunImageName
	// CNBLifecycleCreatorPath is the path to the lifecycle creator binary
	CNBLifecycleCreatorPath = "/lifecycle/creator"

	// Short image names for constructing internal registry references
	cnbBuilderShortName = constants.CNBBuilderImageName
	cnbRunShortName     = constants.CNBRunImageName
	phpBuilderShortName = constants.PHPCNBBuilderImageName
	phpRunShortName     = constants.PHPCNBRunImageName
)

// isOfflineMode checks whether the cluster is in offline/air-gapped mode
// by looking for the same marker file used by getDependencyMirror.
func isOfflineMode() bool {
	_, err := os.Stat(offlineMirrorMarker)
	return err == nil
}

// GetCNBBuilderImage returns the CNB builder image reference.
// Priority: env var > offline (REGISTRYDOMAIN) > default online URL.
func GetCNBBuilderImage() string {
	if v := os.Getenv("CNB_BUILDER_IMAGE"); v != "" {
		return v
	}
	if isOfflineMode() {
		img := path.Join(builder.REGISTRYDOMAIN, cnbBuilderShortName)
		logrus.Infof("Offline mode: using CNB builder image from internal registry: %s", img)
		return img
	}
	return DefaultCNBBuilder
}

// GetCNBRunImage returns the CNB run image reference.
// Priority: env var > offline (REGISTRYDOMAIN) > default online URL.
func GetCNBRunImage() string {
	if v := os.Getenv("CNB_RUN_IMAGE"); v != "" {
		return v
	}
	if isOfflineMode() {
		img := path.Join(builder.REGISTRYDOMAIN, cnbRunShortName)
		logrus.Infof("Offline mode: using CNB run image from internal registry: %s", img)
		return img
	}
	return DefaultCNBRunImage
}

func GetCNBBuilderImageForLanguage(lang code.Lang) string {
	if lang == code.PHP {
		if v := os.Getenv("CNB_BUILDER_IMAGE"); v != "" {
			return v
		}
		if isOfflineMode() {
			img := path.Join(builder.REGISTRYDOMAIN, phpBuilderShortName)
			logrus.Infof("Offline mode: using PHP CNB builder image from internal registry: %s", img)
			return img
		}
		return DefaultPHPCNBBuilder
	}
	return GetCNBBuilderImage()
}

func GetCNBRunImageForLanguage(lang code.Lang) string {
	if lang == code.PHP {
		if v := os.Getenv("CNB_RUN_IMAGE"); v != "" {
			return v
		}
		if isOfflineMode() {
			img := path.Join(builder.REGISTRYDOMAIN, phpRunShortName)
			logrus.Infof("Offline mode: using PHP CNB run image from internal registry: %s", img)
			return img
		}
		return DefaultPHPCNBRunImage
	}
	return GetCNBRunImage()
}