Changed stack packages from all packages in upgrade option

ee/cli/plugins/stack_upgrade.py

@@ -15,43 +15,36 @@ class EEStackUpgradeController(CementBaseController):
         label = 'upgrade'
         stacked_on = 'stack'
         stacked_type = 'nested'
-        description = ('UPgrade stack safely')
+        description = ('Upgrade stack safely')
         arguments = [
+            (['--all'],
+                dict(help='Upgrade all stack', action='store_true')),
+            (['--web'],
+                dict(help='Upgrade web stack', action='store_true')),
+            (['--admin'],
+                dict(help='Upgrade admin tools stack', action='store_true')),
+            (['--mail'],
+                dict(help='Upgrade mail server stack', action='store_true')),
+            (['--mailscanner'],
+                dict(help='Upgrade mail scanner stack', action='store_true')),
+            (['--nginx'],
+                dict(help='Upgrade Nginx stack', action='store_true')),
+            (['--php'],
+                dict(help='Upgrade PHP stack', action='store_true')),
+            (['--mysql'],
+                dict(help='Upgrade MySQL stack', action='store_true')),
+            (['--hhvm'],
+                dict(help='Upgrade HHVM stack', action='store_true')),
             (['--php56'],
                 dict(help="Upgrade to PHP5.6 from PHP5.5",
                      action='store_true')),
             (['--no-prompt'],
                 dict(help="Upgrade Packages without any prompt",
                      action='store_true')),
-
             ]
 
     @expose(hide=True)
     def upgrade_php56(self):
-        pass
-
-    @expose(hide=True)
-    def default(self):
-        # All package update
-        if ((not self.app.pargs.php56)):
-            # apt-get update
-            EEAptGet.update(self)
-            # display packges update
-            EEAptGet.check_upgrade(self)
-            Log.info(self, "During package update process non nginx-cached"
-                     " parts of your site may remain down")
-            # Check prompt
-            if (not self.app.pargs.no_prompt):
-                start_upgrade = input("Do you want to continue:[y/N]")
-                if start_upgrade != "Y" and start_upgrade != "y":
-                    Log.error(self, "Not starting package update")
-            # Update packages
-            Log.info(self, "Updating packages, please wait...")
-            EEAptGet.dist_upgrade(self)
-            Log.info(self, "Successfully updated packages")
-
-        # PHP 5.6 to 5.6
-        elif (self.app.pargs.php56):
         if EEVariables.ee_platform_distro == "Ubuntu":
             if not os.path.isfile("/etc/apt/sources.list.d/"
                                   "ondrej-php5-trusty.list"):
@@ -97,5 +90,103 @@ class EEStackUpgradeController(CementBaseController):
                                                      "/20-xedbug.ini"])
 
         Log.info(self, "Successfully updated from PHP 5.5 to PHP 5.6")
+
+    @expose(hide=True)
+    def default(self):
+        # All package update
+        if ((not self.app.pargs.php56)):
+
+            apt_packages = []
+
+            Log.info(self, "During package update process non nginx-cached"
+                     " parts of your site may remain down")
+            # Check prompt
+            if (not self.app.pargs.no_prompt):
+                start_upgrade = input("Do you want to continue:[y/N]")
+                if start_upgrade != "Y" and start_upgrade != "y":
+                    Log.error(self, "Not starting package update")
+
+            if ((not self.app.pargs.web) and (not self.app.pargs.nginx) and
+               (not self.app.pargs.php) and (not self.app.pargs.mysql) and
+               (not self.app.pargs.postfix) and (not self.app.pargs.hhvm) and
+               (not self.app.pargs.mailscanner) and (not self.app.pargs.all)):
+                self.app.pargs.web = True
+
+            if self.app.pargs.all:
+                self.app.pargs.web = True
+                self.app.pargs.mail = True
+
+            if self.app.pargs.web:
+                self.app.pargs.nginx = True
+                self.app.pargs.php = True
+                self.app.pargs.mysql = True
+                self.app.pargs.postfix = True
+                self.app.pargs.hhvm = True
+
+            if self.app.pargs.mail:
+                self.app.pargs.nginx = True
+                self.app.pargs.php = True
+                self.app.pargs.mysql = True
+                self.app.pargs.postfix = True
+
+                if EEAptGet.is_installed(self, 'dovecot-core'):
+                    apt_packages = apt_packages + EEVariables.ee_mail
+                    self.app.pargs.mailscanner = True
+                else:
+                    Log.info(self, "Mail server is not installed")
+
+            if self.app.pargs.nginx:
+                if EEVariables.ee_platform_distro == 'debian':
+                    check_nginx = 'nginx-extras'
+                else:
+                    check_nginx = 'nginx-custom'
+
+                if EEAptGet.is_installed(self, check_nginx):
+                    apt_packages = apt_packages + EEVariables.ee_nginx
+                else:
+                    Log.info(self, "Nginx is not already installed")
+
+            if self.app.pargs.php:
+                if EEAptGet.is_installed(self, 'php5-fpm'):
+                    apt_packages = apt_packages + EEVariables.ee_php
+                else:
+                    Log.info(self, "PHP is not installed")
+
+            if self.app.pargs.hhvm:
+                if EEAptGet.is_installed(self, 'hhvm'):
+                    apt_packages = apt_packages + EEVariables.ee_hhvm
+                else:
+                    Log.info(self, "HHVM is not installed")
+
+            if self.app.pargs.mysql:
+                if EEAptGet.is_installed(self, 'mariadb-server'):
+                    apt_packages = apt_packages + EEVariables.ee_mysql
+                else:
+                    Log.info(self, "MySQL is not installed")
+
+            if self.app.pargs.postfix:
+                if EEAptGet.is_installed(self, 'postfix'):
+                    apt_packages = apt_packages + EEVariables.ee_postfix
+                else:
+                    Log.info(self, "Postfix is not installed")
+
+            if self.app.pargs.mailscanner:
+                if EEAptGet.is_installed(self, 'amavisd-new'):
+                    apt_packages = (apt_packages + EEVariables.ee_mailscanner)
+                else:
+                    Log.info(self, "MailScanner is not installed")
+
+            if len(apt_packages):
+                # apt-get update
+                EEAptGet.update(self)
+
+                # Update packages
+                Log.info(self, "Updating packages, please wait...")
+                EEAptGet.install(self, apt_packages)
+                Log.info(self, "Successfully updated packages")
+
+        # PHP 5.6 to 5.6
+        elif (self.app.pargs.php56):
+            self.upgrade_php56()
         else:
             self.app.args.print_help()