"""
Business logic reusable functions related to IAM

Exports:
 - service_key_dependency: bool: verifies request headers contain the correct api key for the service
"""
from typing import Annotated

from src.service.models import Service
from src.database import db_dependency
from src.schemas import ResourceName

from fastapi import HTTPException, status, Request, Depends


def valid_service_key(db: db_dependency, request: Request, rn: ResourceName) -> bool:
	api_key = request.headers.get("X-API-Key", None)
	if not api_key:
		raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED)
	service = rn.service
	result = db.query(Service).filter(Service.name == service).filter(Service.api_key == api_key).first()
	if result is None:
		raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED)

	return True

service_key_dependency = Annotated[bool, Depends(valid_service_key)]