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

from __future__ import with_statement

import sys
import tarfile

from module.plugins.internal.Extractor import Extractor, ArchiveError, CRCError
from module.plugins.internal.misc import encode


class UnTar(Extractor):
    __name__    = "UnTar"
    __type__    = "extractor"
    __version__ = "0.01"
    __status__  = "stable"

    __description__ = """TAR extractor plugin"""
    __license__     = "GPLv3"
    __authors__     = [("Walter Purcaro", "vuolter@gmail.com")]


    VERSION = "%s.%s.%s" % (sys.version_info[0], sys.version_info[1], sys.version_info[2])


    @classmethod
    def isarchive(cls, filename):
        return tarfile.is_tarfile(encode(filename))


    @classmethod
    def find(cls):
        return sys.version_info[:2] >= (2, 5)


    def list(self, password=None):
        with tarfile.open(self.target) as t:
            return t.getnames()


    def verify(self, password=None):
        try:
            t = tarfile.open(self.target, errorlevel=1)

        except tarfile.CompressionError, e:
            raise CRCError(e)

        except (OSError, tarfile.TarError), e:
            raise ArchiveError(e)

        else:
            t.close()


    def extract(self, password=None):
        self.verify()

        try:
            with tarfile.open(self.target, errorlevel=2) as t:
                t.extractall(self.dest)

        except tarfile.ExtractError, e:
            self.log_warning(e)

        except tarfile.CompressionError, e:
            raise CRCError(e)

        except (OSError, tarfile.TarError), e:
            raise ArchiveError(e)