Refactor more boilerplate out.

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

diff --git a/wizard/app/wordpress.py b/wizard/app/wordpress.py
index 72cf7314b4a05377d3c9b4f6291cf136a7a38824..742881b93a857d5846b690963a08eced1ca1ddd4 100644 (file)
--- a/wizard/app/wordpress.py
+++ b/wizard/app/wordpress.py
@@ -1,6 +1,89 @@
-from wizard import app
+import os
+import re
+import logging
+import distutils
+import urlparse
+
+from wizard import app, install, resolve, sql, util
 from wizard.app import php
 
+def make_filename_regex_define(var):
+    return 'wp-config.php', php.re_define(var)
+
+seed = util.dictmap(make_filename_regex_define, {
+    # these funny names are due to convention set by MediaWiki
+    'WIZARD_DBSERVER': 'DB_HOST',
+    'WIZARD_DBNAME': 'DB_NAME',
+    'WIZARD_DBUSER': 'DB_USER',
+    'WIZARD_DBPASSWORD': 'DB_PASSWORD',
+    'WIZARD_SECRETKEY': 'SECRET_KEY',
+    'WIZARD_AUTH_KEY': 'AUTH_KEY',
+    'WIZARD_SECURE_AUTH_KEY': 'SECURE_AUTH_KEY',
+    'WIZARD_LOGGED_IN_KEY': 'LOGGED_IN_KEY',
+    'WIZARD_NONCE_KEY': 'NONCE_KEY',
+    })
+
 class Application(app.Application):
+    parametrized_files = ['wp-config.php'] + php.parametrized_files
+    extractors = app.make_extractors(seed)
+    extractors.update(php.extractors)
+    substitutions = app.make_substitutions(seed)
+    substitutions.update(php.substitutions)
+    install_schema = install.ArgSchema("mysql", "email", "title")
+    deprecated_keys = set(['WIZARD_SECRETKEY'])
     def download(self, version):
         return "http://wordpress.org/wordpress-%s.tar.gz" % version
+    def checkConfig(self, deployment):
+        return os.path.isfile("wp-config.php")
+    def checkWeb(self, deployment):
+        # XXX: this sucks pretty hard
+        return self.checkWebPage(deployment, "", "<html")
+    def detectVersion(self, deployment):
+        return self.detectVersionFromFile("wp-includes/version.php", php.re_var("wp_version"))
+    def prepareMerge(self, deployment):
+        # This file shouldn't really be edited by users, but be careful: it's
+        # stored as DOS and not as UNIX, so you'll get conflicts if you add this line:
+        # resolve.fix_newlines("wp-config.php")
+        pass
+    def install(self, version, options):
+        util.soft_unlink("wp-config.php")
+
+        post_setup_config = {
+                'dbhost': options.mysql_host,
+                'uname': options.mysql_user,
+                'dbname': options.mysql_db,
+                'pwd': options.mysql_password,
+                'prefix': '',
+                'submit': 'Submit',
+                'step': '2',
+                }
+        post_install = {
+                'weblog_title': options.title,
+                'admin_email': options.email,
+                'submit': 'Continue',
+                'step': '2',
+                }
+        old_mode = os.stat(".").st_mode
+        os.chmod(".", 0777) # XXX: squick squick
+        result = install.fetch(options, "wp-admin/setup-config.php?step=2", post_setup_config)
+        logging.debug("setup-config.php output\n\n" + result)
+        result = install.fetch(options, "wp-admin/install.php?step=2", post_install)
+        logging.debug("install.php output\n\n" + result)
+        os.chmod(".", old_mode)
+        if "Finished" not in result and "Success" not in result:
+            raise app.InstallFailure()
+
+        # not sure what to do about this
+        meta = sql.mysql_connect(options)
+        wp_options = meta.tables["wp_options"]
+        wp_options.update().where(wp_options.c.option_name == 'siteurl').values(option_value=options.web_path).execute()
+        wp_options.update().where(wp_options.c.option_name == 'home').values(option_value="http://%s%s" % (options.web_host, options.web_path)).execute() # XXX: what if missing leading slash; this should be put in a function
+        # should also set the username and admin password
+    def upgrade(self, d, version, options):
+        result = d.fetch("wp-admin/upgrade.php?step=1")
+        if "Upgrade Complete" not in result and "No Upgrade Required" not in result:
+            raise app.UpgradeFailure(result)
+    def backup(self, deployment, backup_dir, options):
+        app.backup_database(backup_dir, deployment)
+    def restore(self, deployment, backup_dir, options):
+        app.restore_database(backup_dir, deployment)