984f67c26e
- MATRIX_ROOMS_BRIDGE_SPEC.md documentation - Migration 012: Add matrix_room_id/alias to city_rooms - Matrix Gateway service (port 7025) - City-service: auto-create Matrix rooms on room creation - Backfill endpoint for existing rooms - API returns matrix_room_id/alias in room responses
229 lines
6.8 KiB
Python
229 lines
6.8 KiB
Python
"""
|
|
Repository для City Backend (PostgreSQL)
|
|
"""
|
|
|
|
import os
|
|
import asyncpg
|
|
from typing import Optional, List
|
|
from datetime import datetime
|
|
import secrets
|
|
|
|
|
|
# Database connection
|
|
_pool: Optional[asyncpg.Pool] = None
|
|
|
|
|
|
async def get_pool() -> asyncpg.Pool:
|
|
"""Отримати connection pool"""
|
|
global _pool
|
|
|
|
if _pool is None:
|
|
database_url = os.getenv("DATABASE_URL", "postgresql://postgres:postgres@localhost:5432/daarion")
|
|
_pool = await asyncpg.create_pool(database_url, min_size=2, max_size=10)
|
|
|
|
return _pool
|
|
|
|
|
|
async def close_pool():
|
|
"""Закрити connection pool"""
|
|
global _pool
|
|
if _pool is not None:
|
|
await _pool.close()
|
|
_pool = None
|
|
|
|
|
|
def generate_id(prefix: str) -> str:
|
|
"""Генерувати простий ID"""
|
|
return f"{prefix}_{secrets.token_urlsafe(12)}"
|
|
|
|
|
|
# =============================================================================
|
|
# City Rooms Repository
|
|
# =============================================================================
|
|
|
|
async def get_all_rooms(limit: int = 100, offset: int = 0) -> List[dict]:
|
|
"""Отримати всі кімнати"""
|
|
pool = await get_pool()
|
|
|
|
query = """
|
|
SELECT id, slug, name, description, is_default, created_at, created_by,
|
|
matrix_room_id, matrix_room_alias
|
|
FROM city_rooms
|
|
ORDER BY is_default DESC, created_at DESC
|
|
LIMIT $1 OFFSET $2
|
|
"""
|
|
|
|
rows = await pool.fetch(query, limit, offset)
|
|
return [dict(row) for row in rows]
|
|
|
|
|
|
async def get_room_by_id(room_id: str) -> Optional[dict]:
|
|
"""Отримати кімнату по ID"""
|
|
pool = await get_pool()
|
|
|
|
query = """
|
|
SELECT id, slug, name, description, is_default, created_at, created_by,
|
|
matrix_room_id, matrix_room_alias
|
|
FROM city_rooms
|
|
WHERE id = $1
|
|
"""
|
|
|
|
row = await pool.fetchrow(query, room_id)
|
|
return dict(row) if row else None
|
|
|
|
|
|
async def get_room_by_slug(slug: str) -> Optional[dict]:
|
|
"""Отримати кімнату по slug"""
|
|
pool = await get_pool()
|
|
|
|
query = """
|
|
SELECT id, slug, name, description, is_default, created_at, created_by,
|
|
matrix_room_id, matrix_room_alias
|
|
FROM city_rooms
|
|
WHERE slug = $1
|
|
"""
|
|
|
|
row = await pool.fetchrow(query, slug)
|
|
return dict(row) if row else None
|
|
|
|
|
|
async def create_room(
|
|
slug: str,
|
|
name: str,
|
|
description: Optional[str],
|
|
created_by: Optional[str],
|
|
matrix_room_id: Optional[str] = None,
|
|
matrix_room_alias: Optional[str] = None
|
|
) -> dict:
|
|
"""Створити кімнату"""
|
|
pool = await get_pool()
|
|
|
|
room_id = f"room_city_{slug}"
|
|
|
|
query = """
|
|
INSERT INTO city_rooms (id, slug, name, description, created_by, matrix_room_id, matrix_room_alias)
|
|
VALUES ($1, $2, $3, $4, $5, $6, $7)
|
|
RETURNING id, slug, name, description, is_default, created_at, created_by, matrix_room_id, matrix_room_alias
|
|
"""
|
|
|
|
row = await pool.fetchrow(query, room_id, slug, name, description, created_by, matrix_room_id, matrix_room_alias)
|
|
return dict(row)
|
|
|
|
|
|
async def update_room_matrix(room_id: str, matrix_room_id: str, matrix_room_alias: str) -> Optional[dict]:
|
|
"""Оновити Matrix поля кімнати"""
|
|
pool = await get_pool()
|
|
|
|
query = """
|
|
UPDATE city_rooms
|
|
SET matrix_room_id = $2, matrix_room_alias = $3
|
|
WHERE id = $1
|
|
RETURNING id, slug, name, description, is_default, created_at, created_by, matrix_room_id, matrix_room_alias
|
|
"""
|
|
|
|
row = await pool.fetchrow(query, room_id, matrix_room_id, matrix_room_alias)
|
|
return dict(row) if row else None
|
|
|
|
|
|
async def get_rooms_without_matrix() -> List[dict]:
|
|
"""Отримати кімнати без Matrix інтеграції"""
|
|
pool = await get_pool()
|
|
|
|
query = """
|
|
SELECT id, slug, name, description, is_default, created_at, created_by,
|
|
matrix_room_id, matrix_room_alias
|
|
FROM city_rooms
|
|
WHERE matrix_room_id IS NULL
|
|
ORDER BY created_at
|
|
"""
|
|
|
|
rows = await pool.fetch(query)
|
|
return [dict(row) for row in rows]
|
|
|
|
|
|
# =============================================================================
|
|
# City Room Messages Repository
|
|
# =============================================================================
|
|
|
|
async def get_room_messages(room_id: str, limit: int = 50) -> List[dict]:
|
|
"""Отримати повідомлення кімнати"""
|
|
pool = await get_pool()
|
|
|
|
query = """
|
|
SELECT id, room_id, author_user_id, author_agent_id, body, created_at
|
|
FROM city_room_messages
|
|
WHERE room_id = $1
|
|
ORDER BY created_at DESC
|
|
LIMIT $2
|
|
"""
|
|
|
|
rows = await pool.fetch(query, room_id, limit)
|
|
# Reverse для правильного порядку (старі → нові)
|
|
return [dict(row) for row in reversed(rows)]
|
|
|
|
|
|
async def create_room_message(
|
|
room_id: str,
|
|
body: str,
|
|
author_user_id: Optional[str] = None,
|
|
author_agent_id: Optional[str] = None
|
|
) -> dict:
|
|
"""Створити повідомлення в кімнаті"""
|
|
pool = await get_pool()
|
|
|
|
message_id = generate_id("m_city")
|
|
|
|
query = """
|
|
INSERT INTO city_room_messages (id, room_id, author_user_id, author_agent_id, body)
|
|
VALUES ($1, $2, $3, $4, $5)
|
|
RETURNING id, room_id, author_user_id, author_agent_id, body, created_at
|
|
"""
|
|
|
|
row = await pool.fetchrow(query, message_id, room_id, author_user_id, author_agent_id, body)
|
|
return dict(row)
|
|
|
|
|
|
# =============================================================================
|
|
# City Feed Events Repository
|
|
# =============================================================================
|
|
|
|
async def get_feed_events(limit: int = 20, offset: int = 0) -> List[dict]:
|
|
"""Отримати події feed"""
|
|
pool = await get_pool()
|
|
|
|
query = """
|
|
SELECT id, kind, room_id, user_id, agent_id, payload, created_at
|
|
FROM city_feed_events
|
|
ORDER BY created_at DESC
|
|
LIMIT $1 OFFSET $2
|
|
"""
|
|
|
|
rows = await pool.fetch(query, limit, offset)
|
|
return [dict(row) for row in rows]
|
|
|
|
|
|
async def create_feed_event(
|
|
kind: str,
|
|
payload: dict,
|
|
room_id: Optional[str] = None,
|
|
user_id: Optional[str] = None,
|
|
agent_id: Optional[str] = None
|
|
) -> dict:
|
|
"""Створити подію в feed"""
|
|
pool = await get_pool()
|
|
|
|
event_id = generate_id("evt_city")
|
|
|
|
query = """
|
|
INSERT INTO city_feed_events (id, kind, room_id, user_id, agent_id, payload)
|
|
VALUES ($1, $2, $3, $4, $5, $6::jsonb)
|
|
RETURNING id, kind, room_id, user_id, agent_id, payload, created_at
|
|
"""
|
|
|
|
import json
|
|
payload_json = json.dumps(payload)
|
|
|
|
row = await pool.fetchrow(query, event_id, kind, room_id, user_id, agent_id, payload_json)
|
|
return dict(row)
|
|
|