From 2bdb6860c9121ffedf287a9e87ffc7305b1b8d08 Mon Sep 17 00:00:00 2001
From: zYne <zYne@625475ce-881a-0410-a577-b389adb331d8>
Date: Wed, 10 Jan 2007 19:05:58 +0000
Subject: [PATCH] Updated sqlite export driver

---
 lib/Doctrine/Export/Sqlite.php | 46 +++++++++++++++++++++++++++++++++-
 1 file changed, 45 insertions(+), 1 deletion(-)

diff --git a/lib/Doctrine/Export/Sqlite.php b/lib/Doctrine/Export/Sqlite.php
index 847ca4e48..75483650d 100644
--- a/lib/Doctrine/Export/Sqlite.php
+++ b/lib/Doctrine/Export/Sqlite.php
@@ -85,7 +85,51 @@ class Doctrine_Export_Sqlite extends Doctrine_Export
             }
             $fields[] = $fieldString;
         }
-        $query .= ' ('.implode(', ', $fields) . ')';
+        $query .= ' (' . implode(', ', $fields) . ')';
+
         return $this->conn->exec($query);
     }
+    /**
+     * create sequence
+     *
+     * @param string    $seqName        name of the sequence to be created
+     * @param string    $start          start value of the sequence; default is 1
+     * @return boolean
+     */
+    public function createSequence($seqName, $start = 1)
+    {
+        $sequenceName   = $this->conn->quoteIdentifier($this->conn->getSequenceName($seqName), true);
+        $seqcolName     = $this->conn->quoteIdentifier($this->conn->getAttribute(Doctrine::ATTR_SEQCOL_NAME), true);
+        $query          = 'CREATE TABLE ' . $sequenceName . ' (' . $seqcolName . ' INTEGER PRIMARY KEY DEFAULT 0 NOT NULL)';
+        
+        $this->conn->exec($query);
+
+        if ($start == 1) {
+            return true;
+        }
+
+        try {
+            $this->conn->exec('INSERT INTO ' . $sequenceName . ' (' . $seqcolName . ') VALUES (' . ($start-1) . ')');
+            return true;
+        } catch(Doctrine_Connection_Exception $e) {
+            // Handle error
+            try {
+                $result = $db->exec('DROP TABLE ' . $sequenceName);
+            } catch(Doctrine_Connection_Exception $e) {
+                throw new Doctrine_Export_Exception('could not drop inconsistent sequence table');
+            }
+        }
+        throw new Doctrine_Export_Exception('could not create sequence table');
+    }
+    /**
+     * drop existing sequence
+     *
+     * @param string    $seq_name     name of the sequence to be dropped
+     * @return boolean
+     */
+    public function dropSequence($seq_name)
+    {
+        $sequenceName   = $this->conn->quoteIdentifier($this->conn->getSequenceName($seqName), true);
+        return $this->conn->exec('DROP TABLE ' . $sequenceName);
+    }
 }