from testfixtures import log_capture
from tests.base_test import BaseTest
from tests import config
from weevely.core.sessions import SessionURL
from weevely.core import modules
from weevely import utils
from weevely.core import messages
import subprocess
import os

def setUpModule():
    subprocess.check_output("""
BASE_FOLDER="{config.base_folder}/test_file_zip/"
rm -rf "$BASE_FOLDER"

mkdir -p "$BASE_FOLDER/dir1/dir2/dir3/dir4"

echo -n 1 > "$BASE_FOLDER/dir1/f1"
echo -n 1 > "$BASE_FOLDER/dir1/dir2/f2"
echo -n 1 > "$BASE_FOLDER/dir1/dir2/dir3/f3"
echo -n 1 > "$BASE_FOLDER/dir1/dir2/dir3/dir4/f4"

cd "$BASE_FOLDER" && zip -r "test_0.zip" "dir1/"

echo -n 1 > "$BASE_FOLDER/f5"

rm -rf "$BASE_FOLDER/dir1"

chown www-data: -R "$BASE_FOLDER/"

""".format(
config = config
), shell=True)

class FileZip(BaseTest):

    folders_rel = [
        'test_file_zip/dir1',
        'test_file_zip/dir1/dir2',
        'test_file_zip/dir1/dir2/dir3',
        'test_file_zip/dir1/dir2/dir3/dir4',
    ]
    folders_abs = [
        os.path.join(config.base_folder, f)
        for f in folders_rel
    ]

    files_rel = [
        'test_file_zip/dir1/f1',
        'test_file_zip/dir1/dir2/f2',
        'test_file_zip/dir1/dir2/dir3/f3',
        'test_file_zip/dir1/dir2/dir3/dir4/f4',
    ]
    files_abs = [
        os.path.join(config.base_folder, f)
        for f in files_rel
    ]

    zips_rel = [
        'test_file_zip/test_0.zip'
    ]
    zips_abs = [
        os.path.join(config.base_folder, f)
        for f in zips_rel
    ]

    other_file_rel = 'test_file_zip/f5'
    other_file_abs = os.path.join(config.base_folder, other_file_rel)

    def setUp(self):
        self.session = SessionURL(
                    self.url,
                    self.password,
                    volatile = True
                    )

        modules.load_modules(self.session)

        self.run_argv = modules.loaded['file_zip'].run_argv

    def test_compress_decompress(self):

        # Uncompress test.zip
        self.assertTrue(self.run_argv(["--decompress", self.zips_rel[0], 'test_file_zip/' ]));
        for file in self.files_abs:
            self.assertEqual(subprocess.check_output("cat %s" % file, shell=True), b'1')
        for folder in self.folders_abs:
            subprocess.check_call('stat -c %%a "%s"' % folder, shell=True)

        # Compress it again giving szipting folder
        self.assertTrue(self.run_argv(['test_file_zip/test_1.zip', self.folders_rel[0]]));
        self.zips_rel.append('test_file_zip/test_1.zip')
        self.zips_abs.append(os.path.join(config.base_folder, self.zips_rel[-1]))

        # Uncompress the new archive and recheck
        self.assertTrue(self.run_argv(["--decompress", 'test_file_zip/test_1.zip', 'test_file_zip/']));
        for file in self.files_abs:
            self.assertEqual(subprocess.check_output("cat %s" % file, shell=True), b'1')
        for folder in self.folders_abs:
            subprocess.check_call('stat -c %%a "%s"' % folder, shell=True)

    def test_compress_multiple(self):

        # Uncompress test.zip
        self.assertTrue(self.run_argv(["--decompress", self.zips_rel[0], 'test_file_zip/' ]));
        for file in self.files_abs:
            self.assertEqual(subprocess.check_output("cat %s" % file, shell=True), b'1')
        for folder in self.folders_abs:
            subprocess.check_call('stat -c %%a "%s"' % folder, shell=True)

        # Create a new zip adding also other_file
        self.assertTrue(self.run_argv(['test_file_zip/test_2.zip', self.folders_rel[0], self.other_file_rel]));
        self.zips_rel.append('test_file_zip/test_2.zip')
        self.zips_abs.append(os.path.join(config.base_folder, self.zips_rel[-1]))

        # Remove all the files
        subprocess.check_output("rm -rf %s" % self.folders_abs[0], shell=True)
        subprocess.check_output("rm %s" % self.other_file_abs, shell=True)

        # Uncompress the new archive and recheck
        self.assertTrue(self.run_argv(["--decompress", 'test_file_zip/test_2.zip', 'test_file_zip/']));
        for file in self.files_abs:
            self.assertEqual(subprocess.check_output("cat %s" % file, shell=True), b'1')
        for folder in self.folders_abs:
            subprocess.check_call('stat -c %%a "%s"' % folder, shell=True)

        # TODO: here skips the final check since f5 is misplaced
        # on the archive
        # Archive:  /var/www/html/test_file_zip/test_2.zip
        #     testing: dir1/                    OK
        #     testing: dir1/dir2/               OK
        #     testing: dir1/dir2/dir3/          OK
        #     testing: dir1/dir2/dir3/dir4/     OK
        #     testing: dir1/dir2/dir3/dir4/f4   OK
        #     testing: dir1/dir2/dir3/f3        OK
        #     testing: dir1/dir2/f2             OK
        #     testing: dir1/f1                  OK
        #     testing: test_file_zip/f5         OK

        #self.assertEqual(subprocess.check_output("cat %s" % self.other_file_abs, shell=True),'1')


    @log_capture()
    def test_already_exists(self, log_captured):

            # Create a new zip with other_file, with the name test_0.zip
        self.assertIsNone(self.run_argv(['test_file_zip/test_0.zip', self.other_file_rel]));
        self.assertEqual(log_captured.records[-1].msg,
                         "File 'test_file_zip/test_0.zip' already exists, skipping compressing")


    @log_capture()
    def test_unexistant_decompress(self, log_captured):

        self.assertIsNone(self.run_argv(["--decompress", 'bogus', '.']));
        self.assertEqual(log_captured.records[-1].msg,
                         "Skipping file 'bogus', check existance and permission")


    @log_capture()
    def test_unexistant_compress(self, log_captured):

        self.assertIsNone(self.run_argv(['bogus.zip', 'bogus']));
        self.assertEqual(log_captured.records[-1].msg,
                         "File 'bogus.zip' not created, check existance and permission")