mirror of
https://github.com/silverstripe/silverstripe-framework
synced 2024-10-22 14:05:37 +02:00
477cf6b793
The original intention of this method was to be a rewrite check by the installer, but it doesn't belong in DatabaseAdmin, nor is it used by the installer anymore, since the rewrite test is done by the installer directly.
316 lines
9.0 KiB
PHP
316 lines
9.0 KiB
PHP
<?php
|
|
// Include the DB class
|
|
require_once("model/DB.php");
|
|
|
|
/**
|
|
* DatabaseAdmin class
|
|
*
|
|
* Utility functions for administrating the database. These can be accessed
|
|
* via URL, e.g. http://www.yourdomain.com/db/build.
|
|
*
|
|
* @package framework
|
|
* @subpackage model
|
|
*/
|
|
class DatabaseAdmin extends Controller {
|
|
|
|
/// SECURITY ///
|
|
static $allowed_actions = array(
|
|
'index',
|
|
'build',
|
|
'cleanup',
|
|
'import'
|
|
);
|
|
|
|
public function init() {
|
|
parent::init();
|
|
|
|
// We allow access to this controller regardless of live-status or ADMIN permission only
|
|
// if on CLI or with the database not ready. The latter makes it less errorprone to do an
|
|
// initial schema build without requiring a default-admin login.
|
|
// Access to this controller is always allowed in "dev-mode", or of the user is ADMIN.
|
|
$isRunningTests = (class_exists('SapphireTest', false) && SapphireTest::is_running_test());
|
|
$canAccess = (
|
|
Director::isDev()
|
|
|| !Security::database_is_ready()
|
|
// We need to ensure that DevelopmentAdminTest can simulate permission failures when running
|
|
// "dev/tests" from CLI.
|
|
|| (Director::is_cli() && !$isRunningTests)
|
|
|| Permission::check("ADMIN")
|
|
);
|
|
if(!$canAccess) {
|
|
return Security::permissionFailure($this,
|
|
"This page is secured and you need administrator rights to access it. " .
|
|
"Enter your credentials below and we will send you right along.");
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Get the data classes, grouped by their root class
|
|
*
|
|
* @return array Array of data classes, grouped by their root class
|
|
*/
|
|
public function groupedDataClasses() {
|
|
// Get all root data objects
|
|
$allClasses = get_declared_classes();
|
|
foreach($allClasses as $class) {
|
|
if(get_parent_class($class) == "DataObject")
|
|
$rootClasses[$class] = array();
|
|
}
|
|
|
|
// Assign every other data object one of those
|
|
foreach($allClasses as $class) {
|
|
if(!isset($rootClasses[$class]) && is_subclass_of($class, "DataObject")) {
|
|
foreach($rootClasses as $rootClass => $dummy) {
|
|
if(is_subclass_of($class, $rootClass)) {
|
|
$rootClasses[$rootClass][] = $class;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return $rootClasses;
|
|
}
|
|
|
|
|
|
/**
|
|
* Display a simple HTML menu of database admin helpers.
|
|
*/
|
|
public function index() {
|
|
echo "<h2>Database Administration Helpers</h2>";
|
|
echo "<p><a href=\"build\">Add missing database fields (similar to sanity check).</a></p>";
|
|
echo "<p><a href=\"../images/flush\">Flush <b>all</b> of the generated images.</a></p>";
|
|
}
|
|
|
|
|
|
/**
|
|
* Updates the database schema, creating tables & fields as necessary.
|
|
*/
|
|
public function build() {
|
|
// The default time limit of 30 seconds is normally not enough
|
|
increase_time_limit_to(600);
|
|
|
|
// Get all our classes
|
|
SS_ClassLoader::instance()->getManifest()->regenerate();
|
|
|
|
if(isset($_GET['returnURL'])) {
|
|
echo "<p>Setting up the database; you will be returned to your site shortly....</p>";
|
|
$this->doBuild(true);
|
|
echo "<p>Done!</p>";
|
|
$this->redirect($_GET['returnURL']);
|
|
} else {
|
|
$this->doBuild(isset($_REQUEST['quiet']) || isset($_REQUEST['from_installer']),
|
|
!isset($_REQUEST['dont_populate']));
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Check if database needs to be built, and build it if it does.
|
|
*/
|
|
public static function autoBuild() {
|
|
$dataClasses = ClassInfo::subclassesFor('DataObject');
|
|
$lastBuilt = self::lastBuilt();
|
|
foreach($dataClasses as $class) {
|
|
if(filemtime(getClassFile($class)) > $lastBuilt) {
|
|
$da = new DatabaseAdmin();
|
|
$da->doBuild(true);
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Build the default data, calling requireDefaultRecords on all
|
|
* DataObject classes
|
|
*/
|
|
public function buildDefaults() {
|
|
$dataClasses = ClassInfo::subclassesFor('DataObject');
|
|
array_shift($dataClasses);
|
|
foreach($dataClasses as $dataClass){
|
|
singleton($dataClass)->requireDefaultRecords();
|
|
print "Defaults loaded for $dataClass<br/>";
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Returns the timestamp of the time that the database was last built
|
|
*
|
|
* @return string Returns the timestamp of the time that the database was
|
|
* last built
|
|
*/
|
|
public static function lastBuilt() {
|
|
$file = TEMP_FOLDER . '/database-last-generated-' .
|
|
str_replace(array('\\','/',':'), '.' , Director::baseFolder());
|
|
|
|
if(file_exists($file)) {
|
|
return filemtime($file);
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* Updates the database schema, creating tables & fields as necessary.
|
|
*
|
|
* @param boolean $quiet Don't show messages
|
|
* @param boolean $populate Populate the database, as well as setting up its schema
|
|
*/
|
|
public function doBuild($quiet = false, $populate = true, $testMode = false) {
|
|
if($quiet) {
|
|
DB::quiet();
|
|
} else {
|
|
$conn = DB::getConn();
|
|
// Assumes database class is like "MySQLDatabase" or "MSSQLDatabase" (suffixed with "Database")
|
|
$dbType = substr(get_class($conn), 0, -8);
|
|
$dbVersion = $conn->getVersion();
|
|
$databaseName = (method_exists($conn, 'currentDatabase')) ? $conn->currentDatabase() : "";
|
|
|
|
if(Director::is_cli()) {
|
|
echo sprintf("\n\nBuilding database %s using %s %s\n\n", $databaseName, $dbType, $dbVersion);
|
|
} else {
|
|
echo sprintf("<h2>Building database %s using %s %s</h2>", $databaseName, $dbType, $dbVersion);
|
|
}
|
|
}
|
|
|
|
// Set up the initial database
|
|
if(!DB::isActive()) {
|
|
if(!$quiet) {
|
|
echo '<p><b>Creating database</b></p>';
|
|
}
|
|
global $databaseConfig;
|
|
$parameters = $databaseConfig ? $databaseConfig : $_REQUEST['db'];
|
|
$connect = DB::getConnect($parameters);
|
|
$username = $parameters['username'];
|
|
$password = $parameters['password'];
|
|
$database = $parameters['database'];
|
|
|
|
if(!$database) {
|
|
user_error("No database name given; please give a value for \$databaseConfig['database']",
|
|
E_USER_ERROR);
|
|
}
|
|
|
|
DB::createDatabase($connect, $username, $password, $database);
|
|
}
|
|
|
|
// Build the database. Most of the hard work is handled by DataObject
|
|
$dataClasses = ClassInfo::subclassesFor('DataObject');
|
|
array_shift($dataClasses);
|
|
|
|
if(!$quiet) {
|
|
if(Director::is_cli()) echo "\nCREATING DATABASE TABLES\n\n";
|
|
else echo "\n<p><b>Creating database tables</b></p>\n\n";
|
|
}
|
|
|
|
$conn = DB::getConn();
|
|
$conn->beginSchemaUpdate();
|
|
foreach($dataClasses as $dataClass) {
|
|
// Check if class exists before trying to instantiate - this sidesteps any manifest weirdness
|
|
if(class_exists($dataClass)) {
|
|
$SNG = singleton($dataClass);
|
|
if($testMode || !($SNG instanceof TestOnly)) {
|
|
if(!$quiet) {
|
|
if(Director::is_cli()) echo " * $dataClass\n";
|
|
else echo "<li>$dataClass</li>\n";
|
|
}
|
|
$SNG->requireTable();
|
|
}
|
|
}
|
|
}
|
|
$conn->endSchemaUpdate();
|
|
ClassInfo::reset_db_cache();
|
|
|
|
if($populate) {
|
|
if(!$quiet) {
|
|
if(Director::is_cli()) echo "\nCREATING DATABASE RECORDS\n\n";
|
|
else echo "\n<p><b>Creating database records</b></p>\n\n";
|
|
}
|
|
|
|
foreach($dataClasses as $dataClass) {
|
|
// Check if class exists before trying to instantiate - this sidesteps any manifest weirdness
|
|
// Test_ indicates that it's the data class is part of testing system
|
|
if(strpos($dataClass,'Test_') === false && class_exists($dataClass)) {
|
|
if(!$quiet) {
|
|
if(Director::is_cli()) echo " * $dataClass\n";
|
|
else echo "<li>$dataClass</li>\n";
|
|
}
|
|
|
|
singleton($dataClass)->requireDefaultRecords();
|
|
}
|
|
}
|
|
}
|
|
|
|
touch(TEMP_FOLDER . '/database-last-generated-' .
|
|
str_replace(array('\\', '/', ':'), '.', Director::baseFolder()));
|
|
|
|
if(isset($_REQUEST['from_installer'])) {
|
|
echo "OK";
|
|
}
|
|
|
|
if(!$quiet) {
|
|
echo (Director::is_cli()) ? "\n Database build completed!\n\n" :"<p>Database build completed!</p>";
|
|
}
|
|
|
|
ClassInfo::reset_db_cache();
|
|
}
|
|
|
|
/**
|
|
* Clear all data out of the database
|
|
* @todo Move this code into SS_Database class, for DB abstraction
|
|
*/
|
|
public function clearAllData() {
|
|
$tables = DB::getConn()->tableList();
|
|
foreach($tables as $table) {
|
|
if(method_exists(DB::getConn(), 'clearTable')) DB::getConn()->clearTable($table);
|
|
else DB::query("TRUNCATE \"$table\"");
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Remove invalid records from tables - that is, records that don't have
|
|
* corresponding records in their parent class tables.
|
|
*/
|
|
public function cleanup() {
|
|
$allClasses = get_declared_classes();
|
|
foreach($allClasses as $class) {
|
|
if(get_parent_class($class) == 'DataObject') {
|
|
$baseClasses[] = $class;
|
|
}
|
|
}
|
|
|
|
foreach($baseClasses as $baseClass) {
|
|
// Get data classes
|
|
$subclasses = ClassInfo::subclassesFor($baseClass);
|
|
unset($subclasses[0]);
|
|
foreach($subclasses as $k => $subclass) {
|
|
if(DataObject::database_fields($subclass)) {
|
|
unset($subclasses[$k]);
|
|
}
|
|
}
|
|
|
|
if($subclasses) {
|
|
$records = DB::query("SELECT * FROM \"$baseClass\"");
|
|
|
|
|
|
foreach($subclasses as $subclass) {
|
|
$recordExists[$subclass] =
|
|
DB::query("SELECT \"ID\" FROM \"$subclass\"")->keyedColumn();
|
|
}
|
|
|
|
foreach($records as $record) {
|
|
foreach($subclasses as $subclass) {
|
|
$id = $record['ID'];
|
|
if(($record['ClassName'] != $subclass) &&
|
|
(!is_subclass_of($record['ClassName'], $subclass)) &&
|
|
(isset($recordExists[$subclass][$id]))) {
|
|
$sql = "DELETE FROM \"$subclass\" WHERE \"ID\" = $record[ID]";
|
|
echo "<li>$sql";
|
|
DB::query($sql);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
|