From fca8ccdf774af5cece13395e5a799e98fa177706 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Micha=C5=82=20Matyas?= <michal@6irc.net>
Date: Thu, 30 Jun 2011 14:46:58 +0200
Subject: [PATCH] created possibility of writing generated code directly into
 migration

---
 classes/minion/task/db/generate.php        |  4 +++-
 views/minion/task/db/generate/template.php | 10 ++++++++++
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/classes/minion/task/db/generate.php b/classes/minion/task/db/generate.php
index e9bed69..c2fe360 100644
--- a/classes/minion/task/db/generate.php
+++ b/classes/minion/task/db/generate.php
@@ -58,7 +58,7 @@ class Minion_Task_Db_Generate extends Minion_Task
 
 	}
 	
-	public function generate($config)
+	public function generate($config, $up = null, $down = null)
 	{
 		$defaults = array(
 			'location'    => APPPATH,
@@ -89,6 +89,8 @@ class Minion_Task_Db_Generate extends Minion_Task
 		$data = Kohana::FILE_SECURITY.View::factory('minion/task/db/generate/template')
 			->set('class', $class)
 			->set('description', $description)
+			->set('up', $up)
+			->set('down', $down)
 			->render();
 
 		if ( ! is_dir(dirname($file)))
diff --git a/views/minion/task/db/generate/template.php b/views/minion/task/db/generate/template.php
index 549ef4d..d8701ca 100644
--- a/views/minion/task/db/generate/template.php
+++ b/views/minion/task/db/generate/template.php
@@ -12,7 +12,12 @@ class <?php echo $class; ?> extends Minion_Migration_Base {
 	 */
 	public function up(Kohana_Database $db)
 	{
+<?php if(!empty($up)): ?>
+<?php echo $up; ?>
+
+<?php else: ?>
 		// $db->query(NULL, 'CREATE TABLE ... ');
+<?php endif; ?>
 	}
 
 	/**
@@ -22,6 +27,11 @@ class <?php echo $class; ?> extends Minion_Migration_Base {
 	 */
 	public function down(Kohana_Database $db)
 	{
+<?php if(!empty($down)): ?>
+<?php echo $down; ?>
+
+<?php else: ?>
 		// $db->query(NULL, 'DROP TABLE ... ');
+<?php endif; ?>
 	}
 }