Version 4.2.0

applications/downloads/setup/upg_40000/upgrade.php

@@ -1,13 +1,12 @@
 <?php
 /**
  * @brief		4.0.0 Upgrade Code
- * @author		<a href='http://www.invisionpower.com'>Invision Power Services, Inc.</a>
- * @copyright	(c) 2001 - 2016 Invision Power Services, Inc.
- * @license		http://www.invisionpower.com/legal/standards/
- * @package		IPS Community Suite
+ * @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	Downloads
  * @since		19 Nov 2013
- * @version		SVN_VERSION_NUMBER
  */
 
 namespace IPS\downloads\setup\upg_40000;
@@ -320,6 +319,81 @@ class _Upgrade
 	{
 		return "Fixing watermark configuration";
 	}
+
+	/**
+	 * Set a default primary screenshot for each file
+	 *
+	 * @return	array	If returns TRUE, upgrader will proceed to next step. If it returns any other value, it will set this as the value of the 'extra' GET parameter and rerun this step (useful for loops)
+	 */
+	public function step5()
+	{
+		$perCycle	= 500;
+		$did		= 0;
+		$limit		= intval( \IPS\Request::i()->extra );
+		
+		/* Try to prevent timeouts to the extent possible */
+		$cutOff		= \IPS\core\Setup\Upgrade::determineCutoff();
+		
+		foreach( \IPS\Db::i()->select( '*', 'downloads_files', null, 'file_id ASC', array( $limit, $perCycle ) ) as $file )
+		{
+			if ( $cutOff !== NULL AND time() >= $cutOff )
+			{
+				return ( $limit + $did );
+			}
+			
+			$did++;
+			
+			/* Try to get the default screenshot if one is set */
+			try
+			{
+				$defaultScreenshot = \IPS\Db::i()->select( 'record_id', 'downloads_files_records', array( "record_file_id=? AND record_type IN('sslink','ssupload') AND record_default=?", $file['file_id'], 1 ) )->first();
+
+				\IPS\Db::i()->update( 'downloads_files', array( 'file_primary_screenshot' => $defaultScreenshot ), array( 'file_id=?', $file['file_id'] ) );
+			}
+			catch( \UnderflowException $e )
+			{
+				/* Get the first screenshot for the file and set the primary screenshot flag */
+				try
+				{
+					$firstScreenshot = \IPS\Db::i()->select( 'record_id', 'downloads_files_records', array( "record_file_id=? AND record_type IN('sslink','ssupload')", $file['file_id'] ), 'record_id ASC', array( 0, 1 ) )->first();
+
+					\IPS\Db::i()->update( 'downloads_files', array( 'file_primary_screenshot' => $firstScreenshot ), array( 'file_id=?', $file['file_id'] ) );
+				}
+				catch( \UnderflowException $e )
+				{
+					continue;
+				}
+			}
+		}
+		
+		if ( $did )
+		{
+			return ( $limit + $did );
+		}
+		else
+		{
+			unset( $_SESSION['_step5Count_40000_downloads'] );
+			
+			return true;
+		}
+	}
+
+	/**
+	 * Custom title for this step
+	 *
+	 * @return string
+	 */
+	public function step5CustomTitle()
+	{
+		$limit = isset( \IPS\Request::i()->extra ) ? \IPS\Request::i()->extra : 0;
+		
+		if ( !isset( $_SESSION['_step5Count_40000_downloads'] ) )
+		{
+			$_SESSION['_step5Count_40000_downloads'] = \IPS\Db::i()->select( 'COUNT(*)', 'downloads_files' )->first();
+		}
+		
+		return "Updating default screenshots (Updated so far: " . ( ( $limit > $_SESSION['_step5Count_40000_downloads'] ) ? $_SESSION['_step5Count_40000_downloads'] : $limit ) . ' out of ' . $_SESSION['_step5Count_40000_downloads'] . ")";
+	}
 	
 	/**
 	 * Finish - This is run after all apps have been upgraded