#
# Copyright (c) 2007 Elliot Peele <elliot@bentlogic.net>
#
# Licensed under the GNU General Public License Version 2
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#

from conary import callbacks
from conaryBackend import PackageKitConaryBackend

class UpdateCallback(callbacks.UpdateCallback, PackageKitConaryBackend):
    def resolvingDependencies(self):
        self.status('Resolving Dependencies')

    def creatingRollback(self):
        self.status('Creating Rollback')

    def committingTransaction(self):
        self.status('Committing Transaction')

    def downloadingFileContents(self, got, need):
        self.status('Downloading files for changeset')

    def downloadingChangeSet(self, got, need):
        self.status('Downloading')

    def requestingFileContents(self):
        self.status('Requesting File Contents')

    def requestingChangeSet(self):
        self.status('Requesting Changeset')

    def preparingUpdate(self, troveNum, troveCount, add=0):
        if troveNum > 0 and troveCount > 0:
            sub_percent = (add + troveNum) / (2 * float(troveCount)) * 100
            self.sub_percentage(sub_percent)

        if troveNum != 0:
            troveNum -= 1

        job = self.currentJob[troveNum]
        name = job[0]
        version, flavor = job[2]

        id = self.get_package_id(name, version, flavor)

        self.package(id, 1, '')

    def creatingDatabaseTransaction(self, troveNum, troveCount):
        self.preparingUpdate(troveNum, troveCount, add=troveCount)

    def setChangesetHunk(self, num, total):
        pass

    def setUpdateHunk(self, hunk, hunkCount):
        if hunk > 0 and hunkCount > 0:
            percentage = hunk / float(hunkCount) * 100.0
            self.percentage(percentage)

    def setUpdateJob(self, job):
        self.currentJob = job

    def updateDone(self):
        self.currentJob = None

    def done(self):
        self.status('done')

    def tagHandlerOutput(self, tag, msg, stderr = False):
        pass

    def troveScriptOutput(self, typ, msg):
        pass

    def troveScriptFailure(self, typ, errcode):
        pass

    def __init__(self, cfg=None, args=None):
        callbacks.UpdateCallback.__init__(self)
        if cfg:
            self.setTrustThreshold(cfg.trustThreshold)
        PackageKitConaryBackend.__init__(self, args)

        self.currentJob = None