ummaya 0.2.4 → 0.2.6

package/tui/src/tools/DocumentPrimitive/prompt.ts

@@ -0,0 +1,35 @@
+// SPDX-License-Identifier: Apache-2.0
+// UMMAYA-original — DocumentPrimitive prompt strings.
+
+export const DOCUMENT_TOOL_NAME = 'document'
+
+export const DESCRIPTION =
+  'Read, fill, validate, render, and save local public-document files through the document harness primitive.'
+
+export const DOCUMENT_TOOL_PROMPT = `Operate on local public-document files through UMMAYA's document harness.
+
+Use this primitive for HWPX, HWP, DOCX, PDF, XLSX, and PPTX document work when the user asks to understand, write, revise, validate, render, diff, or save a document.
+
+Input:
+- correlation_id: stable request id for evidence join.
+- document: { path?: string, artifact_id?: string, expected_format?: "hwpx" | "hwp" | "docx" | "pdf" | "xlsx" | "pptx" }.
+- operation: "inspect", "extract", "fill", "style", "validate", or "save".
+- instruction: natural-language instruction for the document harness.
+- destination_path: optional explicit local output path.
+- approved_draft_id and approved_draft_sha256: include both only after the user has approved a draft preview for narrative insertion.
+- requires_source_verification/source_support: for externally researched factual patches, include source_support with state "source_supported", citation_handle, source_sha256 matching the patch value, observed_at, and prompt_injection "not_detected".
+
+Rules:
+- For "read only" requests use inspect or extract.
+- If the user says not to write, save, or apply changes yet, do not call fill or save.
+- For "understand and write/fill/save" requests use fill or save; do not split into separate inspect/fill/render tool calls.
+- For question-first authoring requests, inspect the document structure first, ask the user for missing evidence, then make one fill/save call after approval.
+- Use either document.path or document.artifact_id, never both.
+- After a document result returns artifact_refs for an existing local artifact, use document.artifact_id only on follow-up document calls and omit document.path.
+- When a result needs more information, ask the user for missing evidence before another write attempt.
+- For self-introductions, business plans, essays, and narrative fields, collect user evidence, show a draft preview in assistant text, wait for user approval, then write only the approved draft with patches plus approved_draft_id and approved_draft_sha256.
+- Externally researched facts are never enough by themselves: write them only when the patch is source-supported and the user has separately approved the exact draft; if source support is missing, blocked, prompt-injected, stale, or unapproved, leave the field blank/question-waiting.
+- Approved narrative writes must include patches whose target_path comes from the inspected document field path; if you do not know the target path, ask the user instead of calling save.
+- Prefer explicit local paths supplied by the user.
+- Do not use workspace file tools for document-format editing; this primitive owns document edits, save evidence, and inline diff rendering.
+- The tool result is already rendered inline in the TUI; final answers should only summarize the visible outcome.`

package/tui/src/tools/FileEditTool/FileEditTool.ts

@@ -1,59 +1,10 @@
-import { dirname, isAbsolute, sep } from 'path'
-import { logEvent } from 'src/services/analytics/index.js'
-import { getFeatureValue_CACHED_MAY_BE_STALE } from '../../services/analytics/growthbook.js'
-import { diagnosticTracker } from '../../services/diagnosticTracking.js'
-import { clearDeliveredDiagnosticsForFile } from '../../services/lsp/LSPDiagnosticRegistry.js'
-import { getLspServerManager } from '../../services/lsp/manager.js'
-import { notifyVscodeFileUpdated } from '../../services/mcp/vscodeSdkMcp.js'
-import { checkTeamMemSecrets } from '../../services/teamMemorySync/teamMemSecretGuard.js'
-import {
-  activateConditionalSkillsForPaths,
-  addSkillDirectories,
-  discoverSkillDirsForPaths,
-} from '../../skills/loadSkillsDir.js'
-import type { ToolUseContext } from '../../Tool.js'
 import { buildTool, type ToolDef } from '../../Tool.js'
-import { getCwd } from '../../utils/cwd.js'
-import { logForDebugging } from '../../utils/debug.js'
-import { countLinesChanged } from '../../utils/diff.js'
-import { isEnvTruthy } from '../../utils/envUtils.js'
-import { isENOENT } from '../../utils/errors.js'
-import {
-  FILE_NOT_FOUND_CWD_NOTE,
-  findSimilarFile,
-  getFileModificationTime,
-  suggestPathUnderCwd,
-  writeTextContent,
-} from '../../utils/file.js'
-import {
-  fileHistoryEnabled,
-  fileHistoryTrackEdit,
-} from '../../utils/fileHistory.js'
-import { logFileOperation } from '../../utils/fileOperationAnalytics.js'
-import {
-  type LineEndingType,
-  readFileSyncWithMetadata,
-} from '../../utils/fileRead.js'
-import { formatFileSize } from '../../utils/format.js'
-import { getFsImplementation } from '../../utils/fsOperations.js'
-import {
-  fetchSingleFileGitDiff,
-  type ToolUseDiff,
-} from '../../utils/gitDiff.js'
-import { logError } from '../../utils/log.js'
 import { expandPath } from '../../utils/path.js'
-import {
-  checkWritePermissionForTool,
-  matchingRuleForInput,
-} from '../../utils/permissions/filesystem.js'
+import { checkWritePermissionForTool } from '../../utils/permissions/filesystem.js'
 import type { PermissionDecision } from '../../utils/permissions/PermissionResult.js'
 import { matchWildcardPattern } from '../../utils/permissions/shellRuleMatching.js'
-import { validateInputForSettingsFileEdit } from '../../utils/settings/validateEditTool.js'
-import { NOTEBOOK_EDIT_TOOL_NAME } from '../NotebookEditTool/constants.js'
-import {
-  FILE_EDIT_TOOL_NAME,
-  FILE_UNEXPECTEDLY_MODIFIED_ERROR,
-} from './constants.js'
+import { callFileEditTool } from './call.js'
+import { FILE_EDIT_TOOL_NAME } from './constants.js'
 import { getEditToolDescription } from './prompt.js'
 import {
   type FileEditInput,
@@ -69,19 +20,8 @@ import {
   renderToolUseRejectedMessage,
   userFacingName,
 } from './UI.js'
-import {
-  areFileEditsInputsEquivalent,
-  findActualString,
-  getPatchForEdit,
-  preserveQuoteStyle,
-} from './utils.js'
-
-// V8/Bun string length limit is ~2^30 characters (~1 billion). For typical
-// ASCII/Latin-1 files, 1 byte on disk = 1 character, so 1 GiB in stat bytes
-// ≈ 1 billion characters ≈ the runtime string limit. Multi-byte UTF-8 files
-// can be larger on disk per character, but 1 GiB is a safe byte-level guard
-// that prevents OOM without being unnecessarily restrictive.
-const MAX_EDIT_FILE_SIZE = 1024 * 1024 * 1024 // 1 GiB (stat bytes)
+import { areFileEditsInputsEquivalent } from './utils.js'
+import { validateFileEditInput } from './validateInput.js'
 
 export const FileEditTool = buildTool({
   name: FILE_EDIT_TOOL_NAME,
@@ -134,231 +74,8 @@ export const FileEditTool = buildTool({
   renderToolResultMessage,
   renderToolUseRejectedMessage,
   renderToolUseErrorMessage,
-  async validateInput(input: FileEditInput, toolUseContext: ToolUseContext) {
-    const { file_path, old_string, new_string, replace_all = false } = input
-    // Use expandPath for consistent path normalization (especially on Windows
-    // where "/" vs "\" can cause readFileState lookup mismatches)
-    const fullFilePath = expandPath(file_path)
-
-    // Reject edits to team memory files that introduce secrets
-    const secretError = checkTeamMemSecrets(fullFilePath, new_string)
-    if (secretError) {
-      return { result: false, message: secretError, errorCode: 0 }
-    }
-    if (old_string === new_string) {
-      return {
-        result: false,
-        behavior: 'ask',
-        message:
-          'No changes to make: old_string and new_string are exactly the same.',
-        errorCode: 1,
-      }
-    }
-
-    // Check if path should be ignored based on permission settings
-    const appState = toolUseContext.getAppState()
-    const denyRule = matchingRuleForInput(
-      fullFilePath,
-      appState.toolPermissionContext,
-      'edit',
-      'deny',
-    )
-    if (denyRule !== null) {
-      return {
-        result: false,
-        behavior: 'ask',
-        message:
-          'File is in a directory that is denied by your permission settings.',
-        errorCode: 2,
-      }
-    }
-
-    // SECURITY: Skip filesystem operations for UNC paths to prevent NTLM credential leaks.
-    // On Windows, fs.existsSync() on UNC paths triggers SMB authentication which could
-    // leak credentials to malicious servers. Let the permission check handle UNC paths.
-    if (fullFilePath.startsWith('\\\\') || fullFilePath.startsWith('//')) {
-      return { result: true }
-    }
-
-    const fs = getFsImplementation()
-
-    // Prevent OOM on multi-GB files.
-    try {
-      const { size } = await fs.stat(fullFilePath)
-      if (size > MAX_EDIT_FILE_SIZE) {
-        return {
-          result: false,
-          behavior: 'ask',
-          message: `File is too large to edit (${formatFileSize(size)}). Maximum editable file size is ${formatFileSize(MAX_EDIT_FILE_SIZE)}.`,
-          errorCode: 10,
-        }
-      }
-    } catch (e) {
-      if (!isENOENT(e)) {
-        throw e
-      }
-    }
-
-    // Read the file as bytes first so we can detect encoding from the buffer
-    // instead of calling detectFileEncoding (which does its own sync readSync
-    // and would fail with a wasted ENOENT when the file doesn't exist).
-    let fileContent: string | null
-    try {
-      const fileBuffer = await fs.readFileBytes(fullFilePath)
-      const encoding: BufferEncoding =
-        fileBuffer.length >= 2 &&
-        fileBuffer[0] === 0xff &&
-        fileBuffer[1] === 0xfe
-          ? 'utf16le'
-          : 'utf8'
-      fileContent = fileBuffer.toString(encoding).replaceAll('\r\n', '\n')
-    } catch (e) {
-      if (isENOENT(e)) {
-        fileContent = null
-      } else {
-        throw e
-      }
-    }
-
-    // File doesn't exist
-    if (fileContent === null) {
-      // Empty old_string on nonexistent file means new file creation — valid
-      if (old_string === '') {
-        return { result: true }
-      }
-      // Try to find a similar file with a different extension
-      const similarFilename = findSimilarFile(fullFilePath)
-      const cwdSuggestion = await suggestPathUnderCwd(fullFilePath)
-      let message = `File does not exist. ${FILE_NOT_FOUND_CWD_NOTE} ${getCwd()}.`
-
-      if (cwdSuggestion) {
-        message += ` Did you mean ${cwdSuggestion}?`
-      } else if (similarFilename) {
-        message += ` Did you mean ${similarFilename}?`
-      }
-
-      return {
-        result: false,
-        behavior: 'ask',
-        message,
-        errorCode: 4,
-      }
-    }
-
-    // File exists with empty old_string — only valid if file is empty
-    if (old_string === '') {
-      // Only reject if the file has content (for file creation attempt)
-      if (fileContent.trim() !== '') {
-        return {
-          result: false,
-          behavior: 'ask',
-          message: 'Cannot create new file - file already exists.',
-          errorCode: 3,
-        }
-      }
-
-      // Empty file with empty old_string is valid - we're replacing empty with content
-      return {
-        result: true,
-      }
-    }
-
-    if (fullFilePath.endsWith('.ipynb')) {
-      return {
-        result: false,
-        behavior: 'ask',
-        message: `File is a Jupyter Notebook. Use the ${NOTEBOOK_EDIT_TOOL_NAME} to edit this file.`,
-        errorCode: 5,
-      }
-    }
-
-    const readTimestamp = toolUseContext.readFileState.get(fullFilePath)
-    if (!readTimestamp || readTimestamp.isPartialView) {
-      return {
-        result: false,
-        behavior: 'ask',
-        message:
-          'File has not been read yet. Read it first before writing to it.',
-        meta: {
-          isFilePathAbsolute: String(isAbsolute(file_path)),
-        },
-        errorCode: 6,
-      }
-    }
-
-    // Check if file exists and get its last modified time
-    if (readTimestamp) {
-      const lastWriteTime = getFileModificationTime(fullFilePath)
-      if (lastWriteTime > readTimestamp.timestamp) {
-        // Timestamp indicates modification, but on Windows timestamps can change
-        // without content changes (cloud sync, antivirus, etc.). For full reads,
-        // compare content as a fallback to avoid false positives.
-        const isFullRead =
-          readTimestamp.offset === undefined &&
-          readTimestamp.limit === undefined
-        if (isFullRead && fileContent === readTimestamp.content) {
-          // Content unchanged, safe to proceed
-        } else {
-          return {
-            result: false,
-            behavior: 'ask',
-            message:
-              'File has been modified since read, either by the user or by a linter. Read it again before attempting to write it.',
-            errorCode: 7,
-          }
-        }
-      }
-    }
-
-    const file = fileContent
-
-    // Use findActualString to handle quote normalization
-    const actualOldString = findActualString(file, old_string)
-    if (!actualOldString) {
-      return {
-        result: false,
-        behavior: 'ask',
-        message: `String to replace not found in file.\nString: ${old_string}`,
-        meta: {
-          isFilePathAbsolute: String(isAbsolute(file_path)),
-        },
-        errorCode: 8,
-      }
-    }
-
-    const matches = file.split(actualOldString).length - 1
-
-    // Check if we have multiple matches but replace_all is false
-    if (matches > 1 && !replace_all) {
-      return {
-        result: false,
-        behavior: 'ask',
-        message: `Found ${matches} matches of the string to replace, but replace_all is false. To replace all occurrences, set replace_all to true. To replace only one occurrence, please provide more context to uniquely identify the instance.\nString: ${old_string}`,
-        meta: {
-          isFilePathAbsolute: String(isAbsolute(file_path)),
-          actualOldString,
-        },
-        errorCode: 9,
-      }
-    }
-
-    // Additional validation for Claude settings files
-    const settingsValidationResult = validateInputForSettingsFileEdit(
-      fullFilePath,
-      file,
-      () => {
-        // Simulate the edit to get the final content using the exact same logic as the tool
-        return replace_all
-          ? file.replaceAll(actualOldString, new_string)
-          : file.replace(actualOldString, new_string)
-      },
-    )
-
-    if (settingsValidationResult !== null) {
-      return settingsValidationResult
-    }
-
-    return { result: true, meta: { actualOldString } }
+  async validateInput(input: FileEditInput, toolUseContext) {
+    return validateFileEditInput(input, toolUseContext)
   },
   inputsEquivalent(input1, input2) {
     return areFileEditsInputsEquivalent(
@@ -384,193 +101,8 @@ export const FileEditTool = buildTool({
       },
     )
   },
-  async call(
-    input: FileEditInput,
-    {
-      readFileState,
-      userModified,
-      updateFileHistoryState,
-      dynamicSkillDirTriggers,
-    },
-    _,
-    parentMessage,
-  ) {
-    const { file_path, old_string, new_string, replace_all = false } = input
-
-    // 1. Get current state
-    const fs = getFsImplementation()
-    const absoluteFilePath = expandPath(file_path)
-
-    // Discover skills from this file's path (fire-and-forget, non-blocking)
-    // Skip in simple mode - no skills available
-    const cwd = getCwd()
-    if (!isEnvTruthy(process.env.CLAUDE_CODE_SIMPLE)) {
-      const newSkillDirs = await discoverSkillDirsForPaths(
-        [absoluteFilePath],
-        cwd,
-      )
-      if (newSkillDirs.length > 0) {
-        // Store discovered dirs for attachment display
-        for (const dir of newSkillDirs) {
-          dynamicSkillDirTriggers?.add(dir)
-        }
-        // Don't await - let skill loading happen in the background
-        addSkillDirectories(newSkillDirs).catch(() => {})
-      }
-
-      // Activate conditional skills whose path patterns match this file
-      activateConditionalSkillsForPaths([absoluteFilePath], cwd)
-    }
-
-    await diagnosticTracker.beforeFileEdited(absoluteFilePath)
-
-    // Ensure parent directory exists before the atomic read-modify-write section.
-    // These awaits must stay OUTSIDE the critical section below — a yield between
-    // the staleness check and writeTextContent lets concurrent edits interleave.
-    await fs.mkdir(dirname(absoluteFilePath))
-    if (fileHistoryEnabled()) {
-      // Backup captures pre-edit content — safe to call before the staleness
-      // check (idempotent v1 backup keyed on content hash; if staleness fails
-      // later we just have an unused backup, not corrupt state).
-      await fileHistoryTrackEdit(
-        updateFileHistoryState,
-        absoluteFilePath,
-        parentMessage.uuid,
-      )
-    }
-
-    // 2. Load current state and confirm no changes since last read
-    // Please avoid async operations between here and writing to disk to preserve atomicity
-    const {
-      content: originalFileContents,
-      fileExists,
-      encoding,
-      lineEndings: endings,
-    } = readFileForEdit(absoluteFilePath)
-
-    if (fileExists) {
-      const lastWriteTime = getFileModificationTime(absoluteFilePath)
-      const lastRead = readFileState.get(absoluteFilePath)
-      if (!lastRead || lastWriteTime > lastRead.timestamp) {
-        // Timestamp indicates modification, but on Windows timestamps can change
-        // without content changes (cloud sync, antivirus, etc.). For full reads,
-        // compare content as a fallback to avoid false positives.
-        const isFullRead =
-          lastRead &&
-          lastRead.offset === undefined &&
-          lastRead.limit === undefined
-        const contentUnchanged =
-          isFullRead && originalFileContents === lastRead.content
-        if (!contentUnchanged) {
-          throw new Error(FILE_UNEXPECTEDLY_MODIFIED_ERROR)
-        }
-      }
-    }
-
-    // 3. Use findActualString to handle quote normalization
-    const actualOldString =
-      findActualString(originalFileContents, old_string) || old_string
-
-    // Preserve curly quotes in new_string when the file uses them
-    const actualNewString = preserveQuoteStyle(
-      old_string,
-      actualOldString,
-      new_string,
-    )
-
-    // 4. Generate patch
-    const { patch, updatedFile } = getPatchForEdit({
-      filePath: absoluteFilePath,
-      fileContents: originalFileContents,
-      oldString: actualOldString,
-      newString: actualNewString,
-      replaceAll: replace_all,
-    })
-
-    // 5. Write to disk
-    writeTextContent(absoluteFilePath, updatedFile, encoding, endings)
-
-    // Notify LSP servers about file modification (didChange) and save (didSave)
-    const lspManager = getLspServerManager()
-    if (lspManager) {
-      // Clear previously delivered diagnostics so new ones will be shown
-      clearDeliveredDiagnosticsForFile(`file://${absoluteFilePath}`)
-      // didChange: Content has been modified
-      lspManager
-        .changeFile(absoluteFilePath, updatedFile)
-        .catch((err: Error) => {
-          logForDebugging(
-            `LSP: Failed to notify server of file change for ${absoluteFilePath}: ${err.message}`,
-          )
-          logError(err)
-        })
-      // didSave: File has been saved to disk (triggers diagnostics in TypeScript server)
-      lspManager.saveFile(absoluteFilePath).catch((err: Error) => {
-        logForDebugging(
-          `LSP: Failed to notify server of file save for ${absoluteFilePath}: ${err.message}`,
-        )
-        logError(err)
-      })
-    }
-
-    // Notify VSCode about the file change for diff view
-    notifyVscodeFileUpdated(absoluteFilePath, originalFileContents, updatedFile)
-
-    // 6. Update read timestamp, to invalidate stale writes
-    readFileState.set(absoluteFilePath, {
-      content: updatedFile,
-      timestamp: getFileModificationTime(absoluteFilePath),
-      offset: undefined,
-      limit: undefined,
-    })
-
-    // 7. Log events
-    if (absoluteFilePath.endsWith(`${sep}CLAUDE.md`)) {
-      logEvent('tengu_write_claudemd', {})
-    }
-    countLinesChanged(patch)
-
-    logFileOperation({
-      operation: 'edit',
-      tool: 'FileEditTool',
-      filePath: absoluteFilePath,
-    })
-
-    logEvent('tengu_edit_string_lengths', {
-      oldStringBytes: Buffer.byteLength(old_string, 'utf8'),
-      newStringBytes: Buffer.byteLength(new_string, 'utf8'),
-      replaceAll: replace_all,
-    })
-
-    let gitDiff: ToolUseDiff | undefined
-    if (
-      isEnvTruthy(process.env.CLAUDE_CODE_REMOTE) &&
-      getFeatureValue_CACHED_MAY_BE_STALE('tengu_quartz_lantern', false)
-    ) {
-      const startTime = Date.now()
-      const diff = await fetchSingleFileGitDiff(absoluteFilePath)
-      if (diff) gitDiff = diff
-      logEvent('tengu_tool_use_diff_computed', {
-        isEditTool: true,
-        durationMs: Date.now() - startTime,
-        hasDiff: !!diff,
-      })
-    }
-
-    // 8. Yield result
-    const data = {
-      filePath: file_path,
-      oldString: actualOldString,
-      newString: new_string,
-      originalFile: originalFileContents,
-      structuredPatch: patch,
-      userModified: userModified ?? false,
-      replaceAll: replace_all,
-      ...(gitDiff && { gitDiff }),
-    }
-    return {
-      data,
-    }
+  async call(input: FileEditInput, context, _, parentMessage) {
+    return callFileEditTool(input, context, parentMessage)
   },
   mapToolResultToToolResultBlockParam(data: FileEditOutput, toolUseID) {
     const { filePath, userModified, replaceAll } = data
@@ -593,33 +125,3 @@ export const FileEditTool = buildTool({
     }
   },
 } satisfies ToolDef<ReturnType<typeof inputSchema>, FileEditOutput>)
-
-// --
-
-function readFileForEdit(absoluteFilePath: string): {
-  content: string
-  fileExists: boolean
-  encoding: BufferEncoding
-  lineEndings: LineEndingType
-} {
-  try {
-    // eslint-disable-next-line custom-rules/no-sync-fs
-    const meta = readFileSyncWithMetadata(absoluteFilePath)
-    return {
-      content: meta.content,
-      fileExists: true,
-      encoding: meta.encoding,
-      lineEndings: meta.lineEndings,
-    }
-  } catch (e) {
-    if (isENOENT(e)) {
-      return {
-        content: '',
-        fileExists: false,
-        encoding: 'utf8',
-        lineEndings: 'LF',
-      }
-    }
-    throw e
-  }
-}