Run service

This commit is contained in:
Micke Nordin 2024-01-15 16:03:11 +01:00
parent 989682d690
commit db6f3998ac
Signed by untrusted user: Micke
GPG key ID: F53C4CC83EDAB3BE

View file

@ -9,6 +9,50 @@ hub:
db:
pvc:
storageClassName: csi-sc-cinderplugin
extraFiles:
refresh-token.py: |
"""A token refresh service authenticating with the Hub.
This service serves `/services/refresh-token/`,
authenticated with the Hub,
showing the user their own info.
"""
import json
import os
from urllib.parse import urlparse
from tornado.httpserver import HTTPServer
from tornado.ioloop import IOLoop
from tornado.web import Application, RequestHandler, authenticated
from jupyterhub.services.auth import HubAuthenticated
class RefreshHandler(HubAuthenticated, RequestHandler):
@authenticated
def get(self):
user_model = self.get_current_user()
self.set_header('content-type', 'application/json')
self.write(json.dumps(user_model, indent=1, sort_keys=True))
def main():
app = Application(
[
(os.environ['JUPYTERHUB_SERVICE_PREFIX'] + '/?', RefreshHandler),
(r'.*', RefreshHandler),
]
)
http_server = HTTPServer(app)
url = urlparse(os.environ['JUPYTERHUB_SERVICE_URL'])
http_server.listen(url.port, url.hostname)
IOLoop.current().start()
if __name__ == '__main__':
main()
extraConfig:
oauthCode: |
import time
@ -111,21 +155,21 @@ hub:
c.NextcloudOAuthenticator.post_auth_hook = post_auth_hook
serviceCode: |
from jupyter_server.base.handlers import JupyterHandler
import tornado
class MyExtensionHandler(JupyterHandler):
@tornado.web.authenticated
def get(self):
print("Hello World")
def _load_jupyter_server_extension(serverapp: jupyter_server.serverapp.ServerApp):
"""
This function is called when the extension is loaded.
"""
handlers = [("/nextcloud/token", NextcloudExtensionHandler)]
serverapp.web_app.add_handlers(".*$", handlers)
import sys
c.JupyterHub.load_roles = [
{
"name": "refresh-token",
"scopes": [
"admin:auth_state"
]
}
]
c.JupyterHub.services = [
{
'name': 'refresh-token',
'command': [sys.executable, './refresh_token.py']
}
]
extraEnv:
NEXTCLOUD_DEBUG_OAUTH: "yes"