Seditio Source
Root |
./othercms/ips_4.3.4/applications/cms/extensions/core/Queue/RebuildRecords.php
<?php
/**
 * @brief        Background Task: Rebuild database records
 * @author        <a href='https://www.invisioncommunity.com'>Invision Power Services, Inc.</a>
 * @copyright    (c) Invision Power Services, Inc.
 * @license        https://www.invisioncommunity.com/legal/standards/
 * @package        Invision Community
 * @subpackage    Content
 * @since        13 Jun 2014
 */

namespace IPS\cms\extensions\core\Queue;

/* To prevent PHP errors (extending class does not exist) revealing path */
if ( !defined( '\IPS\SUITE_UNIQUE_KEY' ) )
{
   
header( ( isset( $_SERVER['SERVER_PROTOCOL'] ) ? $_SERVER['SERVER_PROTOCOL'] : 'HTTP/1.0' ) . ' 403 Forbidden' );
    exit;
}

/**
 * Background Task: Rebuild database records
 */
class _RebuildRecords
{
   
/**
     * @brief Number of content items to rebuild per cycle
     */
   
public $rebuild    = 50;

   
/**
     * Parse data before queuing
     *
     * @param    array    $data
     * @return    array
     */
   
public function preQueueData( $data )
    {
       
$classname  = $data['class'];
       
$databaseId = mb_substr( $classname, 15 );
       
        \
IPS\Log::debug( "Getting preQueueData for " . $classname, 'rebuildRecords' );

        try
        {
           
$data['count'] = (int) \IPS\Db::i()->select( 'COUNT(*)', 'cms_custom_database_' . $databaseId )->first();
        }
        catch( \
Exception $ex )
        {
            throw new \
OutOfRangeException;
        }
       
        if(
$data['count'] == 0 )
        {
            return
null;
        }
       
        return
$data;
    }

   
/**
     * Run Background Task
     *
     * @param    mixed                        $data    Data as it was passed to \IPS\Task::queue()
     * @param    int                            $offset    Offset
     * @return    int                            New offset
     * @throws    \IPS\Task\Queue\OutOfRangeException    Indicates offset doesn't exist and thus task is complete
     */
   
public function run( $data, $offset )
    {
       
$classname  = $data['class'];
       
$databaseId = mb_substr( $classname, 15 );

       
/* Make sure there's even content to parse */
       
if( !class_exists( $classname ) or !isset( $classname::$databaseColumnMap['content'] ) )
        {
            throw new \
IPS\Task\Queue\OutOfRangeException;
        }
       
       
$fixImage = isset( $data['fixImage'] ) ? (boolean) $data['fixImage'] : TRUE;
       
$fixHtml  = isset( $data['fixHtml'] ) ? (boolean) $data['fixHtml'] : FALSE;
       
$fixFurls = isset( $data['fixFurls']) ? (boolean) $data['fixFurls'] : FALSE;
       
$parsed    = 0;
       
$class  = '\IPS\cms\Records' . $databaseId;
       
        if ( \
IPS\Db::i()->checkForTable( 'cms_custom_database_' . $databaseId ) )
        {
            foreach ( \
IPS\Db::i()->select( '*', 'cms_custom_database_' . $databaseId, NULL, 'primary_id_field asc', array( $offset, $this->rebuild ) ) as $row )
            {
               
$record = $class::constructFromData( $row );
               
$record->resetLastComment();
               
$save = FALSE;
               
                if (
$fixImage )
                {
                    if (
$record->record_image and file_exists( \IPS\ROOT_PATH . '/uploads/' . $record->record_image ) )
                    {
                        try
                        {
                           
$record->record_image = (string) \IPS\File::create( 'cms_Records', $record->record_image, file_get_contents( \IPS\ROOT_PATH . '/uploads/' . $record->record_image ) );
                        }
                        catch ( \
Exception $e )
                        {
                           
$record->record_image = NULL;
                        }
                       
$save = TRUE;
                    }
                }
               
                if ( !
$record->record_publish_date )
                {
                   
$record->record_publish_date = $record->record_saved;
                   
$save = TRUE;
                }
               
                if (
$fixFurls and preg_match( '#%([\d\w]{2})#', $record->record_static_furl ) )
                {
                   
$record->record_static_furl = urldecode( $record->record_static_furl );
                   
$save = TRUE;
                }
               
                if (
$save )
                {
                   
$record->save();
                }
               
                if (
$fixHtml )
                {
                   
$fields = iterator_to_array( \IPS\Db::i()->select( '*', 'cms_database_fields', array( 'field_database_id=? AND field_type=? AND field_html=1', $databaseId, 'Editor' ) ) );
                   
                    if (
count( $fields ) )
                    {
                        foreach(
$fields as $field )
                        {
                           
$column = 'field_' . $field['field_id'];
                           
                            if (
$record->member_id and $record->$column )
                            {
                                try
                                {
                                   
$author = \IPS\Member::load( $record->member_id );
                                   
                                   
/* In 3.x this would have been shown as HTML */
                                   
if ( $author->group['g_dohtml'] )
                                    {
                                       
/* This code is copied from IPB3 to ensure it is compatible with data saved */
                                       
$record->$column = str_replace( "&#39;" , "'", $record->$column );
                                       
$record->$column = str_replace( "&#33;" , "!", $record->$column );
                                       
$record->$column = str_replace( "&#036;", "$", $record->$column );
                                       
$record->$column = str_replace( "&#124;", "|", $record->$column );
                                       
$record->$column = str_replace( "&amp;" , "&", $record->$column );
                                       
$record->$column = str_replace( "&gt;"     , ">", $record->$column );
                                       
$record->$column = str_replace( "&lt;"     , "<", $record->$column );
                                       
$record->$column = str_replace( "&#60;" , "<", $record->$column );
                                       
$record->$column = str_replace( "&#62;" , ">", $record->$column );
                                       
$record->$column = str_replace( "&quot;", '"', $record->$column );
                                       
$record->$column = str_replace( '&quot;', '"', $record->$column );
                                       
$record->$column = str_replace( '&lt;', '<', $record->$column );
                                       
$record->$column = str_replace( '&gt;', '>', $record->$column );
                                       
                                       
$record->save();
                                    }
                                }
                                catch( \
OutOfRangeException $e ) { }
                            }        
                        }
                    }
                }

               
$parsed++;
            }
        }
       
        if(
$parsed != $this->rebuild )
        {
            throw new \
IPS\Task\Queue\OutOfRangeException;
        }

        return (
$offset + $this->rebuild );
    }
   
   
/**
     * Get Progress
     *
     * @param    mixed                    $data    Data as it was passed to \IPS\Task::queue()
     * @param    int                        $offset    Offset
     * @return    array( 'text' => 'Doing something...', 'complete' => 50 )    Text explaining task and percentage complete
     * @throws    \OutOfRangeException    Indicates offset doesn't exist and thus task is complete
     */
   
public function getProgress( $data, $offset )
    {
       
$classname  = $data['class'];
       
$databaseId = mb_substr( $classname, 15 );
       
       
$title = ( \IPS\Application::appIsEnabled('cms') ) ? \IPS\cms\Databases::load( $databaseId )->_title : 'Database #' . $databaseId;
        return array(
'text' => \IPS\Member::loggedIn()->language()->addToStack('rebuilding_cms_database_records', FALSE, array( 'sprintf' => array( $title ) ) ), 'complete' => $data['count'] ? ( round( 100 / $data['count'] * $offset, 2 ) ) : 100 );
    }    
}