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