views.py 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752
  1. import base64
  2. import binascii
  3. from datetime import timedelta
  4. from functools import cached_property
  5. from django.conf import settings
  6. from django.contrib.auth import user_logged_in
  7. from django.contrib.auth.hashers import is_password_usable
  8. from django.core.cache import cache
  9. from django.core.mail import EmailMessage
  10. from django.http import Http404
  11. from django.shortcuts import redirect
  12. from django.template.loader import get_template
  13. from rest_framework import generics, mixins, status, viewsets
  14. from rest_framework.authentication import get_authorization_header
  15. from rest_framework.exceptions import (NotAcceptable, NotFound, PermissionDenied, ValidationError)
  16. from rest_framework.permissions import IsAuthenticated, SAFE_METHODS
  17. from rest_framework.renderers import JSONRenderer, StaticHTMLRenderer
  18. from rest_framework.response import Response
  19. from rest_framework.reverse import reverse
  20. from rest_framework.views import APIView
  21. import desecapi.authentication as auth
  22. from desecapi import metrics, models, serializers
  23. from desecapi.exceptions import ConcurrencyException
  24. from desecapi.pdns import get_serials
  25. from desecapi.pdns_change_tracker import PDNSChangeTracker
  26. from desecapi.permissions import ManageTokensPermission, IsDomainOwner, IsOwner, IsVPNClient, WithinDomainLimitOnPOST
  27. from desecapi.renderers import PlainTextRenderer
  28. def generate_confirmation_link(request, action_serializer, viewname, **kwargs):
  29. action = action_serializer.Meta.model(**kwargs)
  30. action_data = action_serializer(action).data
  31. confirmation_link = reverse(viewname, request=request, args=[action_data['code']])
  32. return confirmation_link, action_serializer.validity_period
  33. class EmptyPayloadMixin:
  34. def initialize_request(self, request, *args, **kwargs):
  35. # noinspection PyUnresolvedReferences
  36. request = super().initialize_request(request, *args, **kwargs)
  37. try:
  38. no_data = request.stream is None
  39. except:
  40. no_data = True
  41. if no_data:
  42. # In this case, data and files are both empty, so we can set request.data=None (instead of the default {}).
  43. # This allows distinguishing missing payload from empty dict payload.
  44. # See https://github.com/encode/django-rest-framework/pull/7195
  45. request._full_data = None
  46. return request
  47. class IdempotentDestroyMixin:
  48. def destroy(self, request, *args, **kwargs):
  49. try:
  50. # noinspection PyUnresolvedReferences
  51. super().destroy(request, *args, **kwargs)
  52. except Http404:
  53. pass
  54. return Response(status=status.HTTP_204_NO_CONTENT)
  55. class DomainViewMixin:
  56. @property
  57. def throttle_scope(self):
  58. return 'dns_api_read' if self.request.method in SAFE_METHODS else 'dns_api_write_rrsets'
  59. @property
  60. def throttle_scope_bucket(self):
  61. # Note: bucket should remain constant even when domain is recreated
  62. return None if self.request.method in SAFE_METHODS else self.kwargs['name']
  63. def get_serializer_context(self):
  64. # noinspection PyUnresolvedReferences
  65. return {**super().get_serializer_context(), 'domain': self.domain}
  66. def initial(self, request, *args, **kwargs):
  67. # noinspection PyUnresolvedReferences
  68. super().initial(request, *args, **kwargs)
  69. try:
  70. # noinspection PyAttributeOutsideInit, PyUnresolvedReferences
  71. self.domain = self.request.user.domains.get(name=self.kwargs['name'])
  72. except models.Domain.DoesNotExist:
  73. raise Http404
  74. class TokenViewSet(IdempotentDestroyMixin, viewsets.ModelViewSet):
  75. serializer_class = serializers.TokenSerializer
  76. permission_classes = (IsAuthenticated, ManageTokensPermission,)
  77. throttle_scope = 'account_management_passive'
  78. def get_queryset(self):
  79. return self.request.user.auth_tokens.all()
  80. def get_serializer(self, *args, **kwargs):
  81. # When creating a new token, return the plaintext representation
  82. if self.request.method == 'POST':
  83. kwargs.setdefault('include_plain', True)
  84. return super().get_serializer(*args, **kwargs)
  85. def perform_create(self, serializer):
  86. serializer.save(user=self.request.user)
  87. class DomainViewSet(IdempotentDestroyMixin,
  88. mixins.CreateModelMixin,
  89. mixins.RetrieveModelMixin,
  90. mixins.DestroyModelMixin,
  91. mixins.ListModelMixin,
  92. viewsets.GenericViewSet):
  93. serializer_class = serializers.DomainSerializer
  94. permission_classes = (IsAuthenticated, IsOwner, WithinDomainLimitOnPOST)
  95. lookup_field = 'name'
  96. lookup_value_regex = r'[^/]+'
  97. @property
  98. def throttle_scope(self):
  99. return 'dns_api_read' if self.request.method in SAFE_METHODS else 'dns_api_write_domains'
  100. def get_queryset(self):
  101. return self.request.user.domains
  102. def get_serializer(self, *args, **kwargs):
  103. include_keys = (self.action in ['create', 'retrieve'])
  104. return super().get_serializer(*args, include_keys=include_keys, **kwargs)
  105. def perform_create(self, serializer):
  106. with PDNSChangeTracker():
  107. domain = serializer.save(owner=self.request.user)
  108. # TODO this line raises if the local public suffix is not in our database!
  109. PDNSChangeTracker.track(lambda: self.auto_delegate(domain))
  110. @staticmethod
  111. def auto_delegate(domain: models.Domain):
  112. if domain.is_locally_registrable:
  113. parent_domain = models.Domain.objects.get(name=domain.parent_domain_name)
  114. parent_domain.update_delegation(domain)
  115. def perform_destroy(self, instance: models.Domain):
  116. with PDNSChangeTracker():
  117. instance.delete()
  118. if instance.is_locally_registrable:
  119. parent_domain = models.Domain.objects.get(name=instance.parent_domain_name)
  120. with PDNSChangeTracker():
  121. parent_domain.update_delegation(instance)
  122. class SerialList(generics.ListAPIView):
  123. permission_classes = (IsVPNClient,)
  124. throttle_classes = [] # don't break slaves when they ask too often (our cached responses are cheap)
  125. def list(self, request, *args, **kwargs):
  126. key = 'desecapi.views.serials'
  127. serials = cache.get(key)
  128. if serials is None:
  129. serials = get_serials()
  130. cache.get_or_set(key, serials, timeout=15)
  131. return Response(serials)
  132. class RRsetDetail(IdempotentDestroyMixin, DomainViewMixin, generics.RetrieveUpdateDestroyAPIView):
  133. serializer_class = serializers.RRsetSerializer
  134. permission_classes = (IsAuthenticated, IsDomainOwner,)
  135. def get_queryset(self):
  136. return self.domain.rrset_set
  137. def get_object(self):
  138. queryset = self.filter_queryset(self.get_queryset())
  139. filter_kwargs = {k: self.kwargs[k] for k in ['subname', 'type']}
  140. obj = generics.get_object_or_404(queryset, **filter_kwargs)
  141. # May raise a permission denied
  142. self.check_object_permissions(self.request, obj)
  143. return obj
  144. def update(self, request, *args, **kwargs):
  145. response = super().update(request, *args, **kwargs)
  146. if response.data is None:
  147. response.status_code = 204
  148. return response
  149. def perform_update(self, serializer):
  150. with PDNSChangeTracker():
  151. super().perform_update(serializer)
  152. def perform_destroy(self, instance):
  153. with PDNSChangeTracker():
  154. super().perform_destroy(instance)
  155. class RRsetList(EmptyPayloadMixin, DomainViewMixin, generics.ListCreateAPIView, generics.UpdateAPIView):
  156. serializer_class = serializers.RRsetSerializer
  157. permission_classes = (IsAuthenticated, IsDomainOwner,)
  158. def get_queryset(self):
  159. rrsets = models.RRset.objects.filter(domain=self.domain)
  160. for filter_field in ('subname', 'type'):
  161. value = self.request.query_params.get(filter_field)
  162. if value is not None:
  163. # TODO consider moving this
  164. if filter_field == 'type' and value in models.RR_SET_TYPES_AUTOMATIC:
  165. raise PermissionDenied("You cannot tinker with the %s RRset." % value)
  166. rrsets = rrsets.filter(**{'%s__exact' % filter_field: value})
  167. return rrsets
  168. def get_object(self):
  169. # For this view, the object we're operating on is the queryset that one can also GET. Serializing a queryset
  170. # is fine as per https://www.django-rest-framework.org/api-guide/serializers/#serializing-multiple-objects.
  171. # We skip checking object permissions here to avoid evaluating the queryset. The user can access all his RRsets
  172. # anyways.
  173. return self.filter_queryset(self.get_queryset())
  174. def get_serializer(self, *args, **kwargs):
  175. kwargs = kwargs.copy()
  176. if 'many' not in kwargs:
  177. if self.request.method in ['POST']:
  178. kwargs['many'] = isinstance(kwargs.get('data'), list)
  179. elif self.request.method in ['PATCH', 'PUT']:
  180. kwargs['many'] = True
  181. return super().get_serializer(*args, **kwargs)
  182. def perform_create(self, serializer):
  183. with PDNSChangeTracker():
  184. super().perform_create(serializer)
  185. def perform_update(self, serializer):
  186. with PDNSChangeTracker():
  187. super().perform_update(serializer)
  188. class Root(APIView):
  189. def get(self, request, *_):
  190. if self.request.user.is_authenticated:
  191. routes = {
  192. 'account': {
  193. 'show': reverse('account', request=request),
  194. 'delete': reverse('account-delete', request=request),
  195. 'change-email': reverse('account-change-email', request=request),
  196. 'reset-password': reverse('account-reset-password', request=request),
  197. },
  198. 'logout': reverse('logout', request=request),
  199. 'tokens': reverse('token-list', request=request),
  200. 'domains': reverse('domain-list', request=request),
  201. }
  202. else:
  203. routes = {
  204. 'register': reverse('register', request=request),
  205. 'login': reverse('login', request=request),
  206. 'reset-password': reverse('account-reset-password', request=request),
  207. }
  208. return Response(routes)
  209. class DynDNS12Update(generics.GenericAPIView):
  210. authentication_classes = (auth.TokenAuthentication, auth.BasicTokenAuthentication, auth.URLParamAuthentication,)
  211. renderer_classes = [PlainTextRenderer]
  212. throttle_scope = 'dyndns'
  213. def _find_ip(self, params, version):
  214. if version == 4:
  215. look_for = '.'
  216. elif version == 6:
  217. look_for = ':'
  218. else:
  219. raise Exception
  220. # Check URL parameters
  221. for p in params:
  222. if p in self.request.query_params:
  223. if not len(self.request.query_params[p]):
  224. return None
  225. if look_for in self.request.query_params[p]:
  226. return self.request.query_params[p]
  227. # Check remote IP address
  228. client_ip = self.request.META.get('REMOTE_ADDR')
  229. if look_for in client_ip:
  230. return client_ip
  231. # give up
  232. return None
  233. @cached_property
  234. def qname(self):
  235. # hostname parameter
  236. try:
  237. if self.request.query_params['hostname'] != 'YES':
  238. return self.request.query_params['hostname'].lower()
  239. except KeyError:
  240. pass
  241. # host_id parameter
  242. try:
  243. return self.request.query_params['host_id'].lower()
  244. except KeyError:
  245. pass
  246. # http basic auth username
  247. try:
  248. domain_name = base64.b64decode(
  249. get_authorization_header(self.request).decode().split(' ')[1].encode()).decode().split(':')[0]
  250. if domain_name and '@' not in domain_name:
  251. return domain_name.lower()
  252. except (binascii.Error, IndexError, UnicodeDecodeError):
  253. pass
  254. # username parameter
  255. try:
  256. return self.request.query_params['username'].lower()
  257. except KeyError:
  258. pass
  259. # only domain associated with this user account
  260. try:
  261. return self.request.user.domains.get().name
  262. except models.Domain.MultipleObjectsReturned:
  263. raise ValidationError(detail={
  264. "detail": "Request does not properly specify domain for update.",
  265. "code": "domain-unspecified"
  266. })
  267. except models.Domain.DoesNotExist:
  268. metrics.get('desecapi_dynDNS12_domain_not_found').inc()
  269. raise NotFound('nohost')
  270. @cached_property
  271. def domain(self):
  272. try:
  273. return models.Domain.objects.filter_qname(self.qname, owner=self.request.user).order_by('-name_length')[0]
  274. except (IndexError, ValueError):
  275. raise NotFound('nohost')
  276. @property
  277. def subname(self):
  278. return self.qname.rpartition(f'.{self.domain.name}')[0]
  279. def get_queryset(self):
  280. return self.domain.rrset_set.filter(subname=self.subname, type__in=['A', 'AAAA'])
  281. def get(self, request, *_):
  282. instances = self.get_queryset().all()
  283. ipv4 = self._find_ip(['myip', 'myipv4', 'ip'], version=4)
  284. ipv6 = self._find_ip(['myipv6', 'ipv6', 'myip', 'ip'], version=6)
  285. data = [
  286. {'type': 'A', 'subname': self.subname, 'ttl': 60, 'records': [ipv4] if ipv4 else []},
  287. {'type': 'AAAA', 'subname': self.subname, 'ttl': 60, 'records': [ipv6] if ipv6 else []},
  288. ]
  289. context = {'domain': self.domain, 'minimum_ttl': 60}
  290. serializer = serializers.RRsetSerializer(instances, data=data, many=True, partial=True, context=context)
  291. try:
  292. serializer.is_valid(raise_exception=True)
  293. except ValidationError as e:
  294. if any(
  295. any(
  296. getattr(non_field_error, 'code', '') == 'unique'
  297. for non_field_error
  298. in err.get('non_field_errors', [])
  299. )
  300. for err in e.detail
  301. ):
  302. raise ConcurrencyException from e
  303. raise e
  304. with PDNSChangeTracker():
  305. serializer.save()
  306. return Response('good', content_type='text/plain')
  307. class DonationList(generics.CreateAPIView):
  308. serializer_class = serializers.DonationSerializer
  309. def perform_create(self, serializer):
  310. instance = serializer.save()
  311. context = {
  312. 'donation': instance,
  313. 'creditoridentifier': settings.SEPA['CREDITOR_ID'],
  314. 'creditorname': settings.SEPA['CREDITOR_NAME'],
  315. }
  316. # internal desec notification
  317. content_tmpl = get_template('emails/donation/desec-content.txt')
  318. subject_tmpl = get_template('emails/donation/desec-subject.txt')
  319. attachment_tmpl = get_template('emails/donation/desec-attachment-jameica.txt')
  320. from_tmpl = get_template('emails/from.txt')
  321. email = EmailMessage(subject_tmpl.render(context),
  322. content_tmpl.render(context),
  323. from_tmpl.render(context),
  324. ['donation@desec.io'],
  325. attachments=[
  326. ('jameica-directdebit.xml',
  327. attachment_tmpl.render(context),
  328. 'text/xml')
  329. ])
  330. email.send()
  331. # donor notification
  332. if instance.email:
  333. content_tmpl = get_template('emails/donation/donor-content.txt')
  334. subject_tmpl = get_template('emails/donation/donor-subject.txt')
  335. footer_tmpl = get_template('emails/footer.txt')
  336. email = EmailMessage(subject_tmpl.render(context),
  337. content_tmpl.render(context) + footer_tmpl.render(),
  338. from_tmpl.render(context),
  339. [instance.email])
  340. email.send()
  341. class AccountCreateView(generics.CreateAPIView):
  342. serializer_class = serializers.RegisterAccountSerializer
  343. throttle_scope = 'account_management_active'
  344. def create(self, request, *args, **kwargs):
  345. # Create user and send trigger email verification.
  346. # Alternative would be to create user once email is verified, but this could be abused for bulk email.
  347. serializer = self.get_serializer(data=request.data)
  348. activation_required = settings.USER_ACTIVATION_REQUIRED
  349. try:
  350. serializer.is_valid(raise_exception=True)
  351. except ValidationError as e:
  352. # Hide existing users
  353. email_detail = e.detail.pop('email', [])
  354. email_detail = [detail for detail in email_detail if detail.code != 'unique']
  355. if email_detail:
  356. e.detail['email'] = email_detail
  357. if e.detail:
  358. raise e
  359. else:
  360. # create user
  361. user = serializer.save(is_active=(not activation_required))
  362. # send email if needed
  363. domain = serializer.validated_data.get('domain')
  364. if domain or activation_required:
  365. link, validity_period = generate_confirmation_link(request,
  366. serializers.AuthenticatedActivateUserActionSerializer,
  367. 'confirm-activate-account', user=user, domain=domain)
  368. user.send_email('activate-with-domain' if domain else 'activate', context={
  369. 'confirmation_link': link,
  370. 'link_expiration_hours': validity_period // timedelta(hours=1),
  371. 'domain': domain,
  372. })
  373. # This request is unauthenticated, so don't expose whether we did anything.
  374. message = 'Welcome! Please check your mailbox.' if activation_required else 'Welcome!'
  375. return Response(data={'detail': message}, status=status.HTTP_202_ACCEPTED)
  376. class AccountView(generics.RetrieveAPIView):
  377. permission_classes = (IsAuthenticated,)
  378. serializer_class = serializers.UserSerializer
  379. throttle_scope = 'account_management_passive'
  380. def get_object(self):
  381. return self.request.user
  382. class AccountDeleteView(generics.GenericAPIView):
  383. authentication_classes = (auth.EmailPasswordPayloadAuthentication,)
  384. permission_classes = (IsAuthenticated,)
  385. response_still_has_domains = Response(
  386. data={'detail': 'To delete your user account, first delete all of your domains.'},
  387. status=status.HTTP_409_CONFLICT,
  388. )
  389. throttle_scope = 'account_management_active'
  390. def post(self, request, *args, **kwargs):
  391. if self.request.user.domains.exists():
  392. return self.response_still_has_domains
  393. link, validity_period = generate_confirmation_link(request,
  394. serializers.AuthenticatedDeleteUserActionSerializer,
  395. 'confirm-delete-account', user=self.request.user)
  396. request.user.send_email('delete-user', context={
  397. 'confirmation_link': link,
  398. 'link_expiration_hours': validity_period // timedelta(hours=1),
  399. })
  400. return Response(data={'detail': 'Please check your mailbox for further account deletion instructions.'},
  401. status=status.HTTP_202_ACCEPTED)
  402. class AccountLoginView(generics.GenericAPIView):
  403. authentication_classes = (auth.EmailPasswordPayloadAuthentication,)
  404. permission_classes = (IsAuthenticated,)
  405. throttle_scope = 'account_management_passive'
  406. def post(self, request, *args, **kwargs):
  407. user = self.request.user
  408. token = models.Token.objects.create(user=user, name="login", perm_manage_tokens=True,
  409. max_age=timedelta(days=7), max_unused_period=timedelta(hours=1))
  410. user_logged_in.send(sender=user.__class__, request=self.request, user=user)
  411. data = serializers.TokenSerializer(token, include_plain=True).data
  412. return Response(data)
  413. class AccountLogoutView(generics.GenericAPIView, mixins.DestroyModelMixin):
  414. authentication_classes = (auth.TokenAuthentication,)
  415. permission_classes = (IsAuthenticated,)
  416. throttle_classes = [] # always allow people to log out
  417. def get_object(self):
  418. # self.request.auth contains the hashed key as it is stored in the database
  419. return models.Token.objects.get(key=self.request.auth)
  420. def post(self, request, *args, **kwargs):
  421. return self.destroy(request, *args, **kwargs)
  422. class AccountChangeEmailView(generics.GenericAPIView):
  423. authentication_classes = (auth.EmailPasswordPayloadAuthentication,)
  424. permission_classes = (IsAuthenticated,)
  425. serializer_class = serializers.ChangeEmailSerializer
  426. throttle_scope = 'account_management_active'
  427. def post(self, request, *args, **kwargs):
  428. # Check password and extract email
  429. serializer = self.get_serializer(data=request.data)
  430. serializer.is_valid(raise_exception=True)
  431. new_email = serializer.validated_data['new_email']
  432. link, validity_period = generate_confirmation_link(request,
  433. serializers.AuthenticatedChangeEmailUserActionSerializer,
  434. 'confirm-change-email', user=request.user, new_email=new_email)
  435. request.user.send_email('change-email', recipient=new_email, context={
  436. 'confirmation_link': link,
  437. 'link_expiration_hours': validity_period // timedelta(hours=1),
  438. 'old_email': request.user.email,
  439. 'new_email': new_email,
  440. })
  441. # At this point, we know that we are talking to the user, so we can tell that we sent an email.
  442. return Response(data={'detail': 'Please check your mailbox to confirm email address change.'},
  443. status=status.HTTP_202_ACCEPTED)
  444. class AccountResetPasswordView(generics.GenericAPIView):
  445. serializer_class = serializers.ResetPasswordSerializer
  446. throttle_scope = 'account_management_active'
  447. def post(self, request, *args, **kwargs):
  448. serializer = self.get_serializer(data=request.data)
  449. serializer.is_valid(raise_exception=True)
  450. try:
  451. email = serializer.validated_data['email']
  452. user = models.User.objects.get(email=email, is_active=True)
  453. except models.User.DoesNotExist:
  454. pass
  455. else:
  456. self.send_reset_token(user, request)
  457. # This request is unauthenticated, so don't expose whether we did anything.
  458. return Response(data={'detail': 'Please check your mailbox for further password reset instructions. '
  459. 'If you did not receive an email, please contact support.'},
  460. status=status.HTTP_202_ACCEPTED)
  461. @staticmethod
  462. def send_reset_token(user, request):
  463. link, validity_period = generate_confirmation_link(request,
  464. serializers.AuthenticatedResetPasswordUserActionSerializer,
  465. 'confirm-reset-password', user=user)
  466. user.send_email('reset-password', context={
  467. 'confirmation_link': link,
  468. 'link_expiration_hours': validity_period // timedelta(hours=1),
  469. })
  470. class AuthenticatedActionView(generics.GenericAPIView):
  471. """
  472. Abstract class. Deserializes the given payload according the serializers specified by the view extending
  473. this class. If the `serializer.is_valid`, `act` is called on the action object.
  474. Summary of the behavior depending on HTTP method and Accept: header:
  475. GET POST other method
  476. Accept: text/html forward to `self.html_url` if any perform action 405 Method Not Allowed
  477. else HTTP 406 Not Acceptable perform action 405 Method Not Allowed
  478. """
  479. action = None
  480. html_url = None # Redirect GET requests to this webapp GUI URL
  481. http_method_names = ['get', 'post'] # GET is for redirect only
  482. renderer_classes = [JSONRenderer, StaticHTMLRenderer]
  483. @property
  484. def authentication_classes(self):
  485. # This prevents both code evaluation and user-specific throttling when we only want a redirect
  486. return () if self.request.method in SAFE_METHODS else (auth.AuthenticatedBasicUserActionAuthentication,)
  487. @property
  488. def throttle_scope(self):
  489. return 'account_management_passive' if self.request.method in SAFE_METHODS else 'account_management_active'
  490. def get_serializer_context(self):
  491. return {
  492. **super().get_serializer_context(),
  493. 'code': self.kwargs['code'],
  494. 'validity_period': self.get_serializer_class().validity_period,
  495. }
  496. def get(self, request, *args, **kwargs):
  497. # Redirect browsers to frontend if available
  498. is_redirect = (request.accepted_renderer.format == 'html') and self.html_url is not None
  499. if is_redirect:
  500. # Careful: This can generally lead to an open redirect if values contain slashes!
  501. # However, it cannot happen for Django view kwargs.
  502. return redirect(self.html_url.format(**kwargs))
  503. else:
  504. raise NotAcceptable
  505. def post(self, request, *args, **kwargs):
  506. super().perform_authentication(request)
  507. serializer = self.get_serializer(data=request.data)
  508. serializer.is_valid(raise_exception=True)
  509. try:
  510. self.action = serializer.Meta.model(**serializer.validated_data)
  511. except ValueError: # this happens when state cannot be verified
  512. ex = ValidationError('This action cannot be carried out because another operation has been performed, '
  513. 'invalidating this one. (Are you trying to perform this action twice?)')
  514. ex.status_code = status.HTTP_409_CONFLICT
  515. raise ex
  516. self.action.act()
  517. return self.finalize()
  518. def finalize(self):
  519. raise NotImplementedError
  520. class AuthenticatedActivateUserActionView(AuthenticatedActionView):
  521. html_url = '/confirm/activate-account/{code}/'
  522. serializer_class = serializers.AuthenticatedActivateUserActionSerializer
  523. def finalize(self):
  524. if not self.action.domain:
  525. return self._finalize_without_domain()
  526. else:
  527. domain = self._create_domain()
  528. return self._finalize_with_domain(domain)
  529. def _create_domain(self):
  530. serializer = serializers.DomainSerializer(
  531. data={'name': self.action.domain},
  532. context=self.get_serializer_context()
  533. )
  534. try:
  535. serializer.is_valid(raise_exception=True)
  536. except ValidationError as e: # e.g. domain name unavailable
  537. self.action.user.delete()
  538. reasons = ', '.join([detail.code for detail in e.detail.get('name', [])])
  539. raise ValidationError(
  540. f'The requested domain {self.action.domain} could not be registered (reason: {reasons}). '
  541. f'Please start over and sign up again.'
  542. )
  543. # TODO the following line is subject to race condition and can fail, as for the domain name, we have that
  544. # time-of-check != time-of-action
  545. return PDNSChangeTracker.track(lambda: serializer.save(owner=self.action.user))
  546. def _finalize_without_domain(self):
  547. if not is_password_usable(self.action.user.password):
  548. AccountResetPasswordView.send_reset_token(self.action.user, self.request)
  549. return Response({
  550. 'detail': 'Success! We sent you instructions on how to set your password.'
  551. })
  552. login_url = self.request.build_absolute_uri(reverse('v1:login'))
  553. return Response({
  554. 'detail': f'Success! Please log in at {login_url}.'
  555. })
  556. def _finalize_with_domain(self, domain):
  557. if domain.is_locally_registrable:
  558. # TODO the following line raises Domain.DoesNotExist under unknown conditions
  559. PDNSChangeTracker.track(lambda: DomainViewSet.auto_delegate(domain))
  560. token = models.Token.objects.create(user=domain.owner, name='dyndns')
  561. return Response({
  562. 'detail': 'Success! Here is the password ("token") to configure your router (or any other dynDNS '
  563. 'client). This password is different from your account password for security reasons.',
  564. 'domain': serializers.DomainSerializer(domain).data,
  565. **serializers.TokenSerializer(token, include_plain=True).data,
  566. })
  567. else:
  568. return Response({
  569. 'detail': 'Success! Please check the docs for the next steps, https://desec.readthedocs.io/.',
  570. 'domain': serializers.DomainSerializer(domain, include_keys=True).data,
  571. })
  572. class AuthenticatedChangeEmailUserActionView(AuthenticatedActionView):
  573. html_url = '/confirm/change-email/{code}/'
  574. serializer_class = serializers.AuthenticatedChangeEmailUserActionSerializer
  575. def finalize(self):
  576. return Response({
  577. 'detail': f'Success! Your email address has been changed to {self.action.user.email}.'
  578. })
  579. class AuthenticatedResetPasswordUserActionView(AuthenticatedActionView):
  580. html_url = '/confirm/reset-password/{code}/'
  581. serializer_class = serializers.AuthenticatedResetPasswordUserActionSerializer
  582. def finalize(self):
  583. login_url = self.request.build_absolute_uri(reverse('v1:login'))
  584. return Response({'detail': f'Success! Your password has been changed. Log in at {login_url}.'})
  585. class AuthenticatedDeleteUserActionView(AuthenticatedActionView):
  586. html_url = '/confirm/delete-account/{code}/'
  587. serializer_class = serializers.AuthenticatedDeleteUserActionSerializer
  588. def post(self, request, *args, **kwargs):
  589. if self.request.user.domains.exists():
  590. return AccountDeleteView.response_still_has_domains
  591. return super().post(request, *args, **kwargs)
  592. def finalize(self):
  593. return Response({'detail': 'All your data has been deleted. Bye bye, see you soon! <3'})
  594. class AuthenticatedRenewDomainBasicUserActionView(AuthenticatedActionView):
  595. html_url = '/confirm/renew-domain/{code}/'
  596. serializer_class = serializers.AuthenticatedRenewDomainBasicUserActionSerializer
  597. def finalize(self):
  598. return Response({'detail': f'We recorded that your domain {self.action.domain} is still in use.'})
  599. class CaptchaView(generics.CreateAPIView):
  600. serializer_class = serializers.CaptchaSerializer
  601. throttle_scope = 'account_management_passive'