DatabaseInstaller.php

Go to the documentation of this file.

<?php
abstract class DatabaseInstaller {

        public $parent;

        public $db = null;

        protected $internalDefaults = array();

        protected $globalNames = array();

        public abstract function getName();

        public abstract function isCompiled();

        public function checkPrerequisites() {
                return Status::newGood();
        }

        public abstract function getConnectForm();

        public abstract function submitConnectForm();

        public function getSettingsForm() {
                return false;
        }

        public function submitSettingsForm() {
                return Status::newGood();
        }

        public abstract function openConnection();

        public abstract function setupDatabase();

        public function getConnection() {
                if ( $this->db ) {
                        return Status::newGood( $this->db );
                }

                $status = $this->openConnection();
                if ( $status->isOK() ) {
                        $this->db = $status->value;
                        // Enable autocommit
                        $this->db->clearFlag( DBO_TRX );
                        $this->db->commit();
                }
                return $status;
        }

        public function createTables() {
                $status = $this->getConnection();
                if ( !$status->isOK() ) {
                        return $status;
                }
                $this->db->selectDB( $this->getVar( 'wgDBname' ) );

                if( $this->db->tableExists( 'archive', __METHOD__ ) ) {
                        $status->warning( 'config-install-tables-exist' );
                        $this->enableLB();
                        return $status;
                }

                $this->db->setFlag( DBO_DDLMODE ); // For Oracle's handling of schema files
                $this->db->begin( __METHOD__ );

                $error = $this->db->sourceFile( $this->db->getSchemaPath() );
                if( $error !== true ) {
                        $this->db->reportQueryError( $error, 0, '', __METHOD__ );
                        $this->db->rollback( __METHOD__ );
                        $status->fatal( 'config-install-tables-failed', $error );
                } else {
                        $this->db->commit( __METHOD__ );
                }
                // Resume normal operations
                if( $status->isOk() ) {
                        $this->enableLB();
                }
                return $status;
        }

        public function createExtensionTables() {
                $status = $this->getConnection();
                if ( !$status->isOK() ) {
                        return $status;
                }

                // Now run updates to create tables for old extensions
                DatabaseUpdater::newForDB( $this->db )->doUpdates( array( 'extensions' ) );

                return $status;
        }

        public abstract function getLocalSettings();

        public function getSchemaVars() {
                return array();
        }

        public function setupSchemaVars() {
                $status = $this->getConnection();
                if ( $status->isOK() ) {
                        $status->value->setSchemaVars( $this->getSchemaVars() );
                } else {
                        throw new MWException( __METHOD__.': unexpected DB connection error' );
                }
        }

        public function enableLB() {
                $status = $this->getConnection();
                if ( !$status->isOK() ) {
                        throw new MWException( __METHOD__.': unexpected DB connection error' );
                }
                LBFactory::setInstance( new LBFactory_Single( array(
                        'connection' => $status->value ) ) );
        }

        public function doUpgrade() {
                $this->setupSchemaVars();
                $this->enableLB();

                $ret = true;
                ob_start( array( $this, 'outputHandler' ) );
                try {
                        $up = DatabaseUpdater::newForDB( $this->db );
                        $up->doUpdates();
                } catch ( MWException $e ) {
                        echo "\nAn error occured:\n";
                        echo $e->getText();
                        $ret = false;
                }
                ob_end_flush();
                return $ret;
        }

        public function preInstall() {

        }

        public function preUpgrade() {

        }

        public function getGlobalNames() {
                return $this->globalNames;
        }

        public function __construct( $parent ) {
                $this->parent = $parent;
        }

        protected static function checkExtension( $name ) {
                wfSuppressWarnings();
                $compiled = wfDl( $name );
                wfRestoreWarnings();
                return $compiled;
        }

        public function getReadableName() {
                return wfMsg( 'config-type-' . $this->getName() );
        }

        public function getGlobalDefaults() {
                return array();
        }

        public function getInternalDefaults() {
                return $this->internalDefaults;
        }

        public function getVar( $var, $default = null ) {
                $defaults = $this->getGlobalDefaults();
                $internal = $this->getInternalDefaults();
                if ( isset( $defaults[$var] ) ) {
                        $default = $defaults[$var];
                } elseif ( isset( $internal[$var] ) ) {
                        $default = $internal[$var];
                }
                return $this->parent->getVar( $var, $default );
        }

        public function setVar( $name, $value ) {
                $this->parent->setVar( $name, $value );
        }

        public function getTextBox( $var, $label, $attribs = array(), $helpData = "" ) {
                $name = $this->getName() . '_' . $var;
                $value = $this->getVar( $var );
                if ( !isset( $attribs ) ) {
                        $attribs = array();
                }
                return $this->parent->getTextBox( array(
                        'var' => $var,
                        'label' => $label,
                        'attribs' => $attribs,
                        'controlName' => $name,
                        'value' => $value,
                        'help' => $helpData
                ) );
        }

        public function getPasswordBox( $var, $label, $attribs = array(), $helpData = "" ) {
                $name = $this->getName() . '_' . $var;
                $value = $this->getVar( $var );
                if ( !isset( $attribs ) ) {
                        $attribs = array();
                }
                return $this->parent->getPasswordBox( array(
                        'var' => $var,
                        'label' => $label,
                        'attribs' => $attribs,
                        'controlName' => $name,
                        'value' => $value,
                        'help' => $helpData
                ) );
        }

        public function getCheckBox( $var, $label, $attribs = array(), $helpData = "" ) {
                $name = $this->getName() . '_' . $var;
                $value = $this->getVar( $var );
                return $this->parent->getCheckBox( array(
                        'var' => $var,
                        'label' => $label,
                        'attribs' => $attribs,
                        'controlName' => $name,
                        'value' => $value,
                        'help' => $helpData
                ));
        }

        public function getRadioSet( $params ) {
                $params['controlName'] = $this->getName() . '_' . $params['var'];
                $params['value'] = $this->getVar( $params['var'] );
                return $this->parent->getRadioSet( $params );
        }

        public function setVarsFromRequest( $varNames ) {
                return $this->parent->setVarsFromRequest( $varNames, $this->getName() . '_' );
        }

        public function needsUpgrade() {
                $status = $this->getConnection();
                if ( !$status->isOK() ) {
                        return false;
                }

                if ( !$this->db->selectDB( $this->getVar( 'wgDBname' ) ) ) {
                        return false;
                }
                return $this->db->tableExists( 'cur', __METHOD__ ) || $this->db->tableExists( 'revision', __METHOD__ );
        }

        public function getInstallUserBox() {
                return
                        Html::openElement( 'fieldset' ) .
                        Html::element( 'legend', array(), wfMsg( 'config-db-install-account' ) ) .
                        $this->getTextBox( '_InstallUser', 'config-db-username', array( 'dir' => 'ltr' ), $this->parent->getHelpBox( 'config-db-install-username' ) ) .
                        $this->getPasswordBox( '_InstallPassword', 'config-db-password', array( 'dir' => 'ltr' ), $this->parent->getHelpBox( 'config-db-install-password' ) ) .
                        Html::closeElement( 'fieldset' );
        }

        public function submitInstallUserBox() {
                $this->setVarsFromRequest( array( '_InstallUser', '_InstallPassword' ) );
                return Status::newGood();
        }

        public function getWebUserBox( $noCreateMsg = false ) {
                $wrapperStyle = $this->getVar( '_SameAccount' ) ? 'display: none' : '';
                $s = Html::openElement( 'fieldset' ) .
                        Html::element( 'legend', array(), wfMsg( 'config-db-web-account' ) ) .
                        $this->getCheckBox(
                                '_SameAccount', 'config-db-web-account-same',
                                array( 'class' => 'hideShowRadio', 'rel' => 'dbOtherAccount' )
                        ) .
                        Html::openElement( 'div', array( 'id' => 'dbOtherAccount', 'style' => $wrapperStyle ) ) .
                        $this->getTextBox( 'wgDBuser', 'config-db-username' ) .
                        $this->getPasswordBox( 'wgDBpassword', 'config-db-password' ) .
                        $this->parent->getHelpBox( 'config-db-web-help' );
                if ( $noCreateMsg ) {
                        $s .= $this->parent->getWarningBox( wfMsgNoTrans( $noCreateMsg ) );
                } else {
                        $s .= $this->getCheckBox( '_CreateDBAccount', 'config-db-web-create' );
                }
                $s .= Html::closeElement( 'div' ) . Html::closeElement( 'fieldset' );
                return $s;
        }

        public function submitWebUserBox() {
                $this->setVarsFromRequest(
                        array( 'wgDBuser', 'wgDBpassword', '_SameAccount', '_CreateDBAccount' )
                );

                if ( $this->getVar( '_SameAccount' ) ) {
                        $this->setVar( 'wgDBuser', $this->getVar( '_InstallUser' ) );
                        $this->setVar( 'wgDBpassword', $this->getVar( '_InstallPassword' ) );
                }

                if( $this->getVar( '_CreateDBAccount' ) && strval( $this->getVar( 'wgDBpassword' ) ) == '' ) {
                        return Status::newFatal( 'config-db-password-empty', $this->getVar( 'wgDBuser' ) );
                }

                return Status::newGood();
        }

        public function populateInterwikiTable() {
                $status = $this->getConnection();
                if ( !$status->isOK() ) {
                        return $status;
                }
                $this->db->selectDB( $this->getVar( 'wgDBname' ) );

                if( $this->db->selectRow( 'interwiki', '*', array(), __METHOD__ ) ) {
                        $status->warning( 'config-install-interwiki-exists' );
                        return $status;
                }
                global $IP;
                wfSuppressWarnings();
                $rows = file( "$IP/maintenance/interwiki.list",
                        FILE_IGNORE_NEW_LINES | FILE_SKIP_EMPTY_LINES );
                wfRestoreWarnings();
                $interwikis = array();
                if ( !$rows ) {
                        return Status::newFatal( 'config-install-interwiki-list' );
                }
                foreach( $rows as $row ) {
                        $row = preg_replace( '/^\s*([^#]*?)\s*(#.*)?$/', '\\1', $row ); // strip comments - whee
                        if ( $row == "" ) continue;
                        $row .= "||";
                        $interwikis[] = array_combine(
                                array( 'iw_prefix', 'iw_url', 'iw_local', 'iw_api', 'iw_wikiid' ),
                                explode( '|', $row )
                        );
                }
                $this->db->insert( 'interwiki', $interwikis, __METHOD__ );
                return Status::newGood();
        }

        public function outputHandler( $string ) {
                return htmlspecialchars( $string );
        }
}