@aigne/doc-smith 0.8.12-beta.7 → 0.8.12-beta.9

package/tests/agents/generate/document-structure-tools/delete-document.test.mjs

@@ -1,410 +0,0 @@
-import { afterEach, beforeEach, describe, expect, spyOn, test } from "bun:test";
-import deleteDocument from "../../../../agents/generate/document-structure-tools/delete-document.mjs";
-
-describe("delete-document", () => {
-  let consoleSpy;
-  let baseDocumentStructure;
-
-  beforeEach(() => {
-    consoleSpy = spyOn(console, "log").mockImplementation(() => {});
-    baseDocumentStructure = [
-      {
-        title: "Getting Started",
-        description: "Introduction to the project",
-        path: "/getting-started",
-        parentId: null,
-        sourceIds: ["intro.md"],
-      },
-      {
-        title: "API Reference",
-        description: "Complete API documentation",
-        path: "/api",
-        parentId: null,
-        sourceIds: ["api.js"],
-      },
-      {
-        title: "Authentication",
-        description: "How to authenticate with the API",
-        path: "/api/auth",
-        parentId: "/api",
-        sourceIds: ["auth.js", "security.md"],
-      },
-      {
-        title: "Rate Limiting",
-        description: "API rate limiting documentation",
-        path: "/api/rate-limiting",
-        parentId: "/api",
-        sourceIds: ["rate-limit.js"],
-      },
-      {
-        title: "OAuth",
-        description: "OAuth authentication flow",
-        path: "/api/auth/oauth",
-        parentId: "/api/auth",
-        sourceIds: ["oauth.js"],
-      },
-    ];
-  });
-
-  afterEach(() => {
-    consoleSpy?.mockRestore();
-  });
-
-  // SUCCESSFUL DELETION TESTS
-  test("should delete a leaf document successfully", async () => {
-    const result = await deleteDocument({
-      documentStructure: baseDocumentStructure,
-      path: "/api/rate-limiting",
-    });
-
-    expect(result.documentStructure).toHaveLength(4);
-    expect(result.deletedDocument).toEqual({
-      title: "Rate Limiting",
-      description: "API rate limiting documentation",
-      path: "/api/rate-limiting",
-      parentId: "/api",
-      sourceIds: ["rate-limit.js"],
-    });
-
-    // Verify the document is actually removed
-    expect(
-      result.documentStructure.find((doc) => doc.path === "/api/rate-limiting"),
-    ).toBeUndefined();
-
-    // Verify other documents remain
-    expect(result.documentStructure.find((doc) => doc.path === "/api")).toBeDefined();
-    expect(result.documentStructure.find((doc) => doc.path === "/api/auth")).toBeDefined();
-    expect(consoleSpy).not.toHaveBeenCalled();
-  });
-
-  test("should delete a deeply nested document successfully", async () => {
-    const result = await deleteDocument({
-      documentStructure: baseDocumentStructure,
-      path: "/api/auth/oauth",
-    });
-
-    expect(result.documentStructure).toHaveLength(4);
-    expect(result.deletedDocument).toEqual({
-      title: "OAuth",
-      description: "OAuth authentication flow",
-      path: "/api/auth/oauth",
-      parentId: "/api/auth",
-      sourceIds: ["oauth.js"],
-    });
-
-    // Verify the parent still exists
-    expect(result.documentStructure.find((doc) => doc.path === "/api/auth")).toBeDefined();
-    expect(consoleSpy).not.toHaveBeenCalled();
-  });
-
-  test("should delete a top-level document with no children", async () => {
-    const result = await deleteDocument({
-      documentStructure: baseDocumentStructure,
-      path: "/getting-started",
-    });
-
-    expect(result.documentStructure).toHaveLength(4);
-    expect(result.deletedDocument).toEqual({
-      title: "Getting Started",
-      description: "Introduction to the project",
-      path: "/getting-started",
-      parentId: null,
-      sourceIds: ["intro.md"],
-    });
-
-    expect(result.documentStructure.find((doc) => doc.path === "/getting-started")).toBeUndefined();
-    expect(consoleSpy).not.toHaveBeenCalled();
-  });
-
-  // VALIDATION ERROR TESTS
-  test("should return error when path is missing", async () => {
-    const result = await deleteDocument({
-      documentStructure: baseDocumentStructure,
-    });
-
-    expect(result.documentStructure).toEqual(baseDocumentStructure);
-    expect(result.deletedDocument).toBeUndefined();
-    expect(consoleSpy).toHaveBeenCalledWith("⚠️  Cannot delete document: path: Required");
-  });
-
-  test("should return error when path is empty string", async () => {
-    const result = await deleteDocument({
-      documentStructure: baseDocumentStructure,
-      path: "",
-    });
-
-    expect(result.documentStructure).toEqual(baseDocumentStructure);
-    expect(result.deletedDocument).toBeUndefined();
-    expect(consoleSpy).toHaveBeenCalledWith("⚠️  Cannot delete document: path: Path is required");
-  });
-
-  test("should return error when document does not exist", async () => {
-    const result = await deleteDocument({
-      documentStructure: baseDocumentStructure,
-      path: "/nonexistent-document",
-    });
-
-    expect(result.documentStructure).toEqual(baseDocumentStructure);
-    expect(result.deletedDocument).toBeUndefined();
-    expect(consoleSpy).toHaveBeenCalledWith(
-      "⚠️  Cannot delete document: Document '/nonexistent-document' does not exist. Please choose an existing document to delete.",
-    );
-  });
-
-  // CHILD DOCUMENTS VALIDATION TESTS
-  test("should return error when trying to delete document with child documents", async () => {
-    const result = await deleteDocument({
-      documentStructure: baseDocumentStructure,
-      path: "/api",
-    });
-
-    expect(result.documentStructure).toEqual(baseDocumentStructure);
-    expect(result.deletedDocument).toBeUndefined();
-    expect(consoleSpy).toHaveBeenCalledWith(
-      "⚠️  Cannot delete document: Document '/api' has 2 child document(s): /api/auth, /api/rate-limiting. Please first move or delete these child documents.",
-    );
-  });
-
-  test("should return error when trying to delete document with single child", async () => {
-    const result = await deleteDocument({
-      documentStructure: baseDocumentStructure,
-      path: "/api/auth",
-    });
-
-    expect(result.documentStructure).toEqual(baseDocumentStructure);
-    expect(result.deletedDocument).toBeUndefined();
-    expect(consoleSpy).toHaveBeenCalledWith(
-      "⚠️  Cannot delete document: Document '/api/auth' has 1 child document(s): /api/auth/oauth. Please first move or delete these child documents.",
-    );
-  });
-
-  test("should allow deletion after children are removed", async () => {
-    // First delete the child
-    const firstResult = await deleteDocument({
-      documentStructure: baseDocumentStructure,
-      path: "/api/auth/oauth",
-    });
-
-    // Then delete the parent
-    const secondResult = await deleteDocument({
-      documentStructure: firstResult.documentStructure,
-      path: "/api/auth",
-    });
-
-    expect(secondResult.documentStructure).toHaveLength(3);
-    expect(secondResult.deletedDocument.path).toBe("/api/auth");
-    expect(consoleSpy).not.toHaveBeenCalled();
-  });
-
-  // EDGE CASES
-  test("should handle empty documentation structure", async () => {
-    const result = await deleteDocument({
-      documentStructure: [],
-      path: "/any-path",
-    });
-
-    expect(result.documentStructure).toEqual([]);
-    expect(result.deletedDocument).toBeUndefined();
-    expect(consoleSpy).toHaveBeenCalledWith(
-      "⚠️  Cannot delete document: Document '/any-path' does not exist. Please choose an existing document to delete.",
-    );
-  });
-
-  test("should handle documentation structure with single document", async () => {
-    const singleDocStructure = [
-      {
-        title: "Only Document",
-        description: "The only document",
-        path: "/only",
-        parentId: null,
-        sourceIds: ["only.md"],
-      },
-    ];
-
-    const result = await deleteDocument({
-      documentStructure: singleDocStructure,
-      path: "/only",
-    });
-
-    expect(result.documentStructure).toEqual([]);
-    expect(result.deletedDocument).toEqual(singleDocStructure[0]);
-    expect(consoleSpy).not.toHaveBeenCalled();
-  });
-
-  test("should handle complex nested hierarchies", async () => {
-    const complexStructure = [
-      ...baseDocumentStructure,
-      {
-        title: "OAuth Scopes",
-        description: "OAuth scope documentation",
-        path: "/api/auth/oauth/scopes",
-        parentId: "/api/auth/oauth",
-        sourceIds: ["scopes.md"],
-      },
-      {
-        title: "JWT Tokens",
-        description: "JWT token handling",
-        path: "/api/auth/oauth/jwt",
-        parentId: "/api/auth/oauth",
-        sourceIds: ["jwt.js"],
-      },
-    ];
-
-    // Should not be able to delete parent with grandchildren
-    const result = await deleteDocument({
-      documentStructure: complexStructure,
-      path: "/api/auth/oauth",
-    });
-
-    expect(result.documentStructure).toEqual(complexStructure);
-    expect(consoleSpy).toHaveBeenCalledWith(
-      "⚠️  Cannot delete document: Document '/api/auth/oauth' has 2 child document(s): /api/auth/oauth/scopes, /api/auth/oauth/jwt. Please first move or delete these child documents.",
-    );
-  });
-
-  test("should correctly identify all direct children", async () => {
-    const structureWithManyChildren = [
-      {
-        title: "Parent",
-        description: "Parent document",
-        path: "/parent",
-        parentId: null,
-        sourceIds: ["parent.md"],
-      },
-      {
-        title: "Child 1",
-        description: "First child",
-        path: "/parent/child1",
-        parentId: "/parent",
-        sourceIds: ["child1.md"],
-      },
-      {
-        title: "Child 2",
-        description: "Second child",
-        path: "/parent/child2",
-        parentId: "/parent",
-        sourceIds: ["child2.md"],
-      },
-      {
-        title: "Child 3",
-        description: "Third child",
-        path: "/parent/child3",
-        parentId: "/parent",
-        sourceIds: ["child3.md"],
-      },
-      {
-        title: "Grandchild",
-        description: "Grandchild document",
-        path: "/parent/child1/grandchild",
-        parentId: "/parent/child1",
-        sourceIds: ["grandchild.md"],
-      },
-    ];
-
-    const result = await deleteDocument({
-      documentStructure: structureWithManyChildren,
-      path: "/parent",
-    });
-
-    expect(result.documentStructure).toEqual(structureWithManyChildren);
-    expect(consoleSpy).toHaveBeenCalledWith(
-      "⚠️  Cannot delete document: Document '/parent' has 3 child document(s): /parent/child1, /parent/child2, /parent/child3. Please first move or delete these child documents.",
-    );
-  });
-
-  // DATA INTEGRITY TESTS
-  test("should not modify original documentation structure", async () => {
-    const originalStructure = [...baseDocumentStructure];
-
-    await deleteDocument({
-      documentStructure: baseDocumentStructure,
-      path: "/api/rate-limiting",
-    });
-
-    expect(baseDocumentStructure).toEqual(originalStructure);
-  });
-
-  test("should preserve document order after deletion", async () => {
-    const result = await deleteDocument({
-      documentStructure: baseDocumentStructure,
-      path: "/api/auth",
-    });
-
-    // Since /api/auth has children, it should not be deleted
-    expect(result.documentStructure).toEqual(baseDocumentStructure);
-
-    // But if we delete a leaf node, order should be preserved
-    const leafResult = await deleteDocument({
-      documentStructure: baseDocumentStructure,
-      path: "/api/rate-limiting",
-    });
-
-    const expectedStructure = baseDocumentStructure.filter(
-      (doc) => doc.path !== "/api/rate-limiting",
-    );
-    expect(leafResult.documentStructure).toEqual(expectedStructure);
-  });
-
-  test("should handle special characters in paths", async () => {
-    const specialStructure = [
-      ...baseDocumentStructure,
-      {
-        title: "Special Document",
-        description: "Document with special characters",
-        path: "/api/special-chars_and.numbers123",
-        parentId: "/api",
-        sourceIds: ["special.md"],
-      },
-    ];
-
-    const result = await deleteDocument({
-      documentStructure: specialStructure,
-      path: "/api/special-chars_and.numbers123",
-    });
-
-    expect(result.documentStructure).toHaveLength(5);
-    expect(result.deletedDocument.path).toBe("/api/special-chars_and.numbers123");
-    expect(consoleSpy).not.toHaveBeenCalled();
-  });
-
-  // RETURN VALUE TESTS
-  test("should return complete deleted document information", async () => {
-    const result = await deleteDocument({
-      documentStructure: baseDocumentStructure,
-      path: "/api/rate-limiting",
-    });
-
-    expect(result.deletedDocument).toEqual({
-      title: "Rate Limiting",
-      description: "API rate limiting documentation",
-      path: "/api/rate-limiting",
-      parentId: "/api",
-      sourceIds: ["rate-limit.js"],
-    });
-
-    // Verify all properties are present
-    expect(result.deletedDocument).toHaveProperty("title");
-    expect(result.deletedDocument).toHaveProperty("description");
-    expect(result.deletedDocument).toHaveProperty("path");
-    expect(result.deletedDocument).toHaveProperty("parentId");
-    expect(result.deletedDocument).toHaveProperty("sourceIds");
-  });
-
-  test("should return updated documentation structure without deleted document", async () => {
-    const originalLength = baseDocumentStructure.length;
-    const result = await deleteDocument({
-      documentStructure: baseDocumentStructure,
-      path: "/getting-started",
-    });
-
-    expect(result.documentStructure).toHaveLength(originalLength - 1);
-    expect(result.documentStructure.every((doc) => doc.path !== "/getting-started")).toBe(true);
-
-    // Verify all other documents are still present
-    const remainingPaths = result.documentStructure.map((doc) => doc.path);
-    expect(remainingPaths).toContain("/api");
-    expect(remainingPaths).toContain("/api/auth");
-    expect(remainingPaths).toContain("/api/rate-limiting");
-    expect(remainingPaths).toContain("/api/auth/oauth");
-  });
-});

package/tests/agents/generate/document-structure-tools/generate-sub-structure.test.mjs

@@ -1,277 +0,0 @@
-import { afterEach, beforeEach, describe, expect, test } from "bun:test";
-import { mkdir, rm, writeFile } from "node:fs/promises";
-import { dirname, join } from "node:path";
-import { fileURLToPath } from "node:url";
-import generateSubStructure from "../../../../agents/generate/document-structure-tools/generate-sub-structure.mjs";
-
-const __dirname = dirname(fileURLToPath(import.meta.url));
-
-describe("generate-sub-structure", () => {
-  let testDir;
-
-  beforeEach(async () => {
-    testDir = join(__dirname, "test-generate-sub-structure");
-    await mkdir(testDir, { recursive: true });
-  });
-
-  afterEach(async () => {
-    try {
-      await rm(testDir, { recursive: true, force: true });
-    } catch {
-      // Ignore cleanup errors
-    }
-  });
-
-  describe("generateSubStructure function", () => {
-    test("should return empty subStructure when subSourcePaths is empty", async () => {
-      const result = await generateSubStructure(
-        {
-          parentDocument: {
-            title: "Test Parent",
-            description: "Test Description",
-            path: "/test",
-            parentId: "parent-1",
-            sourceIds: [],
-          },
-          subSourcePaths: [],
-        },
-        {},
-      );
-
-      expect(result).toBeDefined();
-      expect(result.subStructure).toBeDefined();
-      expect(Array.isArray(result.subStructure)).toBe(true);
-      expect(result.subStructure.length).toBe(0);
-      expect(result.message).toBeUndefined();
-    });
-
-    test("should process single source path with small context", async () => {
-      const testFile = join(testDir, "test.js");
-      await writeFile(testFile, "// Small test file\nconst x = 1;");
-
-      const mockContext = {
-        agents: {
-          generateStructureWithoutTools: "mock-agent-without-tools",
-        },
-        invoke: async (agent, params) => {
-          expect(agent).toBe("mock-agent-without-tools");
-          expect(params.isSubStructure).toBe(true);
-          expect(params.parentDocument).toBeDefined();
-          expect(params.datasources).toBeDefined();
-          expect(params.allFilesPaths).toBeDefined();
-          expect(params.isLargeContext).toBe(false);
-          expect(params.files).toBeDefined();
-          expect(params.totalTokens).toBeGreaterThan(0);
-
-          return {
-            documentStructure: [
-              {
-                title: "Test Document",
-                path: "/test-doc",
-                description: "Generated from test file",
-              },
-            ],
-          };
-        },
-      };
-
-      const result = await generateSubStructure(
-        {
-          parentDocument: {
-            title: "Parent",
-            path: "/parent",
-            description: "Parent doc",
-          },
-          subSourcePaths: [{ path: testFile, reason: "Test file" }],
-        },
-        { context: mockContext },
-      );
-
-      expect(result).toBeDefined();
-      expect(result.subStructure).toBeDefined();
-      expect(Array.isArray(result.subStructure)).toBe(true);
-      expect(result.subStructure.length).toBe(1);
-      expect(result.message).toContain("Generated a sub structure");
-      expect(result.message).toContain("/parent");
-    });
-
-    test("should process multiple source paths", async () => {
-      const testFile1 = join(testDir, "test1.js");
-      const testFile2 = join(testDir, "test2.js");
-      await writeFile(testFile1, "// Test file 1\nconst a = 1;");
-      await writeFile(testFile2, "// Test file 2\nconst b = 2;");
-
-      const mockContext = {
-        agents: {
-          generateStructureWithoutTools: "mock-agent-without-tools",
-        },
-        invoke: async (_agent, params) => {
-          expect(params.files.length).toBe(2);
-          expect(params.allFilesPaths).toContain("test1.js");
-          expect(params.allFilesPaths).toContain("test2.js");
-
-          return {
-            documentStructure: [
-              { title: "Doc 1", path: "/doc1" },
-              { title: "Doc 2", path: "/doc2" },
-            ],
-          };
-        },
-      };
-
-      const result = await generateSubStructure(
-        {
-          parentDocument: { title: "Parent", path: "/parent" },
-          subSourcePaths: [
-            { path: testFile1, reason: "First test" },
-            { path: testFile2, reason: "Second test" },
-          ],
-        },
-        { context: mockContext },
-      );
-
-      expect(result.subStructure.length).toBe(2);
-    });
-
-    test("should use generateStructure agent for large context", async () => {
-      const largeContent = `// Large file\n${"const x = 1;\n".repeat(150000)}`;
-      const testFile = join(testDir, "large.js");
-      await writeFile(testFile, largeContent);
-
-      const mockContext = {
-        agents: {
-          generateStructure: "mock-agent-with-tools",
-          generateStructureWithoutTools: "mock-agent-without-tools",
-        },
-        invoke: async (agent, params) => {
-          expect(agent).toBe("mock-agent-with-tools");
-          expect(params.isLargeContext).toBe(true);
-
-          return {
-            documentStructure: [{ title: "Large Doc", path: "/large" }],
-          };
-        },
-      };
-
-      const result = await generateSubStructure(
-        {
-          parentDocument: { title: "Parent", path: "/parent" },
-          subSourcePaths: [{ path: testFile, reason: "Large file" }],
-        },
-        { context: mockContext },
-      );
-
-      expect(result.subStructure).toBeDefined();
-      expect(result.subStructure.length).toBe(1);
-    });
-
-    test("should handle custom include and exclude patterns", async () => {
-      const srcDir = join(testDir, "src");
-      await mkdir(srcDir, { recursive: true });
-      await writeFile(join(srcDir, "index.js"), "// index");
-      await writeFile(join(srcDir, "test.spec.js"), "// test");
-      await writeFile(join(srcDir, "utils.ts"), "// utils");
-
-      const mockContext = {
-        agents: {
-          generateStructureWithoutTools: "mock-agent",
-        },
-        invoke: async (_agent, params) => {
-          const hasTestFile = params.files.some((f) => f.includes("test.spec.js"));
-          expect(hasTestFile).toBe(false);
-
-          return { documentStructure: [] };
-        },
-      };
-
-      await generateSubStructure(
-        {
-          parentDocument: { title: "Parent", path: "/parent" },
-          subSourcePaths: [{ path: testDir, reason: "Test directory" }],
-          includePatterns: ["**/*.js", "**/*.ts"],
-          excludePatterns: ["**/*.spec.js"],
-        },
-        { context: mockContext },
-      );
-    });
-
-    test("should deduplicate files in result", async () => {
-      const testFile = join(testDir, "test.js");
-      await writeFile(testFile, "// test");
-
-      const mockContext = {
-        agents: {
-          generateStructureWithoutTools: "mock-agent",
-        },
-        invoke: async (_agent, params) => {
-          const uniqueFiles = new Set(params.files);
-          expect(uniqueFiles.size).toBe(params.files.length);
-
-          return { documentStructure: [] };
-        },
-      };
-
-      await generateSubStructure(
-        {
-          parentDocument: { title: "Parent", path: "/parent" },
-          subSourcePaths: [
-            { path: testFile, reason: "First" },
-            { path: testFile, reason: "Second" },
-          ],
-        },
-        { context: mockContext },
-      );
-    });
-
-    test("should pass extra parameters to agent", async () => {
-      const testFile = join(testDir, "test.js");
-      await writeFile(testFile, "// test");
-
-      const mockContext = {
-        agents: {
-          generateStructureWithoutTools: "mock-agent",
-        },
-        invoke: async (_agent, params) => {
-          expect(params.extraParam1).toBe("value1");
-          expect(params.extraParam2).toBe("value2");
-
-          return { documentStructure: [] };
-        },
-      };
-
-      await generateSubStructure(
-        {
-          parentDocument: { title: "Parent", path: "/parent" },
-          subSourcePaths: [{ path: testFile, reason: "Test" }],
-          extraParam1: "value1",
-          extraParam2: "value2",
-        },
-        { context: mockContext },
-      );
-    });
-
-    test("should handle empty documentStructure from agent", async () => {
-      const testFile = join(testDir, "test.js");
-      await writeFile(testFile, "// test");
-
-      const mockContext = {
-        agents: {
-          generateStructureWithoutTools: "mock-agent",
-        },
-        invoke: async () => {
-          return {};
-        },
-      };
-
-      const result = await generateSubStructure(
-        {
-          parentDocument: { title: "Parent", path: "/parent" },
-          subSourcePaths: [{ path: testFile, reason: "Test" }],
-        },
-        { context: mockContext },
-      );
-
-      expect(result.subStructure).toEqual([]);
-    });
-  });
-});