1 files changed, 70 insertions, 0 deletions
diff --git a/library/Director/Db/Migration.php b/library/Director/Db/Migration.php
new file mode 100644
index 0000000..5685121
--- /dev/null
+++ b/library/Director/Db/Migration.php
@@ -0,0 +1,70 @@
+<?php
+
+namespace Icinga\Module\Director\Db;
+
+use Exception;
+use Icinga\Module\Director\Data\Db\DbConnection;
+use RuntimeException;
+
+class Migration
+{
+    /**
+     * @var string
+     */
+    protected $sql;
+
+    /**
+     * @var int
+     */
+    protected $version;
+
+    public function __construct($version, $sql)
+    {
+        $this->version = $version;
+        $this->sql     = $sql;
+    }
+
+    /**
+     * @param DbConnection $connection
+     * @return $this
+     */
+    public function apply(DbConnection $connection)
+    {
+        /** @var \Zend_Db_Adapter_Pdo_Abstract $db */
+        $db = $connection->getDbAdapter();
+
+        // TODO: this is fragile and depends on accordingly written schema files:
+        $queries = preg_split(
+            '/[\n\s\t]*\;[\n\s\t]+/s',
+            $this->sql,
+            -1,
+            PREG_SPLIT_NO_EMPTY
+        );
+
+        if (empty($queries)) {
+            throw new RuntimeException(sprintf(
+                'Migration %d has no queries',
+                $this->version
+            ));
+        }
+
+        try {
+            foreach ($queries as $query) {
+                if (preg_match('/^(?:OPTIMIZE|EXECUTE) /i', $query)) {
+                    $db->query($query);
+                } else {
+                    $db->exec($query);
+                }
+            }
+        } catch (Exception $e) {
+            throw new RuntimeException(sprintf(
+                'Migration %d failed (%s) while running %s',
+                $this->version,
+                $e->getMessage(),
+                $query
+            ));
+        }
+
+        return $this;
+    }
+}