feat: search stations api for destination (#165)

server/base/source.py

@@ -59,8 +59,8 @@ def __init__(self, name, emoji, session, typesense):
         self.session = session
         self.typesense = typesense
 
-    def search_stops(self, name=None, lat=None, lon=None, page=1, limit=4, all_sources=False) -> tuple[
-        list[Station], int]:
+    def search_stops(self, name=None, lat=None, lon=None, page=1, limit=4, all_sources=False,
+                     hide_ids: list[str] = None) -> tuple[list[Station], int]:
         search_config = {'per_page': limit, 'query_by': 'name', 'page': page}
 
         limit_hits = None
@@ -78,6 +78,8 @@ def search_stops(self, name=None, lat=None, lon=None, page=1, limit=4, all_sourc
             })
         if not all_sources:
             search_config['filter_by'] = f'source:{self.name}'
+        if hide_ids:
+            search_config['hidden_hits'] = ','.join(hide_ids)
 
         results = self.typesense.collections['stations'].documents.search(search_config)
 

server/routes.py

@@ -33,8 +33,18 @@ async def home(request: Request) -> Response:
 async def search_stations(request: Request) -> Response:
     query = request.query_params.get('q', '')
     limit = int(request.query_params.get('limit', 4))
+    hide_ids = request.query_params.get('hide_ids')
+    if hide_ids:
+        hide_ids = hide_ids.split(',')
+    only_source = request.query_params.get('only_source')
+    if only_source:
+        source = sources[only_source]
+        all_sources = False
+    else:
+        source = sources['aut']
+        all_sources = True
     limit = max(1, min(limit, 10))
-    stations, count = sources['aut'].search_stops(name=query, all_sources=True, limit=limit)
+    stations, count = source.search_stops(name=query, all_sources=all_sources, limit=limit, hide_ids=hide_ids)
     return JSONResponse([station.as_dict() for station in stations])