]> scripts.mit.edu Git - wizard.git/blobdiff - wizard/app/wordpress.py
Fix WordPress install for 4.7.
[wizard.git] / wizard / app / wordpress.py
index 883f5de2177fa4301508e5ee0a07ddeadde8d61f..e75b899d81db92da58ef86e945992cb4d92cf248 100644 (file)
@@ -2,12 +2,16 @@ import os
 import re
 import logging
 import distutils
+import distutils.version
 import urlparse
+import hashlib
+import sqlalchemy.exc
 
 from wizard import app, install, resolve, sql, util
 from wizard.app import php
 
 def make_filename_regex_define(var):
+    """See :ref:`versioning config <seed>` for more information."""
     return 'wp-config.php', php.re_define(var)
 
 seed = util.dictmap(make_filename_regex_define, {
@@ -21,31 +25,55 @@ seed = util.dictmap(make_filename_regex_define, {
     'WIZARD_SECURE_AUTH_KEY': 'SECURE_AUTH_KEY',
     'WIZARD_LOGGED_IN_KEY': 'LOGGED_IN_KEY',
     'WIZARD_NONCE_KEY': 'NONCE_KEY',
+    'WIZARD_AUTH_SALT': 'AUTH_SALT',
+    'WIZARD_SECURE_AUTH_SALT': 'SECURE_AUTH_SALT',
+    'WIZARD_LOGGED_IN_SALT': 'LOGGED_IN_SALT',
+    'WIZARD_NONCE_SALT': 'NONCE_SALT',
     })
 
 class Application(app.Application):
+    fullname = "WordPress"
     database = "mysql"
     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("db", "email", "title")
+    install_schema = install.ArgSchema("db", "admin", "email", "title")
     deprecated_keys = set(['WIZARD_SECRETKEY'])
+    random_keys = set([
+        'WIZARD_SECRETKEY',
+        'WIZARD_AUTH_KEY',
+        'WIZARD_SECURE_AUTH_KEY',
+        'WIZARD_LOGGED_IN_KEY',
+        'WIZARD_NONCE_KEY',
+        'WIZARD_AUTH_SALT',
+        'WIZARD_SECURE_AUTH_SALT',
+        'WIZARD_LOGGED_IN_SALT',
+        'WIZARD_NONCE_SALT',
+        ])
+    random_blacklist = set(['put your unique phrase here'])
+    def urlFromExtract(self, deployment):
+        try:
+            meta = sql.connect(deployment.dsn)
+            try:
+                wp_options = meta.tables["wp_options"]
+            except KeyError:
+                return None
+            query = wp_options.select(wp_options.c.option_name == 'home')
+            return query.execute().fetchone()['option_value']
+        except sqlalchemy.exc.OperationalError:
+            return None
     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")
+        return self.checkWebPage(deployment, "",
+                outputs=["<html", "WordPress", "feed"],
+                exclude=["Error establishing a database connection", "Account unknown"])
     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")
 
@@ -54,7 +82,7 @@ class Application(app.Application):
                 'uname': options.dsn.username,
                 'dbname': options.dsn.database,
                 'pwd': options.dsn.password,
-                'prefix': '',
+                'prefix': 'wp_', # Changed >= 3.4, now disallows empty prefix
                 'submit': 'Submit',
                 'step': '2',
                 }
@@ -63,30 +91,83 @@ class Application(app.Application):
                 'admin_email': options.email,
                 'submit': 'Continue',
                 'step': '2',
+                # Version >= 3.0
+                'user_name': options.admin_name,
+                'admin_password': options.admin_password,
+                'admin_password2': options.admin_password,
                 }
         old_mode = os.stat(".").st_mode
         os.chmod(".", 0777) # XXX: squick squick
+
+        # we need to disable the wp_mail function in wp-includes/pluggable[-functions].php
+        pluggable_path = os.path.exists('wp-includes/pluggable.php') and 'wp-includes/pluggable.php' or 'wp-includes/pluggable-functions.php'
+        pluggable = open(pluggable_path, 'r').read()
+        wp_mail_noop = "<?php function wp_mail( $to, $subject, $message, $headers = '', $attachments = array() ) { /*noop*/ } ?> \n\n"
+        pluggable_file = open(pluggable_path,'w')
+        pluggable_file.write(wp_mail_noop)
+        pluggable_file.write(pluggable)
+        pluggable_file.close()
+
         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()
+        # Starting in Wordpress 4.7, install now redirects you directly
+        # to login page.  So... don't check.
+        #if "Finished" not in result and "Success" not in result:
+        #    raise app.InstallFailure()
+
+        if version < distutils.version.LooseVersion("3.0"):
+            meta = sql.connect(options.dsn)
+            wp_users = meta.tables["wp_users"]
+            hashed_pass = hashlib.md5(options.admin_password).hexdigest()
+            wp_users.update().where(wp_users.c.ID == 1).values(user_login=options.admin_name,user_nicename=options.admin_name,display_name=options.admin_name,user_pass=hashed_pass).execute()
+            wp_usermeta = meta.tables["wp_usermeta"]
+            wp_usermeta.delete().where(wp_usermeta.c.user_id==1 and wp_usermeta.c.meta_key == "default_password_nag").execute()
+
+        # now we can restore the wp_mail function in wp-includes/pluggable[-functions].php
+        pluggable_file = open(pluggable_path,'w')
+        pluggable_file.write(pluggable)
+        pluggable_file.close()
 
-        # not sure what to do about this
-        meta = sql.connect(options.dsn)
-        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
+        # replace random variable stubs with real values
+        old_config = open('wp-config.php').read()
+        def replace_with_random(s):
+            return s.replace('put your unique phrase here', util.random_key(), 1)
+        config = replace_with_random(old_config)
+        while config != old_config:
+            old_config = config
+            config = replace_with_random(config)
+        open('wp-config.php', 'w').write(config)
+
+        php.ini_replace_vars()
     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:
+        if "Upgrade Complete" not in result and "Update Complete" not in result and \
+                "No Upgrade Required" not in result and "No Update Required" not in result:
             raise app.UpgradeFailure(result)
+    @app.throws_database_errors
     def backup(self, deployment, backup_dir, options):
-        app.backup_database(backup_dir, deployment)
+        sql.backup(backup_dir, deployment)
+    @app.throws_database_errors
     def restore(self, deployment, backup_dir, options):
-        app.restore_database(backup_dir, deployment)
+        sql.restore(backup_dir, deployment)
+    @app.throws_database_errors
     def remove(self, deployment, options):
-        app.remove_database(deployment)
+        sql.drop(deployment.dsn)
+
+Application.resolutions = {
+'wp-config.php': [
+    ("""
+<<<<<<<
+
+/** WordPress absolute path to the Wordpress directory. */
+|||||||
+/** WordPress absolute path to the Wordpress directory. */
+=======
+/** Absolute path to the WordPress directory. */
+>>>>>>>
+""", [0])
+],
+}