$dummy) { if(is_subclass_of($class, $rootClass)) { $rootClasses[$rootClass][] = $class; break; } } } } return $rootClasses; } /** * Display a simple HTML menu of database admin helpers. */ function index() { echo "

Database Administration Helpers

"; echo "

Add missing database fields (similar to sanity check).

"; echo "

Flush all of the generated images.

"; } /** * Updates the database schema, creating tables & fields as necessary. */ function build() { if((Director::isLive() && ClassInfo::hasTable('Member') && ClassInfo::hasTable('Group') && ClassInfo::hasTable('Permission')) && (!Member::currentUser() || !Member::currentUser()->isAdmin())) { 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."); return; } // The default time limit of 30 seconds is normally not enough if(ini_get("safe_mode") != "1") { set_time_limit(600); } $this->doBuild(isset($_REQUEST['quiet']) || isset($_REQUEST['from_installer'])); } /** * Check if database needs to be built, and build it if it does. */ 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; } } } /** * 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 */ 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 */ function doBuild($quiet = false, $populate = true) { $conn = DB::getConn(); if($quiet) { DB::quiet(); } else { echo "

Building Database

"; } // Set up the initial database if(!DB::isActive()) { if(!$quiet) { echo '

Creating database

'; } $parameters = $_REQUEST['db']; $connect = DB::getConnect($parameters); $username = $parameters[username]; $password = $parameters[password]; $database = $parameters[database]; DB::createDatabase($connect, $username, $password, $database); // ManifestBuilder::compileManifest(); } // Get all our classes // ManifestBuilder::compileManifest(); // ManifestBuilder::includeEverything(); // Build the database. Most of the hard work is handled by DataObject $dataClasses = ClassInfo::subclassesFor('DataObject'); array_shift($dataClasses); if(!$quiet) { echo '

Creating database tables

'; } $conn->beginSchemaUpdate(); foreach($dataClasses as $dataClass) { // Test_ indicates that it's the data class is part of testing system if(strpos($dataClass,'Test_') === false) { if(!$quiet) { echo "
  • $dataClass
    • "; } singleton($dataClass)->requireTable(); } } $conn->endSchemaUpdate(); ManifestBuilder::update_db_tables(); if($populate) { if(!$quiet) { echo '

    Creating database records

    '; } foreach($dataClasses as $dataClass) { // Test_ indicates that it's the data class is part of testing system if(strpos($dataClass,'Test_') === false) { if(!$quiet) { echo "
  • $dataClass
    • "; } singleton($dataClass)->requireDefaultRecords(); } } } touch(TEMP_FOLDER . '/database-last-generated-' .str_replace(array('\\','/',':'),'.', Director::baseFolder())); if(isset($_REQUEST['from_installer'])) { echo "OK"; } } /** * Method used to check mod_rewrite is working correctly in the installer. */ function testinstall() { echo "OK"; } /** * Remove invalid records from tables - that is, records that don't have * corresponding records in their parent class tables. */ 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(!singleton($subclass)->databaseFields()) { 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)) && ($recordExists[$subclass][$id])) { $sql = "DELETE FROM `$subclass` WHERE ID = $record[ID]"; echo "
  • $sql"; DB::query($sql); } } } } } } /** * Imports objects based on a specified CSV file in $_GET['FileName'] */ function import(){ $FileName = $_GET['FileName']; $FileName = $_SERVER['DOCUMENT_ROOT'] . substr($_SERVER['PHP_SELF'], 0, strlen($_SERVER['PHP_SELF'])-18) . "/assets/" . $FileName; if(file_exists($FileName)) { $handle = fopen($FileName,'r'); if($handle){ while (($data = fgetcsv($handle, 1000, ",")) !== FALSE) { $num = count($data); $row++; if($row == 1){ for ($c=0; $c < $num; $c++) { $ColumnHeaders[] = str_replace(' ','',$data[$c]); // Have to add code here to remove unsafe chars.. } } else { $Product = new Product(); for ($c=0; $c < $num; $c++) { $Product->$ColumnHeaders[$c] = trim($data[$c]); } $MainCategory = DataObject::get("ProductGroup", "URLSegment LIKE '" . $Product->generateURLSegment( $Product->Category) ."'"); if(!$MainCategory) { // if we can't find a main category, create all three sub // categories, as they must be unique. $ProductGroup = new ProductGroup(); $ProductGroup->Title = $Product->Category; print_r("