package builder

import (
	"encoding/base64"
	"encoding/json"
	"net/http"
	"net/http/httptest"
	"strings"
	"testing"
)

func TestBuildLaunchTemplateArtifacts(t *testing.T) {
	resp, err := BuildLaunchTemplateArtifacts(LaunchTemplateRequest{
		ServerURL: "https://deck.example", BuildID: "build-abc", StrainCardID: "card-99",
		Campaign: "aws-wave", AMIID: "ami-test", InstanceType: "t3.micro", Region: "eu-west-1",
	})
	if err != nil || !resp.Success {
		t.Fatalf("resp=%+v err=%v", resp, err)
	}
	if len(resp.GenesisSnapshotHash) != 64 {
		t.Fatalf("hash len %d", len(resp.GenesisSnapshotHash))
	}
	for _, m := range []string{"aetherforge-strain-genesis", "ami-test", resp.GenesisSnapshotHash, JoinLaneLaunchTemplate} {
		if !strings.Contains(resp.LaunchTemplateJSON, m) {
			t.Fatalf("missing %q in lt json", m)
		}
	}
	for _, m := range []string{LaunchTemplateParentAgentID, "AETHER_SPREAD_GENERATION=0", resp.GenesisSnapshotHash, "card-99"} {
		if !strings.Contains(resp.UserDataSh, m) {
			t.Fatalf("missing %q in user-data", m)
		}
	}
	var lt map[string]interface{}
	_ = json.Unmarshal([]byte(resp.LaunchTemplateJSON), &lt)
	data := lt["LaunchTemplateData"].(map[string]interface{})
	raw, _ := base64.StdEncoding.DecodeString(data["UserData"].(string))
	if string(raw) != resp.UserDataSh {
		t.Fatal("user-data mismatch")
	}
}

func TestGenesisSnapshotHashStable(t *testing.T) {
	m := genesisSnapshotManifest{ServerURL: "https://deck.example", BuildID: "b1"}
	if GenesisSnapshotHash(m) != GenesisSnapshotHash(m) {
		t.Fatal("unstable")
	}
}

func TestServeLaunchTemplateHandler(t *testing.T) {
	h := &Handler{}
	req := httptest.NewRequest(http.MethodPost, "/forge/launch-template", strings.NewReader(`{"server_url":"https://deck.example"}`))
	rec := httptest.NewRecorder()
	h.ServeLaunchTemplate(rec, req)
	if rec.Code != http.StatusOK {
		t.Fatalf("status %d", rec.Code)
	}
}

func TestServeLaunchTemplateRequiresServerURL(t *testing.T) {
	h := &Handler{}
	req := httptest.NewRequest(http.MethodPost, "/forge/launch-template", strings.NewReader(`{}`))
	rec := httptest.NewRecorder()
	h.ServeLaunchTemplate(rec, req)
	if rec.Code != http.StatusBadRequest {
		t.Fatalf("status %d", rec.Code)
	}
}