npm - @crawlith/core - Versions diffs - 0.1.0 - Mend

@crawlith/core 0.1.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (201) hide show

package/CHANGELOG.md +7 -0
package/dist/analysis/analyze.d.ts +70 -0
package/dist/analysis/analyze.js +436 -0
package/dist/analysis/content.d.ts +12 -0
package/dist/analysis/content.js +33 -0
package/dist/analysis/images.d.ts +6 -0
package/dist/analysis/images.js +18 -0
package/dist/analysis/links.d.ts +7 -0
package/dist/analysis/links.js +30 -0
package/dist/analysis/scoring.d.ts +9 -0
package/dist/analysis/scoring.js +42 -0
package/dist/analysis/seo.d.ts +15 -0
package/dist/analysis/seo.js +64 -0
package/dist/analysis/structuredData.d.ts +6 -0
package/dist/analysis/structuredData.js +51 -0
package/dist/audit/dns.d.ts +2 -0
package/dist/audit/dns.js +42 -0
package/dist/audit/headers.d.ts +2 -0
package/dist/audit/headers.js +95 -0
package/dist/audit/index.d.ts +2 -0
package/dist/audit/index.js +50 -0
package/dist/audit/scoring.d.ts +14 -0
package/dist/audit/scoring.js +214 -0
package/dist/audit/transport.d.ts +6 -0
package/dist/audit/transport.js +207 -0
package/dist/audit/types.d.ts +88 -0
package/dist/audit/types.js +1 -0
package/dist/core/network/proxyAdapter.d.ts +6 -0
package/dist/core/network/proxyAdapter.js +19 -0
package/dist/core/network/rateLimiter.d.ts +6 -0
package/dist/core/network/rateLimiter.js +31 -0
package/dist/core/network/redirectController.d.ts +13 -0
package/dist/core/network/redirectController.js +41 -0
package/dist/core/network/responseLimiter.d.ts +4 -0
package/dist/core/network/responseLimiter.js +26 -0
package/dist/core/network/retryPolicy.d.ts +10 -0
package/dist/core/network/retryPolicy.js +41 -0
package/dist/core/scope/domainFilter.d.ts +11 -0
package/dist/core/scope/domainFilter.js +40 -0
package/dist/core/scope/scopeManager.d.ts +14 -0
package/dist/core/scope/scopeManager.js +39 -0
package/dist/core/scope/subdomainPolicy.d.ts +6 -0
package/dist/core/scope/subdomainPolicy.js +35 -0
package/dist/core/security/ipGuard.d.ts +11 -0
package/dist/core/security/ipGuard.js +84 -0
package/dist/crawler/crawl.d.ts +22 -0
package/dist/crawler/crawl.js +336 -0
package/dist/crawler/extract.d.ts +5 -0
package/dist/crawler/extract.js +33 -0
package/dist/crawler/fetcher.d.ts +40 -0
package/dist/crawler/fetcher.js +161 -0
package/dist/crawler/metricsRunner.d.ts +1 -0
package/dist/crawler/metricsRunner.js +108 -0
package/dist/crawler/normalize.d.ts +7 -0
package/dist/crawler/normalize.js +88 -0
package/dist/crawler/parser.d.ts +22 -0
package/dist/crawler/parser.js +158 -0
package/dist/crawler/sitemap.d.ts +8 -0
package/dist/crawler/sitemap.js +70 -0
package/dist/crawler/trap.d.ts +24 -0
package/dist/crawler/trap.js +78 -0
package/dist/db/graphLoader.d.ts +2 -0
package/dist/db/graphLoader.js +96 -0
package/dist/db/index.d.ts +4 -0
package/dist/db/index.js +61 -0
package/dist/db/repositories/EdgeRepository.d.ts +16 -0
package/dist/db/repositories/EdgeRepository.js +17 -0
package/dist/db/repositories/MetricsRepository.d.ts +26 -0
package/dist/db/repositories/MetricsRepository.js +27 -0
package/dist/db/repositories/PageRepository.d.ts +47 -0
package/dist/db/repositories/PageRepository.js +93 -0
package/dist/db/repositories/SiteRepository.d.ts +15 -0
package/dist/db/repositories/SiteRepository.js +22 -0
package/dist/db/repositories/SnapshotRepository.d.ts +22 -0
package/dist/db/repositories/SnapshotRepository.js +55 -0
package/dist/db/schema.d.ts +2 -0
package/dist/db/schema.js +169 -0
package/dist/diff/compare.d.ts +26 -0
package/dist/diff/compare.js +64 -0
package/dist/graph/cluster.d.ts +6 -0
package/dist/graph/cluster.js +173 -0
package/dist/graph/duplicate.d.ts +10 -0
package/dist/graph/duplicate.js +251 -0
package/dist/graph/graph.d.ts +103 -0
package/dist/graph/graph.js +106 -0
package/dist/graph/metrics.d.ts +29 -0
package/dist/graph/metrics.js +74 -0
package/dist/graph/pagerank.d.ts +12 -0
package/dist/graph/pagerank.js +102 -0
package/dist/graph/simhash.d.ts +17 -0
package/dist/graph/simhash.js +56 -0
package/dist/index.d.ts +30 -0
package/dist/index.js +30 -0
package/dist/lock/hashKey.d.ts +1 -0
package/dist/lock/hashKey.js +44 -0
package/dist/lock/lockManager.d.ts +7 -0
package/dist/lock/lockManager.js +112 -0
package/dist/lock/pidCheck.d.ts +1 -0
package/dist/lock/pidCheck.js +14 -0
package/dist/report/html.d.ts +2 -0
package/dist/report/html.js +223 -0
package/dist/report/sitegraphExport.d.ts +3 -0
package/dist/report/sitegraphExport.js +52 -0
package/dist/report/sitegraph_template.d.ts +1 -0
package/dist/report/sitegraph_template.js +630 -0
package/dist/scoring/hits.d.ts +9 -0
package/dist/scoring/hits.js +111 -0
package/dist/scoring/orphanSeverity.d.ts +39 -0
package/dist/scoring/orphanSeverity.js +125 -0
package/dist/utils/version.d.ts +2 -0
package/dist/utils/version.js +15 -0
package/package.json +33 -0
package/src/analysis/analyze.ts +548 -0
package/src/analysis/content.ts +62 -0
package/src/analysis/images.ts +28 -0
package/src/analysis/links.ts +41 -0
package/src/analysis/scoring.ts +59 -0
package/src/analysis/seo.ts +82 -0
package/src/analysis/structuredData.ts +62 -0
package/src/audit/dns.ts +49 -0
package/src/audit/headers.ts +98 -0
package/src/audit/index.ts +66 -0
package/src/audit/scoring.ts +232 -0
package/src/audit/transport.ts +258 -0
package/src/audit/types.ts +102 -0
package/src/core/network/proxyAdapter.ts +21 -0
package/src/core/network/rateLimiter.ts +39 -0
package/src/core/network/redirectController.ts +47 -0
package/src/core/network/responseLimiter.ts +34 -0
package/src/core/network/retryPolicy.ts +57 -0
package/src/core/scope/domainFilter.ts +45 -0
package/src/core/scope/scopeManager.ts +52 -0
package/src/core/scope/subdomainPolicy.ts +39 -0
package/src/core/security/ipGuard.ts +92 -0
package/src/crawler/crawl.ts +382 -0
package/src/crawler/extract.ts +34 -0
package/src/crawler/fetcher.ts +233 -0
package/src/crawler/metricsRunner.ts +124 -0
package/src/crawler/normalize.ts +108 -0
package/src/crawler/parser.ts +190 -0
package/src/crawler/sitemap.ts +73 -0
package/src/crawler/trap.ts +96 -0
package/src/db/graphLoader.ts +105 -0
package/src/db/index.ts +70 -0
package/src/db/repositories/EdgeRepository.ts +29 -0
package/src/db/repositories/MetricsRepository.ts +49 -0
package/src/db/repositories/PageRepository.ts +128 -0
package/src/db/repositories/SiteRepository.ts +32 -0
package/src/db/repositories/SnapshotRepository.ts +74 -0
package/src/db/schema.ts +177 -0
package/src/diff/compare.ts +84 -0
package/src/graph/cluster.ts +192 -0
package/src/graph/duplicate.ts +286 -0
package/src/graph/graph.ts +172 -0
package/src/graph/metrics.ts +110 -0
package/src/graph/pagerank.ts +125 -0
package/src/graph/simhash.ts +61 -0
package/src/index.ts +30 -0
package/src/lock/hashKey.ts +51 -0
package/src/lock/lockManager.ts +124 -0
package/src/lock/pidCheck.ts +13 -0
package/src/report/html.ts +227 -0
package/src/report/sitegraphExport.ts +58 -0
package/src/report/sitegraph_template.ts +630 -0
package/src/scoring/hits.ts +131 -0
package/src/scoring/orphanSeverity.ts +176 -0
package/src/utils/version.ts +18 -0
package/tests/__snapshots__/orphanSeverity.test.ts.snap +49 -0
package/tests/analysis.unit.test.ts +98 -0
package/tests/analyze.integration.test.ts +98 -0
package/tests/audit/dns.test.ts +31 -0
package/tests/audit/headers.test.ts +45 -0
package/tests/audit/scoring.test.ts +133 -0
package/tests/audit/security.test.ts +12 -0
package/tests/audit/transport.test.ts +112 -0
package/tests/clustering.test.ts +118 -0
package/tests/crawler.test.ts +358 -0
package/tests/db.test.ts +159 -0
package/tests/diff.test.ts +67 -0
package/tests/duplicate.test.ts +110 -0
package/tests/fetcher.test.ts +106 -0
package/tests/fetcher_safety.test.ts +85 -0
package/tests/fixtures/analyze-crawl.json +26 -0
package/tests/hits.test.ts +134 -0
package/tests/html_report.test.ts +58 -0
package/tests/lock/lockManager.test.ts +138 -0
package/tests/metrics.test.ts +196 -0
package/tests/normalize.test.ts +101 -0
package/tests/orphanSeverity.test.ts +160 -0
package/tests/pagerank.test.ts +98 -0
package/tests/parser.test.ts +117 -0
package/tests/proxy_safety.test.ts +57 -0
package/tests/redirect_safety.test.ts +73 -0
package/tests/safety.test.ts +114 -0
package/tests/scope.test.ts +66 -0
package/tests/scoring.test.ts +59 -0
package/tests/sitemap.test.ts +88 -0
package/tests/soft404.test.ts +41 -0
package/tests/trap.test.ts +39 -0
package/tests/visualization_data.test.ts +46 -0
package/tsconfig.json +11 -0

package/tests/crawler.test.ts ADDED Viewed

@@ -0,0 +1,358 @@
+import { test, expect, beforeEach, afterEach } from 'vitest';
+import { crawl } from '../src/crawler/crawl.js';
+import { loadGraphFromSnapshot } from '../src/db/graphLoader.js';
+import { closeDb } from '../src/db/index.js';
+import { MockAgent, setGlobalDispatcher } from 'undici';
+let mockAgent: MockAgent;
+beforeEach(() => {
+  process.env.CRAWLITH_DB_PATH = ':memory:';
+  mockAgent = new MockAgent();
+  mockAgent.disableNetConnect();
+  setGlobalDispatcher(mockAgent);
+});
+afterEach(() => {
+  closeDb();
+});
+test('crawler should crawl and build graph', async () => {
+  const client = mockAgent.get('https://example.com');
+  // Root
+  client.intercept({
+    path: '/',
+    method: 'GET'
+  }).reply(200, `
+        <html><body>
+            <a href="/page1">Page 1</a>
+            <a href="/page2">Page 2</a>
+        </body></html>
+    `, {
+    headers: { 'content-type': 'text/html' }
+  });
+  // Page 1
+  client.intercept({
+    path: '/page1',
+    method: 'GET'
+  }).reply(200, `
+        <html><body>
+            <a href="/page2">Page 2</a>
+        </body></html>
+    `, {
+    headers: { 'content-type': 'text/html' }
+  });
+  // Page 2
+  client.intercept({
+    path: '/page2',
+    method: 'GET'
+  }).reply(200, `
+        <html><body>
+            <a href="/">Home</a>
+        </body></html>
+    `, {
+    headers: { 'content-type': 'text/html' }
+  });
+  // Robots.txt
+  client.intercept({
+    path: '/robots.txt',
+    method: 'GET'
+  }).reply(404, 'Not Found');
+  const snapshotId = await crawl('https://example.com', {
+    limit: 10,
+    depth: 2,
+    ignoreRobots: false,
+    rate: 1000
+  });
+  const graph = loadGraphFromSnapshot(snapshotId);
+  const nodes = graph.getNodes();
+  expect(nodes.length).toBe(3);
+  const root = graph.nodes.get('https://example.com/');
+  expect(root).toBeDefined();
+  expect(root?.depth).toBe(0);
+  expect(root?.outLinks).toBe(2);
+  const page1 = graph.nodes.get('https://example.com/page1');
+  expect(page1).toBeDefined();
+  expect(page1?.depth).toBe(1);
+  expect(page1?.inLinks).toBe(1);
+  const page2 = graph.nodes.get('https://example.com/page2');
+  expect(page2).toBeDefined();
+  expect(page2?.inLinks).toBe(2);
+});
+test('hard page limit', async () => {
+  const client = mockAgent.get('https://limit.com');
+  // Robots
+  client.intercept({ path: '/robots.txt', method: 'GET' }).reply(404, '');
+  // Root links to 1, 2, 3
+  client.intercept({ path: '/', method: 'GET' }).reply(200, `
+    <html><a href="/1">1</a><a href="/2">2</a><a href="/3">3</a></html>
+  `, { headers: { 'content-type': 'text/html' } });
+  // 1, 2, 3 return html
+  client.intercept({ path: '/1', method: 'GET' }).reply(200, '<html></html>', { headers: { 'content-type': 'text/html' } });
+  client.intercept({ path: '/2', method: 'GET' }).reply(200, '<html></html>', { headers: { 'content-type': 'text/html' } });
+  client.intercept({ path: '/3', method: 'GET' }).reply(200, '<html></html>', { headers: { 'content-type': 'text/html' } });
+  const snapshotId = await crawl('https://limit.com', {
+    limit: 2, // root + 1 page
+    depth: 5,
+    ignoreRobots: true,
+    rate: 1000
+  });
+  const graph = loadGraphFromSnapshot(snapshotId);
+  // Should have visited root + 1 other page (total 2 nodes with status > 0)
+  const crawledNodes = graph.getNodes().filter(n => n.status > 0);
+  expect(crawledNodes.length).toBeLessThanOrEqual(2);
+});
+test('hard depth cap', async () => {
+  const client = mockAgent.get('https://depth.com');
+  // Robots
+  client.intercept({ path: '/robots.txt', method: 'GET' }).reply(404, '');
+  // Chain of 12 pages
+  for (let i = 0; i < 12; i++) {
+    const path = i === 0 ? '/' : `/p${i}`;
+    const nextPath = `/p${i + 1}`;
+    client.intercept({ path, method: 'GET' }).reply(200, `
+      <html><a href="${nextPath}">Next</a></html>
+    `, { headers: { 'content-type': 'text/html' } });
+  }
+  const snapshotId = await crawl('https://depth.com', {
+    limit: 100,
+    depth: 20, // requested 20, but internal hard cap is 10
+    ignoreRobots: true,
+    rate: 1000
+  });
+  const graph = loadGraphFromSnapshot(snapshotId);
+  const crawledNodes = graph.getNodes().filter(n => n.status > 0);
+  const maxCrawledDepth = crawledNodes.reduce((max, n) => Math.max(max, n.depth), 0);
+  expect(maxCrawledDepth).toBeLessThanOrEqual(10);
+});
+test('parameter explosion control', async () => {
+  const client = mockAgent.get('https://params.com');
+  client.intercept({ path: '/robots.txt', method: 'GET' }).reply(404, '');
+  // Root links to many variations
+  let links = '';
+  for (let i = 0; i < 10; i++) {
+    links += `<a href="/search?q=${i}">q${i}</a>`;
+  }
+  client.intercept({ path: '/', method: 'GET' }).reply(200, `
+    <html>${links}</html>
+  `, { headers: { 'content-type': 'text/html' } });
+  // Intercept all variations
+  for (let i = 0; i < 40; i++) {
+    client.intercept({ path: `/search?q=${i}`, method: 'GET' }).reply(200, '<html></html>', { headers: { 'content-type': 'text/html' } });
+  }
+  const snapshotId = await crawl('https://params.com', {
+    limit: 100,
+    depth: 5,
+    ignoreRobots: true,
+    stripQuery: false,
+    detectTraps: true,
+    rate: 1000
+  });
+  const graph = loadGraphFromSnapshot(snapshotId);
+  // Should only crawl 5 variations + root
+  const nodes = graph.getNodes();
+  // Filter nodes that match /search pathname
+  const searchNodes = nodes.filter(n => n.url.includes('/search') && n.status > 0);
+  expect(searchNodes.length).toBeLessThanOrEqual(31);
+});
+test('redirect safety', async () => {
+  const client = mockAgent.get('https://redirect.com');
+  client.intercept({ path: '/robots.txt', method: 'GET' }).reply(404, '');
+  // Root -> /redir1
+  client.intercept({ path: '/', method: 'GET' }).reply(200, `
+    <html><a href="/redir1">Go</a></html>
+  `, { headers: { 'content-type': 'text/html' } });
+  // /redir1 -> 301 -> /dest
+  client.intercept({ path: '/redir1', method: 'GET' }).reply(301, '', {
+    headers: { 'location': '/dest' }
+  });
+  // /dest -> 200
+  client.intercept({ path: '/dest', method: 'GET' }).reply(200, '<html>Success</html>', { headers: { 'content-type': 'text/html' } });
+  const snapshotId = await crawl('https://redirect.com', {
+    limit: 10,
+    depth: 5,
+    ignoreRobots: true,
+    rate: 1000
+  });
+  const graph = loadGraphFromSnapshot(snapshotId);
+  const destNode = graph.nodes.get('https://redirect.com/dest');
+  expect(destNode).toBeDefined();
+  expect(destNode?.status).toBe(200);
+  // Redirect loop: A -> B -> A
+  const clientLoop = mockAgent.get('https://loop.com');
+  clientLoop.intercept({ path: '/robots.txt', method: 'GET' }).reply(404, '');
+  clientLoop.intercept({ path: '/', method: 'GET' }).reply(200, `
+    <html><a href="/a">Loop</a></html>
+  `, { headers: { 'content-type': 'text/html' } });
+  clientLoop.intercept({ path: '/a', method: 'GET' }).reply(301, '', { headers: { location: '/b' } });
+  clientLoop.intercept({ path: '/b', method: 'GET' }).reply(301, '', { headers: { location: '/a' } });
+  // We might mock /a again if it retries, but it shouldn't infinitely loop
+  const snapshotIdLoop = await crawl('https://loop.com', { limit: 10, depth: 5, ignoreRobots: true, rate: 1000 });
+  const graphLoop = loadGraphFromSnapshot(snapshotIdLoop);
+  // It should eventually stop
+  expect(graphLoop.getNodes().length).toBeGreaterThan(0);
+});
+test('mime check', async () => {
+  const client = mockAgent.get('https://mime.com');
+  client.intercept({ path: '/robots.txt', method: 'GET' }).reply(404, '');
+  client.intercept({ path: '/', method: 'GET' }).reply(200, `
+    <html><a href="/image.png">Img</a></html>
+  `, { headers: { 'content-type': 'text/html' } });
+  client.intercept({ path: '/data', method: 'GET' }).reply(200, `
+    <html><a href="/hidden">Hidden</a></html>
+  `, { headers: { 'content-type': 'application/json' } });
+  // Root links to /data
+  client.intercept({ path: '/start', method: 'GET' }).reply(200, `
+    <html><a href="/data">Data</a></html>
+  `, { headers: { 'content-type': 'text/html' } });
+  const snapshotId = await crawl('https://mime.com/start', { limit: 10, depth: 5, ignoreRobots: true, rate: 1000 });
+  const graph = loadGraphFromSnapshot(snapshotId);
+  // /data should be in graph
+  const dataNode = graph.nodes.get('https://mime.com/data');
+  expect(dataNode).toBeDefined();
+  // But we should NOT have parsed it, so /hidden should NOT be in graph
+  const hiddenNode = graph.nodes.get('https://mime.com/hidden');
+  expect(hiddenNode).toBeUndefined();
+});
+test('self-link guard', async () => {
+  const client = mockAgent.get('https://self.com');
+  client.intercept({ path: '/robots.txt', method: 'GET' }).reply(404, '');
+  client.intercept({ path: '/', method: 'GET' }).reply(200, `
+    <html><a href="/">Self</a><a href="/other">Other</a></html>
+  `, { headers: { 'content-type': 'text/html' } });
+  client.intercept({ path: '/other', method: 'GET' }).reply(200, '', { headers: { 'content-type': 'text/html' } });
+  const snapshotId = await crawl('https://self.com', { limit: 10, depth: 5, ignoreRobots: true, rate: 1000 });
+  const graph = loadGraphFromSnapshot(snapshotId);
+  const edges = graph.getEdges();
+  const selfEdge = edges.find(e => e.source === 'https://self.com/' && e.target === 'https://self.com/');
+  expect(selfEdge).toBeUndefined();
+  const otherEdge = edges.find(e => e.source === 'https://self.com/' && e.target === 'https://self.com/other');
+  expect(otherEdge).toBeDefined();
+});
+test('limit warning', async () => {
+  const client = mockAgent.get('https://warn.com');
+  client.intercept({ path: '/robots.txt', method: 'GET' }).reply(404, '');
+  client.intercept({ path: '/', method: 'GET' }).reply(200, `
+    <html><a href="/1">1</a><a href="/2">2</a></html>
+  `, { headers: { 'content-type': 'text/html' } });
+  client.intercept({ path: '/1', method: 'GET' }).reply(200, '', { headers: { 'content-type': 'text/html' } });
+  const snapshotId = await crawl('https://warn.com', { limit: 2, depth: 5, ignoreRobots: true, rate: 1000 });
+  const graph = loadGraphFromSnapshot(snapshotId);
+  expect(graph.limitReached).toBe(true);
+});
+test('seeds from sitemap', async () => {
+  const client = mockAgent.get('https://sitemap-seed.com');
+  client.intercept({ path: '/robots.txt', method: 'GET' }).reply(404, '');
+  // Sitemap
+  client.intercept({ path: '/sitemap.xml', method: 'GET' }).reply(200, `
+    <urlset><url><loc>https://sitemap-seed.com/page1</loc></url></urlset>
+  `);
+  // Root
+  client.intercept({ path: '/', method: 'GET' }).reply(200, '<html>Root</html>', { headers: { 'content-type': 'text/html' } });
+  // Page 1
+  client.intercept({ path: '/page1', method: 'GET' }).reply(200, '<html>Page 1</html>', { headers: { 'content-type': 'text/html' } });
+  const snapshotId = await crawl('https://sitemap-seed.com', {
+    limit: 10,
+    depth: 5,
+    ignoreRobots: true,
+    sitemap: 'true',
+    rate: 1000
+  });
+  const graph = loadGraphFromSnapshot(snapshotId);
+  const page1 = graph.nodes.get('https://sitemap-seed.com/page1');
+  expect(page1).toBeDefined();
+  expect(page1?.status).toBe(200);
+});
+test('incremental crawl uses etags', async () => {
+  const client = mockAgent.get('https://incremental.com');
+  client.intercept({ path: '/robots.txt', method: 'GET' }).reply(404, '');
+  // First crawl setup
+  client.intercept({ path: '/', method: 'GET' }).reply(200, 'Original', {
+    headers: { 'content-type': 'text/html', 'etag': '"v1"' }
+  });
+  const snapshotId1 = await crawl('https://incremental.com', { limit: 10, depth: 1, ignoreRobots: true, rate: 1000 });
+  const graph1 = loadGraphFromSnapshot(snapshotId1);
+  const node1 = graph1.nodes.get('https://incremental.com/');
+  expect(node1?.etag).toBe('"v1"');
+  // Second crawl setup
+  client.intercept({
+    path: '/',
+    method: 'GET',
+    headers: { 'If-None-Match': '"v1"' }
+  }).reply(304, '', { headers: { 'etag': '"v1"' } });
+  const snapshotId2 = await crawl('https://incremental.com', {
+    limit: 10,
+    depth: 1,
+    ignoreRobots: true,
+    previousGraph: graph1,
+    rate: 1000
+  });
+  const graph2 = loadGraphFromSnapshot(snapshotId2);
+  const node2 = graph2.nodes.get('https://incremental.com/');
+  expect(node2?.incrementalStatus).toBe('unchanged');
+});

package/tests/db.test.ts ADDED Viewed

@@ -0,0 +1,159 @@
+import { describe, it, expect, beforeEach, afterEach } from 'vitest';
+import Database from 'better-sqlite3';
+import { initSchema } from '../src/db/schema.js';
+import { SiteRepository } from '../src/db/repositories/SiteRepository.js';
+import { SnapshotRepository } from '../src/db/repositories/SnapshotRepository.js';
+import { PageRepository } from '../src/db/repositories/PageRepository.js';
+import { EdgeRepository } from '../src/db/repositories/EdgeRepository.js';
+import { MetricsRepository } from '../src/db/repositories/MetricsRepository.js';
+describe('Database Layer', () => {
+  let db: Database.Database;
+  let siteRepo: SiteRepository;
+  let snapshotRepo: SnapshotRepository;
+  let pageRepo: PageRepository;
+  let edgeRepo: EdgeRepository;
+  let metricsRepo: MetricsRepository;
+  beforeEach(() => {
+    db = new Database(':memory:');
+    initSchema(db);
+    siteRepo = new SiteRepository(db);
+    snapshotRepo = new SnapshotRepository(db);
+    pageRepo = new PageRepository(db);
+    edgeRepo = new EdgeRepository(db);
+    metricsRepo = new MetricsRepository(db);
+  });
+  afterEach(() => {
+    db.close();
+  });
+  it('should create and retrieve a site', () => {
+    const domain = 'example.com';
+    const id = siteRepo.createSite(domain);
+    expect(id).toBeGreaterThan(0);
+    const site = siteRepo.getSite(domain);
+    expect(site).toBeDefined();
+    expect(site?.domain).toBe(domain);
+  });
+  it('should create and retrieve a snapshot', () => {
+    const siteId = siteRepo.createSite('example.com');
+    const snapshotId = snapshotRepo.createSnapshot(siteId, 'full', 'running');
+    expect(snapshotId).toBeGreaterThan(0);
+    const snapshot = snapshotRepo.getLatestSnapshot(siteId);
+    expect(snapshot).toBeDefined();
+    expect(snapshot?.status).toBe('running');
+    snapshotRepo.updateSnapshotStatus(snapshotId, 'completed', { node_count: 10, edge_count: 5 });
+    const updated = snapshotRepo.getLatestSnapshot(siteId);
+    expect(updated?.status).toBe('completed');
+    expect(updated?.node_count).toBe(10);
+  });
+  it('should upsert pages', () => {
+    const siteId = siteRepo.createSite('example.com');
+    const snapshotId = snapshotRepo.createSnapshot(siteId, 'full');
+    const url = 'http://example.com';
+    // First insert
+    pageRepo.upsertPage({
+      site_id: siteId,
+      normalized_url: url,
+      last_seen_snapshot_id: snapshotId,
+      http_status: 200,
+      depth: 0
+    });
+    let page = pageRepo.getPage(siteId, url);
+    expect(page).toBeDefined();
+    expect(page?.first_seen_snapshot_id).toBe(snapshotId);
+    expect(page?.last_seen_snapshot_id).toBe(snapshotId);
+    expect(page?.http_status).toBe(200);
+    // Update (second snapshot)
+    const snapshotId2 = snapshotRepo.createSnapshot(siteId, 'incremental');
+    pageRepo.upsertPage({
+      site_id: siteId,
+      normalized_url: url,
+      last_seen_snapshot_id: snapshotId2,
+      http_status: 200, // same status
+      depth: 0
+    });
+    page = pageRepo.getPage(siteId, url);
+    expect(page?.first_seen_snapshot_id).toBe(snapshotId); // Should remain the first one
+    expect(page?.last_seen_snapshot_id).toBe(snapshotId2); // Should update to the second one
+  });
+  it('should persist new columns (nofollow, security_error, retries)', () => {
+    const siteId = siteRepo.createSite('new-cols.com');
+    const snapshotId = snapshotRepo.createSnapshot(siteId, 'full');
+    const url = 'http://new-cols.com';
+    pageRepo.upsertPage({
+      site_id: siteId,
+      normalized_url: url,
+      last_seen_snapshot_id: snapshotId,
+      nofollow: 1,
+      security_error: 'blocked',
+      retries: 3
+    });
+    const page = pageRepo.getPage(siteId, url);
+    expect(page?.nofollow).toBe(1);
+    expect(page?.security_error).toBe('blocked');
+    expect(page?.retries).toBe(3);
+  });
+  it('should insert and retrieve edges', () => {
+    const siteId = siteRepo.createSite('example.com');
+    const snapshotId = snapshotRepo.createSnapshot(siteId, 'full');
+    // Create pages first
+    pageRepo.upsertPage({ site_id: siteId, normalized_url: 'http://example.com/1', last_seen_snapshot_id: snapshotId });
+    pageRepo.upsertPage({ site_id: siteId, normalized_url: 'http://example.com/2', last_seen_snapshot_id: snapshotId });
+    const p1 = pageRepo.getPage(siteId, 'http://example.com/1')!;
+    const p2 = pageRepo.getPage(siteId, 'http://example.com/2')!;
+    edgeRepo.insertEdge(snapshotId, p1.id, p2.id, 1.0, 'internal');
+    const edges = edgeRepo.getEdgesBySnapshot(snapshotId);
+    expect(edges).toHaveLength(1);
+    expect(edges[0].source_page_id).toBe(p1.id);
+    expect(edges[0].target_page_id).toBe(p2.id);
+  });
+  it('should insert and retrieve metrics', () => {
+    const siteId = siteRepo.createSite('example.com');
+    const snapshotId = snapshotRepo.createSnapshot(siteId, 'full');
+    pageRepo.upsertPage({ site_id: siteId, normalized_url: 'http://example.com/1', last_seen_snapshot_id: snapshotId });
+    const p1 = pageRepo.getPage(siteId, 'http://example.com/1')!;
+    metricsRepo.insertMetrics({
+      snapshot_id: snapshotId,
+      page_id: p1.id,
+      authority_score: 0.5,
+      hub_score: 0.2,
+      pagerank: 0.8,
+      pagerank_score: 80.0,
+      link_role: 'authority',
+      crawl_status: 'fetched',
+      word_count: 100,
+      thin_content_score: 0.1,
+      external_link_ratio: 0.0,
+      orphan_score: 0,
+      duplicate_cluster_id: null,
+      duplicate_type: null,
+      is_cluster_primary: 0
+    });
+    const metrics = metricsRepo.getMetricsForPage(snapshotId, p1.id);
+    expect(metrics).toBeDefined();
+    expect(metrics?.authority_score).toBe(0.5);
+  });
+});

package/tests/diff.test.ts ADDED Viewed

@@ -0,0 +1,67 @@
+import { test, expect } from 'vitest';
+import { Graph } from '../src/graph/graph.js';
+import { compareGraphs } from '../src/diff/compare.js';
+test('detects added and removed urls', () => {
+  const oldGraph = new Graph();
+  oldGraph.addNode('https://example.com/a', 0, 200);
+  oldGraph.addNode('https://example.com/b', 1, 200);
+  const newGraph = new Graph();
+  newGraph.addNode('https://example.com/a', 0, 200);
+  newGraph.addNode('https://example.com/c', 1, 200); // Added
+  const diff = compareGraphs(oldGraph, newGraph);
+  expect(diff.addedUrls).toContain('https://example.com/c');
+  expect(diff.removedUrls).toContain('https://example.com/b');
+});
+test('detects status changes', () => {
+  const oldGraph = new Graph();
+  oldGraph.addNode('https://example.com/a', 0, 200);
+  const newGraph = new Graph();
+  newGraph.addNode('https://example.com/a', 0, 404);
+  const diff = compareGraphs(oldGraph, newGraph);
+  expect(diff.changedStatus).toHaveLength(1);
+  expect(diff.changedStatus[0]).toEqual({
+    url: 'https://example.com/a',
+    oldStatus: 200,
+    newStatus: 404
+  });
+});
+test('detects canonical changes', () => {
+  const oldGraph = new Graph();
+  oldGraph.addNode('https://example.com/a', 0, 200);
+  oldGraph.updateNodeData('https://example.com/a', { canonical: 'https://example.com/canon1' });
+  const newGraph = new Graph();
+  newGraph.addNode('https://example.com/a', 0, 200);
+  newGraph.updateNodeData('https://example.com/a', { canonical: 'https://example.com/canon2' });
+  const diff = compareGraphs(oldGraph, newGraph);
+  expect(diff.changedCanonical).toHaveLength(1);
+  expect(diff.changedCanonical[0]).toEqual({
+    url: 'https://example.com/a',
+    oldCanonical: 'https://example.com/canon1',
+    newCanonical: 'https://example.com/canon2'
+  });
+});
+test('calculates metric deltas', () => {
+  const oldGraph = new Graph();
+  // Orphan: A (depth 1, inLinks 0)
+  oldGraph.addNode('https://example.com/a', 1, 200);
+  const newGraph = new Graph();
+  // Not Orphan: Root -> A
+  newGraph.addNode('https://example.com/', 0, 200);
+  newGraph.addNode('https://example.com/a', 1, 200);
+  newGraph.addEdge('https://example.com/', 'https://example.com/a');
+  const diff = compareGraphs(oldGraph, newGraph);
+  // Old orphan count: 1 (A). New: 0. Delta: -1.
+  expect(diff.metricDeltas.orphanCount).toBe(-1);
+});