classes/db/SQLParser.inc.php

<?php

// $Id: SQLParser.inc.php,v 1.8 2008/07/01 01:16:09 asmecher Exp $


class SQLParser {

   var $driver;

   var $dataSource;

   var $debug;

   var $errorMsg;

   var $commentDelim;

   var $statementDelim;

   function SQLParser($driver, &$dataSource, $debug = false) {
      $this->driver = $driver;
      $this->dataSource = &$dataSource;
      $this->debug = $debug;
      $this->errorMsg = array();
      $this->commentDelim = '(\-\-|#)';
      $this->statementDelim = ';';
   }

   function executeFile($file, $failOnError = true) {
      if (!file_exists($file) || !is_readable($file)) {
         array_push($this->errorMsg, "$file does not exist or is not readble!");
         return false;
      }

      // Read file and break up into SQL statements
      $sql = join('', file($file));
      $this->stripComments($sql);
      $statements = &$this->parseStatements($sql);

      // Execute each SQL statement
      for ($i=0, $count=count($statements); $i < $count; $i++) {
         if ($this->debug) {
            echo 'Executing: ', $statements[$i], "\n\n";
         }

         $this->dataSource->execute($statements[$i]);

         if ($this->dataSource->errorNo() != 0) {
            // An error occurred executing the statement
            array_push($this->errorMsg, $this->dataSource->errorMsg());

            if ($failOnError) {
               // Abort if fail on error is enabled
               return false;
            } else {
               $error = true;
            }
         }
      }

      return isset($error) ? false : true;
   }

   function stripComments(&$sql) {
      $sql = trim(String::regexp_replace(sprintf('/^\s*%s(.*)$/m', $this->commentDelim), '', $sql));
   }

   function &parseStatements(&$sql) {
      $statements = array();
      $statementsTmp = explode($this->statementDelim, $sql);

      $currentStatement = '';
      $numSingleQuotes = $numEscapedSingleQuotes = 0;

      // This method for parsing the SQL statements was adapted from one used in phpBB (http://www.phpbb.com/)
      for ($i=0, $count=count($statementsTmp); $i < $count; $i++) {
         // Get total number of single quotes in string
         $numSingleQuotes += String::substr_count($statementsTmp[$i], "'");

         // Get number of escaped single quotes
         $numEscapedSingleQuotes += String::regexp_match_all("/(?<!\\\\)(\\\\\\\\)*\\\\'/", $statementsTmp[$i], $matches);

         $currentStatement .= $statementsTmp[$i];

         if (($numSingleQuotes - $numEscapedSingleQuotes) % 2 == 0) {
            // Even number of unescaped single quotes, so statement must be complete
            if (trim($currentStatement) !== '') {
               array_push($statements, trim($currentStatement));
            }
            $currentStatement = '';
            $numSingleQuotes = $numEscapedSingleQuotes = 0;

         } else {
            // The statement is not complete, the delimiter must be inside the statement
            $currentStatement .= $this->statementDelim;
         }
      }

      return $statements;
   }

   function getErrorMsg() {
      return count($this->errorMsg) == 0 ? null : array_pop($this->errorMsg);
   }

}

?>

---