// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
// Author: arnej

package startcbinary

import (
	"os"
	"strings"

	"github.com/vespa-engine/vespa/client/go/internal/admin/envvars"
	"github.com/vespa-engine/vespa/client/go/internal/admin/prog"
	"github.com/vespa-engine/vespa/client/go/internal/vespa"
)

func configureCommonEnv(spec *prog.Spec) {
	os.Unsetenv(envvars.LD_PRELOAD)
	spec.Setenv(envvars.STD_THREAD_PREVENT_TRY_CATCH, "true")
	spec.Setenv(envvars.GLIBCXX_FORCE_NEW, "1")
	spec.Setenv(envvars.LD_LIBRARY_PATH, vespa.FindHome()+"/lib64:/opt/vespa-deps/lib64")
	spec.Setenv(envvars.MALLOC_ARENA_MAX, "1")

	// fallback from old env.vars:
	spec.ConsiderEnvFallback(envvars.VESPA_USE_HUGEPAGES_LIST, envvars.HUGEPAGES_LIST)
	spec.ConsiderEnvFallback(envvars.VESPA_USE_MADVISE_LIST, envvars.MADVISE_LIST)
	spec.ConsiderEnvFallback(envvars.VESPA_USE_VESPAMALLOC, envvars.VESPAMALLOC_LIST)
	spec.ConsiderEnvFallback(envvars.VESPA_USE_VESPAMALLOC_D, envvars.VESPAMALLOCD_LIST)
	spec.ConsiderEnvFallback(envvars.VESPA_USE_VESPAMALLOC_DST, envvars.VESPAMALLOCDST_LIST)
	spec.ConsiderEnvFallback(envvars.VESPA_USE_NO_VESPAMALLOC, envvars.NO_VESPAMALLOC_LIST)
	// other fallbacks:
	spec.ConsiderFallback(envvars.ROOT, vespa.FindHome())
	spec.ConsiderFallback(envvars.VESPA_USER, vespa.FindVespaUser())
	spec.ConsiderFallback(envvars.VESPA_USE_HUGEPAGES_LIST, "all")
	spec.ConsiderFallback(envvars.VESPA_USE_VESPAMALLOC, "all")
	spec.ConsiderFallback(envvars.VESPA_USE_NO_VESPAMALLOC, strings.Join([]string{
		"vespa-rpc-invoke",
		"vespa-get-config",
		"vespa-sentinel-cmd",
		"vespa-route",
		"vespa-proton-cmd",
		"vespa-configproxy-cmd",
		"vespa-config-status",
	}, " "))

}

func configurePath(spec *prog.Spec) {
	// Prefer newer gdb and pstack:
	prependPath("/opt/rh/gcc-toolset-12/root/usr/bin", spec)
	// Maven is needed for tester applications:
	prependPath(vespa.FindHome()+"/local/maven/bin", spec)
	prependPath(vespa.FindHome()+"/bin64", spec)
	prependPath(vespa.FindHome()+"/bin", spec)
	// how to find the "java" program?
	// should be available in $VESPA_HOME/bin or JAVA_HOME
	if javaHome := spec.Getenv(envvars.JAVA_HOME); javaHome != "" {
		prependPath(javaHome+"/bin", spec)
	}
}