bfcc0b9f29
# What this PR does
Introduces a new class,
`apps.grafana_plugin.ui_url_builder.UIURLBuilder`, which is responsible
for... building UI URLs (😄). The class mainly does two things:
- it will decide if the URL should point to `grafana-oncall-app` or
`grafana-irm-app` based on the value of
`organization.is_grafana_irm_enabled` (**NOTE**: this value isn't yet
being set + defaults to `False`; logic for setting this value will be
done in a subsequent PR)
- Adds `enum`s, `OnCallPage` and `IncidentPage` to DRYify hardcoded UI
URLs (in case we decide to change these slightly in the near future)
## Checklist
- [x] Unit, integration, and e2e (if applicable) tests updated
- [x] Documentation added (or `pr:no public docs` PR label added if not
required)
- [x] Added the relevant release notes label (see labels prefixed w/
`release:`). These labels dictate how your PR will
show up in the autogenerated release notes.
40 lines
2 KiB
Python
40 lines
2 KiB
Python
import logging
|
|
|
|
from django.http import HttpResponse
|
|
from django.shortcuts import redirect
|
|
from rest_framework import status
|
|
from social_core import exceptions
|
|
from social_django.middleware import SocialAuthExceptionMiddleware
|
|
|
|
from apps.grafana_plugin.ui_url_builder import UIURLBuilder
|
|
from apps.social_auth.backends import LoginSlackOAuth2V2
|
|
from apps.social_auth.exceptions import InstallMultiRegionSlackException
|
|
from common.constants.slack_auth import REDIRECT_AFTER_SLACK_INSTALL, SLACK_AUTH_FAILED, SLACK_REGION_ERROR
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class SocialAuthAuthCanceledExceptionMiddleware(SocialAuthExceptionMiddleware):
|
|
def process_exception(self, request, exception):
|
|
backend = getattr(exception, "backend", None)
|
|
url_builder = UIURLBuilder(request.user.organization)
|
|
url_builder_function = url_builder.chatops
|
|
|
|
if backend is not None and isinstance(backend, LoginSlackOAuth2V2):
|
|
url_builder_function = url_builder.user_profile
|
|
|
|
if exception:
|
|
logger.warning(f"SocialAuthAuthCanceledExceptionMiddleware.process_exception: {exception}")
|
|
|
|
if isinstance(exception, exceptions.AuthCanceled):
|
|
# if user canceled authentication, redirect them to the previous page using the same link
|
|
# as we used to redirect after auth/install
|
|
return redirect(url_builder_function())
|
|
elif isinstance(exception, exceptions.AuthFailed):
|
|
# if authentication was failed, redirect user to the plugin page using the same link
|
|
# as we used to redirect after auth/install with error flag
|
|
return redirect(url_builder_function(f"?slack_error={SLACK_AUTH_FAILED}"))
|
|
elif isinstance(exception, KeyError) and REDIRECT_AFTER_SLACK_INSTALL in exception.args:
|
|
return HttpResponse(status=status.HTTP_401_UNAUTHORIZED)
|
|
elif isinstance(exception, InstallMultiRegionSlackException):
|
|
return redirect(url_builder_function(f"?tab=Slack&slack_error={SLACK_REGION_ERROR}"))
|