XmlToSqlTool.inc.php

<?php
 
import('lib.pkp.classes.db.DBDataXMLParser');
 
class XmlToSqlTool extends CommandLineTool {
 
   var $type;
 
   var $command;
 
   var $inputFile;
 
   var $outputFile;
 
   function __construct($argv = array()) {
      parent::__construct($argv);
 
      if (isset($this->argv[0]) && in_array($this->argv[0], array('-schema', '-data'))) {
         $this->type = substr($this->argv[0], 1);
         $argOffset = 1;
      } else {
         $this->type = 'schema';
         $argOffset = 0;
      }
 
      if (!isset($this->argv[$argOffset]) || !in_array($this->argv[$argOffset], array('print', 'save', 'print_upgrade', 'save_upgrade', 'execute'))) {
         $this->usage();
         exit(1);
      }
 
      $this->command = $this->argv[$argOffset];
 
      $file = isset($this->argv[$argOffset+1]) ? $this->argv[$argOffset+1] : DATABASE_XML_FILE;
 
      if (!file_exists($file) && !file_exists(($file2 = PWD . '/' . $file))) {
         printf("Input file \"%s\" does not exist!\n", $file);
         exit(1);
      }
 
      $this->inputFile = isset($file2) ? $file2 : $file;
 
      $this->outputFile = isset($this->argv[$argOffset+2]) ? PWD . '/' . $this->argv[$argOffset+2] : null;
      if (in_array($this->command, array('save', 'save_upgrade')) && ($this->outputFile == null || (file_exists($this->outputFile) && (is_dir($this->outputFile) || !is_writeable($this->outputFile))) || !is_writable(dirname($this->outputFile)))) {
         printf("Invalid output file \"%s\"!\n", $this->outputFile);
         exit(1);
      }
   }
 
   function usage() {
      echo "Script to convert and execute XML-formatted database schema and data files\n"
         . "Usage: {$this->scriptName} [-data|-schema] command [input_file] [output_file]\n"
         . "Supported commands:\n"
         . "    print - print SQL statements\n"
         . "    save - save SQL statements to output_file\n"
         . "    print_upgrade - print upgrade SQL statements for current database\n"
         . "    save_upgrade - save upgrade SQL statements to output_file\n"
         . "    execute - execute SQL statements on current database\n";
   }
 
   function execute() {
      require_once('./lib/pkp/lib/vendor/adodb/adodb-php/adodb-xmlschema.inc.php');
 
      if (in_array($this->command, array('print', 'save'))) {
         // Don't connect to actual database (so parser won't build upgrade XML)
         $conn = new DBConnection(
            Config::getVar('database', 'driver'),
            null,
            null,
            null,
            null,
            true,
            Config::getVar('i18n', 'connection_charset')
         );
         $dbconn = $conn->getDBConn();
 
      } else {
         // Create or upgrade existing database
         $dbconn =& DBConnection::getConn();
      }
 
      $schema = new adoSchema($dbconn);
      $dict =& $schema->dict;
 
      if ($this->type == 'schema') {
         // Parse XML schema files
         $sql = $schema->parseSchema($this->inputFile);
 
         switch ($this->command) {
            case 'execute':
               $schema->ExecuteSchema();
               break;
            case 'save':
            case 'save_upgrade':
               $schema->SaveSQL($this->outputFile);
               break;
            case 'print':
            case 'print_upgrade':
            default:
               echo @$schema->PrintSQL('TEXT') . "\n";
               break;
         }
 
      } else if ($this->type == 'data') {
         // Parse XML data files
         $dataXMLParser = new DBDataXMLParser();
         $dataXMLParser->setDBConn($dbconn);
         $sql = $dataXMLParser->parseData($this->inputFile);
 
         switch ($this->command) {
            case 'execute':
               $schema->addSQL($sql);
               $schema->ExecuteSchema();
               break;
            case 'save':
            case 'save_upgrade':
               $schema->addSQL($sql);
               $schema->SaveSQL($this->outputFile);
               break;
            case 'print':
            case 'print_upgrade':
            default:
               $schema->addSQL($sql);
               echo @$schema->PrintSQL('TEXT') . "\n";
               break;
         }
      }
   }
}