Working version of Database Upgrade/Backup/Restore tool in new wizArd format

Story ID:1244695. 3.7 Database Upgrade Script Committed by: Paul Barrett Reviewed by: Jarrett Jordaan

Working version of Database Upgrade/Backup/Restore tool in new wizArd format
Story ID:1244695. 3.7 Database Upgrade Script Committed by: Paul Barrett Reviewed by: Jarrett Jordaan
unknown
1 parent 0c0833f3
Showing 12 changed files with 535 additions and 879 deletions
setup/upgrade/index.php
setup/upgrade/step.php
setup/upgrade/steps/upgradeBackup.php
setup/upgrade/steps/upgradeComplete.php
setup/upgrade/steps/upgradeDatabase.php
setup/upgrade/steps/upgradeInstallation.php
setup/upgrade/steps/upgradeRestore.php
setup/upgrade/steps/upgradeWelcome.php
setup/upgrade/templates/backup.tpl
setup/upgrade/templates/installation.tpl
setup/upgrade/templates/restore.tpl
setup/upgrade/upgradeUtil.php
 <?php
 /**
-* Upgrader Index.
+* Database Upgrader Index.
 *
 * KnowledgeTree Community Edition
 * Document Management Made Simple
 <?php
 /**
-* Step .
+* Step Controller.
 *
 * KnowledgeTree Community Edition
 * Document Management Made Simple
@@ -119,10 +119,10 @@ class Step
     {
         // if not authenticated, return to step 1
         if (!isset($_SESSION['setup_user'])) {
-            header('index.php?step=welcome');
+            header('Location: index.php?step=welcome');
             exit;
         }
-            
+
         return '';
     }
@@ -254,38 +254,31 @@ class Step
         return false;
     }
-
-	/**
-	* Checks if Confirm button has been clicked
-	*
-	* @author KnowledgeTree Team
-	* @param none
-	* @access public
-	* @return boolean
-	*/
+    
+    /**
+    * Checks if Upgrade button has been clicked
+    *
+    * @author KnowledgeTree Team
+    * @param none
+    * @access public
+    * @return boolean
+    */
     function upgrade() {
-        if(isset($_POST['Upgrade'])) {
-            return true;
-        }
-
-        return false;
+        return isset($_POST['Upgrade']);
     }
-
-	/**
-	* Checks if Confirm button has been clicked
-	*
-	* @author KnowledgeTree Team
-	* @param none
-	* @access public
-	* @return boolean
-	*/
-    function installer() {
-        if(isset($_POST['Install'])) {
-            return true;
-        }
-
-        return false;
+    
+    /**
+    * Checks if Upgrade button has been clicked
+    *
+    * @author KnowledgeTree Team
+    * @param none
+    * @access public
+    * @return boolean
+    */
+    function restore() {
+        return isset($_POST['Restore']);
     }
+
 	/**
 	* Checks if we are currently in this class step
 	*
 <?php
 /**
-* Complete Step Controller. 
+* Backup Step Controller. 
 *
 * KnowledgeTree Community Edition
 * Document Management Made Simple
@@ -52,12 +52,9 @@ class upgradeBackup extends Step {
 	* @var object
 	*/	
     private $_dbhandler = null;
-
-    private $privileges_check = 'tick';
-    private $database_check = 'tick';
-    protected $silent = true;
-    
     protected $util = null;
+    protected $silent = false;
+    protected $temp_variables = array();
     public function __construct() {
     	$this->temp_variables = array("step_name"=>"backup", "silent"=>$this->silent);
@@ -65,7 +62,7 @@ class upgradeBackup extends Step {
     	$this->util = new UpgradeUtil();
     }
-    function doStep() {
+    public function doStep() {
         parent::doStep();
         if(!$this->inStep("backup")) {
             $this->doRun();
@@ -92,17 +89,21 @@ class upgradeBackup extends Step {
         else if($this->previous()) {
             return 'previous';
         }
+        else if ($this->upgrade()) {
+            header('Location: index.php?step_name=database');
+            exit;
+        }
         $this->doRun();
         return 'landing';
     }
-    function backupNow()
+    private function backupNow()
     {
         return isset($_POST['BackupNow']);
     }
-    function doRun($action = null) {
+    private function doRun($action = null) {
         $this->temp_variables['action'] = $action;
         if (is_null($action) || ($action == 'confirm')) {
@@ -127,29 +128,7 @@ class upgradeBackup extends Step {
     private function storeSilent() {
     }
-    /*
-    // these belong in a shared lib
-    function set_state($value)
-{
-    $_SESSION['state'] = $value;
-}
-function check_state($value, $state='Home')
-{
-    if ($_SESSION['state'] != $value)
-    {
-        ?>
-            <script type="text/javascript">
-            document.location="?go=<?php echo $state;?>";
-            </script>
-            <?php
-            exit;
-    }
-}
-*/
-    
     private function backup() {
-//        $this->check_state(1);
-//        $this->set_state(2);
         $targetfile = $_SESSION['backupFile'];
         $stmt = $this->create_backup_stmt($targetfile);
         $dir = $stmt['dir'];
@@ -163,7 +142,7 @@ function check_state($value, $state=&#39;Home&#39;)
             $read = fread($handle, 10240);
             pclose($handle);
             $_SESSION['backupOutput']=$read;
-            $dir = $this->resolveTempDir();
+            $dir = $this->util->resolveTempDir();
             $_SESSION['backupFile'] =   $stmt['target'];
             if (OS_UNIX) {
@@ -177,21 +156,9 @@ function check_state($value, $state=&#39;Home&#39;)
                 $_SESSION['backupStatus'] = false;
             }
         }
-        else
-        {
-    ?>
-    <P>
-        The <i>mysqldump</i> utility was not found in the <?php echo $dir;?> subdirectory.
-    
-    &nbsp;&nbsp; &nbsp; &nbsp;  <input type=button value="back" onclick="javascript:do_start('welcome')">
-    <?php
-        }
     }
     private function backupDone() {
-//        $this->check_state(2);
-//        $this->set_state(3);
-//        title('Backup Status');
         $status = $_SESSION['backupStatus'];
         $filename = $_SESSION['backupFile'];
@@ -200,7 +167,7 @@ function check_state($value, $state=&#39;Home&#39;)
         if ($status)
         {
             $stmt = $this->util->create_restore_stmt($filename);
-            $this->temp_variables['display'] = 'The backup file <nobr><i>"<?php echo $filename;?>"</i></nobr> has been created.
+            $this->temp_variables['display'] = 'The backup file <nobr><i>"' . $filename . '"</i></nobr> has been created.
             <P> It appears as though the <font color=green>backup has been successful</font>.
             <P>';
                 if ($stmt['dir'] != '')
@@ -230,136 +197,78 @@ function check_state($value, $state=&#39;Home&#39;)
         else
         {
             $this->temp_variables['display'] .= 'It appears as though <font color=red>the backup process has failed</font>.<P></P> Unfortunately, it is difficult to diagnose these problems automatically
-    and would recommend that you try to do the backup process manually.
-    <P>
-    We appologise for the inconvenience.
-    <P>
-    <table bgcolor="lightgrey">
-    <tr>
-    <td>' . $_SESSION['backupOutput'] . '</table>';
+            and would recommend that you try to do the backup process manually.
+            <P>
+            We appologise for the inconvenience.
+            <P>
+            <table bgcolor="lightgrey">
+            <tr>
+            <td>' . $_SESSION['backupOutput'] . '</table>';
         }
-}
-
-function create_backup_stmt($targetfile=null)
-{
-    $oKTConfig =& KTConfig::getSingleton();
-
-    $adminUser = $oKTConfig->get('db/dbAdminUser');
-    $adminPwd = $oKTConfig->get('db/dbAdminPass');
-    $dbHost = $oKTConfig->get('db/dbHost');
-    $dbName = $oKTConfig->get('db/dbName');
-
-    $dbPort = trim($oKTConfig->get('db/dbPort'));
-    if (empty($dbPort) || $dbPort=='default') $dbPort = get_cfg_var('mysql.default_port');
-    if (empty($dbPort)) $dbPort='3306';
-    $dbSocket = trim($oKTConfig->get('db/dbSocket'));
-    if (empty($dbSocket) || $dbSocket=='default') $dbSocket = get_cfg_var('mysql.default_socket');
-    if (empty($dbSocket)) $dbSocket='../tmp/mysql.sock';
-
-    $date=date('Y-m-d-H-i-s');
-
-    $dir=$this->resolveMysqlDir();
-
-    $info['dir']=$dir;
-
-    $prefix='';
-    if (OS_UNIX)
-    {
-        $prefix .= "./";
-    }
-
-    if (@stat($dbSocket) !== false)
-    {
-        $mechanism="--socket=\"$dbSocket\"";
-    }
-    else
-    {
-        $mechanism="--port=\"$dbPort\"";
-    }
-
-    $tmpdir=$this->resolveTempDir();
-
-    if (is_null($targetfile))
-    {
-        $targetfile="$tmpdir/kt-backup-$date.sql";
-    }
-
-    $stmt = $prefix . "mysqldump --user=\"$adminUser\" -p $mechanism \"$dbName\" > \"$targetfile\"";
-    $info['display']=$stmt;
-    $info['target']=$targetfile;
-
-
-    $stmt  = $prefix. "mysqldump --user=\"$adminUser\" --password=\"$adminPwd\" $mechanism \"$dbName\" > \"$targetfile\"";
-    $info['cmd']=$stmt;
-    return $info;
-}
-
-function resolveMysqlDir()
-{
-    // possibly detect existing installations:
-
-    if (OS_UNIX)
-    {
-        $dirs = array('/opt/mysql/bin','/usr/local/mysql/bin');
-        $mysqlname ='mysql';
-    }
-    else
-    {
-        $dirs = explode(';', $_SERVER['PATH']);
-        $dirs[] ='c:/Program Files/MySQL/MySQL Server 5.0/bin';
-        $dirs[] = 'c:/program files/ktdms/mysql/bin';
-        $mysqlname ='mysql.exe';
-    }
-
-    $oKTConfig =& KTConfig::getSingleton();
-    $mysqldir = $oKTConfig->get('backup/mysqlDirectory',$mysqldir);
-    $dirs[] = $mysqldir;
-
-    if (strpos(__FILE__,'knowledgeTree') !== false && strpos(__FILE__,'ktdms') != false)
-    {
-        $dirs [] = realpath(dirname($FILE) . '/../../mysql/bin');
     }
-    foreach($dirs as $dir)
+    private function create_backup_stmt($targetfile=null)
     {
-        if (is_file($dir . '/' . $mysqlname))
+        $oKTConfig =& KTConfig::getSingleton();
+    
+        $adminUser = $oKTConfig->get('db/dbAdminUser');
+        $adminPwd = $oKTConfig->get('db/dbAdminPass');
+        $dbHost = $oKTConfig->get('db/dbHost');
+        $dbName = $oKTConfig->get('db/dbName');
+    
+        $dbPort = trim($oKTConfig->get('db/dbPort'));
+        if (empty($dbPort) || $dbPort=='default') $dbPort = get_cfg_var('mysql.default_port');
+        if (empty($dbPort)) $dbPort='3306';
+        $dbSocket = trim($oKTConfig->get('db/dbSocket'));
+        if (empty($dbSocket) || $dbSocket=='default') $dbSocket = get_cfg_var('mysql.default_socket');
+        if (empty($dbSocket)) $dbSocket='../tmp/mysql.sock';
+    
+        $date=date('Y-m-d-H-i-s');
+    
+        $dir=$this->util->resolveMysqlDir();
+    
+        $info['dir']=$dir;
+    
+        $prefix='';
+        if (OS_UNIX)
         {
-            return $dir;
+            $prefix .= "./";
         }
+    
+        if (@stat($dbSocket) !== false)
+        {
+            $mechanism="--socket=\"$dbSocket\"";
+        }
+        else
+        {
+            $mechanism="--port=\"$dbPort\"";
+        }
+    
+        $tmpdir=$this->util->resolveTempDir();
+    
+        if (is_null($targetfile))
+        {
+            $targetfile="$tmpdir/kt-backup-$date.sql";
+        }
+    
+        $stmt = $prefix . "mysqldump --user=\"$adminUser\" -p $mechanism \"$dbName\" > \"$targetfile\"";
+        $info['display']=$stmt;
+        $info['target']=$targetfile;
+    
+    
+        $stmt  = $prefix. "mysqldump --user=\"$adminUser\" --password=\"$adminPwd\" $mechanism \"$dbName\" > \"$targetfile\"";
+        $info['cmd']=$stmt;
+        return $info;
     }
-    return '';
-}
-
-function resolveTempDir()
-{
-    if (OS_UNIX) {
-        $dir='/tmp/kt-db-backup';
-    }
-    else {
-        $dir='c:/kt-db-backup';
-    }
-    $oKTConfig =& KTConfig::getSingleton();
-    $dir = $oKTConfig->get('backup/backupDirectory',$dir);
-
-    if (!is_dir($dir)) {
-            mkdir($dir);
+    private function backupConfirm()
+    {
+        $stmt = $this->create_backup_stmt();
+        $_SESSION['backupFile'] = $stmt['target'];
+    
+        $dir = $stmt['dir'];
+        $this->temp_variables['dir'] = $dir;
+        $this->temp_variables['display'] = $stmt['display'];
     }
-    return $dir;
-}
-
-
-function backupConfirm()
-{
-    $stmt = $this->create_backup_stmt();
-    $_SESSION['backupFile'] = $stmt['target'];
-
-    $dir = $stmt['dir'];
-    $this->temp_variables['dir'] = $dir;
-    $this->temp_variables['display'] = $stmt['display'];
-}
-
-
-
 }
 ?>
 \ No newline at end of file
@@ -44,20 +44,20 @@ require &#39;../../config/dmsDefaults.php&#39;;
 class upgradeComplete extends Step {
-    protected $silent = true;
-    
     protected $util = null;
+    protected $silent = false;
+    protected $temp_variables = array();
     public function __construct() {
     	$this->temp_variables = array("step_name"=>"complete", "silent"=>$this->silent);
     }
-    function doStep() {
+    public function doStep() {
         $this->doRun();
     	return 'landing';
     }
-    function doRun() {
+    private function doRun() {
         $this->storeSilent();// Set silent mode variables
     }
 <?php
 /**
-* Database Step Controller. 
+* Upgrade Step Controller. 
 *
 * KnowledgeTree Community Edition
 * Document Management Made Simple
@@ -40,7 +40,6 @@
 * @version Version 0.1
 */
-// include defaults
 require '../../config/dmsDefaults.php';
 require_once KT_LIB_DIR . '/config/config.inc.php';
 require_once KT_LIB_DIR . '/plugins/pluginutil.inc.php';
@@ -67,105 +66,6 @@ class upgradeDatabase extends Step
     public $_util = null;
 	/**
-	* Database type
-	*
-	* @author KnowledgeTree Team
-	* @access private
-	* @var array
-	*/	
-    private $dtype = '';
-    
-	/**
-	* Database types
-	*
-	* @author KnowledgeTree Team
-	* @access private
-	* @var array
-	*/	
-    private $dtypes = array();
-    
-	/**
-	* Database host
-	*
-	* @author KnowledgeTree Team
-	* @access private
-	* @var string
-	*/
-    private $dhost = '';
-    
-	/**
-	* Database port
-	*
-	* @author KnowledgeTree Team
-	* @access private
-	* @var string
-	*/
-    private $dport = '';
-    
-	/**
-	* Database name
-	*
-	* @author KnowledgeTree Team
-	* @access private
-	* @var string
-	*/
-    private $dname = '';
-    
-	/**
-	* Database root username
-	*
-	* @author KnowledgeTree Team
-	* @access private
-	* @var string
-	*/
-    private $duname = '';
-    
-	/**
-	* Database root password
-	*
-	* @author KnowledgeTree Team
-	* @access private
-	* @var string
-	*/
-    private $dpassword = '';
-    
-	/**
-	* Database dms username
-	*
-	* @author KnowledgeTree Team
-	* @access private
-	* @var string
-	*/
-    private $dmsname = '';
-    
-	/**
-	* Database dms password
-	*
-	* @author KnowledgeTree Team
-	* @access private
-	* @var string
-	*/
-    private $dmspassword = '';
-
-	/**
-	* Default dms user username
-	*
-	* @author KnowledgeTree Team
-	* @access private
-	* @var boolean
-	*/
-    private $dmsusername = '';
-    
-	/**
-	* Default dms user password
-	*
-	* @author KnowledgeTree Team
-	* @access private
-	* @var boolean
-	*/
-	private $dmsuserpassword = '';
-	
-	/**
 	* Location of database binaries.
 	*
 	* @author KnowledgeTree Team
@@ -181,25 +81,7 @@ class upgradeDatabase extends Step
 	* @access private
 	* @var string
 	*/
-    private $dbbinary = ''; // TODO:multiple databases
-    
-	/**
-	* Database table prefix
-	*
-	* @author KnowledgeTree Team
-	* @access private
-	* @var string
-	*/
-    private $tprefix = '';
-    
-	/**
-	* Flag to drop database
-	*
-	* @author KnowledgeTree Team
-	* @access private
-	* @var boolean
-	*/
-    private $ddrop = false;
+    private $dbBinary = ''; // TODO:multiple databases
 	/**
 	* List of errors encountered
@@ -228,26 +110,11 @@ class upgradeDatabase extends Step
 	*/
     public $storeInSession = true;
-	/**
-	* Flag if step needs to be upgraded
-	*
-	* @author KnowledgeTree Team
-	* @access public
-	* @var array
-	*/
-    protected $runUpgrade = true;
-    
-	/**
-	* Flag if step needs to run silently
-	*
-	* @author KnowledgeTree Team
-	* @access public
-	* @var array
-	*/
-    protected $silent = true;
+    protected $silent = false;
+    protected $temp_variables = array();
 	/**
-	* Constructs database object
+	* Constructs database upgrade object
 	*
 	* @author KnowledgeTree Team
 	* @access public
@@ -272,12 +139,12 @@ class upgradeDatabase extends Step
     public function doStep() {
         parent::doStep();
     	$this->initErrors();
-    	$this->setDetails(); // Set any posted variables
-//    	if(!$this->inStep("database")) {
-//    	    $this->doRun();
-//    		return 'landing';
-//    	}
+    	if(!$this->inStep("database")) {
+    	    $this->doRun();
+    		return 'landing';
+    	}
 		if($this->next()) {
+		    $this->doRun('preview');
 			return 'next';
 		} else if($this->previous()) {
 			return 'previous';
@@ -297,15 +164,15 @@ class upgradeDatabase extends Step
         return 'landing';
     }
-    function confirmUpgrade() {
+    private function confirmUpgrade() {
         return isset($_POST['ConfirmUpgrade']);
     }
-    function upgrading() {
+    private function upgrading() {
         return isset($_POST['RunUpgrade']);
     } 
-    function doRun($action = null) {
+    private function doRun($action = null) {
         $this->readConfig(KTConfig::getConfigFilename());
         if($this->dbSettings['dbPort'] == '')  {
@@ -327,8 +194,7 @@ class upgradeDatabase extends Step
         }
         else if ($action == 'runUpgrade') {
             $this->temp_variables['title'] = 'Upgrade In Progress';
-            if (!$this->upgrade()) {
-//                $this->temp_variables['upgradeTable'] = '';//$this->upgradeErrors();
+            if (!$this->upgradeDatabase()) {
                 return false;
             }
         }
@@ -336,7 +202,7 @@ class upgradeDatabase extends Step
         return true;
     }
-    public function generateUpgradeTable() {
+    private function generateUpgradeTable() {
         global $default;
         $this->temp_variables['systemVersion'] = $default->systemVersion;
@@ -367,42 +233,6 @@ class upgradeDatabase extends Step
     }
 	/**
-	* Store options
-	*
-	* @author KnowledgeTree Team
-	* @params object SimpleXmlObject
-	* @access private
-	* @return void
-	*/
-   private function setDetails() {
-        // create lock file to indicate Upgrade mode
-        $this->createUpgradeFile();
-    }
-    
-    /**
-     * Creates miUpgradeock file so that system knows it is supposed to run an upgrade installation
-     * 
-     * @author KnowledgeTree Team
-     * @access private
-     * @return void
-     */
-    private function createUpgradeFile() {
-        @touch($this->wizardLocation . DIRECTORY_SEPARATOR . "upgrade.lock");
-    }
-    
-	/**
-	* Safer way to return post data
-	*
-	* @author KnowledgeTree Team
-	* @params SimpleXmlObject $simplexml
-	* @access public
-	* @return void
-	*/
-    public function getPostSafe($key) {
-    	return isset($_POST[$key]) ? $_POST[$key] : "";
-    }
-    
-	/**
 	* Stores varibles used by template
 	*
 	* @author KnowledgeTree Team
@@ -455,7 +285,7 @@ class upgradeDatabase extends Step
         $this->temp_variables['dbSettings'] = $this->dbSettings;
     }
-    function upgradeConfirm()
+    private function upgradeConfirm()
     {
         if (!isset($_SESSION['backupStatus']) || $_SESSION['backupStatus'] === false) {
             $this->temp_variables['backupStatus'] = false;
@@ -465,158 +295,141 @@ class upgradeDatabase extends Step
         }
     }
-
-function upgrade()
-{
-    global $default;
+    private function upgradeDatabase()
+    {
+        global $default;
+        
+        $this->temp_variables['detail'] = '<p>The table below describes the upgrades that have occurred to
+            upgrade your KnowledgeTree installation to <strong>' . $default->systemVersion . '</strong>';
+      
+        $pre_res = $this->performPreUpgradeActions();
+        if (PEAR::isError($pre_res)) {
+            $this->temp_variables['preUpgrade'] = '<font color="red">Pre-Upgrade actions failed.</font>';
+        }
+        else {
+            $this->temp_variables['preUpgrade'] = '<font color="green">Pre-Upgrade actions succeeded.</font>';
-    $this->temp_variables['detail'] = '<p>The table below describes the upgrades that have occurred to
-        upgrade your KnowledgeTree installation to <strong>' . $default->systemVersion . '</strong>';
-?>
-    <p>
-
-<?php
-  
-    $pre_res = $this->performPreUpgradeActions();
-    if (PEAR::isError($pre_res)) {
-        $this->temp_variables['preUpgrade'] = '<font color="red">Pre-Upgrade actions failed.</font>';
-    }
-    else {
-        $this->temp_variables['preUpgrade'] = '<font color="green">Pre-Upgrade actions succeeded.</font>';
-
+        }
+        
+        $res = $this->performAllUpgrades();
+        if (PEAR::isError($res) || PEAR::isError($pres)) {
+            $this->temp_variables['upgradeStatus'] = '<font color="red">Upgrade failed.</font>';
+        }
+        else {
+            $this->temp_variables['upgradeStatus'] = '<font color="green">Upgrade succeeded.</font>';
+        }
+    
+        $post_pres = $this->performPostUpgradeActions();
+        if (PEAR::isError($post_res)) {
+            $this->temp_variables['postUpgrade'] = '<font color="red">Post-Upgrade actions failed.</font>';
+        }
+        else {
+            $this->temp_variables['postUpgrade'] = '<font color="green">Post-Upgrade actions succeeded.</font>';
+        }
     }
-            ?>
-    <br/>
-<p>
-  <?php
-?>
-<p>
-<?php
-    $res = $this->performAllUpgrades();
-    if (PEAR::isError($res) || PEAR::isError($pres)) {
-        $this->temp_variables['upgradeStatus'] = '<font color="red">Upgrade failed.</font>';
-    }
-    else {
-        $this->temp_variables['upgradeStatus'] = '<font color="green">Upgrade succeeded.</font>';
-    }
-?>
-<p>
-    
-<p>
-  <?php
-    $post_pres = $this->performPostUpgradeActions();
-    if (PEAR::isError($post_res)) {
-        $this->temp_variables['postUpgrade'] = '<font color="red">Post-Upgrade actions failed.</font>';
-    }
-    else {
-        $this->temp_variables['postUpgrade'] = '<font color="green">Post-Upgrade actions succeeded.</font>';
+    private function performPreUpgradeActions() {
+    
+        // This is just to test and needs to be updated to a more sane and error resistent architrcture if it works.
+        // It should idealy work the same as the upgrades.
+    
+        global $default;
+    
+        // Lock the scheduler
+        $lockFile = $default->cacheDirectory . DIRECTORY_SEPARATOR . 'scheduler.lock';
+        touch($lockFile);
+        return true;
+    
     }
-}
-
-function performPreUpgradeActions() {
-
-    // This is just to test and needs to be updated to a more sane and error resistent architrcture if it works.
-    // It should idealy work the same as the upgrades.
-
-    global $default;
-
-    // Lock the scheduler
-    $lockFile = $default->cacheDirectory . DIRECTORY_SEPARATOR . 'scheduler.lock';
-    touch($lockFile);
-    return true;
-
-}
-
-function performPostUpgradeActions() {
-
-    // This is just to test and needs to be updated to a more sane and error resistent architrcture if it works.
-    // It should idealy work the same as the upgrades.
-
-    global $default;
-
-    // Ensure all plugins are re-registered.
-    $sql = "TRUNCATE plugin_helper";
-    $res = DBUtil::runQuery($sql);
-
-    // Clear out all caches and proxies - they need to be regenerated with the new code
-    $proxyDir = $default->proxyCacheDirectory;
-    KTUtil::deleteDirectory($proxyDir);
-
-    $oKTCache = new KTCache();
-    $oKTCache->deleteAllCaches();
-
-    // Clear the configuration cache, it'll regenerate on next load
-    $oKTConfig = new KTConfig();
-    $oKTConfig->clearCache();
-
-    // Unlock the scheduler
-    $lockFile = $default->cacheDirectory . DIRECTORY_SEPARATOR . 'scheduler.lock';
-    if(file_exists($lockFile)){
-        @unlink($lockFile);
+    
+    private function performPostUpgradeActions() {
+    
+        // This is just to test and needs to be updated to a more sane and error resistent architrcture if it works.
+        // It should idealy work the same as the upgrades.
+    
+        global $default;
+    
+        // Ensure all plugins are re-registered.
+        $sql = "TRUNCATE plugin_helper";
+        $res = DBUtil::runQuery($sql);
+    
+        // Clear out all caches and proxies - they need to be regenerated with the new code
+        $proxyDir = $default->proxyCacheDirectory;
+        KTUtil::deleteDirectory($proxyDir);
+    
+        $oKTCache = new KTCache();
+        $oKTCache->deleteAllCaches();
+    
+        // Clear the configuration cache, it'll regenerate on next load
+        $oKTConfig = new KTConfig();
+        $oKTConfig->clearCache();
+    
+        // Unlock the scheduler
+        $lockFile = $default->cacheDirectory . DIRECTORY_SEPARATOR . 'scheduler.lock';
+        if(file_exists($lockFile)){
+            @unlink($lockFile);
+        }
+    
+        return true;
+    
     }
-    return true;
-
-}
-
-function performAllUpgrades () {
-    global $default;
-    $query = sprintf('SELECT value FROM %s WHERE name = "databaseVersion"', $default->system_settings_table);
-    $lastVersion = DBUtil::getOneResultKey($query, 'value');
-    $currentVersion = $default->systemVersion;
-
-    $upgrades = describeUpgrade($lastVersion, $currentVersion);
+    private function performAllUpgrades () {
+        global $default;
+        
+        $row = 1;
+        
+        $query = sprintf('SELECT value FROM %s WHERE name = "databaseVersion"', $default->system_settings_table);
+        $lastVersion = DBUtil::getOneResultKey($query, 'value');
+        $currentVersion = $default->systemVersion;
-    $this->temp_variables['upgradeTable'] = '';
-
-    foreach ($upgrades as $upgrade) {
-        if (($GLOBALS['row'] % 2) == 1) {
-            $class = "odd";
-        } else {
-            $class = "even";
+        $upgrades = describeUpgrade($lastVersion, $currentVersion);
+        
+        $this->temp_variables['upgradeTable'] = '';
+    
+        foreach ($upgrades as $upgrade) {
+            if (($row % 2) == 1) {
+                $class = "odd";
+            } else {
+                $class = "even";
+            }
+            $this->temp_variables['upgradeTable'] .= sprintf('<div class="row %s"><div class="foo">%s</div>' . "\n", $class, htmlspecialchars($upgrade->getDescription()));
+            ++$row;
+            $res = $upgrade->performUpgrade();
+            $this->temp_variables['upgradeTable'] .= sprintf('<div class="bar">%s</div>', $this->showResult($res));
+            $this->temp_variables['upgradeTable'] .= '<br style="clear: both">' . "\n";
+            $this->temp_variables['upgradeTable'] .= "</div>\n";
+            if (PEAR::isError($res)) {
+                if (!is_a($res, 'Upgrade_Already_Applied')) {
+                    break;
+                } else {
+                    $res = true;
+                }
+            }
+            if ($res === false) {
+                $res = PEAR::raiseError("Upgrade returned false");
+                break;
+            }
         }
-        $this->temp_variables['upgradeTable'] .= sprintf('<div class="row %s"><div class="foo">%s</div>' . "\n", $class, htmlspecialchars($upgrade->getDescription()));
-        $GLOBALS['row']++;
-        ob_flush();
-        flush();
-        $res = $upgrade->performUpgrade();
-        $this->temp_variables['upgradeTable'] .= sprintf('<div class="bar">%s</div>', $this->showResult($res));
-        $this->temp_variables['upgradeTable'] .= '<br style="clear: both">' . "\n";
-        ob_flush();
-        flush();
-        $this->temp_variables['upgradeTable'] .= "</div>\n";
+    
+        return $res;
+    }
+    
+    private function showResult($res) {
         if (PEAR::isError($res)) {
-            if (!is_a($res, 'Upgrade_Already_Applied')) {
-                break;
-            } else {
-                $res = true;
+            if (is_a($res, 'Upgrade_Already_Applied')) {
+                return '<span style="color: orange">Already applied</span>';
             }
+            return sprintf('<span style="color: red">%s</span>', htmlspecialchars($res->toString()));
         }
-        if ($res === false) {
-            $res = PEAR::raiseError("Upgrade returned false");
-            break;
+        if ($res === true) {
+            return '<span style="color: green">Success</span>';
         }
-    }
-
-    return $res;
-}
-
-function showResult($res) {
-    if (PEAR::isError($res)) {
-        if (is_a($res, 'Upgrade_Already_Applied')) {
-            return '<span style="color: orange">Already applied</span>';
+        if ($res === false) {
+            return '<span style="color: red">Failure</span>';
         }
-        return sprintf('<span style="color: red">%s</span>', htmlspecialchars($res->toString()));
+        return $res;
     }
-    if ($res === true) {
-        return '<span style="color: green">Success</span>';
-    }
-    if ($res === false) {
-        return '<span style="color: red">Failure</span>';
-    }
-    return $res;
-}
 }
 ?>
 \ No newline at end of file
 <?php
 /**
-* Upgrade Step Controller. 
+* Notification Controller. 
 *
 * KnowledgeTree Community Edition
 * Document Management Made Simple
@@ -50,12 +50,13 @@ class UpgradeInstallation extends step
 	* @var array
 	*/
     protected $silent = false;
+    protected $temp_variables = array();
-    function __construct() {
+    public function __construct() {
         $this->temp_variables = array("step_name"=>"installation");
     }
-    function doStep() {
+    public function doStep() {
         parent::doStep();
         if($this->next()) {
             return 'next';
@@ -72,13 +73,5 @@ class UpgradeInstallation extends step
         return 'landing';
     }
-    function backup() {
-        return isset($_POST['Backup']);
-    }
-     
-    function restore() {
-        return isset($_POST['Restore']);
-    }
-    
 }
 ?>
 \ No newline at end of file
 <?php
 /**
-* Complete Step Controller. 
+* Restore Step Controller. 
 *
 * KnowledgeTree Community Edition
 * Document Management Made Simple
@@ -53,10 +53,8 @@ class upgradeRestore extends Step {
 	*/	
     private $_dbhandler = null;
-    private $privileges_check = 'tick';
-    private $database_check = 'tick';
-    protected $silent = true;
-    
+    protected $silent = false;
+    protected $temp_variables = array();    
     protected $util = null;
     public function __construct() {
@@ -65,8 +63,10 @@ class upgradeRestore extends Step {
     	$this->util = new UpgradeUtil();
     }
-    function doStep() {
+    public function doStep() {
         parent::doStep();
+        $this->temp_variables['restore'] = false;
+        
         if(!$this->inStep("restore")) {
             $this->doRun();
             return 'landing';
@@ -78,24 +78,40 @@ class upgradeRestore extends Step {
         } else if($this->previous()) {
             return 'previous';
         }
+        else if ($this->restoreNow()) {
+            $this->temp_variables['restoreSuccessful'] = false;
+            $this->doRun(true);
+            return 'next';
+        }
         $this->doRun();
         return 'landing';
     }
-    function doRun() {
-        $this->temp_variables['selected'] = false;
-        if ($this->select()) {
-            $this->restoreSelected();
-            $this->temp_variables['selected'] = true;
+    private function restoreNow() {
+        return isset($_POST['RunRestore']);
+    } 
+    
+    private function doRun($restore = false) {
+        if (!$restore) {
+            $this->temp_variables['selected'] = false;
+            if ($this->select()) {
+                $this->restoreSelected();
+                $this->temp_variables['selected'] = true;
+                $this->temp_variables['availableBackups'] = true;
+            }
+            $this->restoreConfirm();
+        } // end not running a restore, just setting up
+        else {
+            $this->restoreDatabase();
         }
-        $this->restoreConfirm();
+            
         $this->storeSilent();// Set silent mode variables
         return true;
     }
-    function select() {
+    private function select() {
         return isset($_POST['RestoreSelect']);
     } 
@@ -105,183 +121,126 @@ class upgradeRestore extends Step {
      */
     private function storeSilent() {
     }
- /*   
-    // these belong in a shared lib
-    function set_state($value)
-{
-    $_SESSION['state'] = $value;
-}
-function check_state($value, $state='Home')
-{
-    if ($_SESSION['state'] != $value)
-    {
-        ?>
-            <script type="text/javascript">
-            document.location="?go=<?php echo $state;?>";
-            </script>
-            <?php
-            exit;
-    }
-}
-*/
-    function restore()
-{
-//    check_state(1);
-//    set_state(5);
-//    title('Restore In Progress');
-    $status = $_SESSION['backupStatus'];
-    $filename = $_SESSION['backupFile']; 
-    $stmt = $this->util->create_restore_stmt($filename);
-    $dir = $stmt['dir'];
-    if (is_file($dir . '/mysql') || is_file($dir . '/mysql.exe'))
+    private function restoreDatabase()
     {
-
-?>
-        The restore is now underway. Please wait till it completes.
-<?php
-        print "\n";
-
-
-        $curdir=getcwd();
-        chdir($dir);
-
-
-        $ok=true;
-        $stmts=explode("\n",$stmt['cmd']);
-        foreach($stmts as $stmt)
+        $this->temp_variables['restore'] = true;
+        $status = $_SESSION['backupStatus'];
+        $filename = $_SESSION['backupFile']; 
+        $stmt = $this->util->create_restore_stmt($filename);
+        $dir = $stmt['dir'];
+    
+        if (is_file($dir . '/mysql') || is_file($dir . '/mysql.exe'))
         {
-
-            $handle = popen($stmt, 'r');
-            if ($handle=='false')
+            $curdir=getcwd();
+            chdir($dir);
+    
+            $ok=true;
+            $stmts=explode("\n",$stmt['cmd']);
+            foreach($stmts as $stmt)
             {
-                $ok=false;
-                break;
+    
+                $handle = popen($stmt, 'r');
+                if ($handle=='false')
+                {
+                    $ok=false;
+                    break;
+                }
+                $read = fread($handle, 10240);
+                pclose($handle);
+                $_SESSION['restoreOutput']=$read;
             }
-            $read = fread($handle, 10240);
-            pclose($handle);
-            $_SESSION['restoreOutput']=$read;
-        }
-
+    
             $_SESSION['restoreStatus'] = $ok;
-
-    }
-    else
-    {
-?>
-<P>
-    The <i>mysql</i> utility was not found in the <?php echo $dir;?> subdirectory.
-
-&nbsp;&nbsp; &nbsp; &nbsp;  <input type=button value="back" onclick="javascript:do_start('welcome')">
-<?php
+            // should be some sort of error checking, really
+            $this->restoreDone();
+        }
     }
-}
-
-
-function restoreDone()
-{
-//    check_state(5);
-//    set_state(6);
-//    title('Restore Status');
-    $status = $_SESSION['restoreStatus'];
-    $filename = $_SESSION['backupFile'];
-
-    if ($status)
-    {
-
-?>
-        The restore of <nobr><i>"<?php echo $filename;?>"</i></nobr> has been completed.
-        <P>
-        It appears as though the <font color=green>restore has been successful</font>.
-        <P>
-
-
-
-<?php
-    }
-    else
+    private function restoreDone()
     {
-?>
-It appears as though <font color=red>the restore process has failed</font>. <P>
-Unfortunately, it is difficult to diagnose these problems automatically
-and would recommend that you try to do the backup process manually.
-<P>
-We appologise for the inconvenience.
-<P>
-<table bgcolor="lightgrey">
-<tr>
-<td>
-<?php echo $_SESSION['restoreOutput'];?>
-</table>
-<?php
-
+        $status = $_SESSION['restoreStatus'];
+        $filename = $_SESSION['backupFile'];
+    
+        if ($status)
+        {
+            $this->temp_variables['display'] = 'The restore of <nobr><i>"' . $filename . '"</i></nobr> has been completed.
+            <P>
+            It appears as though the <font color=green>restore has been successful</font>.
+            <P>';
+            
+            $this->temp_variables['title'] = 'Restore Complete'; 
+            $this->temp_variables['restoreSuccessful'] = true;
+        }
+        else
+        {
+            $this->temp_variables['display'] = 'It appears as though <font color=red>the restore process has failed</font>. <P>
+            Unfortunately, it is difficult to diagnose these problems automatically
+            and would recommend that you try to do the backup process manually.
+            <P>
+            We appologise for the inconvenience.
+            <P>
+            <table bgcolor="lightgrey">
+            <tr>
+            <td>' . $_SESSION['restoreOutput'] . '
+            </table>';
+            $this->temp_variables['title'] = 'Restore Failed';
+            $this->temp_variables['restoreSuccessful'] = false;
+        }
     }
-?>
-
-<br/>
-
-&nbsp;&nbsp; &nbsp; &nbsp;  <input type=button value="back" onclick="javascript:do_start('welcome')">
-
-<?php
-
-}
-
-function restoreSelect()
-{
-    $this->temp_variables['availableBackups'] = false;
-//    title('Select Backup to Restore');
-    $dir = $this->util->resolveTempDir();
-
-    $files = array();
-    if ($dh = opendir($dir))
+    
+    private function restoreSelect()
     {
-        while (($file = readdir($dh)) !== false)
+        $this->temp_variables['availableBackups'] = false;
+        $dir = $this->util->resolveTempDir();
+    
+        $files = array();
+        if ($dh = opendir($dir))
         {
-            if (!preg_match('/kt-backup.+\.sql/',$file)) {
-                continue;
+            while (($file = readdir($dh)) !== false)
+            {
+                if (!preg_match('/kt-backup.+\.sql/',$file)) {
+                    continue;
+                }
+                $files[] = $file;
             }
-            $files[] = $file;
+            closedir($dh);
+        }
+        
+        $this->temp_variables['title'] = 'Select Backup to Restore';
+        $this->temp_variables['dir'] = $dir;
+        if (count($files) != 0) {
+            $this->temp_variables['availableBackups'] = true;
+            $this->temp_variables['files'] = $files;
         }
-        closedir($dh);
     }
-    $this->temp_variables['dir'] = $dir;
-    if (count($files) != 0) {
-        $this->temp_variables['availableBackups'] = true;
-        $this->temp_variables['files'] = $files;
+    private function restoreSelected()
+    {
+        $file=$_REQUEST['file'];
+    
+        $dir = $this->util->resolveTempDir();
+        $_SESSION['backupFile'] = $dir . '/' . $file;
     }
-}
-
-function restoreSelected()
-{
-    $file=$_REQUEST['file'];
-
-    $dir = $this->util->resolveTempDir();
-    $_SESSION['backupFile'] = $dir . '/' . $file;
-?>
-<?php
-
-}
-
-function restoreConfirm()
-{
-    if (!isset($_SESSION['backupFile']) || !is_file($_SESSION['backupFile']) || filesize($_SESSION['backupFile']) == 0)
+    
+    private function restoreConfirm()
     {
-        $this->restoreSelect();
-        return;
+        if (!isset($_SESSION['backupFile']) || !is_file($_SESSION['backupFile']) || filesize($_SESSION['backupFile']) == 0)
+        {
+            $this->restoreSelect();
+            return;
+        }
+    
+        $status = $_SESSION['backupStatus'];
+        $filename = $_SESSION['backupFile'];
+        $stmt = $this->util->create_restore_stmt($filename);
+        
+        $this->temp_variables['title'] = 'Confirm Restore';
+        $this->temp_variables['dir'] = $stmt['dir'];
+        $this->temp_variables['display'] = $stmt['display'];
+        $this->temp_variables['availableBackups'] = true;
+        $this->temp_variables['selected'] = true;
     }
-    $status = $_SESSION['backupStatus'];
-    $filename = $_SESSION['backupFile'];
-    $stmt = $this->util->create_restore_stmt($filename);
-
-    $this->temp_variables['dir'] = $stmt['dir'];
-    $this->temp_variables['display'] = $stmt['display'];
-}
-
-
-
-
 }
 ?>
 \ No newline at end of file
@@ -40,14 +40,13 @@
 * @version Version 0.1
 */
-global $default;
-// include defaults
 include '../../config/dmsDefaults.php';
 require_once KT_LIB_DIR . '/authentication/authenticationutil.inc.php';
 class upgradeWelcome extends step {
-	protected $silent = true;
+    protected $silent = false;
+    protected $temp_variables = array();
     public function __construct() {
         $this->temp_variables = array("step_name"=>"welcome");
@@ -76,7 +75,6 @@ class upgradeWelcome extends step {
         if (!$authenticated)
         {
             session_unset();
-//            loginFailed(_kt('Could not authenticate administrative user'));
             return false;
         }
@@ -44,18 +44,17 @@ You can continue to do the backup manually using the following process:
 ?><nobr><?php echo $display; ?></nobr>
 </table>
 <P>
-	<?php if($silent) { ?>
-		</div>
-	<?php } ?>
 	</div>
-        <?php
+	</div>
+    <?php
 if ($dir != '')
 {
     if (($action == '') || ($action == 'confirm')) {
         ?><input type="submit" name="BackupNow" value="Next" class="button_next"><?php
     }
     else if ($backupStatus) {
-        ?><input type="submit" name="Next" value="Next" class="button_next"><?php
+        ?><input type="submit" name="Next" value="Restore" class="button_next">
+        <input type="submit" name="Upgrade" value="Upgrade" class="button_next"><?php
     }
     else {
         ?><input type="submit" name="Next" value="Next" class="button_next"><?php
@@ -64,5 +63,4 @@ if ($dir != &#39;&#39;)
     ?>
     <input type="submit" name="Previous" value="Back" class="button_previous">
-	</div>
 </form>
 \ No newline at end of file
@@ -2,7 +2,6 @@
 	<p class="title">Current Installation</p>
 	<div id="step_content" class="step">
 		<p class="empty_space">
-        <?php // set_state(1); ?>
         <p class="empty_space"> If you have just updated
         your KnowledgeTree code base, you will need to complete the upgrade process in order to ensure your system is fully operational with the new version.
         <p class="empty_space">
 <form action="index.php?step_name=restore" method="post">
-    <p class="title">Confirm Restore</p>
+    <p class="title"><?php echo $title; ?></p>
     <?php
     if($errors || $warnings){
@@ -12,63 +12,65 @@
     <br/><br/>
     <div>
     <?php
-    if (!$availableBackups) {
-        ?>There don't seem to be any backups to restore from the <i>"<?php echo $dir;?>"</i> directory.<?php
-    }
-    else if (!$selected) {
+    if (!$restore) {
+        if (!$availableBackups) {
+            ?>There don't seem to be any backups to restore from the <i>"<?php echo $dir;?>"</i> directory.<?php
+        }
+        else if (!$selected) {
+        ?>
+        <P>
+        Select a backup to restore from the list below:
+        <P>
+            <form action="index.php?step_name=restore" method="post">
+    
+        <table border=1 cellpadding=1 cellspacing=1>
+                <tr bgcolor="darkgrey">
+                <td>Filename
+                <td>File Size
+                <td>Action
+    <?php
+        $i=0;
+        foreach($files as $file)
+        {
+            $color=((($i++)%2)==0)?'white':'lightgrey';
+    ?>
+            <tr bgcolor="<?php echo $color;?>">
+                <td><?php echo $file;?>
+                <td><?php echo filesize($dir . '/'.$file);?>
+                <td><input type="submit" name="RestoreSelect" value="restore">
+    <?php
+        }
+    ?>
+        </table>
+        <input type="hidden" name="file" value="<?php echo $file; ?>" />
+        </form>
+        <?php
+        }
+    else if ($dir != '') {
     ?>
     <P>
-    Select a backup to restore from the list below:
     <P>
-        <form action="index.php?step_name=restore" method="post">
-
-    <table border=1 cellpadding=1 cellspacing=1>
-            <tr bgcolor="darkgrey">
-            <td>Filename
-            <td>File Size
-            <td>Action
-<?php
-    $i=0;
-    foreach($files as $file)
-    {
-        $color=((($i++)%2)==0)?'white':'lightgrey';
-?>
-        <tr bgcolor="<?php echo $color;?>">
-            <td><?php echo $file;?>
-            <td><?php echo filesize($dir . '/'.$file);?>
-            <td><input type="submit" name="RestoreSelect" value="restore">
-<?php
-    }
-?>
-    </table>
-    <input type="hidden" name="file" value="<?php echo $file; ?>" />
-    </form>
+    Manually, you would do the following to restore the backup:
+    <P>
+    <table bgcolor="lightgrey">
+    <tr>
+    <td>
+    <nobr>cd "<?php echo $dir;?>"</nobr>
+    <br/>
     <?php
-    }
-else if ($dir != '') {
-?>
-<P>
-<P>
-Manually, you would do the following to restore the backup:
-<P>
-<table bgcolor="lightgrey">
-<tr>
-<td>
-<nobr>cd "<?php echo $dir;?>"</nobr>
-<br/>
-<?php
-    }
-    else
-    {
-    ?>
-     The mysql backup utility could not be found automatically. Either do a manual restore, or edit the config.ini and update the backup/mysql Directory entry.
-<P>
-You can continue to do the restore manually using the following command(s):
-<P>
-<table bgcolor="lightgrey">
-<tr>
-<td><?php
-    }
+        }
+        else
+        {
+        ?>
+         The mysql backup utility could not be found automatically. Either do a manual restore, or edit the config.ini and update the backup/mysql Directory entry.
+    <P>
+    You can continue to do the restore manually using the following command(s):
+    <P>
+    <table bgcolor="lightgrey">
+    <tr>
+    <td><?php
+        }
+    } // end not doing a restore, just preliminary steps
 ?>
 <nobr><?php echo $display;?></nobr>
 </td>
@@ -89,11 +91,12 @@ Press &lt;i&gt;Next&lt;/i&gt; to attempt the command(s) above.
 }
     ?>
     </div>
-
-
+    </div>
     <input type="submit" name="Previous" value="Back" class="button_previous">
     <?php if (($dir != '') && ($selected)) { ?>
+    <input type="submit" name="RunRestore" value="Next" class="button_next">
+    <?php }
+    else { ?>
     <input type="submit" name="Next" value="Next" class="button_next">
     <?php } ?>
-    </div>
 </form>
 \ No newline at end of file
@@ -666,115 +666,106 @@ class UpgradeUtil {
         return join(" ", $aSafeArgs);
     }
-    function create_restore_stmt($targetfile)
-{
-    $oKTConfig =& KTConfig::getSingleton();
-
-    $adminUser = $oKTConfig->get('db/dbAdminUser');
-    $adminPwd = $oKTConfig->get('db/dbAdminPass');
-    $dbHost = $oKTConfig->get('db/dbHost');
-    $dbName = $oKTConfig->get('db/dbName');
-    $dbPort = trim($oKTConfig->get('db/dbPort'));
-    if ($dbPort=='' || $dbPort=='default')$dbPort = get_cfg_var('mysql.default_port');
-    if (empty($dbPort)) $dbPort='3306';
-    $dbSocket = trim($oKTConfig->get('db/dbSocket'));
-    if (empty($dbSocket) || $dbSocket=='default') $dbSocket = get_cfg_var('mysql.default_socket');
-    if (empty($dbSocket)) $dbSocket='../tmp/mysql.sock';
-
-    $dir = $this->resolveMysqlDir();
-
-    $info['dir']=$dir;
-
-    $prefix='';
-    if (OS_UNIX)
-    {
-        $prefix .= "./";
-    }
-
-    if (@stat($dbSocket) !== false)
-    {
-        $mechanism="--socket=\"$dbSocket\"";
-    }
-    else
+    public function create_restore_stmt($targetfile)
     {
-        $mechanism="--port=\"$dbPort\"";
-    }
-
-    $tmpdir = $this->resolveTempDir();
-
-    $stmt = $prefix ."mysqladmin --user=\"$adminUser\" -p $mechanism drop  \"$dbName\"<br/>";
-    $stmt .= $prefix ."mysqladmin --user=\"$adminUser\" -p $mechanism create  \"$dbName\"<br/>";
-
-
-    $stmt .= $prefix ."mysql --user=\"$adminUser\" -p $mechanism \"$dbName\" < \"$targetfile\"\n";
-    $info['display']=$stmt;
-
-
-    $stmt = $prefix ."mysqladmin --user=\"$adminUser\" --force --password=\"$adminPwd\" $mechanism drop  \"$dbName\"\n";
-    $stmt .= $prefix ."mysqladmin --user=\"$adminUser\" --password=\"$adminPwd\" $mechanism create  \"$dbName\"\n";
-
-    $stmt .=  $prefix ."mysql --user=\"$adminUser\" --password=\"$adminPwd\" $mechanism \"$dbName\" < \"$targetfile\"";
-    $info['cmd']=$stmt;
-    return $info;
-}
-
-function resolveMysqlDir()
-{
-    // possibly detect existing installations:
-
-    if (OS_UNIX)
-    {
-        $dirs = array('/opt/mysql/bin','/usr/local/mysql/bin');
-        $mysqlname ='mysql';
-    }
-    else
-    {
-        $dirs = explode(';', $_SERVER['PATH']);
-        $dirs[] ='c:/Program Files/MySQL/MySQL Server 5.0/bin';
-        $dirs[] = 'c:/program files/ktdms/mysql/bin';
-        $mysqlname ='mysql.exe';
-    }
-
-    $oKTConfig =& KTConfig::getSingleton();
-    $mysqldir = $oKTConfig->get('backup/mysqlDirectory',$mysqldir);
-    $dirs[] = $mysqldir;
-
-    if (strpos(__FILE__,'knowledgeTree') !== false && strpos(__FILE__,'ktdms') != false)
-    {
-        $dirs [] = realpath(dirname($FILE) . '/../../mysql/bin');
+        $oKTConfig =& KTConfig::getSingleton();
+    
+        $adminUser = $oKTConfig->get('db/dbAdminUser');
+        $adminPwd = $oKTConfig->get('db/dbAdminPass');
+        $dbHost = $oKTConfig->get('db/dbHost');
+        $dbName = $oKTConfig->get('db/dbName');
+        $dbPort = trim($oKTConfig->get('db/dbPort'));
+        if ($dbPort=='' || $dbPort=='default')$dbPort = get_cfg_var('mysql.default_port');
+        if (empty($dbPort)) $dbPort='3306';
+        $dbSocket = trim($oKTConfig->get('db/dbSocket'));
+        if (empty($dbSocket) || $dbSocket=='default') $dbSocket = get_cfg_var('mysql.default_socket');
+        if (empty($dbSocket)) $dbSocket='../tmp/mysql.sock';
+    
+        $dir = $this->resolveMysqlDir();
+    
+        $info['dir']=$dir;
+    
+        $prefix='';
+        if (OS_UNIX) {
+            $prefix .= "./";
+        }
+    
+        if (@stat($dbSocket) !== false) {
+            $mechanism="--socket=\"$dbSocket\"";
+        }
+        else {
+            $mechanism="--port=\"$dbPort\"";
+        }
+    
+        $tmpdir = $this->resolveTempDir();
+    
+        $stmt = $prefix ."mysqladmin --user=\"$adminUser\" -p $mechanism drop  \"$dbName\"<br/>";
+        $stmt .= $prefix ."mysqladmin --user=\"$adminUser\" -p $mechanism create  \"$dbName\"<br/>";
+    
+    
+        $stmt .= $prefix ."mysql --user=\"$adminUser\" -p $mechanism \"$dbName\" < \"$targetfile\"\n";
+        $info['display']=$stmt;
+    
+    
+        $stmt = $prefix ."mysqladmin --user=\"$adminUser\" --force --password=\"$adminPwd\" $mechanism drop  \"$dbName\"\n";
+        $stmt .= $prefix ."mysqladmin --user=\"$adminUser\" --password=\"$adminPwd\" $mechanism create  \"$dbName\"\n";
+    
+        $stmt .=  $prefix ."mysql --user=\"$adminUser\" --password=\"$adminPwd\" $mechanism \"$dbName\" < \"$targetfile\"";
+        $info['cmd']=$stmt;
+        return $info;
     }
-
-    foreach($dirs as $dir)
+    
+    public function resolveMysqlDir()
     {
-        if (is_file($dir . '/' . $mysqlname))
+        // possibly detect existing installations:
+    
+        if (OS_UNIX) {
+            $dirs = array('/opt/mysql/bin','/usr/local/mysql/bin');
+            $mysqlname ='mysql';
+        }
+        else
         {
-            return $dir;
+            $dirs = explode(';', $_SERVER['PATH']);
+            $dirs[] ='c:/Program Files/MySQL/MySQL Server 5.0/bin';
+            $dirs[] = 'c:/program files/ktdms/mysql/bin';
+            $mysqlname ='mysql.exe';
         }
+    
+        $oKTConfig =& KTConfig::getSingleton();
+        $mysqldir = $oKTConfig->get('backup/mysqlDirectory',$mysqldir);
+        $dirs[] = $mysqldir;
+    
+        if (strpos(__FILE__,'knowledgeTree') !== false && strpos(__FILE__,'ktdms') != false) {
+            $dirs [] = realpath(dirname($FILE) . '/../../mysql/bin');
+        }
+    
+        foreach($dirs as $dir)
+        {
+            if (is_file($dir . '/' . $mysqlname))
+            {
+                return $dir;
+            }
+        }
+    
+        return '';
     }
-
-    return '';
-}
-
-function resolveTempDir()
-{
-
-    if (OS_UNIX)
-    {
-        $dir='/tmp/kt-db-backup';
-    }
-    else
-    {
-        $dir='c:/kt-db-backup';
-    }
-    $oKTConfig =& KTConfig::getSingleton();
-    $dir = $oKTConfig->get('backup/backupDirectory',$dir);
-
-    if (!is_dir($dir))
+    
+    public function resolveTempDir()
     {
-            mkdir($dir);
+        if (OS_UNIX) {
+            $dir='/tmp/kt-db-backup';
+        }
+        else {
+            $dir='c:/kt-db-backup';
+        }
+        $oKTConfig =& KTConfig::getSingleton();
+        $dir = $oKTConfig->get('backup/backupDirectory',$dir);
+    
+        if (!is_dir($dir)) {
+                mkdir($dir);
+        }
+        return $dir;
     }
-    return $dir;
-}
 }
 ?>
 \ No newline at end of file