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
|
import humanize
import stat
import os
from flask import session
from functools import lru_cache
from srht.flask import SrhtFlask
from srht.config import cfg
from srht.database import DbSession
db = DbSession(cfg("git.sr.ht", "connection-string"))
from gitsrht.types import User
db.init()
import gitsrht.oauth
from gitsrht.git import commit_time, trim_commit
def lookup_user(email):
return User.query.filter(User.email == email).one_or_none()
class GitApp(SrhtFlask):
def __init__(self):
super().__init__("git.sr.ht", __name__)
from gitsrht.blueprints.api import api
from gitsrht.blueprints.public import public
from gitsrht.blueprints.repo import repo
from gitsrht.blueprints.manage import manage
self.register_blueprint(api)
self.register_blueprint(public)
self.register_blueprint(repo)
self.register_blueprint(manage)
meta_client_id = cfg("git.sr.ht", "oauth-client-id")
meta_client_secret = cfg("git.sr.ht", "oauth-client-secret")
builds_client_id = cfg("builds.sr.ht", "oauth-client-id", default=None)
self.configure_meta_auth(meta_client_id, meta_client_secret,
base_scopes=["profile"] + ([
"{}/jobs:write".format(builds_client_id)
] if builds_client_id else []))
@self.context_processor
def inject():
notice = session.get("notice")
if notice:
del session["notice"]
return {
"commit_time": commit_time,
"trim_commit": trim_commit,
"humanize": humanize,
"lookup_user": lookup_user,
"stat": stat,
"notice": notice,
"path_join": os.path.join
}
@self.login_manager.user_loader
def user_loader(username):
# TODO: Switch to a session token
return User.query.filter(User.username == username).one_or_none()
def lookup_or_register(self, exchange, profile, scopes):
user = User.query.filter(User.username == profile["username"]).first()
if not user:
user = User()
db.session.add(user)
user.username = profile.get("username")
user.email = profile.get("email")
user.paid = profile.get("paid")
user.oauth_token = exchange["token"]
user.oauth_token_expires = exchange["expires"]
user.oauth_token_scopes = scopes
db.session.commit()
return user
app = GitApp()
|
