package httpapi

import (
	"encoding/json"
	"net/http"
	"net/http/httptest"
	"strings"
	"testing"
	"time"

	"evobgp/internal/store"
)

func TestPeerDiscoveryApproveReject(t *testing.T) {
	srv, err := New(Options{InsecureDev: true, SeedDemo: true, BundleSeedHex: testBundleSeed})
	if err != nil {
		t.Fatal(err)
	}
	defer srv.Close()
	tenant, _, _, _, speaker := srv.Store().DemoIDs()
	mustSetTestAPIKeys(t, srv, "edkey|"+tenant+"|operator")
	h := srv.Handler()

	disc, err := srv.Store().UpsertPeerDiscovery(tenant, &store.PeerDiscoveryUpsert{
		SpeakerID:    speaker,
		NeighborID:   "203.0.113.10",
		Neighbor:     "203.0.113.10",
		RemoteASN:    65099,
		ProtocolName: "evobgp_dyn_0001",
		SessionState: "Established",
		SeenAt:       time.Now().UTC(),
	})
	if err != nil {
		t.Fatal(err)
	}

	listReq := httptest.NewRequest(http.MethodGet, "/v1/peers/discovered?status=pending", nil)
	listReq.Header.Set("Authorization", "Bearer edkey")
	listRec := httptest.NewRecorder()
	h.ServeHTTP(listRec, listReq)
	if listRec.Code != http.StatusOK {
		t.Fatalf("list status %d body %s", listRec.Code, listRec.Body.String())
	}

	approveBody := `{"name":"client-a"}`
	approveReq := httptest.NewRequest(http.MethodPost, "/v1/peers/discovered/"+disc.ID+"/approve", strings.NewReader(approveBody))
	approveReq.Header.Set("Authorization", "Bearer edkey")
	approveReq.Header.Set("Content-Type", "application/json")
	approveRec := httptest.NewRecorder()
	h.ServeHTTP(approveRec, approveReq)
	if approveRec.Code != http.StatusOK {
		t.Fatalf("approve status %d body %s", approveRec.Code, approveRec.Body.String())
	}
	var approveOut struct {
		Peer struct {
			ID       string `json:"id"`
			Neighbor string `json:"neighbor"`
			Name     string `json:"name"`
		} `json:"peer"`
	}
	if err := json.Unmarshal(approveRec.Body.Bytes(), &approveOut); err != nil {
		t.Fatal(err)
	}
	if approveOut.Peer.Neighbor != "203.0.113.10" || approveOut.Peer.Name != "client-a" {
		t.Fatalf("unexpected peer: %+v", approveOut.Peer)
	}

	peers := srv.Store().ListPeers(tenant)
	found := false
	for _, p := range peers {
		if p.ID == approveOut.Peer.ID {
			found = true
			break
		}
	}
	if !found {
		t.Fatal("approved peer not in ListPeers")
	}

	disc2, err := srv.Store().UpsertPeerDiscovery(tenant, &store.PeerDiscoveryUpsert{
		SpeakerID:    speaker,
		NeighborID:   "198.51.100.99",
		Neighbor:     "198.51.100.99",
		RemoteASN:    65100,
		ProtocolName: "evobgp_dyn_0002",
		SessionState: "Active",
		SeenAt:       time.Now().UTC(),
	})
	if err != nil {
		t.Fatal(err)
	}
	rejReq := httptest.NewRequest(http.MethodPost, "/v1/peers/discovered/"+disc2.ID+"/reject", nil)
	rejReq.Header.Set("Authorization", "Bearer edkey")
	rejRec := httptest.NewRecorder()
	h.ServeHTTP(rejRec, rejReq)
	if rejRec.Code != http.StatusOK {
		t.Fatalf("reject status %d body %s", rejRec.Code, rejRec.Body.String())
	}
	again, err := srv.Store().UpsertPeerDiscovery(tenant, &store.PeerDiscoveryUpsert{
		NeighborID:   "198.51.100.99",
		Neighbor:     "198.51.100.99",
		RemoteASN:    65100,
		ProtocolName: "evobgp_dyn_0002",
		SessionState: "Established",
		SeenAt:       time.Now().UTC(),
	})
	if err != nil {
		t.Fatal(err)
	}
	if again.Status != store.PeerDiscoveryRejected {
		t.Fatalf("expected rejected, got %s", again.Status)
	}
}