use fuzzy text search in main torrent search

Author: tdjsnelling

api/src/controllers/torrent.js

@@ -1,7 +1,7 @@
 import bencode from "bencode";
 import crypto from "crypto";
 import mongoose from "mongoose";
-import { createNGrams } from "mongoose-fuzzy-searching/helpers";
+import { createNGrams, nGrams } from "mongoose-fuzzy-searching/helpers";
 import slugify from "slugify";
 import Torrent from "../schema/torrent";
 import User from "../schema/user";
@@ -463,7 +463,23 @@ export const getTorrentsPage = async ({
   userId,
   tracker,
 }) => {
+  const queryNGrams = nGrams(query, false, 2, false).join(" ");
+
   const torrents = await Torrent.aggregate([
+    ...(query
+      ? [
+          {
+            $match: {
+              $text: {
+                $search: queryNGrams,
+              },
+            },
+          },
+          {
+            $addFields: { confidenceScore: { $meta: "textScore" } },
+          },
+        ]
+      : []),
     {
       $project: {
         infoHash: 1,
@@ -476,6 +492,7 @@ export const getTorrentsPage = async ({
         created: 1,
         freeleech: 1,
         tags: 1,
+        confidenceScore: 1,
       },
     },
     ...(Array.isArray(ids)
@@ -485,18 +502,6 @@ export const getTorrentsPage = async ({
           },
         ]
       : []),
-    ...(query
-      ? [
-          {
-            $match: {
-              $or: [
-                { name: { $regex: query, $options: "i" } },
-                { description: { $regex: query, $options: "i" } },
-              ],
-            },
-          },
-        ]
-      : []),
     ...(category
       ? [
           {
@@ -534,7 +539,9 @@ export const getTorrentsPage = async ({
         ]
       : []),
     {
-      $sort: { created: -1 },
+      $sort: query
+        ? { confidenceScore: { $meta: "textScore" } }
+        : { created: -1 },
     },
     {
       $skip: skip,
@@ -592,23 +599,24 @@ export const getTorrentsPage = async ({
     { $unwind: { path: "$fetchedBy", preserveNullAndEmptyArrays: true } },
   ]);
 
+  console.log(torrents);
+
   const [count] = await Torrent.aggregate([
-    ...(Array.isArray(ids)
+    ...(query
       ? [
           {
-            $match: { expr: { $in: ["$_id", ids] } },
+            $match: {
+              $text: {
+                $search: queryNGrams,
+              },
+            },
           },
         ]
       : []),
-    ...(query
+    ...(Array.isArray(ids)
       ? [
           {
-            $match: {
-              $or: [
-                { name: { $regex: query, $options: "i" } },
-                { description: { $regex: query, $options: "i" } },
-              ],
-            },
+            $match: { expr: { $in: ["$_id", ids] } },
           },
         ]
       : []),

client/pages/search/[[...query]].js

@@ -35,12 +35,7 @@ const Search = ({ results, error }) => {
         {query ? `Search results for “${query}”` : "Search"}
       </Text>
       <Box as="form" onSubmit={handleSearch} display="flex" mb={5}>
-        <Input
-          placeholder="Search torrents (text or RegExp)"
-          name="query"
-          mr={3}
-          required
-        />
+        <Input name="query" mr={3} required />
         <Button>Search</Button>
       </Box>
       {error ? (