69 lines
2.1 KiB
PHP
69 lines
2.1 KiB
PHP
<?php
|
|
|
|
namespace SilverStripe\UserForms\Task;
|
|
|
|
use SilverStripe\Dev\MigrationTask;
|
|
use SilverStripe\ORM\DataObject;
|
|
use SilverStripe\ORM\DB;
|
|
use SilverStripe\UserForms\Model\EditableFormField;
|
|
|
|
/**
|
|
* UserForms Column Clean Task
|
|
*
|
|
* Column clean up tasks for Userforms
|
|
*
|
|
* @package userforms
|
|
*/
|
|
|
|
class UserFormsColumnCleanTask extends MigrationTask
|
|
{
|
|
protected $title = 'UserForms EditableFormField Column Clean task';
|
|
|
|
protected $description = 'Removes unused columns from EditableFormField for MySQL databases;';
|
|
|
|
protected $tables = [EditableFormField::class];
|
|
|
|
protected $keepColumns = ['ID'];
|
|
|
|
/**
|
|
* Publish the existing forms.
|
|
*/
|
|
public function run($request)
|
|
{
|
|
/** @var \SilverStripe\ORM\DataObjectSchema $schema */
|
|
$schema = DataObject::getSchema();
|
|
|
|
foreach ($this->tables as $db) {
|
|
$columns = $schema->databaseFields($db);
|
|
$query = "SHOW COLUMNS FROM $db";
|
|
$liveColumns = DB::query($query)->column();
|
|
$backedUp = 0;
|
|
$query = "SHOW TABLES LIKE 'Backup_$db'";
|
|
$tableExists = DB::query($query)->value();
|
|
if ($tableExists != null) {
|
|
echo "Tasks run already on $db exiting";
|
|
return;
|
|
}
|
|
$backedUp = 0;
|
|
foreach ($liveColumns as $index => $column) {
|
|
if ($backedUp == 0) {
|
|
echo "Backing up $db <br />";
|
|
echo "Creating Backup_$db <br />";
|
|
// backup table
|
|
$query = "CREATE TABLE Backup_$db LIKE $db";
|
|
DB::query($query);
|
|
echo "Populating Backup_$db <br />";
|
|
$query = "INSERT Backup_$db SELECT * FROM $db";
|
|
DB::query($query);
|
|
$backedUp = 1;
|
|
}
|
|
if (!isset($columns[$column]) && !in_array($column, $this->keepColumns ?? [])) {
|
|
echo "Dropping $column from $db <br />";
|
|
$query = "ALTER TABLE $db DROP COLUMN $column";
|
|
DB::query($query);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|