From 31a80420ec9f4dbd61a7145044f5e1797d4e0dd0 Mon Sep 17 00:00:00 2001
From: Samantaz Fox <coding@samantaz.fr>
Date: Tue, 13 Feb 2024 21:46:12 +0100
Subject: [PATCH] Search: Add URL search inhibition logic

---
 src/invidious/search/query.cr | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/invidious/search/query.cr b/src/invidious/search/query.cr
index f87c243e..b3db0f63 100644
--- a/src/invidious/search/query.cr
+++ b/src/invidious/search/query.cr
@@ -20,6 +20,9 @@ module Invidious::Search
     property region : String?
     property channel : String = ""
 
+    # Flag that indicates if the smart search features have been disabled.
+    @inhibit_ssf : Bool = false
+
     # Return true if @raw_query is either `nil` or empty
     private def empty_raw_query?
       return @raw_query.empty?
@@ -55,6 +58,13 @@ module Invidious::Search
       # Remove surrounding whitespaces. Mostly useful for copy/pasted URLs.
       @raw_query = _raw_query.strip
 
+      # Check for smart features (ex: URL search) inhibitor (exclamation mark).
+      # If inhibitor is present, remove it.
+      if @raw_query.starts_with?('!')
+        @inhibit_ssf = true
+        @raw_query = @raw_query[1..]
+      end
+
       # Get the page number (also common to all search types)
       @page = params["page"]?.try &.to_i? || 1
 
@@ -140,6 +150,9 @@ module Invidious::Search
 
     # Checks if the query is a standalone URL
     def is_url? : Bool
+      # If the smart features have been inhibited, don't go further.
+      return false if @inhibit_ssf
+
       # Only supported in regular search mode
       return false if !@type.regular?