git-serve.py 5.52 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
#!/usr/bin/python3

import configparser
import mysql.connector
import os
import re
import shlex
import sys

config = configparser.RawConfigParser()
Lukas Fleischer's avatar
Lukas Fleischer committed
11
config.read(os.path.dirname(os.path.realpath(__file__)) + "/../conf/config")
12
13
14
15
16

aur_db_host = config.get('database', 'host')
aur_db_name = config.get('database', 'name')
aur_db_user = config.get('database', 'user')
aur_db_pass = config.get('database', 'password')
17
aur_db_socket = config.get('database', 'socket')
18

19
repo_path = config.get('serve', 'repo-path')
20
21
repo_regex = config.get('serve', 'repo-regex')
git_shell_cmd = config.get('serve', 'git-shell-cmd')
22
ssh_cmdline = config.get('serve', 'ssh-cmdline')
23

24
25
26
27
28
29
30
31
32
33
34
def pkgbase_exists(pkgbase):
    db = mysql.connector.connect(host=aur_db_host, user=aur_db_user,
                                 passwd=aur_db_pass, db=aur_db_name,
                                 unix_socket=aur_db_socket)
    cur = db.cursor()

    cur.execute("SELECT COUNT(*) FROM PackageBases WHERE Name = %s ",
                [pkgbase])

    db.close()
    return (cur.fetchone()[0] > 0)
35

36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
def list_repos(user):
    db = mysql.connector.connect(host=aur_db_host, user=aur_db_user,
                                 passwd=aur_db_pass, db=aur_db_name,
                                 unix_socket=aur_db_socket)
    cur = db.cursor()

    cur.execute("SELECT ID FROM Users WHERE Username = %s ", [user])
    userid = cur.fetchone()[0]
    if userid == 0:
        die('%s: unknown user: %s' % (action, user))

    cur.execute("SELECT Name, PackagerUID FROM PackageBases " +
                "WHERE MaintainerUID = %s ", [userid])
    for row in cur:
        print((' ' if row[1] else '*') + row[0])
    db.close()

53
def create_pkgbase(pkgbase, user):
54
55
56
57
    if not re.match(repo_regex, pkgbase):
        die('%s: invalid repository name: %s' % (action, pkgbase))
    if pkgbase_exists(pkgbase):
        die('%s: package base already exists: %s' % (action, pkgbase))
58
59

    db = mysql.connector.connect(host=aur_db_host, user=aur_db_user,
60
61
                                 passwd=aur_db_pass, db=aur_db_name,
                                 unix_socket=aur_db_socket)
62
63
64
65
66
    cur = db.cursor()

    cur.execute("SELECT ID FROM Users WHERE Username = %s ", [user])
    userid = cur.fetchone()[0]
    if userid == 0:
67
        die('%s: unknown user: %s' % (action, user))
68
69

    cur.execute("INSERT INTO PackageBases (Name, SubmittedTS, ModifiedTS, " +
70
                "SubmitterUID, MaintainerUID) VALUES (%s, UNIX_TIMESTAMP(), " +
71
                "UNIX_TIMESTAMP(), %s, %s)", [pkgbase, userid, userid])
72
73
74
75
    pkgbase_id = cur.lastrowid

    cur.execute("INSERT INTO CommentNotify (PackageBaseID, UserID) " +
                "VALUES (%s, %s)", [pkgbase_id, userid])
76
77
78
79
80
81
82

    db.commit()
    db.close()

def check_permissions(pkgbase, user):
    db = mysql.connector.connect(host=aur_db_host, user=aur_db_user,
                                 passwd=aur_db_pass, db=aur_db_name,
83
                                 unix_socket=aur_db_socket, buffered=True)
84
85
    cur = db.cursor()

86
87
88
89
    cur.execute("SELECT AccountTypeID FROM Users WHERE UserName = %s ", [user])
    if cur.fetchone()[0] > 1:
        return True

90
91
92
93
94
95
96
    cur.execute("SELECT COUNT(*) FROM PackageBases " +
                "LEFT JOIN PackageComaintainers " +
                "ON PackageComaintainers.PackageBaseID = PackageBases.ID " +
                "INNER JOIN Users ON Users.ID = PackageBases.MaintainerUID " +
                "OR PackageBases.MaintainerUID IS NULL " +
                "OR Users.ID = PackageComaintainers.UsersID " +
                "WHERE Name = %s AND Username = %s", [pkgbase, user])
97
98
99
100
101
102
    return cur.fetchone()[0] > 0

def die(msg):
    sys.stderr.write("%s\n" % (msg))
    exit(1)

103
104
105
def die_with_help(msg):
    die(msg + "\nTry `%s help` for a list of commands." % (ssh_cmdline))

106
107
108
user = sys.argv[1]
cmd = os.environ.get("SSH_ORIGINAL_COMMAND")
if not cmd:
109
    die_with_help("Interactive shell is disabled.")
110
111
112
113
cmdargv = shlex.split(cmd)
action = cmdargv[0]

if action == 'git-upload-pack' or action == 'git-receive-pack':
114
115
    if len(cmdargv) < 2:
        die_with_help("%s: missing path" % (action))
116
117
118

    path = cmdargv[1].rstrip('/')
    if not path.startswith('/') or not path.endswith('.git'):
119
        die('%s: invalid path: %s' % (action, path))
120
121
    pkgbase = path[1:-4]
    if not re.match(repo_regex, pkgbase):
122
        die('%s: invalid repository name: %s' % (action, pkgbase))
123
124

    if not pkgbase_exists(pkgbase):
125
        create_pkgbase(pkgbase, user)
126

127
128
    if action == 'git-receive-pack':
        if not check_permissions(pkgbase, user):
129
            die('%s: permission denied: %s' % (action, user))
130

131
132
    os.environ["AUR_USER"] = user
    os.environ["AUR_PKGBASE"] = pkgbase
133
134
    os.environ["GIT_NAMESPACE"] = pkgbase
    cmd = action + " '" + repo_path + "'"
135
    os.execl(git_shell_cmd, git_shell_cmd, '-c', cmd)
136
137
138
139
elif action == 'list-repos':
    if len(cmdargv) > 1:
        die_with_help("%s: too many arguments" % (action))
    list_repos(user)
140
141
elif action == 'setup-repo':
    if len(cmdargv) < 2:
142
143
144
        die_with_help("%s: missing repository name" % (action))
    if len(cmdargv) > 2:
        die_with_help("%s: too many arguments" % (action))
145
    create_pkgbase(cmdargv[1], user)
146
147
148
elif action == 'help':
    die("Commands:\n" +
        "  help                 Show this help message and exit.\n" +
149
        "  list-repos           List all your repositories.\n" +
150
151
152
        "  setup-repo <name>    Create an empty repository.\n" +
        "  git-receive-pack     Internal command used with Git.\n" +
        "  git-upload-pack      Internal command used with Git.")
153
else:
154
    die_with_help("invalid command: %s" % (action))