Add a restore command to the SSH interface

Implement a new command that can be used to restore deleted package
bases without having to push a new commit.

Signed-off-by: Lukas Fleischer <lfleischer@archlinux.org>

2 files changed, 31 insertions, 9 deletions

diff --git a/git-interface/git-serve.py b/git-interface/git-serve.py
index 8316cf7..45c9a01 100755
--- a/git-interface/git-serve.py
+++ b/git-interface/git-serve.py
@@ -19,6 +19,7 @@ aur_db_socket = config.get('database', 'socket')
 repo_path = config.get('serve', 'repo-path')
 repo_regex = config.get('serve', 'repo-regex')
 git_shell_cmd = config.get('serve', 'git-shell-cmd')
+git_update_cmd = config.get('serve', 'git-update-cmd')
 ssh_cmdline = config.get('serve', 'ssh-cmdline')
 
 enable_maintenance = config.getboolean('options', 'enable-maintenance')
@@ -152,10 +153,28 @@ elif action == 'setup-repo':
     if len(cmdargv) > 2:
         die_with_help("{:s}: too many arguments".format(action))
     create_pkgbase(cmdargv[1], user)
+elif action == 'restore':
+    if len(cmdargv) < 2:
+        die_with_help("{:s}: missing repository name".format(action))
+    if len(cmdargv) > 2:
+        die_with_help("{:s}: too many arguments".format(action))
+
+    pkgbase = cmdargv[1]
+    if not re.match(repo_regex, pkgbase):
+        die('{:s}: invalid repository name: {:s}'.format(action, pkgbase))
+
+    if pkgbase_exists(pkgbase):
+        die('{:s}: package base exists: {:s}'.format(action, pkgbase))
+    create_pkgbase(pkgbase, user)
+
+    os.environ["AUR_USER"] = user
+    os.environ["AUR_PKGBASE"] = pkgbase
+    os.execl(git_update_cmd, git_update_cmd, 'restore')
 elif action == 'help':
     die("Commands:\n" +
         "  help                 Show this help message and exit.\n" +
         "  list-repos           List all your repositories.\n" +
+        "  restore <name>       Restore a deleted package base.\n" +
         "  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.")
diff --git a/git-interface/git-update.py b/git-interface/git-update.py
index d258385..5ff8b28 100755
--- a/git-interface/git-update.py
+++ b/git-interface/git-update.py
@@ -176,22 +176,25 @@ def die_commit(msg, commit):
     sys.stderr.write("error: {:s}\n".format(msg))
     exit(1)
 
-if len(sys.argv) != 4:
-    die("invalid arguments")
-
-refname = sys.argv[1]
-sha1_old = sys.argv[2]
-sha1_new = sys.argv[3]
+repo = pygit2.Repository(repo_path)
 
 user = os.environ.get("AUR_USER")
 pkgbase = os.environ.get("AUR_PKGBASE")
 privileged = (os.environ.get("AUR_PRIVILEGED", '0') == '1')
 
+if len(sys.argv) == 2 and sys.argv[1] == "restore":
+    if 'refs/heads/' + pkgbase not in repo.listall_references():
+        die('{:s}: repository not found: {:s}'.format(sys.argv[1], pkgbase))
+    refname = "refs/heads/master"
+    sha1_old = sha1_new = repo.lookup_reference('refs/heads/' + pkgbase).target
+elif len(sys.argv) == 4:
+    refname, sha1_old, sha1_new = sys.argv[1:3]
+else:
+    die("invalid arguments")
+
 if refname != "refs/heads/master":
     die("pushing to a branch other than master is restricted")
 
-repo = pygit2.Repository(repo_path)
-
 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, buffered=True)
@@ -291,7 +294,7 @@ if srcinfo_pkgbase != pkgbase:
 
 # Ensure that packages are neither blacklisted nor overwritten.
 cur.execute("SELECT ID FROM PackageBases WHERE Name = %s", [pkgbase])
-pkgbase_id = cur.fetchone()[0]
+pkgbase_id = cur.fetchone()[0] if cur.rowcount == 1 else 0
 
 cur.execute("SELECT Name FROM PackageBlacklist")
 blacklist = [row[0] for row in cur.fetchall()]