MigrationLib to help with the migration procedure

2026-06-24 15:39:28 +00:00 · 2016-07-04 15:36:53 +02:00
parent 4902f41a3f
commit ba0e2bb7ff
2 changed files with 190 additions and 83 deletions
@@ -0,0 +1,146 @@
+<?php
+
+if (! defined("BASEPATH")) exit("No direct script access allowed");
+
+class MigrationLib extends CI_Migration
+{
+	private $MSG_PREFIX = "[-]";
+	private $INFO_PREFIX = "[I]";
+	private $ERROR_PREFIX = "[E]";
+	private $SEPARATOR = "------------------------------";
+	
+	private function printMessage($message)
+	{
+		printf("%s %s" . PHP_EOL, $this->MSG_PREFIX, $message);
+	}
+	
+	private function printInfo($info)
+	{
+		printf("%s %s" . PHP_EOL, $this->INFO_PREFIX, $info);
+	}
+	
+	private function printError($error)
+	{
+		printf("%s %s" . PHP_EOL, $this->ERROR_PREFIX, $error);
+	}
+	
+	protected function startUP()
+	{
+		$this->printInfo(sprintf("%s Start method up of class %s %s", $this->SEPARATOR, get_called_class(), $this->SEPARATOR));
+	}
+	
+	protected function endUP()
+	{
+		$this->printInfo(sprintf("%s End method up of class %s %s", $this->SEPARATOR, get_called_class(), $this->SEPARATOR));
+	}
+	
+	protected function columnExists($column, $schema, $table)
+	{
+		$query = sprintf("SELECT COUNT(%s) FROM %s.%s", $column, $schema, $table);
+		
+		if (! @$this->db->simple_query($query))
+		{
+			return false;
+		}
+		
+		return true;
+	}
+	
+    protected function addColumn($schema, $table, $column, $type)
+	{
+		if (!$this->columnExists($column, $schema, $table))
+		{
+			$query = sprintf("ALTER TABLE %s.%s ADD COLUMN %s %s", $schema, $table, $column, $type);
+  			if (@$this->db->simple_query($query))
+			{
+				$this->printMessage(sprintf("Column %s.%s.%s of type %s added", $schema, $table, $column, $type));
+			}
+			else
+			{
+				$this->printError(sprintf("Error while adding column %s.%s.%s of type %s", $schema, $table, $column, $type));
+			}
+		}
+		else
+		{
+			$this->printInfo(sprintf("Column %s.%s.%s already exists", $schema, $table, $column));
+		}
+	}
+	
+	protected function grantTable($permission, $schema, $table, $user)
+	{
+		$stringPermission = null;
+
+		if (is_array($permission))
+		{
+			if (count($permission) > 0)
+			{
+				$stringPermission = "";
+				for ($i = 0; $i < count($permission); $i++)
+				{
+					$stringPermission .= $permission[$i];
+					if ($i != count($permission) - 1)
+					{
+						$stringPermission .= ", ";
+					}
+				}
+				$query = sprintf("GRANT %s ON TABLE %s.%s TO %s", $stringPermission, $schema, $table, $user);
+			}
+		}
+		else
+		{
+			$query = sprintf("GRANT %s ON TABLE %s.%s TO %s", $permission, $schema, $table, $user);
+		}
+
+		if (@$this->db->simple_query($query))
+		{
+			$this->printMessage(
+				sprintf("Granted permissions %s on table %s.%s to user %s",
+						is_null($stringPermission) ? $permission : $stringPermission,
+						$schema,
+						$table,
+						$user
+			));
+		}
+		else
+		{
+			$this->printError(
+				sprintf("Granting permissions %s on table %s.%s to user %s",
+						is_null($stringPermission) ? $permission : $stringPermission,
+						$schema,
+						$table,
+						$user
+			));
+		}
+	}
+	
+	protected function createTable($schema, $table, $fields)
+	{
+		if (! $this->db->table_exists($schema . "." . $table))
+		{
+			$query = sprintf("CREATE TABLE %s.%s (%s)", $schema, $table, $fields);
+			
+			if (@$this->db->simple_query($query))
+			{
+				$this->printMessage(sprintf("Table %s.%s created", $schema, $table));
+			}
+			else
+			{
+				$this->printError(sprintf("Creating table %s.%s", $schema, $table));
+			}
+		}
+		else
+		{
+			$this->printInfo(sprintf("Table %s.%s already exists", $schema, $table));
+		}
+	}
+	
+	protected function execQuery($query)
+	{
+		if (! @$this->db->simple_query($query))
+		{
+			$this->printError($this->db->error());
+		}
+		
+		$this->printInfo("Query correctly executed");
+	}
+}