mirror of
https://github.com/jupyterhub/jupyterhub.git
synced 2025-10-12 20:43:02 +00:00
122 lines
4.2 KiB
Python
122 lines
4.2 KiB
Python
"""Authorization handlers"""
|
|
|
|
# Copyright (c) Jupyter Development Team.
|
|
# Distributed under the terms of the Modified BSD License.
|
|
|
|
from datetime import datetime
|
|
import json
|
|
from urllib.parse import quote
|
|
|
|
from oauth2.web.tornado import OAuth2Handler
|
|
from tornado import web
|
|
|
|
from .. import orm
|
|
from ..user import User
|
|
from ..utils import token_authenticated
|
|
from .base import BaseHandler, APIHandler
|
|
|
|
|
|
class TokenAPIHandler(APIHandler):
|
|
@token_authenticated
|
|
def get(self, token):
|
|
orm_token = orm.APIToken.find(self.db, token)
|
|
if orm_token is None:
|
|
orm_token = orm.OAuthAccessToken.find(self.db, token)
|
|
if orm_token is None:
|
|
raise web.HTTPError(404)
|
|
|
|
# record activity whenever we see a token
|
|
now = orm_token.last_activity = datetime.utcnow()
|
|
if orm_token.user:
|
|
orm_token.user.last_activity = now
|
|
model = self.user_model(self.users[orm_token.user])
|
|
elif orm_token.service:
|
|
model = self.service_model(orm_token.service)
|
|
else:
|
|
self.log.warning("%s has no user or service. Deleting..." % orm_token)
|
|
self.db.delete(orm_token)
|
|
self.db.commit()
|
|
raise web.HTTPError(404)
|
|
self.db.commit()
|
|
self.write(json.dumps(model))
|
|
|
|
async def post(self):
|
|
warn_msg = (
|
|
"Using deprecated token creation endpoint %s."
|
|
" Use /hub/api/users/:user/tokens instead."
|
|
) % self.request.uri
|
|
self.log.warning(warn_msg)
|
|
requester = user = self.get_current_user()
|
|
if user is None:
|
|
# allow requesting a token with username and password
|
|
# for authenticators where that's possible
|
|
data = self.get_json_body()
|
|
try:
|
|
requester = user = await self.login_user(data)
|
|
except Exception as e:
|
|
self.log.error("Failure trying to authenticate with form data: %s" % e)
|
|
user = None
|
|
if user is None:
|
|
raise web.HTTPError(403)
|
|
else:
|
|
data = self.get_json_body()
|
|
# admin users can request tokens for other users
|
|
if data and data.get('username'):
|
|
user = self.find_user(data['username'])
|
|
if user is not requester and not requester.admin:
|
|
raise web.HTTPError(403, "Only admins can request tokens for other users.")
|
|
if requester.admin and user is None:
|
|
raise web.HTTPError(400, "No such user '%s'" % data['username'])
|
|
|
|
note = (data or {}).get('note')
|
|
if not note:
|
|
note = "Requested via deprecated api"
|
|
if requester is not user:
|
|
kind = 'user' if isinstance(user, User) else 'service'
|
|
note += " by %s %s" % (kind, requester.name)
|
|
|
|
api_token = user.new_api_token(note=note)
|
|
self.write(json.dumps({
|
|
'token': api_token,
|
|
'warning': warn_msg,
|
|
'user': self.user_model(user),
|
|
}))
|
|
|
|
|
|
class CookieAPIHandler(APIHandler):
|
|
@token_authenticated
|
|
def get(self, cookie_name, cookie_value=None):
|
|
cookie_name = quote(cookie_name, safe='')
|
|
if cookie_value is None:
|
|
self.log.warning("Cookie values in request body is deprecated, use `/cookie_name/cookie_value`")
|
|
cookie_value = self.request.body
|
|
else:
|
|
cookie_value = cookie_value.encode('utf8')
|
|
user = self._user_for_cookie(cookie_name, cookie_value)
|
|
if user is None:
|
|
raise web.HTTPError(404)
|
|
self.write(json.dumps(self.user_model(user)))
|
|
|
|
|
|
class OAuthHandler(BaseHandler, OAuth2Handler):
|
|
"""Implement OAuth provider handlers
|
|
|
|
OAuth2Handler sets `self.provider` in initialize,
|
|
but we are already passing the Provider object via settings.
|
|
"""
|
|
@property
|
|
def provider(self):
|
|
return self.settings['oauth_provider']
|
|
|
|
def initialize(self):
|
|
pass
|
|
|
|
|
|
default_handlers = [
|
|
(r"/api/authorizations/cookie/([^/]+)(?:/([^/]+))?", CookieAPIHandler),
|
|
(r"/api/authorizations/token/([^/]+)", TokenAPIHandler),
|
|
(r"/api/authorizations/token", TokenAPIHandler),
|
|
(r"/api/oauth2/authorize", OAuthHandler),
|
|
(r"/api/oauth2/token", OAuthHandler),
|
|
]
|