import os.path
from Logger import Logger
from CommandUtils import CommandUtils
from constants import constants

class ChrootUtils(object):

    def __init__(self, logName=None, logPath=None):
        if logName is None:
            logName = "ChrootUtils"
        if logPath is None:
            logPath = constants.logPath
        self.logName = logName
        self.logPath = logPath
        self.logger = Logger.getLogger(logName, logPath)

    def createChroot(self, chrootName):
        chrootID = constants.buildRootPath + "/" + chrootName
        if os.path.isdir(chrootID):
            if not self.destroyChroot(chrootID):
                self.logger.error("Given chroot " + chrootID +
                                  " is already exists. unable to destroy it ")
                return False, None
        # need to add timeout for this step
        # http://stackoverflow.com/questions/1191374/subprocess-with-timeout
        cmdUtils = CommandUtils()
        returnVal = cmdUtils.runCommandInShell("mkdir -p " + chrootID)
        if not returnVal:
            self.logger.error("Unable to create chroot:" + chrootID + ".Unknown error.")
            return False, None
        return True, chrootID

    def destroyChroot(self, chrootID):
        # need to add timeout for this step
        # http://stackoverflow.com/questions/1191374/subprocess-with-timeout
        cmdUtils = CommandUtils()
        returnVal = cmdUtils.runCommandInShell("./clean-up-chroot.py " + chrootID)
        if not returnVal:
            self.logger.error("Unable to destroy chroot:" + chrootID + ".Unknown error.")
            return False

        returnVal = cmdUtils.runCommandInShell("rm -rf " + chrootID)
        if not returnVal:
            self.logger.error("Unable to destroy chroot:" + chrootID + ".Unknown error.")
            return False
        self.logger.info("Successfully destroyed chroot:" + chrootID)
        return True