<?php namespace DBDiff\SQLGen\DiffToSQL;

use DBDiff\SQLGen\SQLGenInterface;

use DBDiff\SQLGen\Dialect\DialectRegistry;

use DBDiff\SQLGen\Dialect\SQLDialectInterface;

use DBDiff\DB\Data\BinaryValue;

use Diff\DiffOp\DiffOpAdd;

use Diff\DiffOp\DiffOpRemove;

class UpdateDataSQL implements SQLGenInterface {

protected $obj;

protected SQLDialectInterface $dialect;

public function __construct($obj, SQLDialectInterface $dialect = null) {

$this->obj = $obj;

$this->dialect = $dialect ?? DialectRegistry::get();

}

public function getUp(): string {

$t = $this->dialect->quote($this->obj->table);

$d = $this->dialect;

$values = $this->obj->diff['diff'];

array_walk($values, function (&$diff, $column) use ($d) {

$q = $d->quote($column);

if ($diff instanceof DiffOpRemove) {

$diff = "$q = NULL";

} elseif (!method_exists($diff, 'getNewValue') || is_null($diff->getNewValue())) {

$diff = "$q = NULL";

} else {

$diff = "$q = " . BinaryValue::formatSQL($diff->getNewValue());

}

});

$keys = $this->obj->diff['keys'];

array_walk($keys, function (&$value, $column) use ($d) {

$value = BinaryValue::formatCondition($d->quote($column), $value);

});

return "UPDATE $t SET " . implode(', ', $values) . ' WHERE ' . implode(' AND ', $keys) . ';';

}

public function getDown(): string {

$t = $this->dialect->quote($this->obj->table);

$d = $this->dialect;

$values = $this->obj->diff['diff'];

array_walk($values, function (&$diff, $column) use ($d) {

$q = $d->quote($column);

if ($diff instanceof DiffOpAdd) {

$diff = "$q = NULL";

} elseif (!method_exists($diff, 'getOldValue') || is_null($diff->getOldValue())) {

$diff = "$q = NULL";

} else {

$diff = "$q = " . BinaryValue::formatSQL($diff->getOldValue());

}

});

$keys = $this->obj->diff['keys'];

array_walk($keys, function (&$value, $column) use ($d) {

$value = BinaryValue::formatCondition($d->quote($column), $value);

});

return "UPDATE $t SET " . implode(', ', $values) . ' WHERE ' . implode(' AND ', $keys) . ';';

}

}