Version 5.0.0 beta 1

applications/core/setup/upg_106118/upgrade.php

@@ -12,16 +12,23 @@
 namespace IPS\core\setup\upg_106118;
 
 /* To prevent PHP errors (extending class does not exist) revealing path */
-if ( !\defined( '\IPS\SUITE_UNIQUE_KEY' ) )
+
+use IPS\core\AdminNotification;
+use IPS\Db;
+use IPS\Settings;
+use UnderflowException;
+use function defined;
+
+if ( !defined( '\IPS\SUITE_UNIQUE_KEY' ) )
 {
-	header( ( isset( $_SERVER['SERVER_PROTOCOL'] ) ? $_SERVER['SERVER_PROTOCOL'] : 'HTTP/1.0' ) . ' 403 Forbidden' );
+	header( ( $_SERVER['SERVER_PROTOCOL'] ?? 'HTTP/1.0' ) . ' 403 Forbidden' );
 	exit;
 }
 
 /**
  * 4.6.4 Upgrade Code
  */
-class _Upgrade
+class Upgrade
 {
 	/**
 	 * ...
@@ -33,6 +40,33 @@ class _Upgrade
 		
 
 
+		return TRUE;
+	}
+
+	/**
+	 * Finish step
+	 *
+	 * @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 finish()
+	{
+		/* Enable the onboarding system for those that still have the admincp notification active. */
+		if( AdminNotification::find('core', 'ConfigurationError', 'marketplaceSetup') )
+		{
+			/* If the upgrade is from an old version, the settings may not exist yet. */
+			try
+			{
+				Db::i()->select( '*', 'core_sys_conf_settings', [ 'conf_key=?', 'mp_onboard_complete' ] )->setKeyField('conf_key')->setValueField('conf_default' )->first();
+			}
+			catch( UnderflowException $e )
+			{
+				Db::i()->insert( 'core_sys_conf_settings', array( 'conf_key' => 'mp_onboard_complete', 'conf_value' => 0, 'conf_default' => 1, 'conf_app' => 'core' ), TRUE );
+			}
+
+			/* Set onboard complete flag */
+			Settings::i()->changeValues( array( 'mp_onboard_complete' => 0 ) );
+		}
+
 		return TRUE;
 	}
 }