FM / memory

/opt/canhelp/node_modules/@anthropic-ai/sdk/src/tools/memory/

/opt/canhelp/node_modules/@anthropic-ai/sdk/src/tools/memory

Name	Size	Mode	Actions
node.ts	13190	0644	edit dl rm

Edit: /opt/canhelp/node_modules/@anthropic-ai/sdk/src/tools/memory/node.ts (13190B)

export { betaMemoryTool } from '../../helpers/beta/memory';
export type { MemoryToolHandlers } from '../../helpers/beta/memory';

import * as fs from 'fs/promises';
import * as path from 'path';
import { randomUUID } from 'crypto';
import type { MemoryToolHandlers } from '../../helpers/beta/memory';
import {
  BetaMemoryTool20250818CreateCommand,
  BetaMemoryTool20250818DeleteCommand,
  BetaMemoryTool20250818InsertCommand,
  BetaMemoryTool20250818RenameCommand,
  BetaMemoryTool20250818StrReplaceCommand,
  BetaMemoryTool20250818ViewCommand,
} from '../../resources/beta';

async function exists(path: string) {
  return await fs
    .access(path)
    .then(() => true)
    .catch((err) => {
      if (err.code === 'ENOENT') return false;
      throw err;
    });
}

/**
 * Atomically writes content to a file by writing to a temporary file first and then renaming it.
 * This ensures the target file is never in a partially written state, preventing data corruption
 * if the process crashes or is interrupted during the write operation. The rename operation is
 * atomic on most file systems, guaranteeing that readers will only ever see the complete old
 * content or the complete new content, never a mix or partial state.
 *
 * @param targetPath - The path where the file should be written
 * @param content - The content to write to the file
 */
async function atomicWriteFile(targetPath: string, content: string): Promise<void> {
  const dir = path.dirname(targetPath);
  const tempPath = path.join(dir, `.tmp-${process.pid}-${randomUUID()}`);

let handle: fs.FileHandle | undefined;
  try {
    handle = await fs.open(tempPath, 'wx');
    await handle.writeFile(content, 'utf-8');
    await handle.sync();
    await handle.close();
    handle = undefined;

await fs.rename(tempPath, targetPath);
  } catch (err) {
    if (handle) {
      await handle.close().catch(() => {});
    }
    await fs.unlink(tempPath).catch(() => {});
    throw err;
  }
}

/**
 * Validates that a target path doesn't escape the memory root via symlinks.
 *
 * Prevents symlink attacks where a malicious symlink inside /memories points
 * outside (e.g., /memories/foo -> /etc), which would allow operations like
 * creating /memories/foo/passwd to actually write to /etc/passwd.
 *
 * Walks up from the target path to find the deepest existing ancestor,
 * then resolves it to ensure the real path stays within memoryRoot.
 */
async function validateNoSymlinkEscape(targetPath: string, memoryRoot: string): Promise<void> {
  const resolvedRoot = await fs.realpath(memoryRoot);

async function readFileContent(fullPath: string, memoryPath: string): Promise<string> {
  try {
    return await fs.readFile(fullPath, 'utf-8');
  } catch (err: any) {
    if (err.code === 'ENOENT') {
      throw new Error(
        `The file ${memoryPath} no longer exists (may have been deleted or renamed concurrently).`,
      );
    }
    throw err;
  }
}

function formatFileSize(bytes: number): string {
  if (bytes === 0) return '0B';
  const k = 1024;
  const sizes = ['B', 'K', 'M', 'G'];
  const i = Math.floor(Math.log(bytes) / Math.log(k));
  const size = bytes / Math.pow(k, i);

return (size % 1 === 0 ? size.toString() : size.toFixed(1)) + sizes[i];
}

const MAX_LINES = 999999;
const LINE_NUMBER_WIDTH = String(MAX_LINES).length;

export class BetaLocalFilesystemMemoryTool implements MemoryToolHandlers {
  private basePath: string;
  private memoryRoot: string;

constructor(basePath: string = './memory') {
    this.basePath = basePath;
    this.memoryRoot = path.join(this.basePath, 'memories');
  }

static async init(basePath: string = './memory'): Promise<BetaLocalFilesystemMemoryTool> {
    const memory = new BetaLocalFilesystemMemoryTool(basePath);

await fs.mkdir(memory.memoryRoot, { recursive: true });

return memory;
  }

private async validatePath(memoryPath: string): Promise<string> {
    if (!memoryPath.startsWith('/memories')) {
      throw new Error(`Path must start with /memories, got: ${memoryPath}`);
    }

const relativePath = memoryPath.slice('/memories'.length).replace(/^\//, '');
    const fullPath = relativePath ? path.join(this.memoryRoot, relativePath) : this.memoryRoot;

const resolvedPath = path.resolve(fullPath);
    const resolvedRoot = path.resolve(this.memoryRoot);
    if (!resolvedPath.startsWith(resolvedRoot)) {
      throw new Error(`Path ${memoryPath} would escape /memories directory`);
    }

await validateNoSymlinkEscape(resolvedPath, this.memoryRoot);

return resolvedPath;
  }

async view(command: BetaMemoryTool20250818ViewCommand): Promise<string> {
    const fullPath = await this.validatePath(command.path);

let stat;
    try {
      stat = await fs.stat(fullPath);
    } catch (err: any) {
      if (err.code === 'ENOENT') {
        throw new Error(`The path ${command.path} does not exist. Please provide a valid path.`);
      }
      throw err;
    }

if (stat.isDirectory()) {
      const items: Array<{ size: string; path: string }> = [];

const collectItems = async (dirPath: string, relativePath: string, depth: number): Promise<void> => {
        if (depth > 2) return;

const dirContents = await fs.readdir(dirPath);

for (const item of dirContents.sort()) {
          if (item.startsWith('.') || item === 'node_modules') {
            continue;
          }
          const itemPath = path.join(dirPath, item);
          const itemRelativePath = relativePath ? `${relativePath}/${item}` : item;
          let itemStat;
          try {
            itemStat = await fs.stat(itemPath);
          } catch {
            continue;
          }

if (itemStat.isDirectory()) {
            items.push({ size: formatFileSize(itemStat.size), path: `${itemRelativePath}/` });
            if (depth < 2) {
              await collectItems(itemPath, itemRelativePath, depth + 1);
            }
          } else if (itemStat.isFile()) {
            items.push({ size: formatFileSize(itemStat.size), path: itemRelativePath });
          }
        }
      };

await collectItems(fullPath, '', 1);

return `${header}\n${lines.join('\n')}`;
    } else if (stat.isFile()) {
      const content = await readFileContent(fullPath, command.path);
      const lines = content.split('\n');

if (lines.length > MAX_LINES) {
        throw new Error(
          `File ${command.path} has too many lines (${
            lines.length
          }). Maximum is ${MAX_LINES.toLocaleString()} lines.`,
        );
      }

let displayLines = lines;
      let startNum = 1;

if (command.view_range && command.view_range.length === 2) {
        const startLine = Math.max(1, command.view_range[0]!) - 1;
        const endLine = command.view_range[1] === -1 ? lines.length : command.view_range[1];
        displayLines = lines.slice(startLine, endLine);
        startNum = startLine + 1;
      }

const numberedLines = displayLines.map(
        (line, i) => `${String(i + startNum).padStart(LINE_NUMBER_WIDTH, ' ')}\t${line}`,
      );

return `Here's the content of ${command.path} with line numbers:\n${numberedLines.join('\n')}`;
    } else {
      throw new Error(`Unsupported file type for ${command.path}`);
    }
  }

async create(command: BetaMemoryTool20250818CreateCommand): Promise<string> {
    const fullPath = await this.validatePath(command.path);

await fs.mkdir(path.dirname(fullPath), { recursive: true });

let handle: fs.FileHandle | undefined;
    try {
      handle = await fs.open(fullPath, 'wx');
      await handle.writeFile(command.file_text, 'utf-8');
      await handle.sync();
    } catch (err: any) {
      if (err?.code === 'EEXIST') {
        throw new Error(`File ${command.path} already exists`);
      }
      throw err;
    } finally {
      await handle?.close().catch(() => {});
    }

return `File created successfully at: ${command.path}`;
  }

async str_replace(command: BetaMemoryTool20250818StrReplaceCommand): Promise<string> {
    const fullPath = await this.validatePath(command.path);

if (!stat.isFile()) {
      throw new Error(`The path ${command.path} is not a file.`);
    }

const content = await readFileContent(fullPath, command.path);
    const lines = content.split('\n');

const matchingLines: number[] = [];
    lines.forEach((line, index) => {
      if (line.includes(command.old_str)) {
        matchingLines.push(index + 1);
      }
    });

if (matchingLines.length === 0) {
      throw new Error(
        `No replacement was performed, old_str \`${command.old_str}\` did not appear verbatim in ${command.path}.`,
      );
    } else if (matchingLines.length > 1) {
      throw new Error(
        `No replacement was performed. Multiple occurrences of old_str \`${
          command.old_str
        }\` in lines: ${matchingLines.join(', ')}. Please ensure it is unique`,
      );
    }

const newContent = content.replace(command.old_str, command.new_str);
    await atomicWriteFile(fullPath, newContent);

return `The memory file has been edited. Here is the snippet showing the change (with line numbers):\n${snippet.join(
      '\n',
    )}`;
  }

async insert(command: BetaMemoryTool20250818InsertCommand): Promise<string> {
    const fullPath = await this.validatePath(command.path);

if (!stat.isFile()) {
      throw new Error(`The path ${command.path} is not a file.`);
    }

const content = await readFileContent(fullPath, command.path);
    const lines = content.split('\n');

if (command.insert_line < 0 || command.insert_line > lines.length) {
      throw new Error(
        `Invalid \`insert_line\` parameter: ${command.insert_line}. It should be within the range of lines of the file: [0, ${lines.length}]`,
      );
    }

lines.splice(command.insert_line, 0, command.insert_text.replace(/\n$/, ''));
    await atomicWriteFile(fullPath, lines.join('\n'));
    return `The file ${command.path} has been edited.`;
  }

async delete(command: BetaMemoryTool20250818DeleteCommand): Promise<string> {
    const fullPath = await this.validatePath(command.path);

if (command.path === '/memories') {
      throw new Error('Cannot delete the /memories directory itself');
    }

try {
      await fs.rm(fullPath, { recursive: true, force: false });
    } catch (err: any) {
      if (err.code === 'ENOENT') {
        throw new Error(`The path ${command.path} does not exist`);
      }
      throw err;
    }

return `Successfully deleted ${command.path}`;
  }

async rename(command: BetaMemoryTool20250818RenameCommand): Promise<string> {
    const oldFullPath = await this.validatePath(command.old_path);
    const newFullPath = await this.validatePath(command.new_path);

// POSIX rename() silently overwrites existing files without error,
    // so we can't catch this atomically. Best-effort check to warn user.
    if (await exists(newFullPath)) {
      throw new Error(`The destination ${command.new_path} already exists`);
    }

const newDir = path.dirname(newFullPath);
    await fs.mkdir(newDir, { recursive: true });

try {
      await fs.rename(oldFullPath, newFullPath);
    } catch (err: any) {
      if (err.code === 'ENOENT') {
        throw new Error(`The path ${command.old_path} does not exist`);
      }
      throw err;
    }

return `Successfully renamed ${command.old_path} to ${command.new_path}`;
  }
}

cmd: