Endpoints to request reindexing of a library deleted from ElasticSearch

controllers/FullTextController.php

@@ -157,4 +157,48 @@ public function itemContent() {
 
 		$this->end();
 	}
+
+	public function reindex() {
+		$this->allowMethods(['POST', 'GET']);
+
+		// General library access
+		if (!$this->permissions->canAccess($this->objectLibraryID)) {
+			$this->e403();
+		}
+
+		$isDeindexed = Zotero_Libraries::checkEsIndexStatus($this->objectLibraryID);
+
+		// GET - return indexing status of ES: indexing, indexed, deindexed
+		if ($this->method == "GET") {
+			// Current count of records in ES
+			$esCount = Zotero_FullText::countInLibrary($this->objectLibraryID);
+			// Expected count of records in ES
+			$expectedCount = Zotero_Libraries::countIndexableAttachments($this->objectLibraryID);
+
+			if ($esCount === $expectedCount) {
+				$result = ["reindexingStatus" => "indexed"];
+			}
+			else if ($isDeindexed) {
+				$result = ["reindexingStatus" => "deindexed"];
+			}
+			else {
+				$result = ["reindexingStatus" => "indexing", "indexedCount" => $esCount, "expectedCount" => $expectedCount];
+			}
+			echo Zotero_Utilities::formatJSON($result);
+			$this->end();
+		}
+
+		// POST - request reindexing if the library was removed from ES
+		if (!$isDeindexed) {
+            $this->e400("Request was already submitted or the library was not removed from ElasticSearch");
+        }
+
+		// Send event to reindexing queue
+		Z_SQS::send(Z_CONFIG::$REINDEX_QUEUE_URL, json_encode(['libraryID' => $this->objectLibraryID]));
+
+		// Update DB
+		Zotero_Libraries::setEsIndexStatus($this->objectLibraryID, 0); 
+		$this->end();
+	}
+
 }

include/SQS.inc.php

@@ -102,7 +102,7 @@ public static function deleteBatch($queueURL, $batchEntries) {
 
 	private static function load() {
 		if (!self::$sqs) {
-			self::$sqs = Z_Core::$AWS->get('sqs');
+			self::$sqs = Z_Core::$AWS->createSQS();
 		}
 	}
 }

include/config/config.inc.php-sample

@@ -89,5 +89,7 @@ class Z_CONFIG {
 	public static $CACHE_VERSION_ITEM_DATA = 1;
 	public static $CACHE_VERSION_RESPONSE_JSON_COLLECTION = 1;
 	public static $CACHE_VERSION_RESPONSE_JSON_ITEM = 1;
+
+	public static $REINDEX_QUEUE_URL = "";
 }
 ?>

include/config/routes.inc.php

@@ -42,6 +42,8 @@
 $router->map('/users/i:objectUserID/fulltext', array('controller' => 'FullText', 'action' => 'fulltext'));
 //$router->map('/users/i:objectUserID/publications/fulltext', ['controller' => 'FullText', 'action' => 'fulltext', 'extra' => ['publications' => true]]);
 $router->map('/groups/i:objectGroupID/fulltext', array('controller' => 'FullText', 'action' => 'fulltext'));
+$router->map('/groups/i:objectGroupID/fulltext/reindex', array('controller' => 'FullText', 'action' => 'reindex'));
+$router->map('/users/i:objectUserID/fulltext/reindex', array('controller' => 'FullText', 'action' => 'reindex'));
 
 // All trashed items
 $router->map('/users/i:objectUserID/items/trash', array('controller' => 'Items', 'extra' => array('subset' => 'trash')));

misc/db-updates/2024-02-15/add_es_status_to_library

@@ -0,0 +1,6 @@
+#!/usr/bin/php -d mysqlnd.net_read_timeout=3600
+<?php
+set_include_path("../../../include");
+require("header.inc.php");
+
+Zotero_DB::query("ALTER TABLE libraries ADD COLUMN deindexed_from_es tinyint(1) unsigned NOT NULL DEFAULT '0';");

misc/master.sql

@@ -184,6 +184,7 @@ CREATE TABLE `libraries` (
   `lastUpdated` timestamp NOT NULL DEFAULT '0000-00-00 00:00:00',
   `version` int(10) unsigned NOT NULL DEFAULT '0',
   `shardID` smallint(5) unsigned NOT NULL,
+  `deindexed_from_es` tinyint(1) unsigned NOT NULL DEFAULT '0',
   PRIMARY KEY (`libraryID`),
   KEY `shardID` (`shardID`)
 ) ENGINE=InnoDB DEFAULT CHARSET=utf8;

model/FullText.inc.php

@@ -259,6 +259,25 @@ public static function searchInLibrary($libraryID, $searchText) {
 		}
 		return $keys;
 	}
+
+	public static function countInLibrary($libraryID) {
+		$params = [
+			'index' => self::$elasticsearchType . "_index",
+			'body'  => [
+				'query' => [
+					'bool' => [
+						'filter' => [
+							'term' => [
+								'libraryID' => $libraryID
+							]
+						]
+					]
+				]
+			]
+		];
+		$resp = Z_Core::$ES->count($params);
+		return $resp['count'];
+	}
 
 	public static function deleteItemContent(Zotero_Item $item) {
 		$libraryID = $item->libraryID;

model/Libraries.inc.php

@@ -53,6 +53,28 @@ public static function exists($libraryID) {
 		$sql = "SELECT COUNT(*) FROM libraries WHERE libraryID=?";
 		return !!Zotero_DB::valueQuery($sql, $libraryID);
 	}
+
+	public static function countIndexableAttachments($libraryID) {
+		$attachmentIds = Zotero_DB::columnQuery(
+			"SELECT itemTypeID FROM itemTypes "
+			. "WHERE itemTypeName IN ('attachment') "
+		);
+		$sql = "SELECT COUNT(*) as count FROM items INNER JOIN itemAttachments USING (itemID)" 
+				. "WHERE NOT(linkMode='LINKED_URL') AND libraryID=? AND itemTypeID IN (" . implode(",", $attachmentIds) . ")";
+		$count = Zotero_DB::query($sql, $libraryID, Zotero_Shards::getByLibraryID($libraryID));
+		return $count[0]['count'];
+	}
+
+	public static function checkEsIndexStatus($libraryID) {
+		$sql = "SELECT deindexed_from_es FROM libraries WHERE libraryID=?";
+		$isDeleted = Zotero_DB::query($sql, $libraryID);
+		return $isDeleted[0]['deindexed_from_es'] == 1;
+	}
+
+	public static function setEsIndexStatus($libraryID, $deindexed) {
+		$sql = "UPDATE libraries SET deindexed_from_es=? WHERE libraryID=?";
+		Zotero_DB::query($sql, [$deindexed, $libraryID]);
+	}
 
 
 	public static function getName($libraryID) {