import os
import shutil
from calculate.utils.files import join_paths
from calculate.templates.template_engine import Variables
from calculate.utils.package import PackageAtomName, Version, Package
from calculate.templates.template_processor import DirectoryProcessor,\
                                                   TemplateWrapper


CHROOT_PATH = os.path.join(os.getcwd(),
                           'tests/templates/testfiles/test_dir_processor_root')
CONFIG_PATH = os.path.join(CHROOT_PATH, 'var/lib/calculate/config')
CONFIG_ARCHIVE_PATH = os.path.join(CHROOT_PATH,
                                   'var/lib/calculate/config-archive')
EXECUTE_ARCHIVE_PATH = os.path.join(CHROOT_PATH,
                                    'var/lib/calculate/.execute')

test_package_name = PackageAtomName(
                        {'pkg_path': os.path.join(
                                  CHROOT_PATH,
                                  'var/db/pkg/test-category/test-package-1.0'),
                         'version': Version('1.0')})

other_package_name = PackageAtomName(
                        {'pkg_path': os.path.join(
                                 CHROOT_PATH,
                                 'var/db/pkg/test-category/other-package-1.1'),
                         'version': Version('1.1')})

new_package_name = PackageAtomName(
                        {'pkg_path': os.path.join(
                                 CHROOT_PATH,
                                 'var/db/pkg/test-category/new-package-0.1.1'),
                         'version': Version('1.0')})

# Вместо модуля переменных.
group = Variables({'bool': True,
                   'list': [1, 2, 3]})

variables = Variables({'variable_1': 'value_1',
                       'variable_2': 'value_2',
                       'group': group})

install = Variables({'os_disk_dev': 'os_disk_dev_value',
                     'version': 1.5,
                     'number': 128,
                     'boolean': False,
                     'type': 'static',
                     'path': '/usr/sbin'})

merge = Variables({'var_1': 674,
                   'var_2': 48,
                   'version': 1.0,
                   'calculate_domains': 'lists.calculate-linux.org',
                   'ip_value': '127.0.0.0/8'})

main = Variables({'cl_template_path':
                  '{0},{1}'.format(os.path.join(CHROOT_PATH,
                                                'templates'),
                                   os.path.join(CHROOT_PATH,
                                                'var/calculate/templates')),
                  'cl_chroot_path': CHROOT_PATH,
                  'cl_config_path': CONFIG_PATH,
                  'cl_config_archive': CONFIG_ARCHIVE_PATH,
                  'cl_exec_dir_path': EXECUTE_ARCHIVE_PATH,
                  'cl_ignore_files': '*.swp'})

test = ({'test_root': CHROOT_PATH})

datavars = Variables({'install': install,
                      'merge': merge,
                      'variables': variables,
                      'main': main,
                      'test': test,
                      'custom': Variables()})


TemplateWrapper._protected_is_set = False

shutil.copytree(os.path.join(CHROOT_PATH, 'etc.backup'),
                os.path.join(CHROOT_PATH, 'etc'),
                symlinks=True)
shutil.copytree(os.path.join(CHROOT_PATH, 'var.backup'),
                os.path.join(CHROOT_PATH, 'var'),
                symlinks=True)


datavars.main['cl_template_path'] = os.path.join(CHROOT_PATH,
                                                 'templates_20')
directory_processor = DirectoryProcessor('install',
                                         datavars_module=datavars)
directory_processor.process_template_directories()
test_package = Package(test_package_name, chroot_path=CHROOT_PATH)

# Для разнообразия один из шаблонов удаляет файл, а не создает.
# Но в данном случае он ничего не сделает.
assert os.path.exists(join_paths(CHROOT_PATH,
                                 '/etc/dir_20/file_0'))
assert '/etc/dir_20/file_0' not in test_package

assert not os.path.islink(join_paths(CHROOT_PATH, '/etc/dir_19'))
assert os.path.isdir(join_paths(CHROOT_PATH, '/etc/dir_19'))

assert os.path.exists(join_paths(CHROOT_PATH,
                                 '/etc/dir_19/dir_0'))
assert os.path.exists(join_paths(CHROOT_PATH,
                                 '/etc/dir_19/dir_0/file_0'))
assert '/etc/dir_19/dir_0/file_0' in test_package

shutil.rmtree(os.path.join(CHROOT_PATH, 'etc'))
shutil.rmtree(os.path.join(CHROOT_PATH, 'var'))