dsh-locator 0.5.2 → 0.6.0

package/src/index.ts

@@ -7,7 +7,9 @@
 import type { Context } from '@deepseek-ai/cordis'
 import { Config as ConfigSchema } from './config.ts'
 import type { Config } from './config.ts'
+import { ContentIndex } from './content.ts'
 import { makeExportTool } from './export.ts'
+import { makeFindTool } from './find.ts'
 import { registerHttp } from './http.ts'
 import { resolveBinary, resolveRoots, resolveStateDir } from './paths.ts'
 import { LocatorRuntime } from './runtime.ts'
@@ -34,15 +36,26 @@ export function apply(ctx: Context, config: Config): void {
   const binary = resolveBinary(config, cwd)
   const logger = ctx.logger('dsh-locator')
   const rt = new LocatorRuntime(config, roots, stateDir, binary, cwd)
+  const content = config.contentIndex
+    ? new ContentIndex(rt, {
+        stateDir,
+        sessionDirs: config.contentSessionDirs,
+        maxFileBytes: config.contentMaxFileBytes,
+        maxFiles: config.contentMaxFiles,
+        maxSessions: config.contentMaxSessions,
+      })
+    : undefined
 
   ctx.effect(() => {
     // Let the host finish starting before a cold scan starts reading the disk.
     const timer = setTimeout(() => {
       rt.start()
       logger.info(`index roots: ${roots.join(', ')} (state ${stateDir})`)
+      if (content !== undefined) void startContent(rt, content, logger)
     }, 1500)
     return () => {
       clearTimeout(timer)
+      content?.close()
       void rt.stop()
     }
   })
@@ -51,8 +64,24 @@ export function apply(ctx: Context, config: Config): void {
   ctx.tools.register(makeStatusTool(rt))
   ctx.tools.register(makeRebuildTool(rt))
   ctx.tools.register(makeExportTool(rt))
+  if (content !== undefined) ctx.tools.register(makeFindTool(content))
 
   // The Web GUI panel is optional: profiles without a Web server keep the
   // three tools and never mount an HTTP route.
-  registerHttp(ctx, rt)
+  registerHttp(ctx, rt, content)
+}
+
+async function startContent(rt: LocatorRuntime, content: ContentIndex, logger: { info(message: string): void }): Promise<void> {
+  try {
+    await rt.waitReady()
+    for (let i = 0; i < 900; i++) {
+      const status = await rt.engineStatus()
+      if (status === undefined || !status.scanning) break
+      await new Promise(resolve => setTimeout(resolve, 2_000))
+    }
+    content.start()
+    logger.info('content index started')
+  } catch (error) {
+    logger.info(`content index skipped: ${error instanceof Error ? error.message : String(error)}`)
+  }
 }

package/src/runtime.ts

@@ -157,6 +157,7 @@ export class LocatorRuntime {
     if (!this.config.watch) args.push('--no-watch')
     if (this.config.refreshIntervalMs > 0) args.push('--refresh', `${String(this.config.refreshIntervalMs)}ms`)
     if (this.config.scanWorkers > 0) args.push('--workers', String(this.config.scanWorkers))
+    if (this.config.memoryLimitMB > 0) args.push('--memory-limit-mb', String(this.config.memoryLimitMB))
     if (this.config.windowsUsn) args.push('--usn')
 
     let child: ChildProcessWithoutNullStreams