Table: added removing methods

Author: janpecha

src/Table.php

@@ -74,6 +74,17 @@ public function setOption($name, $value)
 		}
 
 
+		/**
+		 * @param  string $name
+		 * @return self
+		 */
+		public function removeOption($name)
+		{
+			unset($this->options[$name]);
+			return $this;
+		}
+
+
 		/**
 		 * @return array<string, string>
 		 */
@@ -110,6 +121,24 @@ public function addColumn($name, $type = NULL, array $parameters = NULL, array $
 		}
 
 
+		/**
+		 * @param  string|Column $name
+		 * @return void
+		 */
+		public function removeColumn($name)
+		{
+			if ($name instanceof Column) {
+				$name = $name->getName();
+			}
+
+			if (!isset($this->columns[$name])) {
+				throw new MissingException("Column '$name' in table '{$this->getName()}' not exists.");
+			}
+
+			unset($this->columns[$name]);
+		}
+
+
 		/**
 		 * @param  string $name
 		 * @return Column|NULL
@@ -159,6 +188,24 @@ public function addIndex($name, $columns = [], $type = Index::TYPE_INDEX)
 		}
 
 
+		/**
+		 * @param  string|Index $name
+		 * @return void
+		 */
+		public function removeIndex($name)
+		{
+			if ($name instanceof Index) {
+				$name = $name->getName();
+			}
+
+			if (!isset($this->indexes[$name])) {
+				throw new MissingException("Index '$name' in table '{$this->getName()}' not exists.");
+			}
+
+			unset($this->indexes[$name]);
+		}
+
+
 		/**
 		 * @param  string $name
 		 * @return Index|NULL
@@ -209,6 +256,24 @@ public function addForeignKey($name, $columns = [], $targetTable = NULL, $target
 		}
 
 
+		/**
+		 * @param  string|ForeignKey $name
+		 * @return void
+		 */
+		public function removeForeignKey($name)
+		{
+			if ($name instanceof ForeignKey) {
+				$name = $name->getName();
+			}
+
+			if (!isset($this->foreignKeys[$name])) {
+				throw new MissingException("Foreign key '$name' in table '{$this->getName()}' not exists.");
+			}
+
+			unset($this->foreignKeys[$name]);
+		}
+
+
 		/**
 		 * @param  string $name
 		 * @return ForeignKey|NULL

src/exceptions.php

@@ -18,6 +18,11 @@ class EmptyException extends Exception
 	}
 
 
+	class MissingException extends Exception
+	{
+	}
+
+
 	class OutOfRangeException extends Exception
 	{
 	}

tests/SqlSchema/Table.phpt

@@ -84,6 +84,22 @@ test(function () {
 });
 
 
+test(function () {
+	Assert::exception(function () {
+		$table = new Table('book');
+		$table->removeColumn('id');
+
+	}, CzProject\SqlSchema\MissingException::class, "Column 'id' in table 'book' not exists.");
+
+	$table = new Table('book');
+	$column = $table->addColumn('id', 'INT');
+	Assert::same(1, count($table->getColumns()));
+
+	$table->removeColumn($column);
+	Assert::same(0, count($table->getColumns()));
+});
+
+
 test(function () {
 	Assert::exception(function () {
 		$table = new Table('book');
@@ -101,6 +117,22 @@ test(function () {
 });
 
 
+test(function () {
+	Assert::exception(function () {
+		$table = new Table('book');
+		$table->removeIndex('id');
+
+	}, CzProject\SqlSchema\MissingException::class, "Index 'id' in table 'book' not exists.");
+
+	$table = new Table('book');
+	$index = $table->addIndex('id', Index::TYPE_INDEX);
+	Assert::same(1, count($table->getIndexes()));
+
+	$table->removeIndex($index);
+	Assert::same(0, count($table->getIndexes()));
+});
+
+
 test(function () {
 	Assert::exception(function () {
 		$table = new Table('book');
@@ -116,3 +148,19 @@ test(function () {
 
 	}, CzProject\SqlSchema\DuplicateException::class, "Foreign key 'author_id' in table 'book' already exists.");
 });
+
+
+test(function () {
+	Assert::exception(function () {
+		$table = new Table('book');
+		$table->removeForeignKey('author_id');
+
+	}, CzProject\SqlSchema\MissingException::class, "Foreign key 'author_id' in table 'book' not exists.");
+
+	$table = new Table('book');
+	$foreignKey = $table->addForeignKey('author_id', [], 'author');
+	Assert::same(1, count($table->getForeignKeys()));
+
+	$table->removeForeignKey($foreignKey);
+	Assert::same(0, count($table->getForeignKeys()));
+});