system_integrity_orphaned_assets.php

<?php
error_reporting(E_ALL);
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();
}

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

$MAP_ASSETID = (isset($_SERVER['argv'][2])) ? $_SERVER['argv'][2] : '0';

$map_asset_info = $GLOBALS['SQ_SYSTEM']->am->getAssetInfo(Array($MAP_ASSETID),'asset',FALSE);

if (empty($MAP_ASSETID) || empty($map_asset_info)) {
    echo "ERROR: You need to supply the assetid of a valid asset that orphaned assets will be mapped to as the second argument\n";
    exit();
} else {
    $map_asset =& $GLOBALS['SQ_SYSTEM']->am->getAsset($MAP_ASSETID);
}

$ROOT_ASSETID = (isset($_SERVER['argv'][3])) ? $_SERVER['argv'][3] : '1';
if ($ROOT_ASSETID == 1) {
    echo "\nWARNING: You are running this integrity checker on the whole system.\nThis is fine, but it may take a long time\n\n";
}

// ask for the root password for the system
echo 'Enter the root password for "'.SQ_CONF_SYSTEM_NAME.'": ';
system('stty -echo');
$root_password = rtrim(fgets(STDIN, 4094));
system('stty echo');

// check that the correct root password was entered
$root_user =& $GLOBALS['SQ_SYSTEM']->am->getSystemAsset('root_user');
if (!$root_user->comparePassword($root_password)) {
    echo "ERROR: The root password entered was incorrect\n";
    exit();
}

// log in as root
if (!$GLOBALS['SQ_SYSTEM']->setCurrentUser($root_user)) {
    echo "ERROR: Failed login in as root user\n";
    exit();
}

$db =& $GLOBALS['SQ_SYSTEM']->db;

$GLOBALS['SQ_SYSTEM']->setRunLevel(SQ_RUN_LEVEL_FORCED);

// go through each child of the specified asset, lock it, validate it, unlock it
$assets = $GLOBALS['SQ_SYSTEM']->am->getChildren($ROOT_ASSETID, 'asset', FALSE);
foreach ($assets as $assetid => $type_code_data) {
    $type_code = $type_code_data[0]['type_code'];

    printAssetName($assetid);

    $sql = 'SELECT
                linkid,
                majorid,
                link_type,
                sort_order
            FROM
                sq_ast_lnk
            WHERE
                minorid = :minorid
                AND
                link_type <> :link_type';

    $query = MatrixDAL::preparePdoQuery($sql);
    MatrixDAL::bindValueToPdo($query, 'minorid', $assetid);
    MatrixDAL::bindValueToPdo($query, 'link_type', SQ_LINK_NOTICE);
    $links = MatrixDAL::executePdoAssoc($query);

    $updated = FALSE;
    $errors = FALSE;

    foreach (array_keys($links) as $linkid) {
        $link =& $links[$linkid];
        if ($link['linkid'] == 1) continue;

        $major_asset_info = $GLOBALS['SQ_SYSTEM']->am->getAssetInfo(Array($link['majorid']),'asset',FALSE);

        if (empty($major_asset_info)) {

            // ARGH, we have a link but our major isn't there anymore... this could be a problem
            // if it was an exclusive link...!
            // basically we need to do the whole rubbish regarding deleting asset links in here
            // because deleteAssetLink() checks to see if the asset is alive... which it isn't >_>;;;

            $GLOBALS['SQ_SYSTEM']->changeDatabaseConnection('db2');
            $db =& $GLOBALS['SQ_SYSTEM']->db;

            // open the transaction
            $GLOBALS['SQ_SYSTEM']->doTransaction('BEGIN');

            // if this is a significant link
            if ($link['link_type'] & SQ_SC_LINK_SIGNIFICANT) {


                // update the parents to tell them that they are going to be one kid less
                $sub_sql = 'SELECT
                                SUBSTR(t.treeid, 1, (LENGTH(t.treeid) - '.SQ_CONF_ASSET_TREE_SIZE.'))
                            FROM
                                sq_ast_lnk_tree t
                            WHERE
                                t.linkid = :linkid';

                $sql = 'UPDATE
                            sq_ast_lnk_tree
                        SET
                            num_kids = num_kids - 1
                        WHERE
                            treeid in ('.$sub_sql.')';

                $query = MatrixDAL::preparePdoQuery($sql);
                MatrixDAL::bindValueToPdo($query, 'linkid', $link['linkid']);
                MatrixDAL::execPdoQuery($query);


                // we can delete all the links under these nodes because it will be a clean start
                // when we insert into the gap's we create below
                $sub_sql   = 'SELECT
                                    ct.treeid
                            FROM
                                sq_ast_lnk_tree pt,
                                sq_ast_lnk_tree ct
                            WHERE
                                    pt.linkid = :linkid
                                AND ct.treeid LIKE pt.treeid || '.MatrixDAL::quote('%').'
                                AND ct.treeid > pt.treeid';

                $sql = 'DELETE FROM
                            sq_ast_lnk_tree
                        WHERE
                            treeid in ('.$sub_sql.')';

                $query = MatrixDAL::preparePdoQuery($sql);
                MatrixDAL::bindValueToPdo($query, 'linkid', $link['linkid']);
                MatrixDAL::execPdoQuery($query);

                // we are going to set the treeid nodes that this link is associated
                // with to zero so that we can find it as a gap when we createLink() later on

                $sql = 'UPDATE
                            sq_ast_lnk_tree
                        SET
                            linkid = :linkid,
                            num_kids = :num_kids

                        WHERE
                            linkid = :old_linkid';

                $query = MatrixDAL::preparePdoQuery($sql);
                MatrixDAL::bindValueToPdo($query, 'linkid',     '0');
                MatrixDAL::bindValueToPdo($query, 'num_kids',   '0');
                MatrixDAL::bindValueToPdo($query, 'old_linkid', $link['linkid']);
                MatrixDAL::execPdoQuery($query);

            }//end if significant link

            // move 'em up, higher
            $sql = 'UPDATE
                        sq_ast_lnk
                    SET
                        sort_order = sort_order - 1
                    WHERE
                            majorid = :majorid
                        AND
                            sort_order > :sort_order';

            $query = MatrixDAL::preparePdoQuery($sql);
            MatrixDAL::bindValueToPdo($query, 'majorid',    $link['majorid']);
            MatrixDAL::bindValueToPdo($query, 'sort_order', $link['sort_order']);
            MatrixDAL::execPdoQuery($query);

            $sql = 'DELETE FROM
                        sq_ast_lnk
                    WHERE
                        linkid  = :linkid';

            $query = MatrixDAL::preparePdoQuery($sql);
            MatrixDAL::bindValueToPdo($query, 'linkid', $link['linkid']);
            MatrixDAL::execPdoQuery($query);

            // tell the asset it has updated
            $asset =& $GLOBALS['SQ_SYSTEM']->am->getAsset($assetid, $type_code);
            $asset->linksUpdated();

            $GLOBALS['SQ_SYSTEM']->doTransaction('COMMIT');
            $GLOBALS['SQ_SYSTEM']->restoreDatabaseConnection();
            unset($links[$linkid]);
        }//end if
    }//end foreach

    if ($errors) {          // no links
        printUpdateStatus('FAILED');
        continue;
    }

    if (empty($links)) {            // no links
        $asset =& $GLOBALS['SQ_SYSTEM']->am->getAsset($assetid, $type_code);
        if (!$GLOBALS['SQ_SYSTEM']->am->createAssetLink($map_asset, $asset, SQ_LINK_TYPE_2, 'Orphaned Asset')) {
            printUpdateStatus('FAILED');
            continue;
        }
        $updated = TRUE;
    }

    if ($updated) {
        printUpdateStatus('OK');
    } else {
        printUpdateStatus('--');
    }
    $GLOBALS['SQ_SYSTEM']->am->forgetAsset($asset);

}//end foreach

$GLOBALS['SQ_SYSTEM']->restoreRunLevel();


function printAssetName($assetid)
{
    $asset_info = $GLOBALS['SQ_SYSTEM']->am->getAssetInfo($assetid);
    $str = '[ #'.$assetid.' ]'.$asset_info[$assetid]['name'];
    if (strlen($str) > 66) {
        $str = substr($str, 0, 66).'...';
    }
    printf ('%s%'.(70 - strlen($str)).'s', $str,'');

}//end printAssetName()


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

}//end printUpdateStatus()


?>