Add modifyColumn function

Add function to Queries and DB classes to allow modification of column types
NamelessMC · Aug 18, 2016 · 902f355 · 902f355
1 parent 315c3d4
commit 902f355
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/core/classes/DB.php b/core/classes/DB.php
@@ -244,6 +244,15 @@ public function alterTable($name, $column, $attributes) {
 		return false;
 	}
 
+	public function modifyColumn($name, $column, $attributes) {
+		$name = $this->_prefix . $name;
+		$sql = "ALTER TABLE `{$name}` MODIFY {$column} {$attributes}";
+		if(!$this->createQuery($sql)->error()) {
+			return $this;
+		}
+		return false;
+	}
+
 	public function removeColumn($name, $column) {
 		$name = $this->_prefix . $name;
 		$sql = "ALTER TABLE `{$name}` DROP COLUMN {$column}";

diff --git a/core/classes/Queries.php b/core/classes/Queries.php
@@ -88,6 +88,12 @@ public function alterTable($table, $column, $attributes){
 		}
 	}
 
+	public function modifyColumn($table, $column, $attributes){
+		if(!$this->_db->modifyColumn($table, $column, $attributes)) {
+			throw new Exception('There was a problem modifying the column ' . htmlspecialchars($column));
+		}
+	}
+
 	public function removeColumn($table, $column){
 		if($this->_db->removeColumn($table, $column)){
 			throw new Exception('Unable to drop column.');