aboutsummaryrefslogtreecommitdiffstats
path: root/python/fatcat_web/__init__.py
blob: 0725c2e20e785bcded815e4c22855fc3701f3e6b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
import sys
from typing import Any

import elasticsearch
import fatcat_openapi_client
import sentry_sdk
from authlib.flask.client import OAuth
from flask import Flask
from flask.logging import create_logger
from flask_debugtoolbar import DebugToolbarExtension
from flask_login import LoginManager
from flask_misaka import Misaka
from flask_mwoauth import MWOAuth
from flask_uuid import FlaskUUID
from flask_wtf.csrf import CSRFProtect
from loginpass import GitHub, Gitlab, ORCiD, create_flask_blueprint
from sentry_sdk.integrations.flask import FlaskIntegration

from fatcat_web.types import AnyResponse
from fatcat_web.web_config import Config  # type: ignore

toolbar = DebugToolbarExtension()

# TODO: annotating Flask object as Any because we insert fields (like log,
# csrf). Should refactor to make this separate globals and not inject them
app: Any = Flask(__name__, static_url_path="/static")
app.config.from_object(Config)
toolbar = DebugToolbarExtension(app)
FlaskUUID(app)
app.csrf = CSRFProtect(app)
app.log = create_logger(app)

# This is the Markdown processor; setting default here
Misaka(
    app,
    autolink=True,
    no_intra_emphasis=True,
    strikethrough=True,
    escape=True,
)

login_manager = LoginManager()
login_manager.init_app(app)
login_manager.login_view = "/auth/login"
oauth = OAuth(app)

# Grabs sentry config from SENTRY_DSN environment variable
sentry_sdk.init(
    integrations=[FlaskIntegration()],
    release=Config.GIT_RELEASE,
    environment=Config.FATCAT_DOMAIN,
)

conf = fatcat_openapi_client.Configuration()
conf.host = Config.FATCAT_API_HOST
api = fatcat_openapi_client.DefaultApi(fatcat_openapi_client.ApiClient(conf))

# remove most jinja2 template whitespace
app.jinja_env.trim_blocks = True
app.jinja_env.lstrip_blocks = True


def auth_api(token: str) -> fatcat_openapi_client.DefaultApi:
    conf = fatcat_openapi_client.Configuration()
    conf.api_key["Authorization"] = token
    conf.api_key_prefix["Authorization"] = "Bearer"
    conf.host = Config.FATCAT_API_HOST
    return fatcat_openapi_client.DefaultApi(fatcat_openapi_client.ApiClient(conf))


if Config.FATCAT_API_AUTH_TOKEN:
    print("Found and using privileged token (eg, for account signup)", file=sys.stderr)
    priv_api = auth_api(Config.FATCAT_API_AUTH_TOKEN)
else:
    print("No privileged token found", file=sys.stderr)
    priv_api = None

# TODO: refactor integration so this doesn't always need to be defined. If
# key/secret are empty, library will not init; if init is skipped, get
# undefined errors elsewhere.
mwoauth = MWOAuth(
    consumer_key=Config.WIKIPEDIA_CLIENT_ID or "dummy",
    consumer_secret=Config.WIKIPEDIA_CLIENT_SECRET or "dummy",
    default_return_to="wp_oauth_finish_login",
)
mwoauth.handshaker.user_agent = "fatcat.wiki;python_web_interface"
app.register_blueprint(mwoauth.bp, url_prefix="/auth/wikipedia")

app.es_client = elasticsearch.Elasticsearch(Config.ELASTICSEARCH_BACKEND, timeout=40.0)

from fatcat_web import auth, cors, editing_routes, forms, ref_routes, routes

# TODO: blocking on ORCID support in loginpass
if Config.ORCID_CLIENT_ID:
    orcid_bp = create_flask_blueprint(ORCiD, oauth, auth.handle_oauth)
    app.register_blueprint(orcid_bp, url_prefix="/auth/orcid")

if Config.GITLAB_CLIENT_ID:
    gitlab_bp = create_flask_blueprint(Gitlab, oauth, auth.handle_oauth)
    app.register_blueprint(gitlab_bp, url_prefix="/auth/gitlab")

if Config.GITHUB_CLIENT_ID:
    github_bp = create_flask_blueprint(GitHub, oauth, auth.handle_oauth)
    app.register_blueprint(github_bp, url_prefix="/auth/github")