90 lines
3.1 KiB
Python
90 lines
3.1 KiB
Python
from fastapi import HTTPException, status
|
|
from pydantic import BaseModel, Field
|
|
from sqlalchemy import select
|
|
from sqlalchemy.orm import Session
|
|
|
|
from .log_service import record_otp_log
|
|
from .message_send_service import SendMessageError, normalize_recipients
|
|
from .models import UrmiOtpProject, UserAccount
|
|
from .urmiotp_catalog import OTP_TEMPLATES
|
|
|
|
TEMPLATE_TYPES = {template["type"] for template in OTP_TEMPLATES}
|
|
|
|
|
|
class GenerateOtpPayload(BaseModel):
|
|
project_name: str = Field(min_length=3, max_length=30)
|
|
template_type: str = Field(min_length=3, max_length=40)
|
|
to: str = Field(min_length=10, max_length=15)
|
|
variables: dict[str, str] = Field(default_factory=dict)
|
|
|
|
|
|
def _build_otp_message(template_type: str, message_label: str, variables: dict[str, str]) -> str:
|
|
template = next((item for item in OTP_TEMPLATES if item["type"] == template_type), None)
|
|
if not template:
|
|
raise HTTPException(status_code=status.HTTP_422_UNPROCESSABLE_ENTITY, detail="Unknown template type")
|
|
|
|
otp_value = variables.get("otp", "").strip()
|
|
validity = variables.get("validity_minutes", "").strip()
|
|
if not otp_value or not validity:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
|
|
detail="variables.otp and variables.validity_minutes are required",
|
|
)
|
|
|
|
content = template["content"].replace("{message_label}", message_label)
|
|
for value in (otp_value, validity):
|
|
content = content.replace("{#var#}", value, 1)
|
|
return content
|
|
|
|
|
|
def generate_otp_for_user(
|
|
db: Session,
|
|
user: UserAccount,
|
|
payload: GenerateOtpPayload,
|
|
) -> dict:
|
|
if payload.template_type not in TEMPLATE_TYPES:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
|
|
detail=f"template_type must be one of: {', '.join(sorted(TEMPLATE_TYPES))}",
|
|
)
|
|
|
|
project = db.scalar(
|
|
select(UrmiOtpProject).where(
|
|
UrmiOtpProject.user_id == user.id,
|
|
UrmiOtpProject.project_name == payload.project_name.strip(),
|
|
)
|
|
)
|
|
if not project:
|
|
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="UrmiOTP project not found")
|
|
if project.status != "approved":
|
|
raise HTTPException(
|
|
status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
|
|
detail="UrmiOTP project is not approved yet",
|
|
)
|
|
|
|
try:
|
|
recipients = normalize_recipients([payload.to])
|
|
except SendMessageError as exc:
|
|
raise HTTPException(status_code=exc.status_code, detail=exc.detail) from exc
|
|
|
|
recipient = recipients[0]
|
|
message = _build_otp_message(payload.template_type, project.message_label, payload.variables)
|
|
|
|
entry = record_otp_log(
|
|
db,
|
|
user,
|
|
project_name=project.project_name,
|
|
template_type=payload.template_type,
|
|
to_number=recipient,
|
|
status="accepted",
|
|
)
|
|
|
|
return {
|
|
"success": True,
|
|
"log_id": str(entry.id),
|
|
"project_name": project.project_name,
|
|
"template_type": payload.template_type,
|
|
"to": recipient,
|
|
"message_preview": message,
|
|
"status": entry.status,
|
|
}
|