import os
import zipfile

from buildstream.utils import sha256sum

from buildstream.testing import Repo


class Zip(Repo):

    def create(self, directory):
        archive = os.path.join(self.repo, 'file.zip')

        old_dir = os.getcwd()
        os.chdir(directory)
        with zipfile.ZipFile(archive, "w") as zipfp:
            for root, dirs, files in os.walk('.'):
                names = dirs + files
                names = [os.path.join(root, name) for name in names]

                for name in names:
                    zipfp.write(name)

        os.chdir(old_dir)

        return sha256sum(archive)

    def source_config(self, ref=None):
        archive = os.path.join(self.repo, 'file.zip')
        config = {
            'kind': 'zip',
            'url': 'file://' + archive,
            'directory': '',
            'base-dir': ''
        }
        if ref is not None:
            config['ref'] = ref

        return config