import_from_xml.php

<?php
define ('SQ_IN_IMPORT', 1);

error_reporting(E_ALL);
if (ini_get('memory_limit') != '-1') ini_set('memory_limit', '-1');

// enable gabage collection
if(function_exists ('gc_enable')) {
    gc_enable();
}

if ((php_sapi_name() != 'cli')) trigger_error("You can only run this script from the command line\n", E_USER_ERROR);

$SYSTEM_ROOT = (isset($_SERVER['argv'][1])) ? $_SERVER['argv'][1] : '';
if (empty($SYSTEM_ROOT)) {
    echo "ERROR: You need to supply the path to the System Root as the first argument\n";
    exit();
}

if (!is_dir($SYSTEM_ROOT) || !is_readable($SYSTEM_ROOT.'/core/include/init.inc')) {
    echo "ERROR: Path provided doesn't point to a Matrix installation's System Root. Please provide correct path and try again.\n";
    exit();
}

$import_file = (isset($_SERVER['argv'][2])) ? $_SERVER['argv'][2] : '';
if (empty($import_file) || !is_file($import_file)) {
    echo "You need to supply the path to the import file as the second argument\n";
    exit();
}

require_once $SYSTEM_ROOT.'/core/include/init.inc';

if (isset($_SERVER['argv'][3]) && $_SERVER['argv'][3] == '--root-node' ) {
    $root_node_id = (isset($_SERVER['argv'][4])) ? $_SERVER['argv'][4] : '';
    if (empty($root_node_id)) {
        echo "you need to supply root node under which the assets are to be imported as fourth argument\n";
        exit();
    }
}

$root_user = $GLOBALS['SQ_SYSTEM']->am->getSystemAsset('root_user');

require_once SQ_LIB_PATH.'/import_export/import.inc';
$import_actions = get_import_actions($import_file);

//done checking authenticity of the system, first thing we check is the valid asset id for root node
if (isset($_SERVER['argv'][3]) && $_SERVER['argv'][3] == '--root-node' ) {
    //we are going to try and get the asset, we dont wanna be throwing errors here, so mute them
    error_reporting(E_NOTICE);
    $root_node = $GLOBALS['SQ_SYSTEM']->am->getAsset($root_node_id);
    if (is_null($root_node)) {
        echo "\nProvided assetid is not valid for given system, Script will stop execution\n";
        exit;
    }
    //restore error reporting
    error_reporting(E_ALL);

    $import_actions['actions'][0]['action'][0]['parentid'][0] = $root_node_id;

}
$total_number = count($import_actions['actions'][0]['action']);

// overcom oracle 'end of communication' bug
_disconnectFromMatrixDatabase();

// temp file to store global data, so child process can access it
// also used as a progress file to resume previous progress
$import_file_name_hash = md5($import_file);
define('TEMP_FILE', SQ_TEMP_PATH.'/import_from_xml_'.$import_file_name_hash.'.tmp');    
if(is_file(TEMP_FILE)) {
    echo "Previous progress file is detected, resuming to previous import.\n";
    echo "Using ".TEMP_FILE."\n\n";
    $actions_done = unserialize(file_get_contents(TEMP_FILE));
}


$import_action_outputs = Array();
$nest_content_to_fix = Array();
$designs_to_fix = Array();
// Loop through the Actions from the XML File
foreach ($import_actions['actions'][0]['action'] as $index => $action) {
    // skip actions that have done before
    // also try those previous failed actions
    if(isset($actions_done[$action['action_id'][0]])) { 
        continue;
    }
    
    
    // remember nest content to fix
    if($action['action_type'][0] === 'create_asset' && $action['type_code'][0] === 'Content_Type_Nest_Content') {   
        $nest_content_to_fix[] = $action['action_id'][0];
    }
    if($action['action_type'][0] === 'create_asset' && $action['type_code'][0] === 'Design') {  
        $designs_to_fix[] = $action['action_id'][0];
    }
    

 
    // Use forked process to make sure memory usage is stable
    $pid = pcntl_fork();
    switch ($pid) {
        case -1: 
            trigger_error('Process failed to fork', E_USER_ERROR);
            exit(1);
            break;
        case 0:
            // process action in forked child process
            // connect to DB within the child process to overcome oracle stupidness
            _connectToMatrixDatabase();          
            $GLOBALS['SQ_SYSTEM']->setRunLevel(SQ_RUN_LEVEL_OPEN);
            $GLOBALS['SQ_SYSTEM']->doTransaction('BEGIN');
            // restore the temp data from file
            if(is_file(TEMP_FILE))
                $import_action_outputs = unserialize(file_get_contents(TEMP_FILE));

            // check if there is hard coded assetid reference which doesn't exist in the target system
            if(!checkAssetExists($action, 'parentid') || !checkAssetExists($action, 'assetid') || !checkAssetExists($action, 'asset')) {
                trigger_error('Action ID "'.$action['action_id'][0].'" contains non-exist assetid reference. Action skipped.', E_USER_WARNING);
                _disconnectFromMatrixDatabase();
                exit(0);
                break;
            }

            // Execute the action
            printActionId($action['action_id'][0]);
            if (!execute_import_action($action, $import_action_outputs)) {
                trigger_error('Action ID "'.$action['action_id'][0].'" could not be executed', E_USER_WARNING);
            } else {
                $count = $index + 1;
                printStatus($count.'/'.$total_number);
            }
            
            // blank out old value and new value returned from set attribute action. 
            // recording the attribute value content will flood our temp data file and memory
            $new_entry = array_slice( $import_action_outputs, -1, 1, TRUE );
            $new_entry_key = key($new_entry);
            if(isset($new_entry[$new_entry_key]['old_value'])) unset($new_entry[$new_entry_key]['old_value']);
            if(isset($new_entry[$new_entry_key]['new_value'])) unset($new_entry[$new_entry_key]['new_value']);
            $import_action_outputs = array_merge($import_action_outputs, $new_entry);
            
            // save global temp data to file
            $temp_string = serialize($import_action_outputs);
            string_to_file($temp_string, TEMP_FILE);
            
            $GLOBALS['SQ_SYSTEM']->doTransaction('COMMIT');
            $GLOBALS['SQ_SYSTEM']->restoreRunLevel();
            // Disconnect from DB
            _disconnectFromMatrixDatabase();
            exit(0);
            break;

        default:
            $status = null;
            pcntl_waitpid(-1, $status);
            break;
    }//end switch
    
}


// fix nest content type, regenerate the bodycopy
foreach ($nest_content_to_fix as $actionid) {
    if(isset($import_action_outputs[$actionid])) {
        $nest_content_id = $import_action_outputs[$actionid]['assetid'];
        $nest_content = $GLOBALS['SQ_SYSTEM']->am->getAsset($nest_content_id);
        $nest_content->_tmp['edit_fns'] = NULL;
        $nest_content->linksUpdated();
    }
}

// we have imported a few design , lets fix them
foreach ($designs_to_fix as $design) {
    if(isset($import_action_outputs[$design])) {
        $design_id = $import_action_outputs[$design]['assetid'];
        $vars = Array('assetid' => $design_id);
        $hh = $GLOBALS['SQ_SYSTEM']->getHipoHerder();
        $hh->freestyleHipo('hipo_job_regenerate_design', $vars);
    }
}

// remove the temp file
if(is_file(TEMP_FILE)) unlink(TEMP_FILE);

// disable gabage collection
if(function_exists ('gc_disable')) {
    gc_disable();
}

function printActionId($action_id)
{
    if (strlen($action_id) > 66) {
        $action_id = substr($action_id, 0, 66).'...';
    }
    printf ('%s%'.(70 - strlen($action_id)).'s', $action_id,'');

}//end printActionId()


function printStatus($status)
{
    echo "[ $status ]\n";

}//end printStatus()


function checkAssetExists($action, $type='asset')
{
        if(isset($action[$type][0]) && preg_match('/^[0-9]+$/', $action[$type][0])){
        return ($GLOBALS['SQ_SYSTEM']->am->assetExists ($action[$type][0]));
    }   
    return TRUE;
}


function _disconnectFromMatrixDatabase()
{
    $conn_id = MatrixDAL::getCurrentDbId();
    if (isset($conn_id) && !empty($conn_id)) {
        MatrixDAL::restoreDb();
        MatrixDAL::dbClose($conn_id);
    }//end if
        
}//end _disconnectFromMatrixDatabase()


function _connectToMatrixDatabase()
{
    $GLOBALS['SQ_SYSTEM']->changeDatabaseConnection('db2');

}//end _connectToMatrixDatabase()


?>