sirosen/SALVE

from salve import logger, paths
from salve.action.copy.base import CopyAction
from salve.filesys import access_codes
from salve.context import ExecutionContext


class DirCopyAction(CopyAction):
    def verify_can_exec(self, filesys):
        """
        Check to ensure that execution can proceed without errors.
        Ensures that the the target directory is writable.
        """
        ExecutionContext().transition(ExecutionContext.phases.VERIFICATION)

        logstr = ('DirCopy: Checking source is readable + traversable, ' +
                  '{0}'.format(self.dst))
        logger.info('{0}: {1}'.format(self.file_context, logstr))

        if not filesys.access(self.src, access_codes.R_OK | access_codes.X_OK):
            return self.verification_codes.UNREADABLE_SOURCE

        logger.info(
            ('{0}: DirCopy: Checking target is writeable, \"{1}\"').format(
                self.file_context, self.dst)
            )

        if not filesys.access(paths.dirname(self.dst), access_codes.W_OK):
            return self.verification_codes.UNWRITABLE_TARGET

        return self.verification_codes.OK

    def execute(self, filesys):
        """
        Copy a directory tree from one location to another.
        """
        vcode = self.verify_can_exec(filesys)

        if vcode == self.verification_codes.UNREADABLE_SOURCE:
            logstr = "DirCopy: Non-Readable source directory \"%s\"" % self.src
            logger.warn('{0}: {1}'.format(self.file_context, logstr))
            return

        if vcode == self.verification_codes.UNWRITABLE_TARGET:
            logstr = "DirCopy: Non-Writable target directory \"%s\"" % self.dst
            logger.warn('{0}: {1}'.format(self.file_context, logstr))
            return

        ExecutionContext().transition(ExecutionContext.phases.EXECUTION)

        logstr = ('Performing Directory Copy \"%s\" -> \"%s\"' %
                  (self.src, self.dst))
        logger.info('{0}: {1}'.format(self.file_context, logstr))

        filesys.copy(self.src, self.dst)