lib/Doctrine/AuditLog.php

@@ -33,20 +33,16 @@ class Doctrine_AuditLog
 {
     protected $_options = array(
                             'className'     => '%CLASS%Version',
-                            'deleteTrigger' => '%TABLE%_ddtr',
-                            'updateTrigger' => '%TABLE%_dutr',
-                            'versionTable'  => '%TABLE%_dvt',
                             'versionColumn' => 'version',
+                            'generateFiles' => false,
+                            'table'         => null,
                             );
 
-    protected $_table;
-    
     protected $_auditTable;
 
-    public function __construct(Doctrine_Table $table)
+    public function __construct($options)
     {
-    	$this->_table = $table;
+        $this->_options = array_merge($this->_options, $options);
-
     }
     /**
      * __get
@@ -111,25 +107,7 @@ class Doctrine_AuditLog
         }
         return null;
     }
-    public function audit()
-    {
-        $conn = $this->_table->getConnection();
 
-        // begin new transaction
-        $conn->beginTransaction();
-        try {
-
-            // create the version table and the triggers
-            $this->createVersionTable();
-            $conn->execute($this->deleteTriggerSql());
-            $conn->execute($this->updateTriggerSql());
-
-            // commit structure changes
-            $conn->commit();
-        } catch(Doctrine_Connection_Exception $e) {
-            $conn->rollback();
-        }
-    }
     public function getVersion(Doctrine_Record $record, $version)
     {
         $className = str_replace('%CLASS%', $this->_table->getComponentName(), $this->_options['className']);
@@ -149,88 +127,31 @@ class Doctrine_AuditLog
                  ->where($where)
                  ->execute($values, Doctrine_HYDRATE::HYDRATE_ARRAY);
     }
-    public function createVersionTable()
+    public function buildDefinition(Doctrine_Table $table)
     {
-        $data = $this->_table->getExportableFormat(false);
+        $name = $table->getComponentName();
-        $conn = $this->_table->getConnection();
-        $data['tableName'] = str_replace('%TABLE%', $data['tableName'], $this->_options['versionTable']);
 
-        foreach ($data['columns'] as $name => $def) {
+        $className = $name . 'Version';
-            unset($data['columns'][$name]['autoinc']);
+        
-            unset($data['columns'][$name]['autoincrement']);
+        if (class_exists($className)) {
-            unset($data['columns'][$name]['sequence']);
+            return false;
-            unset($data['columns'][$name]['seq']);
         }
-        $data['columns'][$this->_options['versionColumn']]['primary'] = true;
 
-        $className  =  str_replace('%CLASS%', $this->_table->getComponentName(), $this->_options['className']);
+        $columns = $table->getColumns();
-        $definition = 'class ' . $className
-                    . ' extends Doctrine_Record { '
-                    . 'public function setTableDefinition() { '
-                    . '$this->hasColumns(' . var_export($data['columns'], true) . ');'
-                    . '$this->option(\'tableName\', \'' . $data['tableName'] . '\'); } }';
 
-        $this->_table->getRelationParser()->bind($className, array(
+        $id = $table->getIdentifier();
-                                  'local'   => $this->_table->getIdentifier(),
-                                  'foreign' => $this->_table->getIdentifier(),
-                                  'type'    => Doctrine_Relation::MANY));
 
+        $options = array('className' => $className);
 
-        $this->_table->addListener(new Doctrine_AuditLog_Listener($this));
+        $builder = new Doctrine_Import_Builder();
 
-        eval($definition);
+        $options['primary'][] = $this->_options['versionColumn'];
 
-        $data['options']['primary'][] = $this->_options['versionColumn'];
+        $def = $builder->buildDefinition($options, $columns);
 
-
+        if ( ! $this->_options['generateFiles']) {
-    	$this->_auditTable = $this->_table->getConnection()->getTable($className);
+            eval($def);
         }
-    /**
+        return true;
-     * deleteTriggerSql
-     *
-     * returns the sql needed for the delete trigger creation
-     */
-    public function deleteTriggerSql()
-    {
-    	$conn = $this->_table->getConnection();
-    	$columnNames = $this->_table->getColumnNames();
-    	$oldColumns  = array_map(array($this, 'formatOld'), $columnNames);
-        $sql  = 'CREATE TRIGGER '
-              . $conn->quoteIdentifier($this->_table->getTableName()) . '_ddt' . ' BEFORE DELETE ON '
-              . $conn->quoteIdentifier($this->_table->getTableName())
-              . ' BEGIN'
-              . ' INSERT INTO ' . $this->_table->getTableName() . '_dvt ('
-              . implode(', ', array_map(array($conn, 'quoteIdentifier'), $columnNames))
-              . ') VALUES ('
-              . implode(', ', array_map(array($conn, 'quoteIdentifier'), $oldColumns))
-              . ');'
-              . ' END;';
-        return $sql;
-    }
-    /**
-     * updateTriggerSql
-     *
-     * returns the sql needed for the update trigger creation
-     */
-    public function updateTriggerSql()
-    {
-    	$conn = $this->_table->getConnection();
-    	$columnNames = $this->_table->getColumnNames();
-    	$oldColumns  = array_map(array($this, 'formatOld'), $columnNames);
-        $sql  = 'CREATE TRIGGER '
-              . $conn->quoteIdentifier($this->_table->getTableName()) . '_dut' . ' UPDATE ON '
-              . $conn->quoteIdentifier($this->_table->getTableName())
-              . ' BEGIN'
-              . ' INSERT INTO ' . $this->_table->getTableName() . '_dvt ('
-              . implode(', ', array_map(array($conn, 'quoteIdentifier'), $columnNames))
-              . ') VALUES ('
-              . implode(', ', array_map(array($conn, 'quoteIdentifier'), $oldColumns))
-              . ');'
-              . ' END;';
-        return $sql;
-    }
-    public function formatOld($column)
-    {
-        return 'old.' . $column;
     }
 }

lib/Doctrine/AuditLog/Listener.php

@@ -18,6 +18,7 @@
  * and is licensed under the LGPL. For more information, see
  * <http://www.phpdoctrine.com>.
  */
+Doctrine::autoload('Doctrine_Record_Listener');
 /**
  * Doctrine_AuditLog_Listener
  *
@@ -29,7 +30,7 @@
  * @version     $Revision$
  * @author      Konsta Vesterinen <kvesteri@cc.hut.fi>
  */
-class Doctrine_AuditLog_Listener extends Doctrine_EventListener
+class Doctrine_AuditLog_Listener extends Doctrine_Record_Listener
 {
     
     protected $_auditLog;