route.py

from fastapi import APIRouter, HTTPException, Depends
from starlette.responses import RedirectResponse

from database import get_database
from model import URLResponse, URLCreate
from service import URLShortenerService

router = APIRouter(tags=["URL Shortener"])
async def get_url_service():
    db = await get_database()
    return URLShortenerService(db)
@router.post("/shorten", response_model=URLResponse, status_code=201)
async def shorten_url(
        url_data: URLCreate,
        service: URLShortenerService = Depends(get_url_service)
):
    try:
        result = await service.get_shortened_url(str(url_data.original_url))

        return URLResponse(
            short_url=result
        )
    except Exception as e:
        raise HTTPException(status_code=500, detail="invalid url")

@router.get("/{short_code}")
async def redirect_to_original(
    short_code: str,
    service: URLShortenerService = Depends(get_url_service)
):
    original_url = await service.get_original_url(short_code)
    return RedirectResponse(url=original_url)