/**
* query command - Search for terms in the index
*/
import { existsSync } from 'fs';
import { join } from 'path';
import { PRODUCT_NAME, INDEX_DIR, TOOL_PREFIX } from '../constants.js';
import { openDatabase, createQueries, type Queries } from '../db/index.js';
import { globToRegex } from '../utils/glob.js';
// ============================================================
// Types
// ============================================================
export type QueryMode = 'exact' | 'contains' | 'starts_with';
export interface QueryParams {
path: string;
term: string;
mode?: QueryMode;
fileFilter?: string;
typeFilter?: string[];
modifiedSince?: string;
modifiedBefore?: string;
limit?: number;
}
export interface QueryMatch {
file: string;
lineNumber: number;
lineType: string;
modified?: number;
}
export interface QueryResult {
success: boolean;
term: string;
mode: QueryMode;
matches: QueryMatch[];
totalMatches: number;
truncated: boolean;
error?: string;
}
// ============================================================
// Main query function
// ============================================================
export function query(params: QueryParams): QueryResult {
const mode = params.mode ?? 'exact';
const limit = params.limit ?? 100;
// Validate project path
const dbPath = join(params.path, INDEX_DIR, 'index.db');
if (!existsSync(dbPath)) {
return {
success: false,
term: params.term,
mode,
matches: [],
totalMatches: 0,
truncated: false,
error: `No ${PRODUCT_NAME} index found at ${params.path}. Run ${TOOL_PREFIX}init first.`,
};
}
// Open database
const db = openDatabase(dbPath, true);
const queries = createQueries(db);
try {
// Search for items
const items = queries.searchItems(params.term, mode, 1000);
if (items.length === 0) {
db.close();
return {
success: true,
term: params.term,
mode,
matches: [],
totalMatches: 0,
truncated: false,
};
}
// Parse time filters
const modifiedSinceTs = params.modifiedSince ? parseTimeOffset(params.modifiedSince) : null;
const modifiedBeforeTs = params.modifiedBefore ? parseTimeOffset(params.modifiedBefore) : null;
// Pre-compile file filter regex
const fileFilterRegex = params.fileFilter ? globToRegex(params.fileFilter) : null;
// Collect all occurrences
let allMatches: QueryMatch[] = [];
for (const item of items) {
const occurrences = queries.getOccurrencesByItem(item.id);
for (const occ of occurrences) {
// Apply file filter
if (fileFilterRegex && !fileFilterRegex.test(occ.path.replace(/\\/g, '/'))) {
continue;
}
// Apply type filter
if (params.typeFilter && params.typeFilter.length > 0) {
if (!params.typeFilter.includes(occ.line_type)) {
continue;
}
}
// Apply time filters
if (modifiedSinceTs !== null && occ.modified !== null) {
if (occ.modified < modifiedSinceTs) {
continue;
}
}
if (modifiedBeforeTs !== null && occ.modified !== null) {
if (occ.modified > modifiedBeforeTs) {
continue;
}
}
allMatches.push({
file: occ.path,
lineNumber: occ.line_number,
lineType: occ.line_type,
modified: occ.modified ?? undefined,
});
}
}
// Remove duplicates (same file + line)
const seen = new Set<string>();
allMatches = allMatches.filter(m => {
const key = `${m.file}:${m.lineNumber}`;
if (seen.has(key)) return false;
seen.add(key);
return true;
});
// Sort by file, then line number
allMatches.sort((a, b) => {
const fileCompare = a.file.localeCompare(b.file);
if (fileCompare !== 0) return fileCompare;
return a.lineNumber - b.lineNumber;
});
const totalMatches = allMatches.length;
const truncated = allMatches.length > limit;
if (truncated) {
allMatches = allMatches.slice(0, limit);
}
db.close();
return {
success: true,
term: params.term,
mode,
matches: allMatches,
totalMatches,
truncated,
};
} catch (error) {
db.close();
return {
success: false,
term: params.term,
mode,
matches: [],
totalMatches: 0,
truncated: false,
error: error instanceof Error ? error.message : String(error),
};
}
}
// ============================================================
// Helper functions
// ============================================================
/**
* Parse time offset string to Unix timestamp
* Supports: "2h" (hours), "30m" (minutes), "1d" (days), "1w" (weeks), or ISO date string
*/
export function parseTimeOffset(input: string): number | null {
if (!input) return null;
// Try relative time format: 2h, 30m, 1d, 1w
const match = input.match(/^(\d+)([mhdw])$/i);
if (match) {
const value = parseInt(match[1], 10);
const unit = match[2].toLowerCase();
const now = Date.now();
switch (unit) {
case 'm': return now - value * 60 * 1000; // minutes
case 'h': return now - value * 60 * 60 * 1000; // hours
case 'd': return now - value * 24 * 60 * 60 * 1000; // days
case 'w': return now - value * 7 * 24 * 60 * 60 * 1000; // weeks
}
}
// Try ISO date string
const date = new Date(input);
if (!isNaN(date.getTime())) {
return date.getTime();
}
return null;
}
