@aigne/doc-smith 0.8.11-beta.3 → 0.8.11-beta.5

package/prompts/translate/translate-document.md

@@ -1,26 +1,26 @@
 <role_and_goal>
-You are a professional translator proficient in multiple languages, skilled in accurate and standardized bilingual conversion.
+
+You are an **Elite Polyglot Localization and Translation Specialist** with extensive professional experience across multiple domains. Your core mission is to produce translations that are not only **100% accurate** to the source meaning but are also **natively fluent, highly readable, and culturally appropriate** in the target language.
+
+Core Mandates:
+
+1. Semantic Fidelity (Accuracy): The translation must perfectly and comprehensively convey the **entire meaning, tone, and nuance** of the source text. **No omission, addition, or distortion of the original content** is permitted.
+2. Native Fluency and Style: The resulting text must adhere strictly to the target language's **grammar, syntax, and idiomatic expressions**. The translation must **sound like it was originally written by a native speaker**, completely **free of grammatical errors** or "translationese" (literal, stiff, or unnatural phrasing).
+3. Readability and Flow: The final output must be **smooth, logical, and highly readable**. Sentences must flow naturally, ensuring a pleasant and coherent reading experience for the target audience.
+4. Localization and Clarity: Where a **literal (word-for-word) translation** of a term, phrase, or idiom would be **uncommon, confusing, or ambiguous** in the target language, you must apply **localization best practices**. This means translating the **concept** into the most **idiomatic, common, and easily understandable expression** in the target language.
+5. Versatility and Scope: You are proficient in handling **any pair of requested languages** (e.g., Chinese $\leftrightarrow$ English, English $\leftrightarrow$ Japanese) and are adept at translating diverse **document types**, including but not limited to: **Technical Manuals, Business Reports, Marketing Copy/Ads, Legal Documents, Academic Papers, and General Correspondence.**
 </role_and_goal>
 
 <translation_rules>
 Translation Requirements:
 
-- **Accurate Conveyance**: Accurately convey the facts and context of the original text, ensuring complete coverage.
-- **Avoid Exaggeration**: Avoid using emotionally charged or subjective words (for example, "excited" or "shocked").
-- **Follow Language Standards**: Ensure correct punctuation and grammar, and express ideas naturally and fluently.
-- **Preserve Original Structure**: Translate only the content portions without modifying tags or introducing any extra content or punctuation. Do not add markdown syntax at the outermost level. Ensure the translated structure matches the original, preserving line breaks and blank lines from the source.
-- **Strictly Protect Markdown Syntax**: All Markdown syntax characters, including but not limited to `|` and `-` in tables, `*` and `-` in lists, `#` in headings, `` ` `` in code blocks, etc., must be **copied exactly**, with no modification, addition, deletion, or merging. Table separators (e.g., `|---|---|---|`) must match the original column count and format exactly, with separator columns matching table data columns.
-- **Follow Translation Process**: Include literal translation, optimization, and omission checking to ensure the final output meets all requirements.
-- **Use Terminology Reference**: Ensure accuracy and consistency of professional terminology.
-- **Preserve Terms**: Retain specific terms in their original form, avoiding translation.
-
-Translation Process:
-
-- **Literal Translation**: Translate the original text word by word and sentence by sentence into the target language, ensuring every word's meaning is accurately conveyed.
-- **Optimization**: Based on the literal translation, ensure the text stays faithful to the original meaning while making it more natural and aligned with **{{ language }}** usage.
-- **Check for Omissions**: Compare the original with the literal translation to correct any meaning distortions or omissions.
-- **Format Check**: Compare the original with the literal translation to ensure the translated content is complete. If the original is in Markdown format, verify that the format matches the original.
-- **Final Output**: Output the optimized translation result, ensuring compliance with the above requirements (do not output the literal translation content).
+- Avoid Exaggeration: Avoid using emotionally charged or subjective words (for example, "excited" or "shocked").
+- Preserve Original Structure: Translate only the content portions without modifying tags or introducing any extra content or punctuation. Do not add markdown syntax at the outermost level. Ensure the translated structure matches the original, preserving line breaks and blank lines from the source.
+- Strictly Protect Markdown Syntax: All Markdown syntax characters, including but not limited to `|` and `-` in tables, `*` and `-` in lists, `#` in headings, `` ` `` in code blocks, etc., must be **copied exactly**, with no modification, addition, deletion, or merging. Table separators (e.g., `|---|---|---|`) must match the original column count and format exactly, with separator columns matching table data columns.
+- Use Terminology Reference: Ensure accuracy and consistency of professional terminology.
+- Preserve Terms: Retain specific terms in their original form, avoiding translation.
+
+{% include "./code-block.md" %}
 </translation_rules>
 
 
@@ -57,6 +57,9 @@ Terms to preserve (do not translate):
 </terms>
 
 <example>
+<example_item>
+**Special Note**: Keep table separators `|---|---|---|` unchanged from the original
+
 <before_translate>
 | Name | Type | Description |
 |---|---|---|
@@ -71,9 +74,10 @@ Terms to preserve (do not translate):
 | `id` | `string` | 要更新的 Webhook 端点的唯一标识符。 |
 | `data` | `PartialDeep<ABTNodeClient.WebhookEndpointStateInput>` | 包含要更新的 Webhook 端点字段的对象。 |
 </after_translate>
+</example_item>
 
-**Special Note**: Keep table separators `|---|---|---|` unchanged from the original
-
+<example_item>
+**Special Note**: All x-field component attributes must maintain the original format. Only translate the description content within data-desc attributes or x-field-desc elements
 <before_translate>
 
 <x-field data-name="teamDid" data-type="string" data-required="true" data-desc="The DID of the team or Blocklet managing the webhook."></x-field>
@@ -89,8 +93,100 @@ Terms to preserve (do not translate):
     <x-field-desc markdown>您的 **API 密钥**，用于身份验证。请从 `设置 > API 密钥` 部分生成一个。</x-field-desc>
 </x-field>
 </after_translate>
+</example_item>
+
+<example_item>
+**Special Note**: In code blocks, only translate comments while keeping all other code content (variables, functions, syntax) unchanged
+
+<before_translate>
+```xxx
+// Initialize the API client
+const client = new APIClient({
+  apiKey: 'your-api-key', // Replace with your actual API key
+  baseUrl: 'https://api.example.com'
+});
+
+const errorMessage = 'Failed to fetch user data';
+const successMessage = 'User data retrieved successfully';
+
+// Send request to get user data
+async function getUserData(userId) {
+  console.log('Starting user data fetch for ID:', userId);
+
+  try {
+    // Fetch user information from the API
+    const result = await client.get(`/users/${userId}`);
+    console.log('API response received');
+    console.log(successMessage);
+    return result;
+  } catch (error) {
+    console.error('Error occurred:', error.message);
+    throw new Error(errorMessage);
+  }
+}
+```
+</before_translate>
+
+<after_translate>
+```xxx
+// 初始化 API 客户端
+const client = new APIClient({
+  apiKey: 'your-api-key', // 替换为您的实际 API 密钥
+  baseUrl: 'https://api.example.com'
+});
+
+const errorMessage = 'Failed to fetch user data';
+const successMessage = 'User data retrieved successfully';
+
+// 发送请求获取用户数据
+async function getUserData(userId) {
+  console.log('Starting user data fetch for ID:', userId);
+
+  try {
+    // 从 API 获取用户信息
+    const result = await client.get(`/users/${userId}`);
+    console.log('API response received');
+    console.log(successMessage);
+    return result;
+  } catch (error) {
+    console.error('Error occurred:', error.message);
+    throw new Error(errorMessage);
+  }
+}
+```
+</after_translate>
+</example_item>
+
+<example_item>
+**Special Note**: **Command execution and log printing** should untranslated
+
+<before_translate>
+```text Timeout Error Message
+Blocklet Server failed to stop within 5 minutes
+You can stop blocklet server with blocklet stop --force
+```
+
+```bash Success Output
+$ cli log
+
+Cache for server cleared: [list of cleared cache keys]
+```
+</before_translate>
+
+<after_translate>
+```text 超时错误消息
+Blocklet Server failed to stop within 5 minutes
+You can stop blocklet server with blocklet stop --force
+```
+
+```bash 成功输出
+$ cli log
+
+Cache for server cleared: [list of cleared cache keys]
+```
+</after_translate>
+</example_item>
 
-**Special Note**: All x-field component attributes must maintain the original format. Only translate the description content within data-desc attributes or x-field-desc elements
 </example>
 
 Original text as follows:

package/tests/agents/init/init.test.mjs

@@ -986,11 +986,12 @@ describe("init", () => {
         const mockResponses = {
           checkbox_1: ["getStarted", "findAnswers"], // Document purpose
           checkbox_2: ["developers"], // Target audience
-          select_3: "domainFamiliar", // Reader knowledge level
-          select_4: "balancedCoverage", // Documentation depth
-          select_5: "en", // Primary language
-          checkbox_6: ["zh", "ja"], // Translation languages
-          input_7: join(tempDir, "docs"), // Documentation directory
+          input_3: "Custom rules for documentation", // Custom rules
+          select_4: "domainFamiliar", // Reader knowledge level
+          select_5: "balancedCoverage", // Documentation depth
+          select_6: "en", // Primary language
+          checkbox_7: ["zh", "ja"], // Translation languages
+          input_8: join(tempDir, "docs"), // Documentation directory
           search: "", // Source paths (empty to finish)
         };
 
@@ -1023,6 +1024,7 @@ describe("init", () => {
 
         expect(config.documentPurpose).toEqual(["getStarted", "findAnswers"]);
         expect(config.targetAudienceTypes).toEqual(["developers"]);
+        expect(config.rules).toBe("Custom rules for documentation");
         expect(config.readerKnowledgeLevel).toBe("domainFamiliar");
         expect(config.documentationDepth).toBe("balancedCoverage");
         expect(config.locale).toBe("en");
@@ -1042,11 +1044,12 @@ describe("init", () => {
           checkbox_1: ["mixedPurpose"], // Document purpose - triggers follow-up
           checkbox: ["completeTasks", "findAnswers"], // Top priorities after mixedPurpose
           checkbox_2: ["developers", "devops"], // Target audience
-          select_3: "experiencedUsers", // Reader knowledge level
-          select_4: "comprehensive", // Documentation depth
-          select_5: "zh-CN", // Primary language
-          checkbox_6: ["en"], // Translation languages
-          input_7: join(tempDir, "documentation"), // Documentation directory
+          input_3: "Custom rules for documentation", // Custom rules
+          select_4: "experiencedUsers", // Reader knowledge level
+          select_5: "comprehensive", // Documentation depth
+          select_6: "zh-CN", // Primary language
+          checkbox_7: ["en"], // Translation languages
+          input_8: join(tempDir, "documentation"), // Documentation directory
           search: "", // Source paths (empty to finish)
         };
 
@@ -1071,6 +1074,7 @@ describe("init", () => {
 
         expect(config.documentPurpose).toEqual(["completeTasks", "findAnswers"]);
         expect(config.targetAudienceTypes).toEqual(["developers", "devops"]);
+        expect(config.rules).toBe("Custom rules for documentation");
         expect(config.readerKnowledgeLevel).toBe("experiencedUsers");
         expect(config.documentationDepth).toBe("comprehensive");
         expect(config.locale).toBe("zh-CN");
@@ -1087,11 +1091,12 @@ describe("init", () => {
         const mockResponses = {
           checkbox_1: ["getStarted"], // Document purpose
           checkbox_2: ["endUsers"], // Target audience
-          select_3: "completeBeginners", // Reader knowledge level
-          select_4: "essentialOnly", // Documentation depth
-          select_5: "en", // Primary language
-          checkbox_6: [], // No translation languages
-          input_7: join(tempDir, "simple-docs"), // Documentation directory
+          input_3: "Custom rules for documentation", // Custom rules
+          select_4: "completeBeginners", // Reader knowledge level
+          select_5: "essentialOnly", // Documentation depth
+          select_6: "en", // Primary language
+          checkbox_7: [], // No translation languages
+          input_8: join(tempDir, "simple-docs"), // Documentation directory
           search: "", // Source paths (empty to finish)
         };
 
@@ -1115,6 +1120,7 @@ describe("init", () => {
 
         expect(config.documentPurpose).toEqual(["getStarted"]);
         expect(config.targetAudienceTypes).toEqual(["endUsers"]);
+        expect(config.rules).toBe("Custom rules for documentation");
         expect(config.readerKnowledgeLevel).toBe("completeBeginners");
         expect(config.documentationDepth).toBe("essentialOnly");
         expect(config.locale).toBe("en");
@@ -1298,7 +1304,7 @@ describe("init", () => {
         let validateCalled = false;
         const mockPrompts = {
           checkbox: (options) => {
-            if (options.message.includes("[1/8]") && options.validate) {
+            if (options.message.includes("[1/9]") && options.validate) {
               // Test the validation function directly
               const validationResult = options.validate([]);
               expect(validationResult).toBe(
@@ -1336,10 +1342,10 @@ describe("init", () => {
         let audienceValidateCalled = false;
         const mockPrompts = {
           checkbox: (options) => {
-            if (options.message.includes("[1/8]")) {
+            if (options.message.includes("[1/9]")) {
               return Promise.resolve(["getStarted"]); // Valid document purpose
             }
-            if (options.message.includes("[2/8]") && options.validate) {
+            if (options.message.includes("[2/9]") && options.validate) {
               // Test the validation function for target audience
               const validationResult = options.validate([]);
               expect(validationResult).toBe("Please choose at least one audience.");
@@ -1374,7 +1380,7 @@ describe("init", () => {
         let priorityValidateCalled = false;
         const mockPrompts = {
           checkbox: (options) => {
-            if (options.message.includes("[1/8]")) {
+            if (options.message.includes("[1/9]")) {
               return Promise.resolve(["mixedPurpose"]); // Trigger follow-up question
             }
             // This is the follow-up priority selection
@@ -1438,6 +1444,128 @@ describe("init", () => {
     });
   });
 
+  describe("checkOnly parameter", () => {
+    test("should exit silently when checkOnly is true and config exists", async () => {
+      const tempDir = await createTempDir();
+
+      try {
+        // Create existing config file
+        const configPath = join(tempDir, "config.yaml");
+        const existingConfig = `projectName: "Test Project"
+locale: "en"
+documentPurpose:
+  - getStarted
+targetAudienceTypes:
+  - developers`;
+        await fs.writeFile(configPath, existingConfig, "utf8");
+
+        // The function should continue normally when config exists
+        // No mocking needed as it doesn't log or exit in this case
+        const result = await init(
+          {
+            outputPath: tempDir,
+            fileName: "config.yaml",
+            checkOnly: true,
+          },
+          { prompts: {} }, // Options not needed for checkOnly
+        );
+
+        // Should return loaded config
+        expect(result).toBeDefined();
+        expect(result.projectName).toBe("Test Project");
+        expect(result.locale).toBe("en");
+      } finally {
+        await cleanupTempDir(tempDir);
+      }
+    });
+
+    test("should show reminder and exit when checkOnly is true and config doesn't exist", async () => {
+      const tempDir = await createTempDir();
+
+      try {
+        // Mock console.log and process.exit
+        const originalLog = console.log;
+        const originalExit = process.exit;
+        const logMessages = [];
+        let exitCalled = false;
+        let exitCode = null;
+
+        console.log = (...args) => logMessages.push(args.join(" "));
+        process.exit = (code) => {
+          exitCalled = true;
+          exitCode = code;
+          throw new Error("process.exit called"); // Prevent actual exit
+        };
+
+        try {
+          await expect(
+            init(
+              {
+                outputPath: tempDir,
+                fileName: "config.yaml",
+                checkOnly: true,
+              },
+              { prompts: {} },
+            ),
+          ).rejects.toThrow("process.exit called");
+
+          // Should log reminder messages
+          expect(logMessages.some((msg) => msg.includes("No configuration found"))).toBe(true);
+          expect(logMessages.some((msg) => msg.includes("aigne doc init"))).toBe(true);
+
+          // Should call process.exit(0)
+          expect(exitCalled).toBe(true);
+          expect(exitCode).toBe(0);
+        } finally {
+          console.log = originalLog;
+          process.exit = originalExit;
+        }
+      } finally {
+        await cleanupTempDir(tempDir);
+      }
+    });
+
+    test("should not interfere with normal workflow when checkOnly is false", async () => {
+      const tempDir = await createTempDir();
+
+      try {
+        const mockPrompts = createMockPrompts({
+          checkbox_1: ["getStarted"],
+          checkbox_2: ["developers"],
+          select_3: "domainFamiliar",
+          select_4: "balancedCoverage",
+          select_5: "en",
+          checkbox_6: [],
+          input_7: join(tempDir, "docs"),
+          search: "",
+        });
+
+        const options = { prompts: mockPrompts };
+
+        const result = await init(
+          {
+            outputPath: tempDir,
+            fileName: "config.yaml",
+            checkOnly: false, // Explicit false
+          },
+          options,
+        );
+
+        expect(result).toBeDefined();
+
+        // Check that config file was created normally
+        const configPath = join(tempDir, "config.yaml");
+        const configExists = await fs
+          .access(configPath)
+          .then(() => true)
+          .catch(() => false);
+        expect(configExists).toBe(true);
+      } finally {
+        await cleanupTempDir(tempDir);
+      }
+    });
+  });
+
   describe("Advanced source path scenarios", () => {
     test("should handle source path validation and duplicate detection", async () => {
       const tempDir = await createTempDir();

package/tests/agents/publish/publish-docs.test.mjs

@@ -15,6 +15,7 @@ import publishDocs from "../../../agents/publish/publish-docs.mjs";
 import * as authUtils from "../../../utils/auth-utils.mjs";
 import * as d2Utils from "../../../utils/d2-utils.mjs";
 import * as utils from "../../../utils/utils.mjs";
+import * as deployUtils from "../../../utils/deploy.mjs";
 
 // Mock all external dependencies
 const mockPublishDocs = {
@@ -51,6 +52,7 @@ describe("publish-docs", () => {
   let getGithubRepoUrlSpy;
   let loadConfigFromFileSpy;
   let saveValueToConfigSpy;
+  let deploySpy;
 
   beforeAll(() => {
     // Apply mocks for external dependencies only
@@ -98,6 +100,10 @@ describe("publish-docs", () => {
     );
     loadConfigFromFileSpy = spyOn(utils, "loadConfigFromFile").mockResolvedValue({});
     saveValueToConfigSpy = spyOn(utils, "saveValueToConfig").mockResolvedValue();
+    deploySpy = spyOn(deployUtils, "deploy").mockResolvedValue({
+      appUrl: "https://deployed.example.com",
+      token: "deploy-token",
+    });
 
     mockOptions = {
       prompts: {
@@ -125,6 +131,7 @@ describe("publish-docs", () => {
     getGithubRepoUrlSpy?.mockRestore();
     loadConfigFromFileSpy?.mockRestore();
     saveValueToConfigSpy?.mockRestore();
+    deploySpy?.mockRestore();
   });
 
   // BASIC FUNCTIONALITY TESTS
@@ -593,6 +600,98 @@ describe("publish-docs", () => {
     expect(mockOptions.prompts.select).not.toHaveBeenCalled();
   });
 
+  // RESUME PREVIOUS WEBSITE SETUP TESTS
+  test("should show resume option when checkoutId exists in config", async () => {
+    loadConfigFromFileSpy.mockResolvedValue({
+      checkoutId: "cached-checkout-123",
+    });
+    mockOptions.prompts.select.mockResolvedValue("default");
+
+    await publishDocs(
+      {
+        docsDir: "./docs",
+        appUrl: "https://docsmith.aigne.io",
+      },
+      mockOptions,
+    );
+
+    expect(mockOptions.prompts.select).toHaveBeenCalledWith(
+      expect.objectContaining({
+        message: "Select platform to publish your documents:",
+        choices: expect.arrayContaining([
+          expect.objectContaining({
+            name: expect.stringContaining("Resume previous website setup"),
+            value: "new-instance-continue",
+          }),
+        ]),
+      }),
+    );
+
+    // Verify the exact text content
+    const selectCall = mockOptions.prompts.select.mock.calls[0][0];
+    const resumeChoice = selectCall.choices.find(
+      (choice) => choice.value === "new-instance-continue",
+    );
+    expect(resumeChoice.name).toContain("Resume previous website setup");
+    expect(resumeChoice.name).toContain("Already paid.");
+    expect(resumeChoice.name).toContain(
+      "Continue where you left off. Your payment has already been processed.",
+    );
+  });
+
+  test("should not show resume option when no checkoutId in config", async () => {
+    loadConfigFromFileSpy.mockResolvedValue({});
+    mockOptions.prompts.select.mockResolvedValue("default");
+
+    await publishDocs(
+      {
+        docsDir: "./docs",
+        appUrl: "https://docsmith.aigne.io",
+      },
+      mockOptions,
+    );
+
+    expect(mockOptions.prompts.select).toHaveBeenCalledWith(
+      expect.objectContaining({
+        message: "Select platform to publish your documents:",
+        choices: expect.not.arrayContaining([
+          expect.objectContaining({
+            value: "new-instance-continue",
+          }),
+        ]),
+      }),
+    );
+  });
+
+  test("should handle resume previous website setup selection", async () => {
+    deploySpy.mockResolvedValue({
+      appUrl: "https://resumed.example.com",
+      token: "resume-token",
+    });
+
+    loadConfigFromFileSpy.mockResolvedValue({
+      checkoutId: "cached-checkout-123",
+      paymentUrl: "https://payment.example.com",
+    });
+    mockOptions.prompts.select.mockResolvedValue("new-instance-continue");
+
+    const consoleSpy = spyOn(console, "log").mockImplementation(() => {});
+
+    await publishDocs(
+      {
+        docsDir: "./docs",
+        appUrl: "https://docsmith.aigne.io",
+      },
+      mockOptions,
+    );
+
+    expect(consoleSpy).toHaveBeenCalledWith("\nResuming your previous website setup...");
+    expect(deploySpy).toHaveBeenCalledWith("cached-checkout-123", "https://payment.example.com");
+    expect(getAccessTokenSpy).toHaveBeenCalledWith("https://resumed.example.com", "resume-token");
+
+    consoleSpy.mockRestore();
+  });
+
   test("should handle URL validation edge cases", async () => {
     loadConfigFromFileSpy.mockResolvedValue({});
     mockOptions.prompts.select.mockResolvedValue("custom");

package/tests/agents/utils/check-detail-result.test.mjs

@@ -229,7 +229,7 @@ set -euo pipefail
 function deploy_app() {
     local app_name="$1"
     local version="$2"
-    
+
     echo "Deploying $app_name version $version"
     docker run --rm "$app_name:$version"
 }
@@ -337,19 +337,6 @@ This document demonstrates various programming language code blocks.`;
     });
   });
 
-  describe("D2 syntax validation", () => {
-    test("should handle D2 syntax errors", async () => {
-      const documentStructure = [];
-      const reviewContent =
-        "```d2\n" +
-        "invalid d2 syntax {{\n" + // Malformed D2
-        "```\n\n" +
-        "This has proper structure.";
-      const result = await checkDetailResult({ documentStructure, reviewContent });
-      expect(result.isApproved).toBe(false);
-    });
-  });
-
   describe("Advanced table edge cases", () => {
     test("should handle empty table cells correctly", async () => {
       const documentStructure = [];
@@ -660,7 +647,7 @@ flowchart TD
     A["1. Create Backend Implementation<br>api/src/providers/"]
     B["2. Add Backend Configuration<br>api/src/providers/models.ts"]
     C["3. Update Frontend Selector<br>src/pages/config/ai-providers/"]
-    
+
     A --> B --> C
 \`\`\`
 

package/tests/agents/utils/choose-docs.test.mjs

@@ -183,12 +183,7 @@ describe("chooseDocs utility", () => {
       locale: "en",
     };
 
-    await expect(chooseDocs(input, mockOptions)).rejects.toThrow(
-      "Please provide a docs parameter to specify which documents to update",
-    );
-    expect(consoleErrorSpy).toHaveBeenCalledWith(
-      new Error("No documents found in the docs directory"),
-    );
+    await expect(chooseDocs(input, mockOptions)).rejects.toThrow();
   });
 
   test("should throw error when no documents selected interactively", async () => {
@@ -202,9 +197,7 @@ describe("chooseDocs utility", () => {
       locale: "en",
     };
 
-    await expect(chooseDocs(input, mockOptions)).rejects.toThrow(
-      "Please provide a docs parameter to specify which documents to update",
-    );
+    await expect(chooseDocs(input, mockOptions)).rejects.toThrow();
   });
 
   // CHECKBOX VALIDATION TESTS

package/tests/utils/auth-utils.test.mjs

@@ -336,7 +336,7 @@ describe("auth-utils", () => {
 
     // Test that openPage calls the mock open function
     await capturedOpenPage("https://auth.example.com");
-    expect(mockOpen).toHaveBeenCalledWith("https://auth.example.com/");
+    expect(mockOpen).toHaveBeenCalledWith("https://auth.example.com/?required_roles=owner%2Cadmin");
   });
 
   test("should handle authorization failure", async () => {

package/tests/utils/d2-utils.test.mjs

@@ -6,7 +6,7 @@ import path from "node:path";
 
 import Debug from "debug";
 
-import { TMP_ASSETS_DIR } from "../../utils/constants/index.mjs";
+import { DOC_SMITH_DIR, TMP_ASSETS_DIR, TMP_DIR } from "../../utils/constants/index.mjs";
 import {
   beforePublishHook,
   checkContent,
@@ -346,7 +346,7 @@ E -> F
       try {
         await checkContent({ content });
 
-        const assetDir = path.join(process.cwd(), ".aigne", "doc-smith", ".tmp", "assets", "d2");
+        const assetDir = path.join(process.cwd(), DOC_SMITH_DIR, TMP_DIR, TMP_ASSETS_DIR, "d2");
         const files = await readdir(assetDir);
         const d2File = files.find((file) => file.endsWith(".d2"));
         expect(d2File).toBeDefined();
@@ -365,7 +365,7 @@ E -> F
       try {
         await ensureTmpDir();
 
-        const tmpDir = path.join(tempDir, ".aigne", "doc-smith", ".tmp");
+        const tmpDir = path.join(tempDir, DOC_SMITH_DIR, TMP_DIR);
         const gitignorePath = path.join(tmpDir, ".gitignore");
 
         expect(existsSync(tmpDir)).toBe(true);
@@ -386,7 +386,7 @@ E -> F
         // First call
         await ensureTmpDir();
 
-        const tmpDir = path.join(tempDir, ".aigne", "doc-smith", ".tmp");
+        const tmpDir = path.join(tempDir, DOC_SMITH_DIR, TMP_DIR);
         const gitignorePath = path.join(tmpDir, ".gitignore");
 
         // Modify .gitignore to test if it gets overwritten