package api

import (
	"encoding/json"
	"net/http"
	"net/http/httptest"
	"sort"
	"testing"

	"github.com/metazla/meta-core/internal/storage"
)

func newFileTuplesTestServer(t *testing.T) (*Server, *storage.Client) {
	t.Helper()
	c := newTestStorage(t)
	return &Server{storage: c}, c
}

func getFileTuples(t *testing.T, s *Server, query string) (int, FileTuplesResponse) {
	t.Helper()
	req := httptest.NewRequest(http.MethodGet, "/api/files/tuples"+query, nil)
	rr := httptest.NewRecorder()
	s.handleGetFileTuples(rr, req)
	var resp FileTuplesResponse
	_ = json.Unmarshal(rr.Body.Bytes(), &resp)
	return rr.Code, resp
}

func mustMint(t *testing.T, c *storage.Client, path string, size, mtime int64) string {
	t.Helper()
	id, _, err := c.RegisterFileTuple(path, size, mtime, "bafkmid"+path)
	if err != nil {
		t.Fatalf("register %s: %v", path, err)
	}
	return id
}

// Only local files count: a gateway record — even one whose claims carry a
// filePath and size — is left out.
func TestFileTuples_OnlyFileBackedRoots(t *testing.T) {
	s, c := newFileTuplesTestServer(t)
	a := mustMint(t, c, "/files/watch/a.mkv", 1000, 1)
	b := mustMint(t, c, "/files/watch/b.mkv", 2500, 2)
	gw := testSigner(t, 1)
	storeClaims(t, c,
		signed(t, gw, "bagcsgatewayrecord", "title", "Some Release", 1),
		signed(t, gw, "bafkpartial", "filePath", "/files/cache/x.mkv", 1),
		signed(t, gw, "bafkpartial", "sizeByte", "999", 1),
	)

	code, resp := getFileTuples(t, s, "")
	if code != http.StatusOK {
		t.Fatalf("want 200, got %d", code)
	}
	if resp.Count != 2 || resp.TotalSize != 3500 {
		t.Fatalf("want count 2 / totalSize 3500, got %d / %d", resp.Count, resp.TotalSize)
	}
	if resp.Files == nil || len(*resp.Files) != 2 {
		t.Fatalf("want 2 rows, got %+v", resp.Files)
	}
	got := []string{(*resp.Files)[0].HashID, (*resp.Files)[1].HashID}
	want := []string{a, b}
	sort.Strings(got)
	sort.Strings(want)
	if got[0] != want[0] || got[1] != want[1] {
		t.Fatalf("want roots %v, got %v", want, got)
	}
}

func TestFileTuples_SummaryOmitsRows(t *testing.T) {
	s, c := newFileTuplesTestServer(t)
	mustMint(t, c, "/files/watch/a.mkv", 1000, 1)
	mustMint(t, c, "/files/watch/b.mkv", 2500, 2)

	code, resp := getFileTuples(t, s, "?summary=1")
	if code != http.StatusOK {
		t.Fatalf("want 200, got %d", code)
	}
	if resp.Files != nil {
		t.Fatalf("summary must omit rows, got %d", len(*resp.Files))
	}
	if resp.Count != 2 || resp.TotalSize != 3500 {
		t.Fatalf("want count 2 / totalSize 3500, got %d / %d", resp.Count, resp.TotalSize)
	}
}

// An empty library still carries "files": [] — clients treat an absent key as
// "this meta-core cannot list files", so it must not disappear when empty.
func TestFileTuples_EmptyLibraryKeepsFilesKey(t *testing.T) {
	s, _ := newFileTuplesTestServer(t)

	code, resp := getFileTuples(t, s, "")
	if code != http.StatusOK {
		t.Fatalf("want 200, got %d", code)
	}
	if resp.Files == nil || len(*resp.Files) != 0 || resp.Count != 0 {
		t.Fatalf("want an empty files array and count 0, got %+v", resp)
	}
}

func TestFileTuples_StorageNotConnected(t *testing.T) {
	s := &Server{storage: storage.NewClient()}

	code, _ := getFileTuples(t, s, "")
	if code != http.StatusServiceUnavailable {
		t.Fatalf("want 503, got %d", code)
	}
}