array( 'desc' => 'prefix for the password (eg. fred > xxx4fred - prefix is xxx4)', 'short' => 'p', 'default' => '', 'min' => 1, 'max' => 1, ), 'data-only' => array( 'desc' => 'only run the updateData - do not run import the tables and procedures.', 'short' => 'p', 'default' => '', 'min' => 1, 'max' => 1, ), 'add-company' => array( 'desc' => 'add a company name of the company', 'short' => 'n', 'default' => '', 'min' => 1, 'max' => 1, ), 'add-company-with-type' => array( 'desc' => 'the type of company (default OWNER)', 'short' => 't', 'default' => 'OWNER', 'min' => 1, 'max' => 1, ), 'init' => array( 'desc' => 'Initialize the database (pg only supported)', 'short' => 'i', 'default' => '', 'min' => 1, 'max' => 1, ), 'only-module-sql' => array( 'desc' => 'Only run sql import on this modules - eg. Core', 'default' => '', 'min' => 1, 'max' => 1, ), 'procedures-only' => array( 'desc' => 'Only import procedures (not supported by most modules yet) - ignores sql directory', 'default' => '', 'min' => 1, 'max' => 1, ), 'json-person' => array( 'desc' => 'Person JSON file', 'default' => '', 'min' => 1, 'max' => 1, ), ); static function cli_opts() { $ret = self::$cli_opts; $ff = HTML_FlexyFramework::get(); $a = new Pman(); $mods = $a->modulesList(); foreach($mods as $m) { $fd = $ff->rootDir. "/Pman/$m/UpdateDatabase.php"; if (!file_exists($fd)) { continue; } require_once $fd; $cls = new ReflectionClass('Pman_'. $m . '_UpdateDatabase'); $ret = array_merge($ret, $cls->getStaticPropertyValue('cli_opts')); } return $ret; } var $opts = false; var $disabled = array(); var $cli = false; function getAuth() { $ff = HTML_FlexyFramework::get(); if (!empty($ff->cli)) { $this->cli = true; return true; } parent::getAuth(); // load company! $au = $this->getAuthUser(); if (!$au || $au->company()->comptype != 'OWNER') { $this->jerr("Not authenticated", array('authFailure' => true)); } $this->authUser = $au; return true; } function get($args, $opts) { PEAR::setErrorHandling(PEAR_ERROR_CALLBACK, array($this, 'onPearError')); $this->checkSystem(); $ff = HTML_FlexyFramework::get(); if(!empty($ff->Core_Notify)){ require_once 'Pman/Core/NotifySmtpCheck.php'; $x = new Pman_Core_NotifySmtpCheck(); $x->check(); } $this->disabled = explode(',', $ff->disable); //$this->fixSequencesPgsql();exit; $this->opts = $opts; // ask all the modules to verify the opts $this->checkOpts($opts); if (empty($opts['data-only'])) { $this->importSQL(); } if (!empty($opts['only-module-sql'])) { return; } $this->runUpdateModulesData(); if (!empty($opts['add-company']) && !in_array('Core', $this->disabled)) { // make sure we have a good cache...? DB_DataObject::factory('companies')->initCompanies($this, $opts); } $this->runExtensions(); } function output() { return ''; } /** * imports SQL files from all DataObjects directories.... * * except any matching /migrate/ */ function importSQL() { // loop through all the modules, and see if they have a importSQL method? $ff = HTML_Flexyframework::get(); $dburl = parse_url($ff->DB_DataObject['database']); //$this->{'import' . $url['scheme']}($url); $dbtype = $dburl['scheme']; $dirmethod = 'import' . $dburl['scheme'] . 'dir'; $ar = $this->modulesList(); foreach($ar as $m) { if(in_array($m, $this->disabled)){ echo "module $m is disabled \n"; continue; } echo "Importing SQL from module $m\n"; if (!empty($this->opts['only-module-sql']) && $m != $this->opts['only-module-sql']) { continue; } // check to see if the class has $file = $this->rootDir. "/Pman/$m/UpdateDatabase.php"; if($m != 'Core' && file_exists($file)){ require_once $file; $class = "Pman_{$m}_UpdateDatabase"; $x = new $class; if(method_exists($x, 'importModuleSQL')){ echo "Importing SQL from module $m using Module::importModuleSQL\n"; $x->opts = $this->opts; $x->rootDir = $this->rootDir; $x->importModuleSQL($dburl); continue; } }; echo "Importing SQL from module $m\n"; // if init has been called // look in pgsql.ini if (!empty($this->opts['init'])) { $this->{$dirmethod}($dburl, $this->rootDir. "/Pman/$m/{$dbtype}.init"); } $fd = $this->rootDir. "/Pman/$m/DataObjects"; $this->{$dirmethod}($dburl, $fd); // new -- sql directory.. // new style will not support migrate ... they have to go into mysql-migrate.... directories.. // new style will not support pg.sql etc.. naming - that's what the direcotries are for.. $this->{$dirmethod}($dburl, $this->rootDir. "/Pman/$m/sql"); $this->{$dirmethod}($dburl, $this->rootDir. "/Pman/$m/{$dbtype}"); if (!empty($this->opts['init']) && file_exists($this->rootDir. "/Pman/$m/{$dbtype}.initdata")) { HTML_FlexyFramework::get()->generateDataobjectsCache(true); $this->{$dirmethod}($dburl, $this->rootDir. "/Pman/$m/{$dbtype}.initdata"); $this->{'fixSequences'. $dbtype}(); } } } /** -------------- code to handle importing a whole directory of files into the database ------- **/ function importpgsqldir($url, $dir, $disable_triggers = false) { $ff = HTML_FlexyFramework::get(); require_once 'System.php'; $cat = System::which('cat'); $psql = System::which('psql'); if (!empty($url['pass'])) { putenv("PGPASSWORD=". $url['pass']); } $psql_cmd = $psql . ' -h ' . $url['host'] . ' -U' . escapeshellarg($url['user']) . ' ' . basename($url['path']); echo $psql_cmd . "\n" ; echo "scan : $dir\n"; if (is_file($dir)) { $files = array($dir); } else { $files = glob($dir.'/*.sql'); uksort($files, 'strcasecmp'); } //$lsort = create_function('$a,$b','return strlen($a) > strlen($b) ? 1 : -1;'); //usort($files, $lsort); foreach($files as $bfn) { if (preg_match('/migrate/i', basename($bfn))) { // skip migration scripts at present.. continue; } if (preg_match('#\.[a-z]{2}\.sql#i', basename($bfn)) && !preg_match('#\.pg\.sql#i', basename($bfn)) ) { // skip migration scripts at present.. continue; } $fn = false; if (!preg_match('/pgsql/', basename($dir) )) { if ( !preg_match('#\.pg\.sql$#', basename($bfn))) { $fn = $this->convertToPG($bfn); } } // files ending in .pg.sql are native postgres files.. ## depricated $cmd = "$psql_cmd < " . escapeshellarg($fn ? $fn : $bfn) . ' 2>&1' ; echo "$bfn: $cmd ". ($ff->cli ? "\n" : "
\n"); passthru($cmd); if ($fn) { unlink($fn); } } } /** * mysql - does not support conversions. * * */ function importmysqldir($dburl, $dir) { $this->fixMysqlInnodb(); /// run once echo "Import MYSQL :: $dir\n"; require_once 'System.php'; $cat = System::which('cat'); $mysql = System::which('mysql'); $mysql_cmd = $mysql . ' -h ' . $dburl['host'] . ' -u' . escapeshellarg($dburl['user']) . (!empty($dburl['pass']) ? ' -p' . escapeshellarg($dburl['pass']) : '') . ' ' . basename($dburl['path']); //echo $mysql_cmd . "\n" ; $files = glob($dir.'/*.sql'); uksort($files, 'strcasecmp'); foreach($files as $fn) { if (preg_match('/migrate/i', basename($fn))) { // skip migration scripts at present.. continue; } // .my.sql but not .pg.sql if (preg_match('#\.[a-z]{2}\.sql#i', basename($fn)) && !preg_match('#\.my\.sql#i', basename($fn)) ) { // skip migration scripts at present.. continue; } if (!strlen(trim($fn))) { continue; } $cmd = "$mysql_cmd -f < " . escapeshellarg($fn) ." 2>&1" ; echo basename($dir).'/'. basename($fn) . '::' . $cmd. ($this->cli ? "\n" : "
\n"); $fp = popen($cmd, "r"); while(!feof($fp)) { // send the current file part to the browser $line = trim(fgets($fp, 1024)); if (empty($line)) { continue; } $matches = array(); if (!preg_match('/^ERROR\s+([0-9]+)/', $line, $matches)) { echo " ---- {$line}\n"; flush(); continue; } $continue =0; switch($matches[1]) { case 1017: // cause by renaming table -- old one does not exist.. case 1050: // create tables triggers this.. case 1060: // Duplicate column name case 1061: // Duplicate key name - triggered by add index.. but could hide error. - unlikely though. case 1091: // drop index -- name does not exist.. might hide errors.. case 1146: // drop a index on an unknown table.. - happens rarely... case 1054: // Unknown column -- triggered by CHANGE COLUMN - but may hide other errrors.. $continue = 1; break; } if ($continue) { echo " ---- {$line}\n"; flush(); continue; } // real errors... // 1051: // Unknown table -- normally drop = add iff exists.. echo "File: $fn\n$line\n"; exit; } } } /** * simple regex based convert mysql to pgsql... */ function convertToPG($src) { //echo "Convert $src\n"; $fn = $this->tempName('sql'); $ret = array( ); // pad it a bit. $extra = array("", "" ); $tbl = false; foreach(file($src) as $l) { $l = trim($l); if (!strlen($l) || $l[0] == '#') { continue; } $m = array(); if (preg_match('#create\s+table\s+([a-z0-9_]+)#i', $l, $m)) { $tbl = $m[1]; } if (preg_match('#create\s+table\s+\`([a-z0-9_]+)\`#i', $l, $m)) { $tbl = 'shop_' . strtolower($m[1]); $l = preg_replace('#create\s+table\s+\`([a-z0-9_]+)\`#i', "CREATE TABLE {$tbl}", $l); } if (preg_match('#\`([a-z0-9_]+)\`#i', $l, $m) && !preg_match('#alter\s+table\s+#i', $l)) { $l = preg_replace('#\`([a-z0-9_]+)\`#i', "{$m[1]}_name", $l); } // autoinc if ($tbl && preg_match('#auto_increment#i', $l, $m)) { $l = preg_replace('#auto_increment#i', "default nextval('{$tbl}_seq')", $l); $extra[] = "create sequence {$tbl}_seq;"; } if (preg_match('#alter\s+table\s+(\`[a-z0-9_]+\`)#i', $l, $m)){ $l = preg_replace('#alter\s+table\s+(\`[a-z0-9_]+\`)#i', "ALTER TABLE {$tbl}", $l); } // enum value -- use the text instead.. if ($tbl && preg_match('#([\w]+)\s+(enum\([\w|\W]+\))#i', $l, $m)) { $l = preg_replace('#enum\([\w|\W]+\)#i', "TEXT", $l); } // ignore the alter enum if ($tbl && preg_match('#alter\s+table\s+([\w|\W]+)\s+enum\([\w|\W]+\)#i', $l, $m)) { continue; } // UNIQUE KEY .. ignore if ($tbl && preg_match('#UNIQUE KEY#i', $l, $m)) { $last = array_pop($ret); $ret[] = trim($last, ","); continue; } if ($tbl && preg_match('#RENAME\s+TO#i', $l, $m)) { continue; } if ($tbl && preg_match('#change\s+column#i', $l, $m)) { continue; } // INDEX lookup ..ignore if ($tbl && preg_match('#INDEX lookup+([\w|\W]+)#i', $l, $m)) { $last = array_pop($ret); $ret[] = trim($last, ","); continue; } // CREATE INDEX ..ignore if (preg_match('#alter\s+table\s+([a-z0-9_]+)\s+add\s+index\s+#i', $l, $m)) { // $l = "CREATE INDEX {$m[1]}_{$m[2]} ON {$m[1]} {$m[3]}"; continue; } // basic types.. $l = preg_replace('#int\([0-9]+\)#i', 'INT', $l); $l = preg_replace('# datetime#i', ' TIMESTAMP WITHOUT TIME ZONE', $l); $l = preg_replace('# blob#i', ' TEXT', $l); $l = preg_replace('# longtext#i', ' TEXT', $l); $l = preg_replace('# tinyint#i', ' INT', $l); $ret[] = $l; } $ret = array_merge($extra,$ret); // echo implode("\n", $ret); exit; file_put_contents($fn, implode("\n", $ret)); return $fn; } function checkOpts($opts) { foreach($opts as $o=>$v) { if (!preg_match('/^json-/', $o) || empty($v)) { continue; } if (!file_exists($v)) { die("File does not exist : OPTION --{$o} = {$v} \n"); } } $modules = array_reverse($this->modulesList()); // move 'project' one to the end... foreach ($modules as $module){ $file = $this->rootDir. "/Pman/$module/UpdateDatabase.php"; if($module == 'Core' || !file_exists($file)){ continue; } require_once $file; $class = "Pman_{$module}_UpdateDatabase"; $x = new $class; if(!method_exists($x, 'checkOpts')){ continue; }; $x->checkOpts($opts); } } static function jsonImportFromArray($opts) { foreach($opts as $o=>$v) { if (!preg_match('/^json-/', $o) || empty($v)) { continue; } $type = str_replace('_', '-', substr($o,5)); $data= json_decode(file_get_contents($v),true); $pg = HTML_FlexyFramework::get()->page; DB_DataObject::factory($type)->importFromArray($pg ,$data,$opts); } } function runUpdateModulesData() { HTML_FlexyFramework::get()->generateDataobjectsCache(true); if(!in_array('Core', $this->disabled)){ echo "Running jsonImportFromArray\n"; Pman_Core_UpdateDatabase::jsonImportFromArray($this->opts); echo "Running updateData on modules\n"; // runs core... echo "Core\n"; $this->updateData(); } $modules = array_reverse($this->modulesList()); // move 'project' one to the end... foreach ($modules as $module){ if(in_array($module, $this->disabled)){ continue; } $file = $this->rootDir. "/Pman/$module/UpdateDatabase.php"; if($module == 'Core' || !file_exists($file)){ continue; } require_once $file; $class = "Pman_{$module}_UpdateDatabase"; $x = new $class; if(!method_exists($x, 'updateData')){ continue; }; echo "$module\n"; $x->updateData(); } } function updateDataEnums() { $enum = DB_DataObject::Factory('core_enum'); //DB_DAtaObject::debugLevel(1); $enum->initEnums( array( array( 'etype' => '', 'name' => 'COMPTYPE', 'display_name' => 'Company Types', 'is_system_enum' => 1, 'cn' => array( array( 'name' => 'OWNER', 'display_name' => 'Owner', 'seqid' => 999, // last... 'is_system_enum' => 1, ) ) ), array( 'etype' => '', 'name' => 'HtmlEditor.font-family', 'display_name' => 'HTML Editor font families', 'is_system_enum' => 1, 'cn' => array( array( 'name' => 'Helvetica,Arial,sans-serif', 'display_name' => 'Helvetica', ), array( 'name' => 'Courier New', 'display_name' => 'Courier', ), array( 'name' => 'Tahoma', 'display_name' => 'Tahoma', ), array( 'name' => 'Times New Roman,serif', 'display_name' => 'Times', ), array( 'name' => 'Verdana', 'display_name' => 'Verdana', ), ) ), ) ); } function updateDataGroups() { $groups = DB_DataObject::factory('groups'); $groups->initGroups(); $groups->initDatabase($this,array( array( 'name' => 'bcc-email', // group who are bcc'ed on all requests. 'type' => 0, // system ), array( 'name' => 'system-email-from', 'type' => 0, // system ), array( 'name' => 'core-person-signup-bcc', 'type' => 0, // system ), )); } function updateDataCompanies() { // fix comptypes enums.. $c = DB_DataObject::Factory('Companies'); $c->selectAdd(); $c->selectAdd('distinct(comptype) as comptype'); $c->whereAdd("comptype != ''"); $ctb = array(); foreach($c->fetchAll('comptype') as $cts) { $ctb[]= array( 'etype'=>'COMPTYPE', 'name' => $cts, 'display_name' => ucfirst(strtolower($cts))); } $c = DB_DataObject::Factory('core_enum'); $c->initEnums($ctb); //DB_DataObject::debugLevel(1); // fix comptypeid $c = DB_DataObject::Factory('Companies'); $c->query(" UPDATE Companies SET comptype_id = (SELECT id FROM core_enum where etype='comptype' and name=Companies.comptype LIMIT 1) WHERE comptype_id = 0 AND LENGTH(comptype) > 0 "); } function initEmails($templateDir, $emails) { $pg = HTML_FlexyFramework::get()->page; foreach($emails as $name=>$data) { $cm = DB_DataObject::factory('core_email'); $update = $cm->get('name', $name); $old = clone($cm); if (empty($cm->bcc_group)) { if (empty($data['bcc_group'])) { $this->jerr("missing bcc_group for template $name"); } $g = DB_DataObject::Factory('Groups')->lookup('name',$data['bcc_group']); if (!$g) { $this->jerr("bcc_group {$data['bcc_group']} does not exist when importing template $name"); } if (!$g->members('email')) { $this->jerr("bcc_group {$data['bcc_group']} does not have any members"); } $cm->bcc_group = $g->id; } if (empty($cm->test_class)) { if (empty($data['test_class'])) { $this->jerr("missing test_class for template $name"); } $cm->test_class = $data['test_class']; } require_once $cm->test_class . '.php'; $clsname = str_replace('/','_', $cm->test_class); try { $method = new ReflectionMethod($clsname , 'test_'. $name) ; $got_it = $method->isStatic(); } catch(Exception $e) { $got_it = false; } if (!$got_it) { $this->jerr("template {$name} does not have a test method {$clsname}::test_{$name}"); } if ($update) { $cm->update($old); echo "email: {$name} - checked\n"; continue; /// we do not import the body content of templates that exist... } else { $cm->insert(); } // $basedir = $this->bootLoader->rootDir . $mail_template_dir; $opts = array( 'update' => 1, 'file' => $templateDir. $name .'.html' ); if (!empty($data['master'])) { $opts['master'] = $templateDir . $master .'.html'; } require_once 'Pman/Core/Import/Core_email.php'; $x = new Pman_Core_Import_Core_email(); $x->get('', $opts); echo "email: {$name} - CREATED\n"; } } function updateData() { // fill i18n data.. HTML_FlexyFramework::get()->generateDataobjectsCache(true); $this->updateDataEnums(); $this->updateDataGroups(); $this->updateDataCompanies(); $c = DB_DataObject::Factory('I18n'); $c->buildDB(); } function fixMysqlInnodb() { static $done_check = false; if ($done_check) { return; } // innodb in single files is far more efficient that MYD or one big innodb file. // first check if database is using this format. $db = DB_DataObject::factory('core_enum'); $db->query("show variables like 'innodb_file_per_table'"); $db->fetch(); if ($db->Value == 'OFF') { die("Error: set innodb_file_per_table = 1 in my.cnf\n\n"); } $done_check = true;; } /** ------------- schema fixing ... there is an issue with data imported having the wrong sequence names... --- */ function fixSequencesMysql() { // not required... } function fixSequencesPgsql() { //DB_DataObject::debugLevel(1); $cs = DB_DataObject::factory('core_enum'); $cs->query(" SELECT 'ALTER SEQUENCE '|| CASE WHEN strpos(seq_name, '.') > 0 THEN min(seq_name) ELSE quote_ident(min(schema_name)) ||'.'|| quote_ident(min(seq_name)) END ||' OWNED BY '|| quote_ident(min(schema_name)) || '.' || quote_ident(min(table_name)) ||'.'|| quote_ident(min(column_name)) ||';' as cmd FROM ( SELECT n.nspname AS schema_name, c.relname AS table_name, a.attname AS column_name, regexp_replace(regexp_replace(d.adsrc, E'nextval\\\\(+[''\\\"]*', ''),E'[''\\\"]*::.*\$','') AS seq_name FROM pg_class c JOIN pg_attribute a ON (c.oid=a.attrelid) JOIN pg_attrdef d ON (a.attrelid=d.adrelid AND a.attnum=d.adnum) JOIN pg_namespace n ON (c.relnamespace=n.oid) WHERE has_schema_privilege(n.oid,'USAGE') AND n.nspname NOT LIKE 'pg!_%' escape '!' AND has_table_privilege(c.oid,'SELECT') AND (NOT a.attisdropped) AND d.adsrc ~ '^nextval' ) seq WHERE CASE WHEN strpos(seq_name, '.') > 0 THEN substring(seq_name, 1,strpos(seq_name,'.')-1) ELSE schema_name END = schema_name GROUP BY seq_name HAVING count(*)=1 "); $cmds = array(); while ($cs->fetch()) { $cmds[] = $cs->cmd; } foreach($cmds as $cmd) { $cs = DB_DataObject::factory('core_enum'); echo "$cmd\n"; $cs->query($cmd); } $cs = DB_DataObject::factory('core_enum'); $cs->query(" SELECT 'SELECT SETVAL(' || quote_literal(quote_ident(nspname) || '.' || quote_ident(S.relname)) || ', MAX(' || quote_ident(C.attname)|| ')::integer ) FROM ' || nspname || '.' || quote_ident(T.relname)|| ';' as cmd FROM pg_class AS S, pg_depend AS D, pg_class AS T, pg_attribute AS C, pg_namespace AS NS WHERE S.relkind = 'S' AND S.oid = D.objid AND D.refobjid = T.oid AND D.refobjid = C.attrelid AND D.refobjsubid = C.attnum AND NS.oid = T.relnamespace ORDER BY S.relname "); $cmds = array(); while ($cs->fetch()) { $cmds[] = $cs->cmd; } foreach($cmds as $cmd) { $cs = DB_DataObject::factory('core_enum'); echo "$cmd\n"; $cs->query($cmd); } } var $extensions = array( 'EngineCharset', 'Links', ); function runExtensions() { $ff = HTML_Flexyframework::get(); $dburl = parse_url($ff->DB_DataObject['database']); $dbtype = $dburl['scheme']; foreach($this->extensions as $ext) { $scls = ucfirst($dbtype). $ext; $cls = __CLASS__ . '_'. $scls; $fn = implode('/',explode('_', $cls)).'.php'; if (!file_exists(__DIR__.'/UpdateDatabase/'. $scls .'.php')) { return; } require_once $fn; $c = new $cls(); } } function checkSystem() { // most of these are from File_Convert... // these are required - and have simple dependancies. require_once 'System.php'; $req = array( 'convert', 'grep', 'pdfinfo', 'pdftoppm', 'rsvg-convert', //librsvg2-bin 'strings', ); // these are prefered - but may have complicated depenacies $pref= array( 'abiword', 'faad', 'ffmpeg', 'html2text', // not availabe in debian squeeze 'pdftocairo', //poppler-utils - not available in debian squeeze. 'lame', 'ssconvert', 'unoconv', 'wkhtmltopdf', 'xvfb-run', ); $res = array(); $fail = false; foreach($req as $r) { if (!System::which($r)) { $res[] = $r; } $fail = true; } if ($res) { $this->jerr("Missing these programs - need installing\n" . implode("\n",$res)); } foreach($pref as $r) { if (!System::which($r)) { $res[] = $r; } $fail = true; } if ($res) { echo "WARNING: Missing these programs - they may need installing\n". implode("\n",$res); sleep(5); } } }