Migration_Actions

Complexity

Total Complexity

44

Size/Duplication

Total Lines	215
Duplicated Lines	0 %

Coupling/Cohesion

Components	2
Dependencies	2

Test Coverage

Coverage

65.25%

Importance

Changes

0

12 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	4	1
A	parse()	0	16	4
A	template()	0	9	2
A	add_columns()	0	8	2
A	remove_columns()	0	18	3
A	create_table()	0	8	2
B	drop_table()	0	30	5
A	rename_table()	0	5	1
A	rename_column()	0	16	4
A	change_column()	0	18	3
D	field_params_to_string()	0	29	9
C	guess_column_type()	0	25	8

<?php defined('SYSPATH') OR die('No direct script access.');

class Migration_Actions
{
	public $up = array();
	public $down = array();
	private $driver = NULL;

	static private $patterns = array(
		'add_columns'    => '/^add_(.+)_to_(.+)$/',
		'remove_columns' => '/^remove_(.+)_from_(.+)$/',
		'create_table'   => '/^create_table_(.+)$/',
		'drop_table'     => '/^drop_table_(.+)$/',
		'rename_table'   => '/^rename_table_(.+)_to_(.+)$/',
		'rename_column'  => '/^rename_(.+)_in_(.+)$/',
		'change_column'  => '/^change_(.+)_in_(.+)$/',
	);

	public function __construct(Migration_Driver $driver)
	{
		$this->driver = $driver;
	}

	public function parse($string)
	{
		foreach (explode('_also_', $string) as $part)
		{
			foreach (self::$patterns as $method => $pattern)
			{
				if (preg_match($pattern, $part, $matches))
				{
					call_user_func_array(array($this, $method), array_slice($matches, 1));
					break;
				}
			}
		}

		return $this;
	}

	public function template($template)
	{
		if ( ! is_file($template))
			throw new Migration_Exception("Template file :template does not exist", array($template));

		list($this->up, $this->down) = explode('--- DOWN ---', file_get_contents($template));

		return $this;
	}

	public function add_columns($columns, $table)
	{
		foreach (explode('_and_', $columns) as $column)
		{
			$this->up[] = "\$this->add_column('$table', '$column', '".self::guess_column_type($column)."');";
			$this->down[] = "\$this->remove_column('$table', '$column');";
		}
	}

	public function remove_columns($columns, $table)
	{
		foreach (explode('_and_', $columns) as $column)
		{
			$this->up[] = "\$this->remove_column('$table', '$column');";

			try
			{
				$field_params = $this->driver->column($column)->load($table);

				$this->down[] = "\$this->add_column('$table', '$column', ".self::field_params_to_string($field_params).");";
			}
			catch (Migration_Exception $e)
			{
				$this->down[] = "\$this->add_column('$table', '$column', 'string');";
			}
		}
	}

	public function create_table($tables)
	{
		foreach (explode('_and_', $tables) as $table_name)
		{
			$this->up[] = "\$this->create_table('{$table_name}', array( ));";
			$this->down[] = "\$this->drop_table('{$table_name}');";
		}
	}

	public function drop_table($tables)
	{
		foreach (explode('_and_', $tables) as $table_name)
		{
			$this->up[] = "\$this->drop_table('{$table_name}');";

			try
			{
				$table = $this->driver->table($table_name)->load();
				$fields = array();
				$options = array();

				foreach ($table->columns as $name => $field_params)
				{
					$fields[] = "\n\t\t\t'$name' => ".self::field_params_to_string($field_params);
				}

				foreach($table->options as $name => $option)
				{
					$options[] = "'$name' => '$option'";
				}

				$this->down[] = "\$this->create_table('{$table->name()}', array( ".join(',', $fields)." \n\t\t), array( ".join(',', $options)." )); ";
			}
			catch (Migration_Exception $e)
			{
				$this->down[] = "\$this->create_table('{$table_name}', array( ));";
			}
		}
	}

	public function rename_table($old_name, $new_name)
	{
		$this->up[] = "\$this->rename_table('$old_name', '$new_name');";
		$this->down[] = "\$this->rename_table('$new_name', '$old_name');";
	}

	public function rename_column($columns, $table)
	{
		foreach (explode('_and_', $columns) as $column)
		{
			if (preg_match('/^(.+)_to_(.+)$/', $column, $matches))
			{
				if (count($matches) === 3)
				{
					$old_name = $matches[1];
					$new_name = $matches[2];
					$this->up[] = "\$this->rename_column('$table', '$old_name', '$new_name');";
					$this->down[] = "\$this->rename_column('$table', '$new_name', '$old_name');";
				}
			}
		}
	}

	public function change_column($columns, $table)
	{
		foreach (explode('_and_', $columns) as $column)
		{
			$this->up[] = "\$this->change_column('$table', '$column', '".self::guess_column_type($column)."');";

			try
			{
				$field_params = $this->driver->column($column)->load($table);

				$this->down[] = "\$this->change_column('$table', '$column', ".self::field_params_to_string($field_params).");";
			}
			catch (Migration_Exception $e)
			{
				$this->down[] = "\$this->change_column('$table', '$column', 'string');";
			}
		}
	}

	static public function field_params_to_string($column)
	{
		$options = array();

		foreach ($column->params() as $option_name => $option)
		{
			if ($option) {
				if (is_array($option))
				{
					foreach ($option as & $param)
					{
						$param = "'$param'";
					}
					$option = 'array('.join(', ', $option).')';
				}
				elseif (is_bool($option))
				{
					$option = $option ? 'TRUE' : 'FALSE';
				}
				elseif (is_string($option))
				{
					$option = "'$option'";
				}
				$options[] = (is_numeric($option_name) ? '' : '"' . $option_name . '" => ') . $option;
			}
		}

		return "array( ".join(', ', $options).')';
	}


	static public function guess_column_type($column)
	{
		if (preg_match('/_id$/', $column))
			return 'integer';

		if (preg_match('/^(id|position)$/', $column))
			return 'integer';

		if (preg_match('/(_count|_width|_height|_x|_y)$/', $column))
			return 'integer';

		if (preg_match('/_at$/', $column))
			return 'datetime';

		if (preg_match('/^is_/', $column))
			return 'boolean';

		if (preg_match('/_on$/', $column))
			return 'date';

			if (preg_match('/^(description|text)$/', $column))
				return 'text';

		return 'string';
	}
}