54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
import os
|
|
|
|
import requests
|
|
from flask import Flask, request
|
|
from oauthlib.oauth2 import WebApplicationClient
|
|
from ovos_local_backend.database.oauth import OAuthTokenDatabase, OAuthApplicationDatabase
|
|
from pywebio.platform.flask import webio_view
|
|
|
|
from ovos_backend_manager.menu import start
|
|
|
|
os.environ['OAUTHLIB_INSECURE_TRANSPORT'] = '1'
|
|
|
|
app = Flask(__name__)
|
|
|
|
# `task_func` is PyWebIO task function
|
|
app.add_url_rule('/', 'webio_view', webio_view(start),
|
|
methods=['GET', 'POST', 'OPTIONS']) # need GET,POST and OPTIONS methods
|
|
|
|
|
|
@app.route("/auth/callback/<oauth_id>", methods=['GET'])
|
|
def oauth_callback(oauth_id):
|
|
""" user completed oauth, save token to db """
|
|
params = dict(request.args)
|
|
code = params["code"]
|
|
|
|
data = OAuthApplicationDatabase()[oauth_id]
|
|
client_id = data["client_id"]
|
|
client_secret = data["client_secret"]
|
|
token_endpoint = data["token_endpoint"]
|
|
|
|
# Prepare and send a request to get tokens! Yay tokens!
|
|
client = WebApplicationClient(client_id)
|
|
token_url, headers, body = client.prepare_token_request(
|
|
token_endpoint,
|
|
authorization_response=request.url,
|
|
redirect_url=request.base_url,
|
|
code=code
|
|
)
|
|
token_response = requests.post(
|
|
token_url,
|
|
headers=headers,
|
|
data=body,
|
|
auth=(client_id, client_secret),
|
|
).json()
|
|
|
|
with OAuthTokenDatabase() as db:
|
|
db.add_token(oauth_id, token_response)
|
|
|
|
return params
|
|
|
|
|
|
def main(port=36535, debug=False):
|
|
app.run(host="0.0.0.0", port=port, debug=debug)
|
|
# start_server(app, port=port, debug=debug)
|