from ee.core.shellexec import EEShellExec
from ee.core.aptget import EEAptGet
from ee.core.services import EEService
from cement.core.controller import CementBaseController, expose
from cement.core import handler, hook
import os


def clean_plugin_hook(app):
    # do something with the ``app`` object here.
    pass


class EECleanController(CementBaseController):
    class Meta:
        label = 'clean'
        stacked_on = 'base'
        stacked_type = 'nested'
        description = 'clean command cleans different cache with following \
                        options'
        arguments = [
            (['--all'],
                dict(help='clean all cache', action='store_true')),
            (['--fastcgi'],
                dict(help='clean fastcgi cache', action='store_true')),
            (['--memcache'],
                dict(help='clean memcache', action='store_true')),
            (['--opcache'],
                dict(help='clean opcode cache cache', action='store_true'))
            ]

    @expose(hide=True)
    def default(self):
        # TODO Default action for ee clean command here
            if (not (self.app.pargs.all or self.app.pargs.fastcgi or
                     self.app.pargs.memcache or self.app.pargs.opcache)):
                self.clean_fastcgi()
            if self.app.pargs.all:
                        self.clean_memcache()
                        self.clean_fastcgi()
                        self.clean_opcache()
            if self.app.pargs.fastcgi:
                self.clean_fastcgi()
            if self.app.pargs.memcache:
                self.clean_memcache()
            if self.app.pargs.opcache:
                self.clean_opcache()

    @expose(hide=True)
    def clean_memcache(self):
        print("in memcache..")
        pkg = EEAptGet()
        if(pkg.is_installed("memcached")):
            print("memcache is installed...")
            EEService.restart_service(self, "memcached")
            print("Cleaning memcache..")
        else:
            print("memcache is not installed..")

    @expose(hide=True)
    def clean_fastcgi(self):
        if(os.path.isdir("/var/run/nginx-cache")):
            print("Cleaning fastcgi...")
            EEShellExec.cmd_exec(self, "rm -rf /var/run/nginx-cache/*")
        else:
            print("Error occur while Cleaning fastcgi..")

    @expose(hide=True)
    def clean_opcache(self):
        try:
            print("Cleaning opcache.... ")
            EEShellExec.cmd_exec(self, "wget --no-check-certificate --spider"
                                 " -q https://127.0.0.1:22222/cache/opcache"
                                 "/opgui.php?page=reset")
        except OSError:
            print("Unable to clean opache..")


def load(app):
    # register the plugin class.. this only happens if the plugin is enabled
    handler.register(EECleanController)
    # register a hook (function) to run after arguments are parsed.
    hook.register('post_argument_parsing', clean_plugin_hook)