package db

import (
	"testing"

	"github.com/bluesky-social/indigo/atproto/syntax"
)

func subjectsOf(t *testing.T, d *DB, repoDid syntax.DID) []syntax.DID {
	t.Helper()
	rows, err := d.ListCollaboratorsByRepoDid(repoDid)
	if err != nil {
		t.Fatalf("ListCollaboratorsByRepoDid: %v", err)
	}
	out := make([]syntax.DID, 0, len(rows))
	for _, r := range rows {
		out = append(out, r.Subject)
	}
	return out
}

func TestAddKnotCollaborator_PersistsAndIsIdempotent(t *testing.T) {
	d := newTestDB(t)
	repo := syntax.DID("did:plc:repo")
	bob := syntax.DID("did:plc:bob")

	if err := d.AddKnotCollaborator(repo, bob); err != nil {
		t.Fatalf("add: %v", err)
	}
	if err := d.AddKnotCollaborator(repo, bob); err != nil {
		t.Fatalf("re-add: %v", err)
	}

	got := subjectsOf(t, d, repo)
	if len(got) != 1 || got[0] != bob {
		t.Fatalf("collaborators = %v, want exactly [bob]", got)
	}
}

func TestDeleteRepoCollaboratorBySubjectRepo(t *testing.T) {
	d := newTestDB(t)
	repo := syntax.DID("did:plc:repo")
	bob := syntax.DID("did:plc:bob")
	carol := syntax.DID("did:plc:carol")

	if err := d.AddKnotCollaborator(repo, bob); err != nil {
		t.Fatalf("add bob: %v", err)
	}
	if err := d.AddKnotCollaborator(repo, carol); err != nil {
		t.Fatalf("add carol: %v", err)
	}

	if err := d.DeleteRepoCollaboratorBySubjectRepo(bob, repo); err != nil {
		t.Fatalf("delete bob: %v", err)
	}
	got := subjectsOf(t, d, repo)
	if len(got) != 1 || got[0] != carol {
		t.Fatalf("after removing bob, collaborators = %v, want [carol]", got)
	}

	if err := d.DeleteRepoCollaboratorBySubjectRepo(bob, repo); err != nil {
		t.Fatalf("idempotent delete: %v", err)
	}
}

func TestKnotCollaborator_NoCollisionAcrossReposAndSubjects(t *testing.T) {
	d := newTestDB(t)
	repoA := syntax.DID("did:plc:repoA")
	repoB := syntax.DID("did:plc:repoB")
	bob := syntax.DID("did:plc:bob")
	carol := syntax.DID("did:plc:carol")

	for _, c := range []struct{ repo, subj syntax.DID }{
		{repoA, bob}, {repoB, bob}, {repoA, carol},
	} {
		if err := d.AddKnotCollaborator(c.repo, c.subj); err != nil {
			t.Fatalf("add %s/%s: %v", c.repo, c.subj, err)
		}
	}

	if got := subjectsOf(t, d, repoA); len(got) != 2 {
		t.Errorf("repoA collaborators = %v, want bob+carol", got)
	}
	if got := subjectsOf(t, d, repoB); len(got) != 1 || got[0] != bob {
		t.Errorf("repoB collaborators = %v, want [bob]", got)
	}

	if err := d.DeleteRepoCollaboratorBySubjectRepo(bob, repoA); err != nil {
		t.Fatalf("delete bob@repoA: %v", err)
	}
	if got := subjectsOf(t, d, repoB); len(got) != 1 || got[0] != bob {
		t.Errorf("repoB after removing bob@repoA = %v, want still [bob]", got)
	}
}