package hostapi

import (
	"context"
	"net/http"
	"net/http/httptest"
	"testing"

	core "agent-system-core"
)

// EXT-8 余项：hostapi 检索面——SharedSearch 可用时整体切换（本人+适配器放行的
// 共享追加）；不可用时维持现状（仅本人回调）。
func TestSharedSearchFace(t *testing.T) {
	own := []core.RetrievalHit{{Doc: core.KnowledgeDoc{Owner: "bob", Chunk: "本人命中"}}}
	sharedHit := core.RetrievalHit{Doc: core.KnowledgeDoc{Owner: "alice", ProjectID: "prj-s", Chunk: "共享命中"}}

	srv := &Server{Token: "t", Search: func(ctx context.Context, owner, text string) ([]core.RetrievalHit, error) {
		return own, nil
	}}
	run := func() int {
		rec := httptest.NewRecorder()
		req := httptest.NewRequest(http.MethodGet, "/host/v1/knowledge/search?q=x", nil)
		req.Header.Set("Authorization", "Bearer "+srv.Token)
		srv.Handler().ServeHTTP(rec, req)
		return rec.Code
	}
	if code := run(); code != http.StatusOK {
		t.Fatalf("plain search: %d", code)
	}

	srv.SharedSearch = func(ctx context.Context, owner, text string) ([]core.RetrievalHit, error) {
		return append(own, sharedHit), nil
	}
	rec := httptest.NewRecorder()
	req := httptest.NewRequest(http.MethodGet, "/host/v1/knowledge/search?q=x", nil)
	req.Header.Set("Authorization", "Bearer "+srv.Token)
	srv.Handler().ServeHTTP(rec, req)
	if rec.Code != http.StatusOK {
		t.Fatalf("shared search: %d %s", rec.Code, rec.Body.String())
	}
	var hits []core.RetrievalHit
	if err := jsonUnmarshal(rec.Body.Bytes(), &hits); err != nil {
		t.Fatal(err)
	}
	if len(hits) != 2 || hits[1].Doc.Owner != "alice" {
		t.Fatalf("hits = %+v, want 本人+共享", hits)
	}
}