from couchpotato.core.event import addEvent, fireEvent
from couchpotato.core.logger import CPLog
from couchpotato.core.plugins.base import Plugin
from couchpotato.environment import Env
from git.repository import LocalRepository
import time

log = CPLog(__name__)


class Updater(Plugin):

    git = 'git://github.com/CouchPotato/CouchPotato.git'

    running = False
    version = None
    updateFailed = False
    updateAvailable = False
    updateVersion = None
    lastCheck = 0

    def __init__(self):

        self.repo = LocalRepository(Env.get('app_dir'))

        fireEvent('schedule.interval', 'updater.check', self.check, hours = 6)

        addEvent('app.load', self.check)

    def getVersion(self):

        if not self.version:
            try:
                output = self.repo.getHead() # Yes, please
                log.debug('Git version output: %s' % output.hash)
                self.version = output.hash
            except Exception, e:
                log.error('Failed using GIT updater, running from source, you need to have GIT installed. %s' % e)
                return 'No GIT'

        return self.version

    def check(self):

        if self.updateAvailable or self.isDisabled():
            return

        current_branch = self.repo.getCurrentBranch().name

        for branch in self.repo.getRemoteByName('origin').getBranches():
            if current_branch == branch.name:

                local = self.repo.getHead()
                remote = branch.getHead()

                if local.getDate() < remote.getDate():
                    if self.conf('automatic') and not self.updateFailed:
                        self.doUpdate()
                    else:
                        self.updateAvailable = True
                        self.updateVersion = remote.hash

        self.lastCheck = time.time()

    def doUpdate(self):
        try:
            log.info('Updating to latest version');
            self.repo.pull()
            return True
        except Exception, e:
            log.error('Failed updating via GIT: %s' % e)

        self.updateFailed = True

        return False

    def isEnabled(self):
        return Plugin.isEnabled(self) and Env.get('uses_git')