from dj_kaos.contrib.users.rest import UserSerializer
from dj_kaos.models.rest import KaosModelSerializer, KaosModelViewSet
from dj_kaos.rest.routers import KaosRestRouter

from .models import Role, RoleMember


class RoleSerializer(KaosModelSerializer):
    class Meta:
        model = Role


class RoleMemberSerializer(KaosModelSerializer):
    user = UserSerializer()
    role = RoleSerializer()

    class Meta:
        model = RoleMember


class RoleViewSet(KaosModelViewSet):
    model = Role
    serializer_class = RoleSerializer


class RoleMemberViewSet(KaosModelViewSet):
    model = RoleMember
    serializer_class = RoleMemberSerializer


router = KaosRestRouter()
router.register(r'roles', RoleViewSet)
router.register(r'user-roles', RoleMemberViewSet)

iam_router = router

__all__ = (
    'RoleSerializer',
    'RoleMemberSerializer',
    'RoleViewSet',
    'RoleMemberViewSet',
    'router',
    'iam_router',
)