# -*- coding: utf-8 -*-

from __future__ import with_statement

import os
import time
import subprocess
import sys

try:
    import caffeine
except ImportError:
    pass

from module.plugins.internal.Addon import Addon
from module.plugins.internal.misc import encode, Expose, fsjoin


class Kernel32(object):
    ES_AWAYMODE_REQUIRED = 0x00000040
    ES_CONTINUOUS        = 0x80000000
    ES_DISPLAY_REQUIRED  = 0x00000002
    ES_SYSTEM_REQUIRED   = 0x00000001
    ES_USER_PRESENT      = 0x00000004


class AntiStandby(Addon):
    __name__    = "AntiStandby"
    __type__    = "hook"
    __version__ = "0.17"
    __status__  = "testing"

    __config__ = [("activated", "bool", "Activated"                       , False),
                  ("hdd"      , "bool", "Prevent HDD standby"             , True ),
                  ("system"   , "bool", "Prevent OS standby"              , True ),
                  ("display"  , "bool", "Prevent display standby"         , False),
                  ("interval" , "int" , "HDD touching interval in seconds", 25   )]

    __description__ = """Prevent OS, HDD and display standby"""
    __license__     = "GPLv3"
    __authors__     = [("Walter Purcaro", "vuolter@gmail.com")]


    TMP_FILE = ".antistandby"


    def init(self):
        self.pid   = None
        self.mtime = 0


    def activate(self):
        hdd     = self.config.get('hdd')
        system  = not self.config.get('system')
        display = not self.config.get('display')

        if hdd:
            self.periodical.start(self.config.get('interval'), threaded=True)

        if os.name == "nt":
            self.win_standby(system, display)

        elif sys.platform == "darwin":
            self.osx_standby(system, display)

        else:
            self.linux_standby(system, display)


    def deactivate(self):
        self.remove(self.TMP_FILE, trash=False)

        if os.name == "nt":
            self.win_standby(True)

        elif sys.platform == "darwin":
            self.osx_standby(True)

        else:
            self.linux_standby(True)


    @Expose
    def win_standby(self, system=True, display=True):
        import ctypes

        set = ctypes.windll.kernel32.SetThreadExecutionState

        if system:
            if display:
                set(Kernel32.ES_CONTINUOUS)
            else:
                set(Kernel32.ES_CONTINUOUS | Kernel32.ES_DISPLAY_REQUIRED)
        else:
            if display:
                set(Kernel32.ES_CONTINUOUS | Kernel32.ES_SYSTEM_REQUIRED)
            else:
                set(Kernel32.ES_CONTINUOUS | Kernel32.ES_SYSTEM_REQUIRED | Kernel32.ES_DISPLAY_REQUIRED)


    @Expose
    def osx_standby(self, system=True, display=True):
        try:
            if system:
                caffeine.off()
            else:
                caffeine.on(display)

        except NameError:
            self.log_warning(_("Unable to change power state"),
                             _("caffeine lib not found"))

        except Exception, e:
            self.log_warning(_("Unable to change power state"), e)


    @Expose
    def linux_standby(self, system=True, display=True):
        try:
            if system:
                if self.pid:
                    self.pid.kill()

            elif not self.pid:
                self.pid = subprocess.Popen(["caffeine"])

        except Exception, e:
            self.log_warning(_("Unable to change system power state"), e)

        try:
            if display:
                subprocess.call(["xset", "+dpms", "s", "default"])
            else:
                subprocess.call(["xset", "-dpms", "s", "off"])

        except Exception, e:
            self.log_warning(_("Unable to change display power state"), e)


    @Expose
    def touch(self, path):
        with open(path, 'w'):
            os.utime(path, None)

        self.mtime = time.time()


    @Expose
    def max_mtime(self, path):
        return max(0, 0,
                   *(os.path.getmtime(fsjoin(root, file))
                        for root, dirs, files in os.walk(encode(path), topdown=False)
                            for file in files))


    def periodical_task(self):
        if self.config.get('hdd') is False:
            return

        if (self.pyload.threadManager.pause or
            not self.pyload.api.isTimeDownload() or
            not self.pyload.threadManager.getActiveFiles()):
            return

        dl_folder = self.pyload.config.get('general', 'download_folder')
        if (self.max_mtime(dl_folder) - self.mtime) < self.periodical.interval:
            return

        self.touch(self.TMP_FILE)