package runtimeflags

import (
	"os"
	"testing"

	"github.com/kandev/kandev/internal/common/config"
	"github.com/kandev/kandev/internal/profiles"
)

func TestApplyStatesToConfigClearsProfileDebugEnvWhenDisabled(t *testing.T) {
	for _, name := range []string{
		"KANDEV_DEBUG_DEV_MODE",
		"KANDEV_DEBUG_PPROF_ENABLED",
		"KANDEV_DEBUG_AGENT_MESSAGES",
	} {
		preserveEnv(t, name)
	}
	_ = os.Unsetenv("KANDEV_DEBUG_PPROF_ENABLED")
	_ = os.Unsetenv("KANDEV_DEBUG_AGENT_MESSAGES")
	t.Setenv("KANDEV_DEBUG_DEV_MODE", "true")

	if _, _, err := profiles.ApplyProfile(); err != nil {
		t.Fatalf("ApplyProfile: %v", err)
	}
	if os.Getenv("KANDEV_DEBUG_AGENT_MESSAGES") != "true" {
		t.Fatal("profile did not enable agent message debug logs")
	}

	cfg := &config.Config{}
	ApplyStatesToConfig(cfg, []RuntimeFlagState{{
		Key:            "debug.devMode",
		EffectiveValue: false,
	}})

	if cfg.Debug.DevMode {
		t.Fatal("Debug.DevMode = true, want false")
	}
	if cfg.Debug.PprofEnabled {
		t.Fatal("Debug.PprofEnabled = true, want false")
	}
	if _, ok := os.LookupEnv("KANDEV_DEBUG_AGENT_MESSAGES"); ok {
		t.Fatal("KANDEV_DEBUG_AGENT_MESSAGES remained set after disabled override")
	}
	if _, ok := os.LookupEnv("KANDEV_DEBUG_PPROF_ENABLED"); ok {
		t.Fatal("KANDEV_DEBUG_PPROF_ENABLED remained set after disabled override")
	}
}

func TestOptionsFromConfigParsesUppercaseTruthyEnv(t *testing.T) {
	preserveEnv(t, "KANDEV_FEATURES_OFFICE")
	t.Setenv("KANDEV_FEATURES_OFFICE", "TRUE")

	opts := OptionsFromConfig(&config.Config{})

	if !opts.EnvValues["KANDEV_FEATURES_OFFICE"] {
		t.Fatal("KANDEV_FEATURES_OFFICE TRUE parsed false, want true")
	}
}

func TestApplyStatesToConfigMarksImpliedDebugEnvAsApplied(t *testing.T) {
	for _, name := range []string{
		"KANDEV_DEBUG_PPROF_ENABLED",
		"KANDEV_DEBUG_AGENT_MESSAGES",
	} {
		preserveEnv(t, name)
	}

	cfg := &config.Config{}
	ApplyStatesToConfig(cfg, []RuntimeFlagState{{
		Key:            "debug.devMode",
		EffectiveValue: true,
	}})
	opts := OptionsFromConfig(cfg)

	for _, name := range []string{
		"KANDEV_DEBUG_PPROF_ENABLED",
		"KANDEV_DEBUG_AGENT_MESSAGES",
	} {
		if !opts.EnvValues[name] {
			t.Fatalf("%s was not enabled", name)
		}
		if opts.IsExplicitEnv(name) {
			t.Fatalf("%s reported explicit, want profile-applied", name)
		}
	}
}

func preserveEnv(t *testing.T, name string) {
	t.Helper()
	value, ok := os.LookupEnv(name)
	_ = os.Unsetenv(name)
	t.Cleanup(func() {
		if ok {
			_ = os.Setenv(name, value)
			return
		}
		_ = os.Unsetenv(name)
	})
}