silverstripe-userforms/code/Task/UserFormsColumnCleanTask.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\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);
}
}
}
}
}