package server

import (
	"database/sql"
	"net/http"
	"strings"
	"testing"

	"github.com/git-pkgs/proxy/internal/database"
)

// seedLodash records lodash with 4.17.21 as its latest and only cached
// version. The syft fixture uses 4.17.20.
func seedLodash(t *testing.T, db *database.DB) {
	t.Helper()
	if err := db.UpsertPackage(&database.Package{
		PURL: "pkg:npm/lodash", Ecosystem: "npm", Name: "lodash",
		LatestVersion: sql.NullString{String: "4.17.21", Valid: true},
	}); err != nil {
		t.Fatal(err)
	}
	if err := db.UpsertVersion(&database.Version{PURL: "pkg:npm/lodash@4.17.21", PackagePURL: "pkg:npm/lodash"}); err != nil {
		t.Fatal(err)
	}
}

func (ts projectsTestServer) page(t *testing.T, target string) string {
	t.Helper()
	rec := ts.do(t, http.MethodGet, target, "", nil)
	if rec.Code != http.StatusOK {
		t.Fatalf("GET %s = %d: %s", target, rec.Code, rec.Body)
	}
	return rec.Body.String()
}

func assertContains(t *testing.T, page string, want ...string) {
	t.Helper()
	for _, w := range want {
		if !strings.Contains(page, w) {
			t.Errorf("page does not contain %q", w)
		}
	}
}

func TestProjectPageShowsComponents(t *testing.T) {
	ts := newProjectsTestServer(t)
	seedLodash(t, ts.db)
	ts.upload(t, "/api/projects/acme/web/sbom?branch=main", syftSBOM(t))

	page := ts.page(t, "/ui/projects/acme/web")
	assertContains(t, page,
		"acme/web",
		`href="/ui/package/npm/lodash"`, // cached, so linked
		"outdated, latest 4.17.21",
		"1 outdated",
		"symfony/console", // not cached: listed without a link
		"not cached",
		`data-project="acme/web"`, // delete control, asks for the token
		`/ui/static/project-delete.js`,
	)
	if strings.Contains(page, `href="/ui/package/composer/symfony/console"`) {
		t.Error("links to the package page of a package the proxy has no record of")
	}

	// Encoded as one segment, too.
	assertContains(t, ts.page(t, "/ui/projects/acme%2Fweb"), "symfony/console")

	if rec := ts.do(t, http.MethodGet, "/ui/projects/acme/other", "", nil); rec.Code != http.StatusNotFound {
		t.Errorf("unknown project status = %d, want 404", rec.Code)
	}
	if rec := ts.do(t, http.MethodGet, "/ui/projects/acme/../web", "", nil); rec.Code != http.StatusBadRequest {
		t.Errorf("invalid project status = %d, want 400", rec.Code)
	}

	list := ts.page(t, "/ui/projects")
	assertContains(t, list, `href="/ui/projects/acme/web"`, "9 packages")
}

func TestPackagePageShowsUsedBy(t *testing.T) {
	ts := newProjectsTestServer(t)
	seedLodash(t, ts.db)
	ts.upload(t, "/api/projects/acme/web/sbom", syftSBOM(t))
	ts.upload(t, "/api/projects/acme/api/sbom", []byte(`{"bomFormat":"CycloneDX","specVersion":"1.5","components":[
		{"type":"library","name":"lodash","version":"4.17.21","purl":"pkg:npm/lodash@4.17.21"}]}`))

	page := ts.page(t, "/ui/package/npm/lodash")
	assertContains(t, page,
		"Used by",
		`href="/ui/package/npm/lodash/4.17.21"`,
		`href="/ui/projects/acme/api"`,
		`href="/ui/projects/acme/web"`,
	)
	// 4.17.20 is used but not cached.
	start := strings.Index(page, "Used by")
	if start < 0 {
		t.Fatal("no Used by section")
	}
	used := page[start:]
	if i := strings.Index(used, "4.17.20"); i < 0 || !strings.Contains(used[i:], "not cached") {
		t.Error("the uncached version 4.17.20 is not listed as not cached")
	}
	if strings.Index(used, "4.17.21") > strings.Index(used, "4.17.20") {
		t.Error("versions are not listed highest first")
	}
}

func TestProjectPageFailsWhenComponentsCannotBeRead(t *testing.T) {
	ts := newProjectsTestServer(t)
	ts.upload(t, "/api/projects/acme/web/sbom", syftSBOM(t))
	// The components query joins packages; without the table it fails while
	// the project itself can still be read.
	if _, err := ts.db.Exec("ALTER TABLE packages RENAME TO packages_gone"); err != nil {
		t.Fatal(err)
	}
	if rec := ts.do(t, http.MethodGet, "/ui/projects/acme/web", "", nil); rec.Code != http.StatusInternalServerError {
		t.Errorf("status = %d, want 500 rather than an empty project", rec.Code)
	}
}

func TestProjectsPageLinksProjectsWhenEmpty(t *testing.T) {
	ts := newProjectsTestServer(t)

	page := ts.page(t, "/ui/projects")
	assertContains(t, page, `href="/ui/projects"`, "0 projects")
}