from cement.core.controller import CementBaseController, expose
from cement.core import handler, hook
from ee.core.logging import Log
from ee.core.variables import EEVariables
from ee.core.aptget import EEAptGet
from ee.core.apt_repo import EERepo
from ee.core.services import EEService
from ee.core.fileutils import EEFileUtils
import configparser
import os


class EEStackUpgradeController(CementBaseController):
    class Meta:
        label = 'upgrade'
        stacked_on = 'stack'
        stacked_type = 'nested'
        description = ('UPgrade stack safely')
        arguments = [
            (['--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"):
                    Log.error(self, "Unable to find PHP 5.5")
            else:
                if not(os.path.isfile(EEVariables.ee_repo_file_path) and
                       EEFileUtils.grep(self, EEVariables.ee_repo_file_path,
                                        "php55")):
                    Log.error(self, "Unable to find PHP 5.5")

            Log.info(self, "During PHP 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 PHP package update")

            if EEVariables.ee_platform_distro == "Ubuntu":
                EERepo.remove(self, ppa="ppa:ondrej/php5")
                EERepo.add(self, ppa=EEVariables.ee_php_repo)
            else:
                EEAptGet.purge(self, ["php5-xdebug"])
                EEFileUtils.searchreplace(self, EEVariables.ee_repo_file_path,
                                          "php55", "php56")

            Log.info(self, "Updating apt-cache, please wait...")
            EEAptGet.update(self)
            Log.info(self, "Installing packages, please wait ...")
            EEAptGet.install(self, EEVariables.ee_php)

            if EEVariables.ee_platform_distro == "debian":
                EEShellExec.cmd_exec(self, "pear install xdebug")

                with open("/etc/php5/mods-available/xdebug.ini",
                          encoding='utf-8', mode='a') as myfile:
                    myfile.write(";zend_extension=/usr/lib/php5/20131226/"
                                 "xdebug.so")

                EEFileUtils.create_symlink(self, ["/etc/php5/mods-available/"
                                           "xdebug.ini", "/etc/php5/fpm/conf.d"
                                                         "/20-xedbug.ini"])

            Log.info(self, "Successfully updated from PHP 5.5 to PHP 5.6")
        else:
            self.app.args.print_help()