src/web/auth.py

   1 #!/usr/bin/python
   2
   3 import logging
   4 import tornado.web
   5
   6 from . import base
   7
   8 class CacheMixin(object):
   9         def prepare(self):
  10                 # Mark this as private when someone is logged in
  11                 if self.current_user:
  12                         self.add_header("Cache-Control", "private")
  13
  14                 self.add_header("Vary", "Cookie")
  15
  16
  17 class AuthenticationMixin(CacheMixin):
  18         def authenticate(self, username, password):
  19                 # Find account
  20                 account = self.backend.accounts.find_account(username)
  21                 if not account:
  22                         raise tornado.web.HTTPError(401, "Unknown user: %s" % username)
  23
  24                 # Check credentials
  25                 if not account.check_password(password):
  26                         raise tornado.web.HTTPError(401, "Invalid password for %s" % account)
  27
  28                 return self.login(account)
  29
  30         def login(self, account):
  31                 # User has logged in, create a session
  32                 session_id, session_expires = self.backend.accounts.create_session(
  33                         account, self.request.host)
  34
  35                 # Check if a new session was created
  36                 if not session_id:
  37                         raise tornado.web.HTTPError(500, "Could not create session")
  38
  39                 # Send session cookie to the client
  40                 self.set_cookie("session_id", session_id,
  41                         domain=self.request.host, expires=session_expires)
  42
  43         def logout(self):
  44                 session_id = self.get_cookie("session_id")
  45                 if not session_id:
  46                         return
  47
  48                 success = self.backend.accounts.destroy_session(session_id, self.request.host)
  49                 if success:
  50                         self.clear_cookie("session_id")
  51
  52
  53 class LoginHandler(AuthenticationMixin, base.BaseHandler):
  54         @base.blacklisted
  55         def get(self):
  56                 next = self.get_argument("next", None)
  57
  58                 self.render("auth/login.html", next=next)
  59
  60         @base.blacklisted
  61         @base.ratelimit(minutes=60, requests=5)
  62         def post(self):
  63                 username = self.get_argument("username")
  64                 password = self.get_argument("password")
  65
  66                 with self.db.transaction():
  67                         self.authenticate(username, password)
  68
  69                 # Determine the page we should redirect to
  70                 next = self.get_argument("next", None)
  71
  72                 return self.redirect(next or "/")
  73
  74
  75 class LogoutHandler(AuthenticationMixin, base.BaseHandler):
  76         def get(self):
  77                 with self.db.transaction():
  78                         self.logout()
  79
  80                 # Get back to the start page
  81                 self.redirect("/")
  82
  83
  84 class RegisterHandler(base.BaseHandler):
  85         @base.blacklisted
  86         def get(self):
  87                 # Redirect logged in users away
  88                 if self.current_user:
  89                         self.redirect("/")
  90
  91                 self.render("auth/register.html")
  92
  93         @base.blacklisted
  94         @base.ratelimit(minutes=24*60, requests=5)
  95         def post(self):
  96                 uid   = self.get_argument("uid")
  97                 email = self.get_argument("email")
  98
  99                 first_name = self.get_argument("first_name")
 100                 last_name  = self.get_argument("last_name")
 101
 102                 # Register account
 103                 try:
 104                         with self.db.transaction():
 105                                 self.backend.accounts.register(uid, email,
 106                                         first_name=first_name, last_name=last_name)
 107                 except ValueError as e:
 108                         raise tornado.web.HTTPError(400, "%s" % e) from e
 109
 110                 self.render("auth/register-success.html")
 111
 112
 113 class ActivateHandler(AuthenticationMixin, base.BaseHandler):
 114         def get(self, uid, activation_code):
 115                 self.render("auth/activate.html")
 116
 117         def post(self, uid, activation_code):
 118                 password1 = self.get_argument("password1")
 119                 password2 = self.get_argument("password2")
 120
 121                 if not password1 == password2:
 122                         raise tornado.web.HTTPError(400, "Passwords do not match")
 123
 124                 with self.db.transaction():
 125                         account = self.backend.accounts.activate(uid, activation_code)
 126                         if not account:
 127                                 raise tornado.web.HTTPError(400, "Account not found: %s" % uid)
 128
 129                         # Set the new password
 130                         account.passwd(password1)
 131
 132                         # Create session
 133                         self.login(account)
 134
 135                 # Redirect to success page
 136                 self.render("auth/activated.html", account=account)