from lti1p3platform.ltiplatform import LTI1P3PlatformConfAbstract
from lti1p3platform.registration import Registration

class LTIPlatformConf(LTI1P3PlatformConfAbstract):
    def init_platform_config(self, platform_settings, platform_key_set):
        """
        register platform configuration
        """
        registration = Registration() \
            .set_client_id(platform_settings.client_id) \
            .set_deployment_id(platform_settings.deployment_id) \
            .set_launch_url(platform_settings.launch_url) \
            .set_deeplink_launch_url(platform_settings.deeplink_launch_url) \
            .set_oidc_login_url(platform_settings.oidc_login_url) \
            .set_tool_key_set_url(platform_settings.key_set_url) \
            .set_platform_public_key(platform_key_set.public_key) \
            .set_platform_private_key(platform_key_set.private_key) \
            .set_tool_redirect_uris([
                "https://tool.example.com/lti/launch",
                "https://tool.example.com/lti/deeplink",
            ])

        self._registration = registration

def get_registered_platform(*args, **kwargs):
    ...

    return LTIPlatformConf(*args, **kwargs)

# public JWK endpoint
def get_jwks(request, *args, **kwargs):
    platform = get_registered_platform(*args, **kwargs)

    return HttpResponseJSON(platform.get_jwks())

from lti1p3platform.oidc_login import OIDCLoginAbstract

class OIDCLogin(OIDCLoginAbstract):
    def set_lti_message_hint(self, **kwargs):
        """ set your own lti_message_hint """
        pass

    def get_lti_message_hint(self):
        """ get your lti_message_hint """
        pass

    def get_redirect(self, url):
        """
        This will be invoked in initiate_login, and it depends on which web framework you are using.
        Here is an example for Django framework:
        """
        return HttpResponseRedirect(url)

    def render_error_page(self, message, status_code):
        """
        This will be invoked when the library decides the error must not be
        returned as an OAuth redirect.
        """
        return HttpResponse(message, status=status_code)

# Initiate login endpoint
def preflight_lti_1p3_launch(request, user_id, *args, **kwargs):
    platform = get_registered_platform(*args, **kwargs)
    oidc_login = OIDCLogin(request, platform)

    # Redirect the current login user to the tool provider,
    return oidc_login.initiate_login(user_id)

from lti1p3platform.message_launch import MessageLaunchAbstract

class LTI1p3MessageLaunch(MessageLaunchAbstract):
    def render_launch_form(self, launch_data, **kwargs):
        """
        This will be invoked in the last step of `lti_launch`.
        So you could render a template in this method. This template should render a form, and then submit it to the tool's launch URL. There is a django example in framework/django/message_launch.py
        """
        pass

    def get_redirect(self, url):
        """
        This will be invoked when launch validation fails with a redirectable
        OAuth/OIDC error.
        """
        return HttpResponseRedirect(url)

    def render_error_page(self, message, status_code):
        """
        This will be invoked when launch validation fails with a local-only
        error such as `invalid_request_uri` or `server_error`.
        """
        return HttpResponse(message, status=status_code)

    def prepare_launch(self, preflight_response, **kwargs):
        """
        You could do some other checks and get some contexts from `lti_message_hint` you've set in previous request
        Also you could call these methods to prepare your own jwt payload:
            - set_user_data
            - set_resource_link_claim
            - set_launch_context_claim
            - set_custom_parameters_claim

        Make sure do these things before lti_launch, it could send necessary launch parameters to the tool.
        """
        pass

def lti_resource_link_launch(request, *args, **kwargs):
    platform = get_registered_platform(*args, **kwargs)
    message_launch = LTI1p3MessageLaunch(request, platform)

    return message_launch.lti_launch(*args, **kwargs)

from lti1p3platform.message_launch import LTIAdvantageMessageLaunchAbstract

class LTI1p3MessageLaunch(LTIAdvantageMessageLaunchAbstract):
    # same abstract method implementations as MessageLaunchAbstract …
    pass

message_launch.set_dl(
    deep_link_return_url="https://platform.example.com/lti/deeplink/return",
    title="Select content",
    description="Choose a resource to embed",
    accept_multiple=True,          # allow selecting more than one item
    auto_create=True,              # platform auto-creates items without extra confirmation
    accept_types={"ltiResourceLink", "link", "file"},
    accept_presentation_document_targets={"iframe", "window"},
)

message_launch.set_ags(
    lineitems_url="https://platform.example.com/lti/lineitems/",
    lineitem_url="https://platform.example.com/lti/lineitems/42",  # optional
    allow_creating_lineitems=True,
    results_service_enabled=True,
    scores_service_enabled=True,
)

message_launch.set_nrps(
    context_memberships_url="https://platform.example.com/lti/memberships/",
)

def lti_advantage_launch(request, *args, **kwargs):
    platform = get_registered_platform(*args, **kwargs)
    message_launch = LTI1p3MessageLaunch(request, platform)

    message_launch \
        .set_dl(
            deep_link_return_url="https://platform.example.com/lti/deeplink/return",
        ) \
        .set_ags(
            lineitems_url="https://platform.example.com/lti/lineitems/",
        ) \
        .set_nrps(
            context_memberships_url="https://platform.example.com/lti/memberships/",
        )

    return message_launch.lti_launch(*args, **kwargs)