refs #2976 updates can be marked a major

git-svn-id: http://dev.piwik.org/svn/trunk@6790 59fd770c-687e-43c8-a1e3-f5a4ff64c105

1 files changed, 27 insertions, 0 deletions

diff --git a/core/Updater.php b/core/Updater.php
index ae3821d080..b993c9b46a 100644
--- a/core/Updater.php
+++ b/core/Updater.php
@@ -98,6 +98,33 @@ class Piwik_Updater
 	}
 
 	/**
+	 * Does one of the new versions involve a major database update?
+	 * 
+	 * @return bool
+	 */
+	public function hasMajorDbUpdate()
+	{
+		foreach($this->componentsWithUpdateFile as $componentName => $componentUpdateInfo) 
+		{
+			foreach($componentUpdateInfo as $file => $fileVersion)
+			{
+				require_once $file;
+				
+				$className = $this->getUpdateClassName($componentName, $fileVersion);
+				if(class_exists($className, false))
+				{
+					$isMajor = call_user_func( array($className, 'isMajorUpdate'));
+					if ($isMajor) {
+						return true;
+					}
+				}
+			}
+		}
+		
+		return false;
+	}
+
+	/**
 	 * Returns the list of SQL queries that would be executed during the update
 	 * 
 	 * @return array of SQL queries