2020-10-22 21:45:05 +02:00
|
|
|
|
from django.contrib import admin
|
|
|
|
|
|
from .models import *
|
|
|
|
|
|
from .api import create_app
|
|
|
|
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
|
|
from requests.exceptions import Timeout
|
|
|
|
|
|
from django.core.exceptions import ObjectDoesNotExist
|
|
|
|
|
|
|
|
|
|
|
|
# Register your models here.
|
|
|
|
|
|
@admin.register(MastodonApplication)
|
|
|
|
|
|
class MastodonApplicationModelAdmin(admin.ModelAdmin):
|
|
|
|
|
|
|
|
|
|
|
|
def add_view(self, request, form_url='', extra_context=None):
|
|
|
|
|
|
"""
|
|
|
|
|
|
Dirty code here, use POST['domain_name'] to pass error message to user.
|
|
|
|
|
|
"""
|
|
|
|
|
|
if request.method == 'POST':
|
|
|
|
|
|
if not request.POST.get('client_id') and not request.POST.get('client_secret'):
|
|
|
|
|
|
# make the post data mutable
|
|
|
|
|
|
request.POST = request.POST.copy()
|
|
|
|
|
|
# (is_proxy xor proxy_to) or (proxy_to!=null and is_proxy=false)
|
|
|
|
|
|
if (bool(request.POST.get('is_proxy')) or bool(request.POST.get('proxy_to'))) and\
|
|
|
|
|
|
not (bool(request.POST.get('is_proxy')) and bool(request.POST.get('proxy_to'))) or\
|
|
|
|
|
|
(not bool(request.POST.get('is_proxy')) and bool(request.POST.get('proxy_to'))):
|
|
|
|
|
|
request.POST['domain_name'] = _("请同时填写is_proxy和proxy_to。")
|
|
|
|
|
|
else:
|
2020-10-22 22:50:56 +02:00
|
|
|
|
if request.POST.get("is_proxy"):
|
2020-10-22 21:45:05 +02:00
|
|
|
|
try:
|
|
|
|
|
|
origin = MastodonApplication.objects.get(domain_name=request.POST['proxy_to'])
|
|
|
|
|
|
# set proxy credentials to those of its original site
|
|
|
|
|
|
request.POST['app_id'] = origin.app_id
|
|
|
|
|
|
request.POST['client_id'] = origin.client_id
|
|
|
|
|
|
request.POST['client_secret'] = origin.client_secret
|
|
|
|
|
|
request.POST['vapid_key'] = origin.vapid_key
|
|
|
|
|
|
except ObjectDoesNotExist:
|
|
|
|
|
|
request.POST['domain_name'] = _("proxy_to所指域名不存在，请先添加原站点。")
|
|
|
|
|
|
else:
|
|
|
|
|
|
# create mastodon app
|
|
|
|
|
|
try:
|
|
|
|
|
|
response = create_app(request.POST.get('domain_name'))
|
|
|
|
|
|
except (Timeout, ConnectionError):
|
2022-02-13 17:00:55 -05:00
|
|
|
|
request.POST['domain_name'] = _("联邦网络请求超时。")
|
2020-10-22 21:45:05 +02:00
|
|
|
|
except Exception as e:
|
|
|
|
|
|
request.POST['domain_name'] = str(e)
|
|
|
|
|
|
else:
|
|
|
|
|
|
# fill the form with returned data
|
|
|
|
|
|
data = response.json()
|
|
|
|
|
|
if response.status_code != 200:
|
|
|
|
|
|
request.POST['domain_name'] = str(data)
|
|
|
|
|
|
else:
|
|
|
|
|
|
request.POST['app_id'] = data['id']
|
|
|
|
|
|
request.POST['client_id'] = data['client_id']
|
|
|
|
|
|
request.POST['client_secret'] = data['client_secret']
|
|
|
|
|
|
request.POST['vapid_key'] = data['vapid_key']
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
return super().add_view(request, form_url=form_url, extra_context=extra_context)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
admin.site.register(CrossSiteUserInfo)
|
