2 import distutils.version
9 from wizard import app, deploy, install, scripts, shell, util
10 from wizard.app import php
12 def make_filename_regex(var):
13 return 'LocalSettings.php', re.compile('^(\$' + app.expand_re(var) + r'''\s*=\s*)(.*)(;)''', re.M)
15 make_extractor = app.filename_regex_extractor(make_filename_regex)
16 make_substitution = app.filename_regex_substitution(make_filename_regex)
18 'WIZARD_IP': 'IP', # obsolete, remove after we're done
19 'WIZARD_SITENAME': 'wgSitename',
20 'WIZARD_SCRIPTPATH': 'wgScriptPath',
21 'WIZARD_EMERGENCYCONTACT': ('wgEmergencyContact', 'wgPasswordSender'),
22 'WIZARD_DBSERVER': 'wgDBserver',
23 'WIZARD_DBNAME': 'wgDBname',
24 'WIZARD_DBUSER': 'wgDBuser',
25 'WIZARD_DBPASSWORD': 'wgDBpassword',
26 'WIZARD_SECRETKEY': ('wgSecretKey', 'wgProxyKey'),
29 class Application(deploy.Application):
30 parametrized_files = ['LocalSettings.php', 'php.ini']
31 deprecated_keys = set(['WIZARD_IP']) | php.deprecated_keys
34 if not self._extractors:
35 self._extractors = util.dictmap(make_extractor, seed)
36 self._extractors.update(php.extractors)
37 return self._extractors
39 def substitutions(self):
40 if not self._substitutions:
41 self._substitutions = util.dictkmap(make_substitution, seed)
42 self._substitutions.update(php.substitutions)
43 return self._substitutions
45 def install_handler(self):
46 handler = install.ArgHandler("mysql", "admin", "email")
47 handler.add(install.Arg("title", help="Title of your new MediaWiki install"))
49 def checkConfig(self, deployment):
50 return os.path.isfile(os.path.join(deployment.location, "LocalSettings.php"))
51 def detectVersion(self, deployment):
52 contents = deployment.read("includes/DefaultSettings.php")
53 regex = make_filename_regex("wgVersion")[1]
54 match = regex.search(contents)
55 if not match: return None
56 return distutils.version.LooseVersion(match.group(2)[1:-1])
57 def install(self, version, options):
59 os.unlink("LocalSettings.php")
63 os.chmod("config", 0777) # XXX: vaguely sketchy
66 'Sitename': options.title,
67 'EmergencyContact': options.email,
69 'DBserver': options.mysql_host,
70 'DBname': options.mysql_db,
71 'DBuser': options.mysql_user,
72 'DBpassword': options.mysql_password,
73 'DBpassword2': options.mysql_password,
74 'defaultEmail': options.email,
75 'SysopName': options.admin_name,
76 'SysopPass': options.admin_password,
77 'SysopPass2': options.admin_password,
79 result = install.fetch(options, 'config/index.php', post=postdata)
80 if options.verbose: print result
81 if result.find("Installation successful") == -1:
82 raise install.Failure()
83 os.rename('config/LocalSettings.php', 'LocalSettings.php')
84 def upgrade(self, version, options):
86 if not os.path.isfile("AdminSettings.php"):
87 sh.call("git", "checkout", "mediawiki-" + str(version), "--", "AdminSettings.php")
88 result = sh.eval("php", "maintenance/update.php", "--quick", log=True)
89 if not result.rstrip().split()[-1] == "Done.":
90 raise app.UpgradeFailure(result)
91 def backup(self, deployment, options):
93 # XXX: duplicate code, refactor, also, race condition
94 backupdir = os.path.join(".scripts", "backups")
95 outdir = os.path.join(backupdir, str(deployment.version) + "-" + datetime.date.today().isoformat())
96 if not os.path.exists(backupdir):
98 if os.path.exists(outdir):
99 util.safe_unlink(outdir)
101 outfile = os.path.join(outdir, "db.sql")
103 sh.call("mysqldump", "--compress", "-r", outfile, *get_mysql_args(deployment))
104 sh.call("gzip", "--best", outfile)
105 except shell.CallError as e:
106 raise app.BackupFailure(e.stderr)
107 def restore(self, deployment, backup, options):
109 backup_dir = os.path.join(".scripts", "backups", backup)
110 if not os.path.exists(backup_dir):
111 raise app.RestoreFailure("Backup %s doesn't exist", backup)
112 sql = open(os.path.join(backup_dir, "db.sql"), 'w+')
113 sh.call("gunzip", "-c", os.path.join(backup_dir, "db.sql.gz"), stdout=sql)
115 sh.call("mysql", *get_mysql_args(deployment), stdin=sql)
118 def get_mysql_args(d):
119 # XXX: add support for getting these out of options
121 if 'WIZARD_DBNAME' not in vars:
122 raise app.BackupFailure("Could not determine database name")
123 triplet = scripts.get_sql_credentials()
125 if triplet is not None:
126 server, user, password = triplet
127 args += ["-h", server, "-u", user, "-p" + password]
128 name = shlex.split(vars['WIZARD_DBNAME'])[0]