//go:build linux

package host

import (
	"context"
	"encoding/json"
	"github.com/carlosframework/platform/internal/provision"
	"os"
	"strings"
	"testing"
	"time"

	"github.com/carlosframework/platform/internal/hostbox"
	"github.com/carlosframework/platform/internal/hostguard"
	"github.com/carlosframework/platform/internal/registry"
)

func TestPlan14BoundUnitInputs(t *testing.T) {
	for _, unit := range []string{"console.service", "woodstar-atbridge.service", "carlos-app@shop.example.service"} {
		t.Run(unit, func(t *testing.T) {
			f, _, _ := generationLaunchFixture(t)
			b, _, shape := plan14UnitBindingFixture(t, unit, "", "")
			if unit == "woodstar-atbridge.service" {
				shape.Addr = ""
				shape.Socket = "/run/carlos/woodstar-atbridge.sock"
			}
			if unit == "carlos-app@shop.example.service" {
				shape.Backing = "unit"
			}
			in, err := f.root.Guard.Root.Installation(t.Context(), f.target.Host, hostguard.Captured{Incarnation: f.target.Incarnation, Stamp: f.target.Stamp})
			if err != nil {
				t.Fatal(err)
			}
			program := claimUnitProgram{Schema: 1, Host: in.Host, Args: []string{"serve"}, Env: []string{"SECRET=root-config"}}
			raw, _ := json.Marshal(program)
			shape.Program = claimUnitDigest(raw)
			b.installation = func(context.Context, string) (hostguard.RootInstallation, error) { return in, nil }
			b.canonical = func(context.Context, string) (claimUnitShape, error) { return shape, nil }
			if _, err = b.establish(t.Context(), in.Host); err != nil {
				t.Fatal(err)
			}
			f.root.unitBindings = b
			f.root.unitProgram = func(s claimUnitShape) (claimUnitProgram, error) { return decodeClaimUnitProgram(raw, s) }
			route := shape.route()
			locked, cancel := context.WithTimeout(t.Context(), time.Second)
			defer cancel()
			if err := f.root.Guard.WithLocal(locked, in.Host, func() error { _, err := f.root.boundInputs(locked, route); return err }); err != nil {
				t.Fatal("input view reacquired host lock", err)
			}
			q := hostbox.RoleLaunchRequest{V: hostbox.RoleLaunchV, Op: hostbox.RoleLaunchInputs, Route: &route}
			out, err := f.root.Serve(t.Context(), f.peer, q)
			if err != nil || out.Inputs == nil || out.Inputs.Unit != unit || out.Inputs.Program != shape.Program || out.Inputs.Env[0] != program.Env[0] {
				t.Fatal(out, err)
			}
			for _, change := range []func(*registry.Route){
				func(r *registry.Route) { r.Unit = "other.service" }, func(r *registry.Route) { r.Unit = ""; r.Backing = "" },
				func(r *registry.Route) { r.Backing = "exec" }, func(r *registry.Route) { r.Socket = "/run/carlos/other.sock" },
				func(r *registry.Route) { r.Addr = "127.0.0.1:9999" }, func(r *registry.Route) { r.DBPath = "/data/other.db" },
				func(r *registry.Route) { r.Tunnel = "tunnel" }, func(r *registry.Route) { r.Seed = "seed" }, func(r *registry.Route) { r.Replica = true },
			} {
				forged := route
				change(&forged)
				q.Route = &forged
				if _, err = f.root.Serve(t.Context(), f.peer, q); err == nil {
					t.Fatal("changed route accepted", forged)
				}
			}
			q.Route = &route
			raw = append(raw, ' ')
			if _, err = f.root.Serve(t.Context(), f.peer, q); err == nil {
				t.Fatal("changed root program accepted")
			}
		})
	}
}

func TestPlan14BoundColdSelection(t *testing.T) {
	f := newAdminTokenFixture(t, "box-1")
	unit := &hostbox.UnitLaunchInputs{Unit: "woodstar.service", Socket: "/run/carlos/woodstar.sock", DBPath: provision.NestedDBPath("/data", f.rt.Host), Program: strings.Repeat("a", 64), Args: []string{"serve"}, Env: []string{"ROOT_PROGRAM=yes"}}
	f.rt.Unit = unit.Unit
	f.rt.Socket = unit.Socket
	f.rt.DBPath = unit.DBPath
	source := claimServingLaunchSource(f.backend, f.tokens, func(context.Context, registry.Route) (*hostbox.UnitLaunchInputs, error) { return unit, nil })
	launch, err := source(t.Context(), f.rt)
	if err != nil {
		t.Fatal(err)
	}
	if launch.Unit != unit || launch.Identity.Config != unit.Program || launch.Identity.Release != launch.Staged.Digest || launch.ExecutableSHA256 != launch.Staged.Digest {
		t.Fatal(launch)
	}
	in := ClaimLaunchInputs{Unit: unit, Roles: launch.Roles, Identity: launch.Identity, Executable: launch.Executable, ExecutableSHA256: launch.ExecutableSHA256, Args: launch.Args, Env: launch.Env, DBPath: unit.DBPath, Limit: launch.Limit}
	if err = shapeAssembly().PreflightLaunch(f.rt.Host, in); err != nil {
		t.Fatal(err)
	}
	for _, change := range []func(*ClaimLaunchInputs){
		func(in *ClaimLaunchInputs) { in.Args = []string{"shell"} }, func(in *ClaimLaunchInputs) { in.Env = []string{"ROOT_PROGRAM=no"} },
		func(in *ClaimLaunchInputs) { in.Identity.Config = "different" }, func(in *ClaimLaunchInputs) { in.DBPath = "/data/sibling.db" },
	} {
		bad := in
		change(&bad)
		if err = shapeAssembly().PreflightLaunch(f.rt.Host, bad); err == nil {
			t.Fatal("changed bound inputs accepted")
		}
	}
	if _, err = os.Stat(f.envPath()); !os.IsNotExist(err) {
		t.Fatal("bound selection minted ordinary environment", err)
	}
}