chg: [search] catch unreachable and timeout error

Author: Terrtia

bin/lib/search_engine.py

@@ -526,8 +526,22 @@ def api_search(data):
         except:
             return {"status": "error", "reason": "Invalid date from"}, 400
 
-    result = Engine.search(indexes, to_search, page=page, nb=nb_per_page, timestamp_from=timestamp_from, sort=sort,
-                           timestamp_to=timestamp_to)
+    try:
+        result = Engine.search(indexes, to_search, page=page, nb=nb_per_page, timestamp_from=timestamp_from, sort=sort,
+                               timestamp_to=timestamp_to)
+    except MeilisearchTimeoutError:
+        return {
+            "status": "error",
+            "error_type": "meilisearch_timeout",
+            "reason": "The connection to Meilisearch timed out. Please try again in a few minutes."
+        }, 503
+    except MeilisearchCommunicationError:
+        return {
+            "status": "error",
+            "error_type": "meilisearch_unreachable",
+            "reason": "Meilisearch is unreachable. Install/start Meilisearch or check the connection settings."
+        }, 503
+
     objs = []
     pagination = {}
     if result.get("hits"):

var/www/blueprints/search_b.py

@@ -38,6 +38,7 @@ def create_json_response(data, status_code):
 def log(user_id, index, to_search):
     logger.warning(f'{user_id} search: {index} - {to_search}')
 
+
 # ============ FUNCTIONS ============
 
 # ============= ROUTES ==============
@@ -65,12 +66,30 @@ def search_dashboard():
 
     # selected_scopes -> scope_human
 
+    search_error = None
     if search:
         r = search_engine.api_search({'indexes': indexes, 'search': search, 'page': page, 'user_id': user_id,
                                       'from': last_seen_from, 'to': last_seen_to, 'sort': sort})
         if r[1] != 200:
-            return create_json_response(r[0], r[1])
-        result, pagination = r[0]
+            error_type = r[0].get('error_type')
+            if error_type == 'meilisearch_timeout':
+                search_error = {
+                    'title': 'Search service timeout',
+                    'message': r[0].get('reason')
+                }
+                result = None
+                pagination = None
+            elif error_type == 'meilisearch_unreachable':
+                search_error = {
+                    'title': 'Search service unavailable',
+                    'message': r[0].get('reason')
+                }
+                result = None
+                pagination = None
+            else:
+                return create_json_response(r[0], r[1])
+        else:
+            result, pagination = r[0]
     else:
         result = None
         pagination = None
@@ -83,6 +102,7 @@ def search_dashboard():
                            sort=sort,
                            last_seen_from=last_seen_from,
                            last_seen_to=last_seen_to,
+                           search_error=search_error,
                            result=result, pagination=pagination)
 
 # username_subtypes=ail_core.get_object_all_subtypes('username')

var/www/templates/search/search_dashboard.html

@@ -382,6 +382,20 @@ <h6 class="text-uppercase text-muted mb-3">Search scope</h6>
 
 </form>
 
+    {% if search_error %}
+        <div class="alert alert-warning border-left border-warning shadow-sm mt-3" role="alert">
+          <div class="d-flex align-items-start">
+            <div class="mr-3 mt-1">
+              <i class="fas fa-exclamation-triangle fa-lg text-warning"></i>
+            </div>
+            <div>
+              <h6 class="alert-heading mb-1">{{ search_error.title }}</h6>
+              <p class="mb-0">{{ search_error.message }}</p>
+            </div>
+          </div>
+        </div>
+    {% endif %}
+
 {#        <small class="text-muted d-block mt-2">#}
 {#          Examples: <code>@username</code>, <code>example.onion</code>, <code>invoice.pdf</code>, <code>ransomware</code>#}
 {#        </small>#}