mirror of
https://github.com/Monadical-SAS/reflector.git
synced 2025-12-20 12:19:06 +00:00
140 lines
4.0 KiB
Python
140 lines
4.0 KiB
Python
from datetime import datetime
|
|
|
|
import sqlalchemy
|
|
from fastapi import HTTPException
|
|
from pydantic import BaseModel, ConfigDict, Field
|
|
from reflector.db import database, metadata
|
|
from reflector.db.transcripts import generate_uuid4
|
|
from sqlalchemy.sql import false
|
|
|
|
rooms = sqlalchemy.Table(
|
|
"room",
|
|
metadata,
|
|
sqlalchemy.Column("id", sqlalchemy.String, primary_key=True),
|
|
sqlalchemy.Column("name", sqlalchemy.String, nullable=False),
|
|
sqlalchemy.Column("user_id", sqlalchemy.String, nullable=False),
|
|
sqlalchemy.Column("created_at", sqlalchemy.DateTime, nullable=False),
|
|
sqlalchemy.Column(
|
|
"zulip_auto_post", sqlalchemy.Boolean, nullable=False, server_default=false()
|
|
),
|
|
sqlalchemy.Column("zulip_stream", sqlalchemy.String),
|
|
sqlalchemy.Column("zulip_topic", sqlalchemy.String),
|
|
)
|
|
|
|
|
|
class Room(BaseModel):
|
|
id: str = Field(default_factory=generate_uuid4)
|
|
name: str
|
|
user_id: str
|
|
created_at: datetime = Field(default_factory=datetime.utcnow)
|
|
zulip_auto_post: bool = False
|
|
zulip_stream: str = ""
|
|
zulip_topic: str = ""
|
|
|
|
|
|
class RoomController:
|
|
async def get_all(
|
|
self,
|
|
user_id: str | None = None,
|
|
order_by: str | None = None,
|
|
return_query: bool = False,
|
|
) -> list[Room]:
|
|
"""
|
|
Get all rooms
|
|
|
|
If `user_id` is specified, only return rooms that belong to the user.
|
|
Otherwise, return all rooms.
|
|
|
|
Parameters:
|
|
- `order_by`: field to order by, e.g. "-created_at"
|
|
"""
|
|
query = rooms.select()
|
|
if user_id is not None:
|
|
query = query.where(rooms.c.user_id == user_id)
|
|
|
|
if order_by is not None:
|
|
field = getattr(rooms.c, order_by[1:])
|
|
if order_by.startswith("-"):
|
|
field = field.desc()
|
|
query = query.order_by(field)
|
|
|
|
if return_query:
|
|
return query
|
|
|
|
results = await database.fetch_all(query)
|
|
return results
|
|
|
|
async def add(
|
|
self,
|
|
name: str,
|
|
user_id: str,
|
|
):
|
|
"""
|
|
Add a new room
|
|
"""
|
|
room = Room(
|
|
name=name,
|
|
user_id=user_id,
|
|
)
|
|
query = rooms.insert().values(**room.model_dump())
|
|
await database.execute(query)
|
|
return room
|
|
|
|
async def get_by_id(self, room_id: str, **kwargs) -> Room | None:
|
|
"""
|
|
Get a room by id
|
|
"""
|
|
query = rooms.select().where(rooms.c.id == room_id)
|
|
if "user_id" in kwargs:
|
|
query = query.where(rooms.c.user_id == kwargs["user_id"])
|
|
result = await database.fetch_one(query)
|
|
if not result:
|
|
return None
|
|
return Room(**result)
|
|
|
|
async def get_by_name(self, room_name: str, **kwargs) -> Room | None:
|
|
"""
|
|
Get a room by name
|
|
"""
|
|
query = rooms.select().where(rooms.c.name == room_name)
|
|
if "user_id" in kwargs:
|
|
query = query.where(rooms.c.user_id == kwargs["user_id"])
|
|
result = await database.fetch_one(query)
|
|
if not result:
|
|
return None
|
|
return Room(**result)
|
|
|
|
async def get_by_id_for_http(self, meeting_id: str, user_id: str | None) -> Room:
|
|
"""
|
|
Get a room by ID for HTTP request.
|
|
|
|
If not found, it will raise a 404 error.
|
|
"""
|
|
query = rooms.select().where(rooms.c.id == meeting_id)
|
|
result = await database.fetch_one(query)
|
|
if not result:
|
|
raise HTTPException(status_code=404, detail="Room not found")
|
|
|
|
room = Room(**result)
|
|
|
|
return room
|
|
|
|
async def remove_by_id(
|
|
self,
|
|
room_id: str,
|
|
user_id: str | None = None,
|
|
) -> None:
|
|
"""
|
|
Remove a room by id
|
|
"""
|
|
room = await self.get_by_id(room_id, user_id=user_id)
|
|
if not room:
|
|
return
|
|
if user_id is not None and room.user_id != user_id:
|
|
return
|
|
query = rooms.delete().where(rooms.c.id == room_id)
|
|
await database.execute(query)
|
|
|
|
|
|
rooms_controller = RoomController()
|