import { afterEach, beforeEach, describe, expect, it } from 'vitest';

import { IdentityDB } from '../src/core/identity-db';

async function seedMemoryGraph(db: IdentityDB): Promise<void> {
  await db.addFact({
    statement: 'I have worked with TypeScript since 2025.',
    topics: [
      { name: 'I', category: 'entity', granularity: 'concrete', role: 'subject' },
      { name: 'TypeScript', category: 'entity', granularity: 'concrete', role: 'object' },
      { name: '2025', category: 'temporal', granularity: 'concrete', role: 'time' },
    ],
  });

  await db.addFact({
    statement: 'TypeScript is a programming language.',
    topics: [
      { name: 'TypeScript', category: 'entity', granularity: 'concrete', role: 'subject' },
      { name: 'programming language', category: 'concept', granularity: 'abstract', role: 'classification' },
    ],
  });
}

describe('IdentityDB queries', () => {
  let db: IdentityDB;

  beforeEach(async () => {
    db = await IdentityDB.connect({ client: 'sqlite', filename: ':memory:' });
    await db.initialize();
    await seedMemoryGraph(db);
  });

  afterEach(async () => {
    await db.close();
  });

  it('gets a topic with its facts', async () => {
    const topic = await db.getTopicByName('TypeScript', { includeFacts: true });

    expect(topic).not.toBeNull();
    expect(topic?.name).toBe('TypeScript');
    expect(topic?.facts).toHaveLength(2);
    expect(topic?.facts.map((fact) => fact.statement)).toEqual([
      'I have worked with TypeScript since 2025.',
      'TypeScript is a programming language.',
    ]);
  });

  it('gets only the facts linked to another topic', async () => {
    const facts = await db.getTopicFactsLinkedTo('TypeScript', '2025');

    expect(facts).toHaveLength(1);
    expect(facts[0]?.statement).toBe('I have worked with TypeScript since 2025.');
  });

  it('lists topics without expanding facts', async () => {
    const topics = await db.listTopics({ includeFacts: false });

    expect(topics.map((topic) => topic.name)).toEqual([
      '2025',
      'I',
      'programming language',
      'TypeScript',
    ]);
    expect('facts' in topics[0]!).toBe(false);
  });

  it('finds connected topics with shared fact counts', async () => {
    const connectedTopics = await db.findConnectedTopics('TypeScript');

    expect(connectedTopics).toEqual([
      expect.objectContaining({ name: '2025', sharedFactCount: 1 }),
      expect.objectContaining({ name: 'I', sharedFactCount: 1 }),
      expect.objectContaining({ name: 'programming language', sharedFactCount: 1 }),
    ]);
  });

  it('finds facts that connect all requested topics', async () => {
    const facts = await db.findFactsConnectingTopics(['I', 'TypeScript', '2025']);

    expect(facts).toHaveLength(1);
    expect(facts[0]?.statement).toBe('I have worked with TypeScript since 2025.');
  });
});