Refinements from running in practice.

[wizard.git] / wizard / app / mediawiki.py

diff --git a/wizard/app/mediawiki.py b/wizard/app/mediawiki.py
index 5c18b159594eec4b8480f1437efe9b7ac1fb7023..a1a1eec56908f32700aa3bee89c824b8ef9152d6 100644 (file)
--- a/wizard/app/mediawiki.py
+++ b/wizard/app/mediawiki.py
@@ -1,11 +1,16 @@
 import re
+import distutils.version
 import os
+import datetime
+import logging
+import shlex
+import shutil
 
-from wizard import app, deploy, install, util
+from wizard import app, deploy, install, scripts, shell, util
 from wizard.app import php
 
 def make_filename_regex(var):
-    return 'LocalSettings.php', re.compile('^(\$' + app.expand_re(var) + r'''\s*=\s*)(.*)(;)$''', re.M)
+    return 'LocalSettings.php', re.compile('^(\$' + app.expand_re(var) + r'''\s*=\s*)(.*)(;)''', re.M)
 
 make_extractor = app.filename_regex_extractor(make_filename_regex)
 make_substitution = app.filename_regex_substitution(make_filename_regex)
@@ -41,13 +46,25 @@ class Application(deploy.Application):
         handler = install.ArgHandler("mysql", "admin", "email")
         handler.add(install.Arg("title", help="Title of your new MediaWiki install"))
         return handler
-    def install(self, options):
+    def checkConfig(self, deployment):
+        return os.path.isfile(os.path.join(deployment.location, "LocalSettings.php"))
+    def detectVersion(self, deployment):
+        contents = deployment.read("includes/DefaultSettings.php")
+        regex = make_filename_regex("wgVersion")[1]
+        match = regex.search(contents)
+        if not match: return None
+        return distutils.version.LooseVersion(match.group(2)[1:-1])
+    def checkWeb(self, d):
+        page = d.fetch("index.php")
+        return page.find("<!-- Served by") != -1
+    def install(self, version, options):
         try:
             os.unlink("LocalSettings.php")
         except OSError:
             pass
 
         os.chmod("config", 0777) # XXX: vaguely sketchy
+
         postdata = {
             'Sitename': options.title,
             'EmergencyContact': options.email,
@@ -67,4 +84,56 @@ class Application(deploy.Application):
         if result.find("Installation successful") == -1:
             raise install.Failure()
         os.rename('config/LocalSettings.php', 'LocalSettings.php')
+    def upgrade(self, d, version, options):
+        sh = shell.Shell()
+        if not os.path.isfile("AdminSettings.php"):
+            sh.call("git", "checkout", "mediawiki-" + str(version), "--", "AdminSettings.php")
+        try:
+            result = sh.eval("php", "maintenance/update.php", "--quick", log=True)
+        except shell.CallError as e:
+            raise app.UpgradeFailure("Update script returned non-zero exit code\nSTDOUT: %s\nSTDERR: %s" % (e.stdout, e.stderr))
+        results = result.rstrip().split()
+        if not results or not results[-1] == "Done.":
+            raise app.UpgradeFailure(result)
+    def backup(self, deployment, options):
+        sh = shell.Shell()
+        # XXX: duplicate code, refactor, also, race condition
+        backupdir = os.path.join(".scripts", "backups")
+        backup = str(deployment.version) + "-" + datetime.date.today().isoformat()
+        outdir = os.path.join(backupdir, backup)
+        if not os.path.exists(backupdir):
+            os.mkdir(backupdir)
+        if os.path.exists(outdir):
+            util.safe_unlink(outdir)
+        os.mkdir(outdir)
+        outfile = os.path.join(outdir, "db.sql")
+        try:
+            sh.call("mysqldump", "--compress", "-r", outfile, *get_mysql_args(deployment))
+            sh.call("gzip", "--best", outfile)
+        except shell.CallError as e:
+            raise app.BackupFailure(e.stderr)
+        return backup
+    def restore(self, deployment, backup, options):
+        sh = shell.Shell()
+        backup_dir = os.path.join(".scripts", "backups", backup)
+        if not os.path.exists(backup_dir):
+            raise app.RestoreFailure("Backup %s doesn't exist", backup)
+        sql = open(os.path.join(backup_dir, "db.sql"), 'w+')
+        sh.call("gunzip", "-c", os.path.join(backup_dir, "db.sql.gz"), stdout=sql)
+        sql.seek(0)
+        sh.call("mysql", *get_mysql_args(deployment), stdin=sql)
+        sql.close()
 
+def get_mysql_args(d):
+    # XXX: add support for getting these out of options
+    vars = d.extract()
+    if 'WIZARD_DBNAME' not in vars:
+        raise app.BackupFailure("Could not determine database name")
+    triplet = scripts.get_sql_credentials(vars)
+    args = []
+    if triplet is not None:
+        server, user, password = triplet
+        args += ["-h", server, "-u", user, "-p" + password]
+    name = shlex.split(vars['WIZARD_DBNAME'])[0]
+    args.append(name)
+    return args