From 6534692b73102729b99d7bedb84e30358ab8e048 Mon Sep 17 00:00:00 2001
From: DebaucheryLibrarian <moonloop.adult@protonmail.com>
Date: Fri, 16 Jun 2023 00:20:24 +0200
Subject: [PATCH] Improved search efficiency.

---
 migrations/20230616000529_search.js | 38 +++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)
 create mode 100644 migrations/20230616000529_search.js

diff --git a/migrations/20230616000529_search.js b/migrations/20230616000529_search.js
new file mode 100644
index 000000000..f9bca5e59
--- /dev/null
+++ b/migrations/20230616000529_search.js
@@ -0,0 +1,38 @@
+exports.up = async (knex) => {
+	await knex.raw(`
+		CREATE OR REPLACE FUNCTION curate_search_query(query text) RETURNS tsquery AS $$
+			SELECT to_tsquery('english', array_to_string(array(SELECT * FROM regexp_matches(query, '[A-Za-zÀ-ÖØ-öø-ÿ0-9]+', 'g')), '|'));
+		$$ LANGUAGE SQL STABLE;
+
+		CREATE OR REPLACE FUNCTION search_releases(query text) RETURNS SETOF releases_search_results AS $$
+			SELECT results.release_id, ts_rank(results.document::tsvector, curate_search_query(query)) as rank
+			FROM (
+				SELECT releases_search.release_id, document
+				FROM releases_search
+				WHERE document::tsvector @@ curate_search_query(query)
+			) AS results
+			ORDER BY rank DESC;
+		$$ LANGUAGE SQL STABLE;
+
+		COMMENT ON FUNCTION search_releases IS E'@sortable';
+	`);
+};
+
+exports.down = async (knex) => {
+	await knex.raw(`
+		CREATE OR REPLACE FUNCTION search_releases(query text) RETURNS SETOF releases_search_results AS $$
+			SELECT releases.id, ranks.rank FROM (
+				SELECT
+					releases_search.release_id,
+					ts_rank(releases_search.document, to_tsquery('english', array_to_string(array(SELECT * FROM regexp_matches(query, '[A-Za-zÀ-ÖØ-öø-ÿ0-9]+', 'g')), '|'))) AS rank
+				FROM releases_search
+				) ranks
+			LEFT JOIN releases ON releases.id = ranks.release_id
+			WHERE ranks.rank > 0
+			ORDER BY ranks.rank DESC;
+		$$ LANGUAGE SQL STABLE;
+
+		DROP FUNCTION curate_search_query;
+		COMMENT ON FUNCTION search_releases IS E'@sortable';
+	`);
+};