from typing import List

from django.contrib.auth import get_user_model
from ninja import Router, Schema, Field
from ninja.pagination import paginate

from epapi.v1.auth import BearerTokenAuth

from .logic import PackageManager
from .models import User, Compound


def _anonymous_or_real(request):
    if request.user.is_authenticated and not request.user.is_anonymous:
        return request.user
    return get_user_model().objects.get(username="anonymous")


router = Router(auth=BearerTokenAuth())


class UserSchema(Schema):
    email: str
    username: str
    id: str = Field(None, alias="url")


class PackageIn(Schema):
    name: str
    description: str


class PackageOut(Schema):
    id: str = Field(None, alias="url")
    name: str
    reviewed: bool
    compound_links: str = None

    @staticmethod
    def resolve_compound_links(obj):
        return f"{obj.url}/compound"


class Error(Schema):
    message: str


class CompoundSchema(Schema):
    name: str = Field(None, alias="name")
    id: str = Field(None, alias="url")
    smiles: str = Field(None, alias="default_structure.smiles")
    reviewed: bool = Field(None, alias="package.reviewed")


@router.get("/user", response={200: List[UserSchema], 403: Error})
def get_users(request):
    return User.objects.all()


@router.get("/package", response={200: List[PackageOut], 403: Error})
def get_packages(request):
    return PackageManager.get_all_readable_packages(request.user, include_reviewed=True)


@router.post("/package", response=PackageOut)
def create_package(request, package: PackageIn):
    user = request.auth
    name = package.name.strip()
    description = package.description.strip()

    p = PackageManager.create_package(user, name, description=description)
    return p


@router.get("/package/{uuid:package_uuid}", response={200: PackageOut, 403: Error})
def get_package(request, package_uuid):
    try:
        return PackageManager.get_package_by_id(request.auth, package_id=package_uuid)
    except ValueError:
        return 403, {
            "message": f"Getting Package with id {package_uuid} failed due to insufficient rights!"
        }


@router.get("/compound", response={200: List[CompoundSchema], 403: Error})
@paginate
def get_compounds(request):
    qs = Compound.objects.none()
    for p in PackageManager.get_reviewed_packages():
        qs |= Compound.objects.filter(package=p)
    return qs


@router.get(
    "/package/{uuid:package_uuid}/compound", response={200: List[CompoundSchema], 403: Error}
)
@paginate
def get_package_compounds(request, package_uuid):
    try:
        p = PackageManager.get_package_by_id(request.auth, package_uuid)
        return Compound.objects.filter(package=p)
    except ValueError:
        return 403, {
            "message": f"Getting Compounds for Package with id {package_uuid} failed due to insufficient rights!"
        }