51 lines
1.8 KiB
Python
51 lines
1.8 KiB
Python
|
import logging
|
||
|
|
from urllib.parse import urljoin
|
||
|
|
|
||
|
|
from django.contrib.auth import REDIRECT_FIELD_NAME
|
||
|
|
from django.http import HttpResponseRedirect
|
||
|
|
from django.views.decorators.cache import never_cache
|
||
|
|
from django.views.decorators.csrf import csrf_exempt
|
||
|
|
from rest_framework.decorators import api_view, authentication_classes
|
||
|
|
from rest_framework.response import Response
|
||
|
|
from social_core.actions import do_auth, do_complete
|
||
|
|
from social_django.utils import psa
|
||
|
|
from social_django.views import _do_login
|
||
|
|
|
||
|
|
from apps.auth_token.auth import PluginAuthentication, SlackTokenAuthentication
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
@api_view(["GET"])
|
||
|
|
@authentication_classes([PluginAuthentication])
|
||
|
|
@never_cache
|
||
|
|
@psa("social:complete")
|
||
|
|
def overridden_login_slack_auth(request, backend):
|
||
|
|
# We can't just redirect frontend here because we need to make a API call and pass tokens to this view from JS.
|
||
|
|
# So frontend can't follow our redirect.
|
||
|
|
# So wrapping and returning URL to redirect as a string.
|
||
|
|
url_to_redirect_to = do_auth(request.backend, redirect_name=REDIRECT_FIELD_NAME).url
|
||
|
|
|
||
|
|
return Response(url_to_redirect_to, 200)
|
||
|
|
|
||
|
|
|
||
|
|
@api_view(["GET"])
|
||
|
|
@authentication_classes([SlackTokenAuthentication])
|
||
|
|
@never_cache
|
||
|
|
@csrf_exempt
|
||
|
|
@psa("social:complete")
|
||
|
|
def overridden_complete_slack_auth(request, backend, *args, **kwargs):
|
||
|
|
"""Authentication complete view"""
|
||
|
|
do_complete(
|
||
|
|
request.backend,
|
||
|
|
_do_login,
|
||
|
|
user=request.user,
|
||
|
|
redirect_name=REDIRECT_FIELD_NAME,
|
||
|
|
request=request,
|
||
|
|
*args,
|
||
|
|
**kwargs,
|
||
|
|
)
|
||
|
|
# We build the frontend url using org url since multiple stacks could be connected to one backend.
|
||
|
|
return_to = urljoin(request.user.organization.grafana_url, "/a/grafana-oncall-app/?page=chat-ops")
|
||
|
|
return HttpResponseRedirect(return_to)
|