views.py 29 KB

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