mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-06-28 21:49:40 +02:00
refactor(podcasts): mount lifecycle router, drop legacy routes
This commit is contained in:
parent
3eb7cdb2d8
commit
8b52cd0ac9
2 changed files with 1 additions and 212 deletions
|
|
@ -4,6 +4,7 @@ from app.automations.api import router as automations_router
|
||||||
from app.file_storage.api import router as file_storage_router
|
from app.file_storage.api import router as file_storage_router
|
||||||
from app.gateway import require_gateway_enabled
|
from app.gateway import require_gateway_enabled
|
||||||
from app.notifications.api import router as notifications_router
|
from app.notifications.api import router as notifications_router
|
||||||
|
from app.podcasts.api import router as podcasts_router
|
||||||
|
|
||||||
from .agent_action_log_route import router as agent_action_log_router
|
from .agent_action_log_route import router as agent_action_log_router
|
||||||
from .agent_flags_route import router as agent_flags_router
|
from .agent_flags_route import router as agent_flags_router
|
||||||
|
|
@ -50,7 +51,6 @@ from .notes_routes import router as notes_router
|
||||||
from .notion_add_connector_route import router as notion_add_connector_router
|
from .notion_add_connector_route import router as notion_add_connector_router
|
||||||
from .obsidian_plugin_routes import router as obsidian_plugin_router
|
from .obsidian_plugin_routes import router as obsidian_plugin_router
|
||||||
from .onedrive_add_connector_route import router as onedrive_add_connector_router
|
from .onedrive_add_connector_route import router as onedrive_add_connector_router
|
||||||
from .podcasts_routes import router as podcasts_router
|
|
||||||
from .prompts_routes import router as prompts_router
|
from .prompts_routes import router as prompts_router
|
||||||
from .public_chat_routes import router as public_chat_router
|
from .public_chat_routes import router as public_chat_router
|
||||||
from .rbac_routes import router as rbac_router
|
from .rbac_routes import router as rbac_router
|
||||||
|
|
|
||||||
|
|
@ -1,211 +0,0 @@
|
||||||
"""
|
|
||||||
Podcast routes for CRUD operations and audio streaming.
|
|
||||||
|
|
||||||
These routes support the podcast generation feature in new-chat.
|
|
||||||
Frontend polls GET /podcasts/{podcast_id} to check status field.
|
|
||||||
"""
|
|
||||||
|
|
||||||
import os
|
|
||||||
from pathlib import Path
|
|
||||||
|
|
||||||
from fastapi import APIRouter, Depends, HTTPException
|
|
||||||
from fastapi.responses import StreamingResponse
|
|
||||||
from sqlalchemy import select
|
|
||||||
from sqlalchemy.exc import SQLAlchemyError
|
|
||||||
from sqlalchemy.ext.asyncio import AsyncSession
|
|
||||||
|
|
||||||
from app.db import (
|
|
||||||
Permission,
|
|
||||||
Podcast,
|
|
||||||
SearchSpace,
|
|
||||||
SearchSpaceMembership,
|
|
||||||
User,
|
|
||||||
get_async_session,
|
|
||||||
)
|
|
||||||
from app.schemas import PodcastRead
|
|
||||||
from app.users import current_active_user
|
|
||||||
from app.utils.rbac import check_permission
|
|
||||||
|
|
||||||
router = APIRouter()
|
|
||||||
|
|
||||||
|
|
||||||
@router.get("/podcasts", response_model=list[PodcastRead])
|
|
||||||
async def read_podcasts(
|
|
||||||
skip: int = 0,
|
|
||||||
limit: int = 100,
|
|
||||||
search_space_id: int | None = None,
|
|
||||||
session: AsyncSession = Depends(get_async_session),
|
|
||||||
user: User = Depends(current_active_user),
|
|
||||||
):
|
|
||||||
"""
|
|
||||||
List podcasts the user has access to.
|
|
||||||
Requires PODCASTS_READ permission for the search space(s).
|
|
||||||
"""
|
|
||||||
if skip < 0 or limit < 1:
|
|
||||||
raise HTTPException(status_code=400, detail="Invalid pagination parameters")
|
|
||||||
try:
|
|
||||||
if search_space_id is not None:
|
|
||||||
# Check permission for specific search space
|
|
||||||
await check_permission(
|
|
||||||
session,
|
|
||||||
user,
|
|
||||||
search_space_id,
|
|
||||||
Permission.PODCASTS_READ.value,
|
|
||||||
"You don't have permission to read podcasts in this search space",
|
|
||||||
)
|
|
||||||
result = await session.execute(
|
|
||||||
select(Podcast)
|
|
||||||
.filter(Podcast.search_space_id == search_space_id)
|
|
||||||
.offset(skip)
|
|
||||||
.limit(limit)
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
# Get podcasts from all search spaces user has membership in
|
|
||||||
result = await session.execute(
|
|
||||||
select(Podcast)
|
|
||||||
.join(SearchSpace)
|
|
||||||
.join(SearchSpaceMembership)
|
|
||||||
.filter(SearchSpaceMembership.user_id == user.id)
|
|
||||||
.offset(skip)
|
|
||||||
.limit(limit)
|
|
||||||
)
|
|
||||||
return result.scalars().all()
|
|
||||||
except HTTPException:
|
|
||||||
raise
|
|
||||||
except SQLAlchemyError:
|
|
||||||
raise HTTPException(
|
|
||||||
status_code=500, detail="Database error occurred while fetching podcasts"
|
|
||||||
) from None
|
|
||||||
|
|
||||||
|
|
||||||
@router.get("/podcasts/{podcast_id}", response_model=PodcastRead)
|
|
||||||
async def read_podcast(
|
|
||||||
podcast_id: int,
|
|
||||||
session: AsyncSession = Depends(get_async_session),
|
|
||||||
user: User = Depends(current_active_user),
|
|
||||||
):
|
|
||||||
"""
|
|
||||||
Get a specific podcast by ID.
|
|
||||||
|
|
||||||
Requires authentication with PODCASTS_READ permission.
|
|
||||||
For public podcast access, use /public/{share_token}/podcasts/{podcast_id}/stream
|
|
||||||
"""
|
|
||||||
try:
|
|
||||||
result = await session.execute(select(Podcast).filter(Podcast.id == podcast_id))
|
|
||||||
podcast = result.scalars().first()
|
|
||||||
|
|
||||||
if not podcast:
|
|
||||||
raise HTTPException(
|
|
||||||
status_code=404,
|
|
||||||
detail="Podcast not found",
|
|
||||||
)
|
|
||||||
|
|
||||||
await check_permission(
|
|
||||||
session,
|
|
||||||
user,
|
|
||||||
podcast.search_space_id,
|
|
||||||
Permission.PODCASTS_READ.value,
|
|
||||||
"You don't have permission to read podcasts in this search space",
|
|
||||||
)
|
|
||||||
|
|
||||||
return PodcastRead.from_orm_with_entries(podcast)
|
|
||||||
except HTTPException as he:
|
|
||||||
raise he
|
|
||||||
except SQLAlchemyError:
|
|
||||||
raise HTTPException(
|
|
||||||
status_code=500, detail="Database error occurred while fetching podcast"
|
|
||||||
) from None
|
|
||||||
|
|
||||||
|
|
||||||
@router.delete("/podcasts/{podcast_id}", response_model=dict)
|
|
||||||
async def delete_podcast(
|
|
||||||
podcast_id: int,
|
|
||||||
session: AsyncSession = Depends(get_async_session),
|
|
||||||
user: User = Depends(current_active_user),
|
|
||||||
):
|
|
||||||
"""
|
|
||||||
Delete a podcast.
|
|
||||||
Requires PODCASTS_DELETE permission for the search space.
|
|
||||||
"""
|
|
||||||
try:
|
|
||||||
result = await session.execute(select(Podcast).filter(Podcast.id == podcast_id))
|
|
||||||
db_podcast = result.scalars().first()
|
|
||||||
|
|
||||||
if not db_podcast:
|
|
||||||
raise HTTPException(status_code=404, detail="Podcast not found")
|
|
||||||
|
|
||||||
# Check permission for the search space
|
|
||||||
await check_permission(
|
|
||||||
session,
|
|
||||||
user,
|
|
||||||
db_podcast.search_space_id,
|
|
||||||
Permission.PODCASTS_DELETE.value,
|
|
||||||
"You don't have permission to delete podcasts in this search space",
|
|
||||||
)
|
|
||||||
|
|
||||||
await session.delete(db_podcast)
|
|
||||||
await session.commit()
|
|
||||||
return {"message": "Podcast deleted successfully"}
|
|
||||||
except HTTPException as he:
|
|
||||||
raise he
|
|
||||||
except SQLAlchemyError:
|
|
||||||
await session.rollback()
|
|
||||||
raise HTTPException(
|
|
||||||
status_code=500, detail="Database error occurred while deleting podcast"
|
|
||||||
) from None
|
|
||||||
|
|
||||||
|
|
||||||
@router.get("/podcasts/{podcast_id}/stream")
|
|
||||||
@router.get("/podcasts/{podcast_id}/audio")
|
|
||||||
async def stream_podcast(
|
|
||||||
podcast_id: int,
|
|
||||||
session: AsyncSession = Depends(get_async_session),
|
|
||||||
user: User = Depends(current_active_user),
|
|
||||||
):
|
|
||||||
"""
|
|
||||||
Stream a podcast audio file.
|
|
||||||
|
|
||||||
Requires authentication with PODCASTS_READ permission.
|
|
||||||
For public podcast access, use /public/{share_token}/podcasts/{podcast_id}/stream
|
|
||||||
|
|
||||||
Note: Both /stream and /audio endpoints are supported for compatibility.
|
|
||||||
"""
|
|
||||||
try:
|
|
||||||
result = await session.execute(select(Podcast).filter(Podcast.id == podcast_id))
|
|
||||||
podcast = result.scalars().first()
|
|
||||||
|
|
||||||
if not podcast:
|
|
||||||
raise HTTPException(status_code=404, detail="Podcast not found")
|
|
||||||
|
|
||||||
await check_permission(
|
|
||||||
session,
|
|
||||||
user,
|
|
||||||
podcast.search_space_id,
|
|
||||||
Permission.PODCASTS_READ.value,
|
|
||||||
"You don't have permission to access podcasts in this search space",
|
|
||||||
)
|
|
||||||
|
|
||||||
file_path = podcast.file_location
|
|
||||||
|
|
||||||
if not file_path or not os.path.isfile(file_path):
|
|
||||||
raise HTTPException(status_code=404, detail="Podcast audio file not found")
|
|
||||||
|
|
||||||
def iterfile():
|
|
||||||
with open(file_path, mode="rb") as file_like:
|
|
||||||
yield from file_like
|
|
||||||
|
|
||||||
return StreamingResponse(
|
|
||||||
iterfile(),
|
|
||||||
media_type="audio/mpeg",
|
|
||||||
headers={
|
|
||||||
"Accept-Ranges": "bytes",
|
|
||||||
"Content-Disposition": f"inline; filename={Path(file_path).name}",
|
|
||||||
},
|
|
||||||
)
|
|
||||||
|
|
||||||
except HTTPException as he:
|
|
||||||
raise he
|
|
||||||
except Exception as e:
|
|
||||||
raise HTTPException(
|
|
||||||
status_code=500, detail=f"Error streaming podcast: {e!s}"
|
|
||||||
) from e
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue