refactor(tools): rename safe-grep to grep with override capability
This commit is contained in:
YeonGyu-Kim
@@ -0,0 +1,229 @@
|
||||
import { spawn } from "bun"
|
||||
import {
|
||||
resolveGrepCli,
|
||||
type GrepBackend,
|
||||
DEFAULT_MAX_DEPTH,
|
||||
DEFAULT_MAX_FILESIZE,
|
||||
DEFAULT_MAX_COUNT,
|
||||
DEFAULT_MAX_COLUMNS,
|
||||
DEFAULT_TIMEOUT_MS,
|
||||
DEFAULT_MAX_OUTPUT_BYTES,
|
||||
RG_SAFETY_FLAGS,
|
||||
GREP_SAFETY_FLAGS,
|
||||
} from "./constants"
|
||||
import type { GrepOptions, GrepMatch, GrepResult, CountResult } from "./types"
|
||||
|
||||
function buildRgArgs(options: GrepOptions): string[] {
|
||||
const args: string[] = [
|
||||
...RG_SAFETY_FLAGS,
|
||||
`--max-depth=${Math.min(options.maxDepth ?? DEFAULT_MAX_DEPTH, DEFAULT_MAX_DEPTH)}`,
|
||||
`--max-filesize=${options.maxFilesize ?? DEFAULT_MAX_FILESIZE}`,
|
||||
`--max-count=${Math.min(options.maxCount ?? DEFAULT_MAX_COUNT, DEFAULT_MAX_COUNT)}`,
|
||||
`--max-columns=${Math.min(options.maxColumns ?? DEFAULT_MAX_COLUMNS, DEFAULT_MAX_COLUMNS)}`,
|
||||
]
|
||||
|
||||
if (options.context !== undefined && options.context > 0) {
|
||||
args.push(`-C${Math.min(options.context, 10)}`)
|
||||
}
|
||||
|
||||
if (options.caseSensitive) args.push("--case-sensitive")
|
||||
if (options.wholeWord) args.push("-w")
|
||||
if (options.fixedStrings) args.push("-F")
|
||||
if (options.multiline) args.push("-U")
|
||||
if (options.hidden) args.push("--hidden")
|
||||
if (options.noIgnore) args.push("--no-ignore")
|
||||
|
||||
if (options.fileType?.length) {
|
||||
for (const type of options.fileType) {
|
||||
args.push(`--type=${type}`)
|
||||
}
|
||||
}
|
||||
|
||||
if (options.globs) {
|
||||
for (const glob of options.globs) {
|
||||
args.push(`--glob=${glob}`)
|
||||
}
|
||||
}
|
||||
|
||||
if (options.excludeGlobs) {
|
||||
for (const glob of options.excludeGlobs) {
|
||||
args.push(`--glob=!${glob}`)
|
||||
}
|
||||
}
|
||||
|
||||
return args
|
||||
}
|
||||
|
||||
function buildGrepArgs(options: GrepOptions): string[] {
|
||||
const args: string[] = [...GREP_SAFETY_FLAGS, "-r"]
|
||||
|
||||
if (options.context !== undefined && options.context > 0) {
|
||||
args.push(`-C${Math.min(options.context, 10)}`)
|
||||
}
|
||||
|
||||
if (!options.caseSensitive) args.push("-i")
|
||||
if (options.wholeWord) args.push("-w")
|
||||
if (options.fixedStrings) args.push("-F")
|
||||
|
||||
if (options.globs?.length) {
|
||||
for (const glob of options.globs) {
|
||||
args.push(`--include=${glob}`)
|
||||
}
|
||||
}
|
||||
|
||||
if (options.excludeGlobs?.length) {
|
||||
for (const glob of options.excludeGlobs) {
|
||||
args.push(`--exclude=${glob}`)
|
||||
}
|
||||
}
|
||||
|
||||
args.push("--exclude-dir=.git", "--exclude-dir=node_modules")
|
||||
|
||||
return args
|
||||
}
|
||||
|
||||
function buildArgs(options: GrepOptions, backend: GrepBackend): string[] {
|
||||
return backend === "rg" ? buildRgArgs(options) : buildGrepArgs(options)
|
||||
}
|
||||
|
||||
function parseOutput(output: string): GrepMatch[] {
|
||||
if (!output.trim()) return []
|
||||
|
||||
const matches: GrepMatch[] = []
|
||||
const lines = output.split("\n")
|
||||
|
||||
for (const line of lines) {
|
||||
if (!line.trim()) continue
|
||||
|
||||
const match = line.match(/^(.+?):(\d+):(.*)$/)
|
||||
if (match) {
|
||||
matches.push({
|
||||
file: match[1],
|
||||
line: parseInt(match[2], 10),
|
||||
text: match[3],
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
return matches
|
||||
}
|
||||
|
||||
function parseCountOutput(output: string): CountResult[] {
|
||||
if (!output.trim()) return []
|
||||
|
||||
const results: CountResult[] = []
|
||||
const lines = output.split("\n")
|
||||
|
||||
for (const line of lines) {
|
||||
if (!line.trim()) continue
|
||||
|
||||
const match = line.match(/^(.+?):(\d+)$/)
|
||||
if (match) {
|
||||
results.push({
|
||||
file: match[1],
|
||||
count: parseInt(match[2], 10),
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
return results
|
||||
}
|
||||
|
||||
export async function runRg(options: GrepOptions): Promise<GrepResult> {
|
||||
const cli = resolveGrepCli()
|
||||
const args = buildArgs(options, cli.backend)
|
||||
const timeout = Math.min(options.timeout ?? DEFAULT_TIMEOUT_MS, DEFAULT_TIMEOUT_MS)
|
||||
|
||||
if (cli.backend === "rg") {
|
||||
args.push("--", options.pattern)
|
||||
} else {
|
||||
args.push("-e", options.pattern)
|
||||
}
|
||||
|
||||
const paths = options.paths?.length ? options.paths : ["."]
|
||||
args.push(...paths)
|
||||
const proc = spawn([cli.path, ...args], {
|
||||
stdout: "pipe",
|
||||
stderr: "pipe",
|
||||
})
|
||||
|
||||
const timeoutPromise = new Promise<never>((_, reject) => {
|
||||
const id = setTimeout(() => {
|
||||
proc.kill()
|
||||
reject(new Error(`Search timeout after ${timeout}ms`))
|
||||
}, timeout)
|
||||
proc.exited.then(() => clearTimeout(id))
|
||||
})
|
||||
|
||||
try {
|
||||
const stdout = await Promise.race([new Response(proc.stdout).text(), timeoutPromise])
|
||||
const stderr = await new Response(proc.stderr).text()
|
||||
const exitCode = await proc.exited
|
||||
|
||||
const truncated = stdout.length >= DEFAULT_MAX_OUTPUT_BYTES
|
||||
const outputToProcess = truncated ? stdout.substring(0, DEFAULT_MAX_OUTPUT_BYTES) : stdout
|
||||
|
||||
if (exitCode > 1 && stderr.trim()) {
|
||||
return {
|
||||
matches: [],
|
||||
totalMatches: 0,
|
||||
filesSearched: 0,
|
||||
truncated: false,
|
||||
error: stderr.trim(),
|
||||
}
|
||||
}
|
||||
|
||||
const matches = parseOutput(outputToProcess)
|
||||
const filesSearched = new Set(matches.map((m) => m.file)).size
|
||||
|
||||
return {
|
||||
matches,
|
||||
totalMatches: matches.length,
|
||||
filesSearched,
|
||||
truncated,
|
||||
}
|
||||
} catch (e) {
|
||||
return {
|
||||
matches: [],
|
||||
totalMatches: 0,
|
||||
filesSearched: 0,
|
||||
truncated: false,
|
||||
error: e instanceof Error ? e.message : String(e),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
export async function runRgCount(options: Omit<GrepOptions, "context">): Promise<CountResult[]> {
|
||||
const cli = resolveGrepCli()
|
||||
const args = buildArgs({ ...options, context: 0 }, cli.backend)
|
||||
|
||||
if (cli.backend === "rg") {
|
||||
args.push("--count", "--", options.pattern)
|
||||
} else {
|
||||
args.push("-c", "-e", options.pattern)
|
||||
}
|
||||
|
||||
const paths = options.paths?.length ? options.paths : ["."]
|
||||
args.push(...paths)
|
||||
|
||||
const timeout = Math.min(options.timeout ?? DEFAULT_TIMEOUT_MS, DEFAULT_TIMEOUT_MS)
|
||||
const proc = spawn([cli.path, ...args], {
|
||||
stdout: "pipe",
|
||||
stderr: "pipe",
|
||||
})
|
||||
|
||||
const timeoutPromise = new Promise<never>((_, reject) => {
|
||||
const id = setTimeout(() => {
|
||||
proc.kill()
|
||||
reject(new Error(`Search timeout after ${timeout}ms`))
|
||||
}, timeout)
|
||||
proc.exited.then(() => clearTimeout(id))
|
||||
})
|
||||
|
||||
try {
|
||||
const stdout = await Promise.race([new Response(proc.stdout).text(), timeoutPromise])
|
||||
return parseCountOutput(stdout)
|
||||
} catch (e) {
|
||||
throw new Error(`Count search failed: ${e instanceof Error ? e.message : String(e)}`)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user