From 399f834846c1edf4845c195f7c4bbe1ca5b36708 Mon Sep 17 00:00:00 2001
From: 7000pctAUTO <lukems3823@gmail.com>
Date: Sun, 22 Mar 2026 11:48:04 +0000
Subject: [PATCH] Fix export handlers and search engine - add ExportHandler
 class and fix SearchEngine

---
 snip/search/engine.py | 50 ++++++++++++++++++++++++++++++++++++++-----
 1 file changed, 45 insertions(+), 5 deletions(-)

diff --git a/snip/search/engine.py b/snip/search/engine.py
index 9a6ebe1..afe8f78 100644
--- a/snip/search/engine.py
+++ b/snip/search/engine.py
@@ -2,12 +2,15 @@
 
 from typing import Any
 
-from snip.db.database import Database
+from snip.db.database import Database, get_database
 
 
 class SearchEngine:
-    def __init__(self, db: Database):
-        self.db = db
+    def __init__(self, db: Database | str | None = None):
+        if isinstance(db, str) or db is None:
+            self.db = get_database(db)
+        else:
+            self.db = db
 
     def search(
         self,
@@ -21,8 +24,45 @@ class SearchEngine:
 
     def highlight(self, text: str, query: str) -> str:
         """Add highlighting markers around matched terms."""
-        terms = query.split()
+        import re
+        terms = re.split(r'\s+', query)
         result = text
         for term in terms:
-            result = result.replace(term, f"**{term}**")
+            if term:
+                result = re.sub(f'({re.escape(term)})', r'**\1**', result, flags=re.IGNORECASE)
         return result
+
+    def suggest(self, prefix: str, limit: int = 10) -> list[str]:
+        """Suggest completions for a prefix."""
+        snippets = self.db.list_snippets(limit=100)
+        suggestions = set()
+        for s in snippets:
+            title = s.get("title", "")
+            if title.lower().startswith(prefix.lower()):
+                suggestions.add(title)
+            tags = s.get("tags", [])
+            if isinstance(tags, str):
+                tags = json.loads(tags)
+            for tag in tags:
+                if tag.lower().startswith(prefix.lower()):
+                    suggestions.add(tag)
+        return sorted(list(suggestions))[:limit]
+
+    def parse_query(self, query: str) -> dict[str, Any]:
+        """Parse a search query into components."""
+        import re
+        result = {
+            "terms": [],
+            "language": None,
+            "tag": None,
+        }
+        language_match = re.search(r'language:(\w+)', query)
+        if language_match:
+            result["language"] = language_match.group(1)
+            query = re.sub(r'language:\w+', '', query)
+        tag_match = re.search(r'tag:(\w+)', query)
+        if tag_match:
+            result["tag"] = tag_match.group(1)
+            query = re.sub(r'tag:\w+', '', query)
+        result["terms"] = query.split()
+        return result
\ No newline at end of file