import re
import os

from wizard import app, deploy, install, 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)

make_extractor = app.filename_regex_extractor(make_filename_regex)
make_substitution = app.filename_regex_substitution(make_filename_regex)
seed = {
        'WIZARD_IP': 'IP', # obsolete, remove after we're done
        'WIZARD_SITENAME': 'wgSitename',
        'WIZARD_SCRIPTPATH': 'wgScriptPath',
        'WIZARD_EMERGENCYCONTACT': ('wgEmergencyContact', 'wgPasswordSender'),
        'WIZARD_DBSERVER': 'wgDBserver',
        'WIZARD_DBNAME': 'wgDBname',
        'WIZARD_DBUSER': 'wgDBuser',
        'WIZARD_DBPASSWORD': 'wgDBpassword',
        'WIZARD_SECRETKEY': ('wgSecretKey', 'wgProxyKey'),
        }

class Application(deploy.Application):
    parametrized_files = ['LocalSettings.php', 'php.ini']
    deprecated_keys = set(['WIZARD_IP']) | php.deprecated_keys
    @property
    def extractors(self):
        if not self._extractors:
            self._extractors = util.dictmap(make_extractor, seed)
            self._extractors.update(php.extractors)
        return self._extractors
    @property
    def substitutions(self):
        if not self._substitutions:
            self._substitutions = util.dictkmap(make_substitution, seed)
            self._substitutions.update(php.substitutions)
        return self._substitutions
    @property
    def install_handler(self):
        handler = install.ArgHandler("mysql", "admin", "email")
        handler.add(install.Arg("title", help="Title of your new MediaWiki install"))
        return handler
    def install(self, options):
        try:
            os.unlink("LocalSettings.php")
        except OSError:
            pass

        os.chmod("config", 0777) # XXX: vaguely sketchy

        postdata = {
            'Sitename': options.title,
            'EmergencyContact': options.email,
            'LanguageCode': 'en',
            'DBserver': options.mysql_host,
            'DBname': options.mysql_db,
            'DBuser': options.mysql_user,
            'DBpassword': options.mysql_password,
            'DBpassword2': options.mysql_password,
            'defaultEmail': options.email,
            'SysopName': options.admin_name,
            'SysopPass': options.admin_password,
            'SysopPass2': options.admin_password,
            }
        result = install.fetch(options, 'config/index.php', post=postdata)
        if options.verbose: print result
        if result.find("Installation successful") == -1:
            raise install.Failure()
        os.rename('config/LocalSettings.php', 'LocalSettings.php')
    def upgrade(self, options):
        sh = shell.Shell()
        sh.call("php", "maintenance/update.php")