@arabold/docs-mcp-server 1.2.0 → 1.3.0

package/dist/{chunk-Y3EE54SV.js → chunk-75KNYK47.js}

@@ -1279,6 +1279,17 @@ var VersionNotFoundError = class extends ToolError {
     return this.availableVersions.sort((a, b) => semver.compare(b.version, a.version))[0];
   }
 };
+var LibraryNotFoundError = class extends ToolError {
+  constructor(requestedLibrary, suggestions = []) {
+    let message = `Library '${requestedLibrary}' not found.`;
+    if (suggestions.length > 0) {
+      message += ` Did you mean one of these: ${suggestions.join(", ")}?`;
+    }
+    super(message, "SearchTool");
+    this.requestedLibrary = requestedLibrary;
+    this.suggestions = suggestions;
+  }
+};
 
 // src/tools/SearchTool.ts
 var SearchTool = class {
@@ -1292,6 +1303,7 @@ var SearchTool = class {
       `\u{1F50D} Searching ${library}@${version} for: ${query}${exactMatch ? " (exact match)" : ""}`
     );
     try {
+      await this.docService.validateLibraryExists(library);
       let versionToSearch = version;
       if (!exactMatch) {
         const versionResult = await this.docService.findBestVersion(library, version);
@@ -1306,6 +1318,16 @@ var SearchTool = class {
       logger.info(`\u2705 Found ${results.length} matching results`);
       return { results };
     } catch (error) {
+      if (error instanceof LibraryNotFoundError) {
+        logger.info(`\u2139\uFE0F Library not found: ${error.message}`);
+        return {
+          results: [],
+          error: {
+            message: error.message,
+            suggestions: error.suggestions
+          }
+        };
+      }
       if (error instanceof VersionNotFoundError) {
         logger.info(`\u2139\uFE0F Version not found: ${error.message}`);
         return {
@@ -1659,6 +1681,7 @@ var RemoveTool = class {
 // src/store/DocumentManagementService.ts
 import { mkdirSync } from "node:fs";
 import path3 from "node:path";
+import Fuse from "fuse.js";
 import semver3 from "semver";
 
 // src/splitter/SemanticMarkdownSplitter.ts
@@ -10912,6 +10935,14 @@ var DocumentStore = class {
       batchSize: 512
     });
   }
+  /**
+   * Escapes a query string for use with SQLite FTS5 MATCH operator.
+   * Wraps the query in double quotes and escapes internal double quotes.
+   */
+  escapeFtsQuery(query) {
+    const escapedQuotes = query.replace(/"/g, '""');
+    return `"${escapedQuotes}"`;
+  }
   /**
    * Initializes database connection and ensures readiness
    */
@@ -11057,6 +11088,7 @@ var DocumentStore = class {
   async findByContent(library, version, query, limit) {
     try {
       const embedding = await this.embeddings.embedQuery(query);
+      const ftsQuery = this.escapeFtsQuery(query);
       const stmt = this.db.prepare(`
         WITH vec_scores AS (
           SELECT
@@ -11099,7 +11131,8 @@ var DocumentStore = class {
         limit,
         library.toLowerCase(),
         version.toLowerCase(),
-        query,
+        ftsQuery,
+        // Use the escaped query
         limit
       );
       const rankedResults = this.assignRanks(rawResults);
@@ -11254,16 +11287,53 @@ var DocumentManagementService = class {
     );
     this.splitter = greedySplitter;
   }
+  /**
+   * Initializes the underlying document store.
+   */
   async initialize() {
     await this.store.initialize();
   }
+  /**
+   * Shuts down the underlying document store.
+   */
   async shutdown() {
     logger.info("\u{1F50C} Shutting down store manager");
     await this.store.shutdown();
   }
   /**
-   * Returns a list of all available versions for a library.
-   * Only returns versions that follow semver format.
+   * Validates if a library exists in the store (either versioned or unversioned).
+   * Throws LibraryNotFoundError with suggestions if the library is not found.
+   * @param library The name of the library to validate.
+   * @throws {LibraryNotFoundError} If the library does not exist.
+   */
+  async validateLibraryExists(library) {
+    logger.info(`\u{1F50E} Validating existence of library: ${library}`);
+    const normalizedLibrary = library.toLowerCase();
+    const versions = await this.listVersions(normalizedLibrary);
+    const hasUnversioned = await this.exists(normalizedLibrary, "");
+    if (versions.length === 0 && !hasUnversioned) {
+      logger.warn(`\u26A0\uFE0F Library '${library}' not found.`);
+      const allLibraries = await this.listLibraries();
+      const libraryNames = allLibraries.map((lib) => lib.library);
+      let suggestions = [];
+      if (libraryNames.length > 0) {
+        const fuse = new Fuse(libraryNames, {
+          // Configure fuse.js options if needed (e.g., threshold)
+          // isCaseSensitive: false, // Handled by normalizing library names
+          // includeScore: true,
+          threshold: 0.4
+          // Adjust threshold for desired fuzziness (0=exact, 1=match anything)
+        });
+        const results = fuse.search(normalizedLibrary);
+        suggestions = results.slice(0, 3).map((result) => result.item);
+        logger.info(`\u{1F50D} Found suggestions: ${suggestions.join(", ")}`);
+      }
+      throw new LibraryNotFoundError(library, suggestions);
+    }
+    logger.info(`\u2705 Library '${library}' confirmed to exist.`);
+  }
+  /**
+   * Returns a list of all available semantic versions for a library.
    */
   async listVersions(library) {
     const versions = await this.store.queryUniqueVersions(library);
@@ -11428,4 +11498,4 @@ export {
   RemoveTool,
   DocumentManagementService
 };
-//# sourceMappingURL=chunk-Y3EE54SV.js.map
+//# sourceMappingURL=chunk-75KNYK47.js.map