57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
from urllib.parse import urlparse
|
|
|
|
from django.http import HttpRequest, HttpResponse, HttpResponseForbidden
|
|
from pyrogram import Chat as PyrogramChat
|
|
from pyrogram.api.types.chat import Chat as MTProtoChat
|
|
from pyrogram.api.types.user import User as MTProtoUser
|
|
from pyrogram.api.types.channel import Channel as MTProtoChannel
|
|
|
|
|
|
def parse_mtproto_chat(client, chat):
|
|
if isinstance(chat, MTProtoChat):
|
|
return PyrogramChat._parse_chat_chat(client, chat)
|
|
elif isinstance(chat, MTProtoUser):
|
|
return PyrogramChat._parse_user_chat(client, chat)
|
|
return PyrogramChat._parse_channel_chat(client, chat)
|
|
|
|
|
|
def same_origin(current_uri, redirect_uri):
|
|
a = urlparse(current_uri)
|
|
if not a.scheme:
|
|
return True
|
|
b = urlparse(redirect_uri)
|
|
return (a.scheme, a.hostname, a.port) == (b.scheme, b.hostname, b.port)
|
|
|
|
|
|
def turbolinks(request: HttpRequest):
|
|
ctx = {}
|
|
if request.META.get('HTTP_TURBOLINKS_REFERRER') is not None:
|
|
ctx['is_turbolinks'] = True
|
|
return ctx
|
|
|
|
|
|
class TurbolinksMiddleware:
|
|
def process_request(self, request: HttpRequest):
|
|
referrer = request.META.get('HTTP_TURBOLINKS_REFERRER')
|
|
if referrer:
|
|
request.META['HTTP_REFERER'] = referrer
|
|
return
|
|
|
|
def process_response(self, request: HttpRequest, response: HttpResponse):
|
|
referrer = request.META.get('HTTP_TURBOLINKS_REFERRER')
|
|
if referrer is None:
|
|
return response
|
|
|
|
if response.has_header('Location'):
|
|
loc = response['Location']
|
|
request.session['_turbolinks_redirect_to'] = loc
|
|
|
|
if referrer and not same_origin(loc, referrer):
|
|
return HttpResponseForbidden()
|
|
else:
|
|
if request.session.get('_turbolinks_redirect_to'):
|
|
loc = request.session.pop('_turbolinks_redirect_to')
|
|
response['Turbolinks-Location'] = loc
|
|
|
|
return response
|