<?php namespace DBDiff\SQLGen\DiffToSQL;

use DBDiff\SQLGen\SQLGenInterface;

use DBDiff\SQLGen\Dialect\DialectRegistry;

use DBDiff\SQLGen\Dialect\SQLDialectInterface;

class AlterRoutineSQL 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 {

$drop = $this->buildDrop($this->obj->targetDefinition, $this->obj->name);

return $drop . "\n" . $this->obj->sourceDefinition . ';';

}

public function getDown(): string {

$drop = $this->buildDrop($this->obj->sourceDefinition, $this->obj->name);

return $drop . "\n" . $this->obj->targetDefinition . ';';

}

private function buildDrop(string $definition, string $name): string {

$type = 'FUNCTION';

if (preg_match('/\bPROCEDURE\b/i', $definition)) {

$type = 'PROCEDURE';

}

return "DROP $type IF EXISTS " . $this->dialect->quote($name) . ';';

}

}