//go:build linux && rootrunner

package host

import (
	"crypto/ed25519"
	"encoding/json"
	"os"
	"os/exec"
	"strings"
	"testing"

	"github.com/carlosframework/platform/internal/hostbox"
	"github.com/carlosframework/platform/internal/hostfloor"
	"github.com/carlosframework/platform/internal/release"
)

func lifecycleArmGuard(t *testing.T, store release.Store, key ed25519.PrivateKey, actor hostbox.Actor, pointer release.SystemPointer, resume bool) {
	t.Helper()
	endpoint := activationS3Fixture(t, store)
	env := []string{"PLAN14_LIFECYCLE_GUARD=1", "CARLOS_DEPLOYMENT_BUCKET=" + actor.ControlBucket, "CARLOS_S3_ENDPOINT=" + endpoint, "AWS_REGION=eu-west-1", "AWS_DEFAULT_REGION=eu-west-1"}
	for _, name := range []string{"CARLOS_RELEASE_PUBKEY", "CARLOS_DATA_DIR", "CARLOS_OPT_DIR", "AWS_ACCESS_KEY_ID", "AWS_SECRET_ACCESS_KEY", "AWS_EC2_METADATA_DISABLED", "AWS_CONFIG_FILE", "AWS_SHARED_CREDENTIALS_FILE"} {
		env = append(env, name+"="+os.Getenv(name))
	}
	lifecycleCheck(t, "guard environment", os.WriteFile("/etc/carlos/host.env", []byte(strings.Join(env, "\n")+"\n"), 0600))
	for _, unit := range []string{"carlos-edge.service", "carlos-system-update.service", "carlos-system-recover.service"} {
		path := "/etc/systemd/system/" + unit + ".d"
		if !resume {
			if _, err := os.Lstat(path); !os.IsNotExist(err) {
				t.Fatal("pre-existing guard drop-in", path, err)
			}
		}
		t.Cleanup(func() {
			if os.Getenv("PLAN14_LIFECYCLE_PHASE") != "1" || t.Failed() {
				os.RemoveAll(path)
			}
			exec.Command("systemctl", "daemon-reload").Run()
		})
	}
	if !resume {
		floor, err := json.Marshal(hostfloor.Floor{Min: 1, MinRelease: "v2.0.0"})
		lifecycleCheck(t, "guard floor", err)
		auth, err := release.SignRootEnvelope(release.RootSigner(key), hostfloor.Authorization{Schema: 1, Purpose: "moves-protocol", DeploymentBucket: actor.ControlBucket, Key: hostfloor.Key, SHA256: hostfloor.Digest(floor)})
		lifecycleCheck(t, "guard authorization", err)
		policy, err := json.Marshal(hostfloor.Policy{Schema: 1, DeploymentBucket: actor.ControlBucket, Mode: "armed", Floor: floor, Authorization: auth})
		lifecycleCheck(t, "guard policy", err)
		identity, err := release.SignRootEnvelope(release.RootSigner(key), pointer)
		lifecycleCheck(t, "guard signed identity", err)
		enrollment, err := json.Marshal(hostfloor.Enrollment{Box: actor.Box, Enrollment: actor.Enrollment, InstanceID: actor.Cloud.InstanceID})
		lifecycleCheck(t, "guard enrollment", err)
		for name, raw := range map[string][]byte{"policy": policy, "identity": identity, "enrollment": enrollment} {
			lifecycleCheck(t, "guard input", os.WriteFile("/etc/carlos/"+name+".json", raw, 0600))
		}
		cmd := exec.Command("/opt/carlos/carlos", "system", "guard-arm", "--policy", "/etc/carlos/policy.json", "--identity", "/etc/carlos/identity.json", "--enrollment", "/etc/carlos/enrollment.json")
		cmd.Env = append(env, "PATH=/usr/sbin:/usr/bin:/sbin:/bin")
		out, err := cmd.CombinedOutput()
		lifecycleCheck(t, "real guard-arm: "+string(out), err)
	}
	raw, err := os.ReadFile("/etc/systemd/system/carlos-edge.service.d/90-host-floor.conf")
	lifecycleCheck(t, "guard drop-in", err)
	if string(raw) != "[Unit]\nRequires=carlos-system-recover.service\nAfter=carlos-system-recover.service\n[Service]\nExecStart=\nExecStart=/opt/carlos/system/guard/carlos system guard-exec edge\n" {
		t.Fatal("edge does not retain guard-arm drop-in", string(raw))
	}
	guard, err := os.Stat("/opt/carlos/system/guard/carlos")
	lifecycleCheck(t, "guard inode", err)
	current, err := os.Stat("/opt/carlos/carlos")
	lifecycleCheck(t, "current inode", err)
	if os.SameFile(guard, current) {
		t.Fatal("guard must be an independent copy")
	}
}