Function load

__tests__/lombok.test.ts:32–46

()

  type Row = { name: string; kind: string; decorators: string | null; docstring: string | null; signature: string | null };
  const load = async () => {
    const cg = await CodeGraph.init(dir, { silent: true });
    await cg.indexAll();
    const db = (cg as any).db.db;
    const nodes: Row[] = db.prepare(`SELECT name, kind, decorators, docstring, signature FROM nodes`).all();
    const calls: { src: string; tgt: string }[] = db
      .prepare(
        `SELECT s.name src, t.name tgt FROM edges e
         JOIN nodes s ON s.id = e.source JOIN nodes t ON t.id = e.target
         WHERE e.kind = 'calls'`
      )
      .all();
    cg.close?.();
    return { nodes, calls };
  };

  const isLombok = (n: Row | undefined) => !!n && (n.decorators ?? '').includes('lombok');