From 777ca49c18cc8a26073098c0a090e21d77a1ae57 Mon Sep 17 00:00:00 2001
From: Axel Rafn <axel@axelrafn.is>
Date: Fri, 12 Jul 2024 18:06:13 +0200
Subject: [PATCH] feat: Add search engine selection functionality

---
 app.py | 49 ++++++++++++++++++++++++++++++++++++++++---------
 1 file changed, 40 insertions(+), 9 deletions(-)

diff --git a/app.py b/app.py
index ce2a3ba..52927e0 100644
--- a/app.py
+++ b/app.py
@@ -88,6 +88,17 @@ class URLManager(QWidget):
         # Load settings
         self.load_settings()
 
+        # Search engines
+        self.search_engines = {
+            "Bing": "https://www.bing.com/search?q=",
+            "Brave": "https://search.brave.com/search?q=",
+            "DuckDuckGo": "https://duckduckgo.com/?q=",
+            "Ecosia": "https://www.ecosia.org/search?q=",
+            "Google": "https://www.google.com/search?q=",
+            "Startpage": "https://www.startpage.com/do/search?q=",
+            "Swisscows": "https://swisscows.com/web?query="
+        }
+
     def update_icon_color(self):
         # Heuristic to check if the application is in dark mode
         palette = self.palette()
@@ -141,8 +152,8 @@ class URLManager(QWidget):
     def search_url(self):
         search_term = self.search_input.text()
         if search_term:
-            search_engine = "https://www.google.com/search?q="
-            search_url = search_engine + search_term
+            search_engine_url = self.search_engines.get(self.search_engine, "https://www.google.com/search?q=")
+            search_url = search_engine_url + search_term
             QMessageBox.information(self, "Search", f"Searching for: {search_term}\nURL: {search_url}")
             self.search_input.clear()
 
@@ -180,13 +191,15 @@ class URLManager(QWidget):
             if os.path.getsize(settings_path) > 0:
                 with open(settings_path, 'r') as file:
                     settings = json.load(file)
-                    # Use settings (for now, we only have the data_directory)
                     self.data_directory = settings.get('data_directory', 'data/')
+                    self.search_engine = settings.get('search_engine', 'Google')
             else:
                 self.data_directory = 'data/'
+                self.search_engine = 'Google'
                 self.save_settings()
         else:
             self.data_directory = 'data/'
+            self.search_engine = 'Google'
             self.save_settings()
 
     def save_settings(self):
@@ -194,7 +207,8 @@ class URLManager(QWidget):
         os.makedirs(os.path.dirname(settings_path), exist_ok=True)
         with open(settings_path, 'w') as file:
             settings = {
-                'data_directory': self.data_directory
+                'data_directory': self.data_directory,
+                'search_engine': self.search_engine
             }
             json.dump(settings, file)
 
@@ -203,18 +217,35 @@ class SettingsDialog(QDialog):
     def __init__(self, parent=None):
         super().__init__(parent)
         self.setWindowTitle("Settings")
+        self.setFixedWidth(360)  # Set the width of the window to 360 pixels
 
         self.parent = parent
 
         layout = QVBoxLayout()
 
-        # File/Directory selection
-        form_layout = QFormLayout()
+        # Data Directory label
+        directory_label = QLabel("Data Directory:")
+        layout.addWidget(directory_label)
+
+        # File/Directory selection layout
+        directory_layout = QHBoxLayout()
         self.directory_input = QLineEdit(self.parent.data_directory)
         self.browse_button = QPushButton("Browse...")
         self.browse_button.clicked.connect(self.browse_directory)
-        form_layout.addRow("Data Directory:", self.directory_input)
-        form_layout.addWidget(self.browse_button)
+        directory_layout.addWidget(self.directory_input)
+        directory_layout.addWidget(self.browse_button)
+        layout.addLayout(directory_layout)
+
+        # Add 4 pixels of empty space
+        layout.addSpacerItem(QSpacerItem(0, 4, QSizePolicy.Minimum, QSizePolicy.Fixed))
+
+        # Search engine selection
+        form_layout = QFormLayout()
+        self.search_engine_combobox = QComboBox()
+        search_engines = sorted(self.parent.search_engines.keys())
+        self.search_engine_combobox.addItems(search_engines)
+        self.search_engine_combobox.setCurrentText(self.parent.search_engine)
+        form_layout.addRow("Search Engine:", self.search_engine_combobox)
 
         layout.addLayout(form_layout)
 
@@ -232,10 +263,10 @@ class SettingsDialog(QDialog):
 
     def save_settings(self):
         self.parent.data_directory = self.directory_input.text()
+        self.parent.search_engine = self.search_engine_combobox.currentText()
         self.parent.save_settings()
         self.accept()
 
-
 if __name__ == '__main__':
     app = QApplication(sys.argv)
     app.setWindowIcon(QIcon('assets/logo.png'))  # Set the application icon using PNG for compatibility