作為庫使用 Repomix
除了作為 CLI 工具使用 Repomix 外,您還可以將其功能直接集成到 Node.js 應用程序中。
安裝
在您的項目中安裝 Repomix 作為依賴項:
bash
npm install repomix
基本用法
使用 Repomix 最簡單的方法是通過 runCli
函數,它提供與命令行界面相同的功能:
javascript
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
:每個文件的令牌數
處理遠端倉庫
您可以克隆並處理遠端倉庫:
javascript
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:
javascript
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 中查看實現。