6bd769ef40
- Add migration 013_city_map_coordinates.sql with map coordinates, zones, and agents table - Add /city/map API endpoint in city-service - Add /city/agents and /city/agents/online endpoints - Extend presence aggregator to include agents[] in snapshot - Add AgentsSource for fetching agent data from DB - Create CityMap component with interactive room tiles - Add useCityMap hook for fetching map data - Update useGlobalPresence to include agents - Add map/list view toggle on /city page - Add agent badges to room cards and map tiles
71 lines
1.9 KiB
Python
71 lines
1.9 KiB
Python
"""Room source - reads rooms from database or static config"""
|
|
from sqlalchemy import create_engine, text
|
|
from typing import List, Dict
|
|
import logging
|
|
import yaml
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class RoomsSource:
|
|
"""Reads room list from PostgreSQL database"""
|
|
|
|
def __init__(self, db_dsn: str):
|
|
self.engine = create_engine(db_dsn)
|
|
|
|
def get_rooms(self) -> List[Dict]:
|
|
"""
|
|
Get all rooms with matrix_room_id set.
|
|
|
|
Expected table structure:
|
|
- id (text)
|
|
- slug (text)
|
|
- name (text)
|
|
- matrix_room_id (text, nullable)
|
|
"""
|
|
query = text(
|
|
"""
|
|
SELECT id, slug, name, matrix_room_id
|
|
FROM city_rooms
|
|
WHERE matrix_room_id IS NOT NULL
|
|
"""
|
|
)
|
|
try:
|
|
with self.engine.connect() as conn:
|
|
rows = conn.execute(query).mappings().all()
|
|
|
|
return [
|
|
{
|
|
"room_id": str(r["id"]),
|
|
"slug": r["slug"],
|
|
"title": r["name"],
|
|
"matrix_room_id": r["matrix_room_id"],
|
|
}
|
|
for r in rows
|
|
]
|
|
except Exception as e:
|
|
logger.error(f"Failed to get rooms from database: {e}")
|
|
return []
|
|
|
|
|
|
class StaticRoomsSource:
|
|
"""Reads room list from YAML config file"""
|
|
|
|
def __init__(self, config_path: str):
|
|
self.config_path = config_path
|
|
self._rooms = self._load_config()
|
|
|
|
def _load_config(self) -> List[Dict]:
|
|
try:
|
|
with open(self.config_path, 'r') as f:
|
|
data = yaml.safe_load(f)
|
|
return data.get('rooms', [])
|
|
except Exception as e:
|
|
logger.error(f"Failed to load rooms config: {e}")
|
|
return []
|
|
|
|
def get_rooms(self) -> List[Dict]:
|
|
return self._rooms
|
|
|
|
|