1: 2: 3: 4: 5: 6: 7: 8: 9: 10: 11: 12: 13: 14: 15: 16: 17: 18: 19: 20: 21: 22: 23: 24: 25: 26: 27: 28: 29: 30: 31: 32: 33: 34: 35: 36: 37: 38: 39: 40: 41: 42: 43: 44: 45: 46: 47: 48: 49: 50: 51: 52: 53: 54: 55: 56: 57: 58: 59: 60: 61: 62: 63: 64: 65: 66: 67: 68: 69: 70: 71: 72: 73: 74: 75: 76: 77: 78: 79: 80: 81: 82: 83: 84: 85: 86: 87: 88: 89: 90: 91: 92: 93: 94: 95: 96: 97: 98: 99: 100: 101: 102: 103: 104: 105: 106: 107: 108: 109: 110: 111: 112: 113: 114: 115: 116: 117: 118: 119: 120: 121: 122: 123: 124: 125: 126: 127: 128: 129: 130: 131: 132: 133: 134: 135: 136: 137: 138: 139: 140: 141: 142: 143: 144: 145: 146: 147: 148: 149: 150: 151: 152: 153: 154: 155: 156: 157: 158: 159: 160: 161: 162:
<?php
namespace CzProject\SqlGenerator\Statements;
use CzProject\SqlGenerator\DuplicateException;
use CzProject\SqlGenerator\Drivers;
use CzProject\SqlGenerator\Helpers;
use CzProject\SqlGenerator\IDriver;
use CzProject\SqlGenerator\IStatement;
class ColumnDefinition implements IStatement
{
private $name;
private $type;
private $parameters = array();
private $options = array();
private $nullable = FALSE;
private $defaultValue;
private $autoIncrement = FALSE;
private $comment;
public function __construct($name, $type, array $parameters = NULL, array $options = array())
{
$this->name = $name;
$this->type = $type;
$this->parameters = ($parameters !== NULL) ? $parameters : array();
$this->options = $options;
}
public function setNullable($nullable = TRUE)
{
$this->nullable = $nullable;
return $this;
}
public function setDefaultValue($defaultValue)
{
$this->defaultValue = $defaultValue;
return $this;
}
public function setAutoIncrement($autoIncrement = TRUE)
{
$this->autoIncrement = $autoIncrement;
return $this;
}
public function setComment($comment)
{
$this->comment = $comment;
return $this;
}
public function toSql(IDriver $driver)
{
$output = $driver->escapeIdentifier($this->name) . ' ' . $this->type;
if (!empty($this->parameters)) {
$parameters = $this->parameters;
array_walk($parameters, function (&$value) use ($driver) {
$value = Helpers::formatValue($value, $driver);
});
$output .= '(' . implode(', ', $parameters) . ')';
}
$options = $this->options;
$specialOptions = array();
if ($driver instanceof Drivers\MysqlDriver) {
$specialOptions = array(
'CHARACTER SET',
'COLLATE',
);
}
foreach ($specialOptions as $option) {
if (isset($options[$option])) {
$output .= ' ' . self::formatOption($option, $options[$option]);
unset($options[$option]);
}
}
foreach ($options as $option => $value) {
$output .= ' ' . self::formatOption($option, $value);
}
$output .= ' ' . ($this->nullable ? 'NULL' : 'NOT NULL');
if (isset($this->defaultValue)) {
$output .= ' DEFAULT ' . Helpers::formatValue($this->defaultValue, $driver);
}
if ($this->autoIncrement) {
$output .= ' AUTO_INCREMENT';
}
if (isset($this->comment)) {
$output .= ' COMMENT ' . $driver->escapeText($this->comment);
}
return $output;
}
private static function formatOption($name, $value)
{
return $name . ($value !== NULL ? (' ' . $value) : '');
}
}