作為庫使用 Repomix

除了作為 CLI 工具使用 Repomix 外，您還可以將其功能直接集成到 Node.js 應用程序中。

安裝

在您的項目中安裝 Repomix 作為依賴項：

npm install repomix

基本用法

使用 Repomix 最簡單的方法是通過 runCli 函數，它提供與命令行界面相同的功能：

import { runCli, type CliOptions } from 'repomix';

// 使用自定義選項處理當前目錄
async function packProject() {
  const options = {
    output: 'output.xml',
    style: 'xml',
    compress: true,
    quiet: true
  } as CliOptions;
  
  const result = await runCli(['.'], process.cwd(), options);
  return result.packResult;
}

result.packResult 包含有關處理文件的信息，包括：

• totalFiles：處理的文件數量
• totalCharacters：總字符數
• totalTokens：總令牌數（對 LLM 上下文限制有用）
• fileCharCounts：每個文件的字符數
• fileTokenCounts：每個文件的令牌數

處理遠端倉庫

您可以克隆並處理遠端倉庫：

import { runCli, type CliOptions } from 'repomix';

// 克隆並處理 GitHub 倉庫
async function processRemoteRepo(repoUrl) {
  const options = {
    remote: repoUrl,
    output: 'output.xml',
    compress: true
  } as CliOptions;
  
  return await runCli(['.'], process.cwd(), options);
}

使用核心組件

要獲得更多控制，您可以直接使用 Repomix 的低級 API：

import { searchFiles, collectFiles, processFiles, TokenCounter } from 'repomix';

async function analyzeFiles(directory) {
  // 查找並收集文件
  const { filePaths } = await searchFiles(directory, { /* 配置 */ });
  const rawFiles = await collectFiles(filePaths, directory);
  const processedFiles = await processFiles(rawFiles, { /* 配置 */ });
  
  // 計算令牌
  const tokenCounter = new TokenCounter('o200k_base');
  
  // 返回分析結果
  return processedFiles.map(file => ({
    path: file.path,
    tokens: tokenCounter.countTokens(file.content)
  }));
}

實際示例

Repomix 網站（repomix.com）使用 Repomix 作為庫來處理遠端倉庫。您可以在 website/server/src/remoteRepo.ts 中查看實現。