dsh-locator 0.6.2 → 0.7.0

package/src/index.ts

@@ -10,6 +10,7 @@ import type { Config } from './config.ts'
 import { ContentIndex } from './content.ts'
 import { makeExportTool } from './export.ts'
 import { makeFindTool } from './find.ts'
+import { makeReindexTool } from './reindex.ts'
 import { registerHttp } from './http.ts'
 import { resolveBinary, resolveRoots, resolveStateDir } from './paths.ts'
 import { LocatorRuntime } from './runtime.ts'
@@ -39,8 +40,9 @@ export function apply(ctx: Context, config: Config): void {
   const content = config.contentIndex
     ? new ContentIndex(rt, {
         stateDir,
-        sessionDirs: config.contentSessionDirs,
+        sessionDirs: config.contentSessionDirs.length > 0 ? config.contentSessionDirs : undefined,
         maxFileBytes: config.contentMaxFileBytes,
+        maxTotalBytes: config.contentMaxTotalBytes,
         maxFiles: config.contentMaxFiles,
         maxSessions: config.contentMaxSessions,
       })
@@ -51,7 +53,7 @@ export function apply(ctx: Context, config: Config): void {
     const timer = setTimeout(() => {
       rt.start()
       logger.info(`index roots: ${roots.join(', ')} (state ${stateDir})`)
-      if (content !== undefined) void startContent(rt, content, logger)
+      if (content !== undefined) void startContent(rt, content, logger, config.contentRefreshMs)
     }, 1500)
     return () => {
       clearTimeout(timer)
@@ -64,14 +66,17 @@ 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))
+  if (content !== undefined) {
+    ctx.tools.register(makeFindTool(content))
+    ctx.tools.register(makeReindexTool(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, content)
 }
 
-async function startContent(rt: LocatorRuntime, content: ContentIndex, logger: { info(message: string): void }): Promise<void> {
+async function startContent(rt: LocatorRuntime, content: ContentIndex, logger: { info(message: string): void }, refreshMs: number): Promise<void> {
   try {
     await rt.waitReady()
     for (let i = 0; i < 900; i++) {
@@ -80,6 +85,7 @@ async function startContent(rt: LocatorRuntime, content: ContentIndex, logger: {
       await new Promise(resolve => setTimeout(resolve, 2_000))
     }
     content.start()
+    content.startAutoRefresh(refreshMs)
     logger.info('content index started')
   } catch (error) {
     logger.info(`content index skipped: ${error instanceof Error ? error.message : String(error)}`)

package/src/reindex.ts

@@ -0,0 +1,39 @@
+/** locator_reindex — full or incremental content-index maintenance. */
+import { defineTool } from '@deepseek-ai/dsh-tools'
+import type { ContentIndex } from './content.ts'
+
+export function makeReindexTool(content: ContentIndex) {
+  return defineTool({
+    name: 'locator_reindex',
+    description: [
+      'Rebuild or refresh the dsh-locator content index (file bodies + DSH session messages).',
+      '',
+      '`full: true` drops the content database and re-reads everything; `full: false` (default) only re-reads files/sessions whose mtime changed. Name/path search keeps working during either mode. Use `locator_status` or the content state in `locator_find` to watch progress.',
+    ].join('\n'),
+    parameters: {
+      full: { type: 'boolean', description: 'Drop and rebuild the content database (default false = incremental).' },
+    },
+    output: {
+      schema: {
+        type: 'object',
+        additionalProperties: false,
+        properties: {
+          ok: { type: 'boolean', required: true },
+          started: { type: 'boolean' },
+          full: { type: 'boolean' },
+          error: { type: 'string' },
+        },
+      },
+      render: (_args, value) => {
+        const v = value as { ok: boolean; started?: boolean; full?: boolean; error?: string }
+        return [{ type: 'text', text: v.ok ? `${v.full ? 'Full' : 'Incremental'} content rebuild started.` : `locator_reindex failed: ${v.error ?? 'unknown'}` }]
+      },
+    },
+    async execute(args) {
+      const full = args.full === true
+      const started = full ? content.rebuild() : content.refreshNow()
+      if (!started) return { ok: false as const, error: 'a content build is already running' }
+      return { ok: true as const, started: true, full }
+    },
+  })
+}