Initial public release
This commit is contained in:
@@ -0,0 +1,129 @@
|
||||
import { create } from 'zustand';
|
||||
import { devtools } from 'zustand/middleware';
|
||||
import { opencodeClient, type ProjectFileSearchHit } from '@/lib/opencode/client';
|
||||
|
||||
const CACHE_TTL_MS = 30_000;
|
||||
const MAX_CACHE_ENTRIES = 40;
|
||||
const DEFAULT_SEARCH_LIMIT = 60;
|
||||
|
||||
interface FileSearchCacheEntry {
|
||||
files: ProjectFileSearchHit[];
|
||||
timestamp: number;
|
||||
}
|
||||
|
||||
interface FileSearchStoreState {
|
||||
cache: Record<string, FileSearchCacheEntry>;
|
||||
cacheKeys: string[];
|
||||
inFlight: Record<string, Promise<ProjectFileSearchHit[]>>;
|
||||
searchFiles: (directory: string, query: string, limit?: number) => Promise<ProjectFileSearchHit[]>;
|
||||
invalidateDirectory: (directory?: string | null) => void;
|
||||
}
|
||||
|
||||
const buildCacheKey = (directory: string, query: string, limit: number) => {
|
||||
const normalizedDirectory = directory.trim();
|
||||
const normalizedQuery = query.trim().toLowerCase();
|
||||
return `${normalizedDirectory}::${normalizedQuery}::${limit}`;
|
||||
};
|
||||
|
||||
export const useFileSearchStore = create<FileSearchStoreState>()(
|
||||
devtools(
|
||||
(set, get) => ({
|
||||
cache: {},
|
||||
cacheKeys: [],
|
||||
inFlight: {},
|
||||
async searchFiles(directory, query, limit = DEFAULT_SEARCH_LIMIT) {
|
||||
if (!directory || directory.trim().length === 0) {
|
||||
return [];
|
||||
}
|
||||
|
||||
const normalizedDirectory = directory.trim();
|
||||
const normalizedQuery = typeof query === 'string' ? query.trim() : '';
|
||||
const key = buildCacheKey(normalizedDirectory, normalizedQuery, limit);
|
||||
const now = Date.now();
|
||||
const cached = get().cache[key];
|
||||
|
||||
if (cached && now - cached.timestamp < CACHE_TTL_MS) {
|
||||
return cached.files;
|
||||
}
|
||||
|
||||
const inflight = get().inFlight[key];
|
||||
if (inflight) {
|
||||
return inflight;
|
||||
}
|
||||
|
||||
const searchPromise = opencodeClient
|
||||
.searchFiles(normalizedQuery, { directory: normalizedDirectory, limit })
|
||||
.then((files) => {
|
||||
set((state) => {
|
||||
const nextCache = { ...state.cache, [key]: { files, timestamp: Date.now() } };
|
||||
const nextKeys = state.cacheKeys.filter((cacheKey) => cacheKey !== key);
|
||||
nextKeys.push(key);
|
||||
|
||||
while (nextKeys.length > MAX_CACHE_ENTRIES) {
|
||||
const oldestKey = nextKeys.shift();
|
||||
if (oldestKey) {
|
||||
delete nextCache[oldestKey];
|
||||
}
|
||||
}
|
||||
|
||||
return {
|
||||
cache: nextCache,
|
||||
cacheKeys: nextKeys,
|
||||
};
|
||||
});
|
||||
return files;
|
||||
})
|
||||
.finally(() => {
|
||||
set((state) => {
|
||||
const nextInFlight = { ...state.inFlight };
|
||||
delete nextInFlight[key];
|
||||
return { inFlight: nextInFlight };
|
||||
});
|
||||
});
|
||||
|
||||
set((state) => ({
|
||||
inFlight: {
|
||||
...state.inFlight,
|
||||
[key]: searchPromise,
|
||||
},
|
||||
}));
|
||||
|
||||
return searchPromise;
|
||||
},
|
||||
invalidateDirectory(directory) {
|
||||
if (!directory || directory.trim().length === 0) {
|
||||
set({ cache: {}, cacheKeys: [], inFlight: {} });
|
||||
return;
|
||||
}
|
||||
|
||||
const normalizedDirectory = directory.trim();
|
||||
const prefix = `${normalizedDirectory}::`;
|
||||
|
||||
set((state) => {
|
||||
const nextCache = { ...state.cache };
|
||||
const nextKeys = state.cacheKeys.filter((cacheKey) => {
|
||||
if (cacheKey.startsWith(prefix)) {
|
||||
delete nextCache[cacheKey];
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
});
|
||||
|
||||
const nextInFlightEntries = Object.entries(state.inFlight).filter(
|
||||
([key]) => !key.startsWith(prefix)
|
||||
);
|
||||
const nextInFlight = Object.fromEntries(nextInFlightEntries);
|
||||
|
||||
return {
|
||||
cache: nextCache,
|
||||
cacheKeys: nextKeys,
|
||||
inFlight: nextInFlight,
|
||||
};
|
||||
});
|
||||
},
|
||||
}),
|
||||
{
|
||||
name: 'file-search-store',
|
||||
}
|
||||
)
|
||||
);
|
||||
Reference in New Issue
Block a user