feat: postgres search engine

server/modules/search/postgres/engine.js

@@ -1,26 +1,31 @@
 const _ = require('lodash')
+const tsquery = require('pg-tsquery')()
 
 module.exports = {
-  activate() {
+  async activate() {
     // not used
   },
-  deactivate() {
+  async deactivate() {
     // not used
   },
   /**
    * INIT
    */
-  init() {
-    // not used
-  },
-  /**
-   * SUGGEST
-   *
-   * @param {String} q Query
-   * @param {Object} opts Additional options
-   */
-  async suggest(q, opts) {
-
+  async init() {
+    // -> Create Index
+    const indexExists = await WIKI.models.knex.schema.hasTable('pagesVector')
+    if (!indexExists) {
+      await WIKI.models.knex.schema.createTable('pagesVector', table => {
+        table.increments()
+        table.string('path')
+        table.string('locale')
+        table.string('title')
+        table.string('description')
+        table.specificType('titleTk', 'TSVECTOR')
+        table.specificType('descriptionTk', 'TSVECTOR')
+        table.specificType('contentTk', 'TSVECTOR')
+      })
+    }
   },
   /**
    * QUERY
@@ -29,6 +34,21 @@ module.exports = {
    * @param {Object} opts Additional options
    */
   async query(q, opts) {
+    try {
+      const results = await WIKI.models.knex.raw(`
+        SELECT id, path, locale, title, description
+        FROM "pagesVector", to_tsquery(?) query
+        WHERE (query @@ "titleTk") OR (query @@ "descriptionTk") OR (query @@ "contentTk")
+      `, [tsquery(q)])
+      return {
+        results: results.rows,
+        suggestions: [],
+        totalHits: results.rows.length
+      }
+    } catch (err) {
+      WIKI.logger.warn('Search Engine Error:')
+      WIKI.logger.warn(err)
+    }
 
   },
   /**
@@ -37,7 +57,11 @@ module.exports = {
    * @param {Object} page Page to create
    */
   async created(page) {
-    // not used
+    await WIKI.models.knex.raw(`
+      INSERT INTO "pagesVector" (path, locale, title, description, "titleTk", "descriptionTk", "contentTk") VALUES (
+        '?', '?', '?', '?', to_tsvector('?'), to_tsvector('?'), to_tsvector('?')
+      )
+    `, [page.path, page.locale, page.title, page.description, page.title, page.description, page.content])
   },
   /**
    * UPDATE
@@ -45,7 +69,15 @@ module.exports = {
    * @param {Object} page Page to update
    */
   async updated(page) {
-    // not used
+    await WIKI.models.knex.raw(`
+      UPDATE "pagesVector" SET
+        title = '?',
+        description = '?',
+        "titleTk" = to_tsvector('?'),
+        "descriptionTk" = to_tsvector('?'),
+        "contentTk" = to_tsvector('?')
+      WHERE path = '?' AND locale = '?' LIMIT 1
+    `, [page.title, page.description, page.title, page.description, page.content, page.path, page.locale])
   },
   /**
    * DELETE
@@ -53,7 +85,10 @@ module.exports = {
    * @param {Object} page Page to delete
    */
   async deleted(page) {
-    // not used
+    await WIKI.models.knex('pagesVector').where({
+      locale: page.locale,
+      path: page.path
+    }).del().limit(1)
   },
   /**
    * RENAME
@@ -61,12 +96,23 @@ module.exports = {
    * @param {Object} page Page to rename
    */
   async renamed(page) {
-    // not used
+    await WIKI.models.knex('pagesVector').where({
+      locale: page.locale,
+      path: page.sourcePath
+    }).update({
+      locale: page.locale,
+      path: page.destinationPath
+    }).limit(1)
   },
   /**
    * REBUILD INDEX
    */
   async rebuild() {
-    // not used
+    await WIKI.models.knex('pagesVector').truncate()
+    await WIKI.models.knex.raw(`
+      INSERT INTO "pagesVector" (path, locale, title, description, "titleTk", "descriptionTk", "contentTk")
+        SELECT path, "localeCode" AS locale, title, description, to_tsvector(title) AS "titleTk", to_tsvector(description) AS "descriptionTk", to_tsvector(content) AS "contentTk"
+        FROM "pages"
+        WHERE pages."isPublished" AND NOT pages."isPrivate"`)
   }
 }