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

# Copyright (c) 2024 Vladimir Botka <vbotka@gmail.com>
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
# SPDX-License-Identifier: GPL-3.0-or-later

from __future__ import (absolute_import, division, print_function)
__metaclass__ = type

import pytest
import yaml

from ansible.inventory.data import InventoryData
from ansible.template import Templar
from ansible_collections.community.general.plugins.inventory.iocage import InventoryModule
from ansible_collections.community.internal_test_tools.tests.unit.utils.trust import make_trusted


@pytest.fixture
def inventory():
    inv = InventoryModule()
    inv.inventory = InventoryData()
    inv.templar = Templar(None)
    inv.jails = load_txt_data('tests/unit/plugins/inventory/fixtures/iocage/iocage_jails.txt')
    inv.js_ok = load_yml_data('tests/unit/plugins/inventory/fixtures/iocage/iocage_jails.yml')
    inv.jails_dhcp = load_txt_data('tests/unit/plugins/inventory/fixtures/iocage/iocage_jails_dhcp.txt')
    inv.js_dhcp_ok = load_yml_data('tests/unit/plugins/inventory/fixtures/iocage/iocage_jails_dhcp.yml')
    inv.jails_dhcp_nr = load_txt_data('tests/unit/plugins/inventory/fixtures/iocage/iocage_jails_dhcp_not_running.txt')
    inv.js_dhcp_nr_ok = load_yml_data('tests/unit/plugins/inventory/fixtures/iocage/iocage_jails_dhcp_not_running.yml')
    prpts_101 = load_txt_data('tests/unit/plugins/inventory/fixtures/iocage/iocage_properties_test_101.txt')
    prpts_102 = load_txt_data('tests/unit/plugins/inventory/fixtures/iocage/iocage_properties_test_102.txt')
    prpts_103 = load_txt_data('tests/unit/plugins/inventory/fixtures/iocage/iocage_properties_test_103.txt')
    inv.prpts = {'test_101': prpts_101, 'test_102': prpts_102, 'test_103': prpts_103}
    inv.ps_ok = load_yml_data('tests/unit/plugins/inventory/fixtures/iocage/iocage_properties.yml')
    inv.ok = load_yml_data('tests/unit/plugins/inventory/fixtures/iocage/iocage_inventory.yml')
    return inv


def load_txt_data(path):
    with open(path, 'r') as f:
        s = f.read()
    return s


def load_yml_data(path):
    with open(path, 'r') as f:
        d = yaml.safe_load(f)
    return d


def get_option(option):
    groups = {}
    groups['test'] = make_trusted("inventory_hostname.startswith('test')")

    if option == 'groups':
        return groups
    elif option == 'keyed_groups':
        return []
    elif option == 'compose':
        return {}
    elif option == 'strict':
        return False
    else:
        return None


def test_verify_file_bad_config(inventory):
    assert inventory.verify_file('foobar.iocage.yml') is False


def test_verify_file(tmp_path, inventory):
    file = tmp_path / "foobar.iocage.yml"
    file.touch()
    assert inventory.verify_file(str(file))


def test_get_jails(inventory):

    # jails
    results = {'_meta': {'hostvars': {}}}
    inventory.get_jails(inventory.jails, results)
    assert results == inventory.js_ok

    # jails_dhcp
    results = {'_meta': {'hostvars': {}}}
    inventory.get_jails(inventory.jails_dhcp, results)
    assert results == inventory.js_dhcp_ok

    # jails_dhcp_not_running
    results = {'_meta': {'hostvars': {}}}
    inventory.get_jails(inventory.jails_dhcp_nr, results)
    assert results == inventory.js_dhcp_nr_ok


def test_get_properties(inventory):
    results = {'_meta': {'hostvars': {}}}
    inventory.get_jails(inventory.jails, results)
    for hostname, host_vars in results['_meta']['hostvars'].items():
        inventory.get_properties(inventory.prpts[hostname], results, hostname)
    assert results == inventory.ps_ok


def test_populate(inventory, mocker):
    results = {'_meta': {'hostvars': {}}}
    inventory.get_jails(inventory.jails, results)
    for hostname, host_vars in results['_meta']['hostvars'].items():
        inventory.get_properties(inventory.prpts[hostname], results, hostname)
    inventory.get_option = mocker.MagicMock(side_effect=get_option)
    inventory.populate(results)

    # test
    hosts = ('test_101', 'test_102', 'test_103')
    vars = ('iocage_basejail', 'iocage_boot', 'iocage_ip4', 'iocage_ip6', 'iocage_properties',
            'iocage_release', 'iocage_state', 'iocage_template', 'iocage_type')

    # test host_vars
    for host in hosts:
        h = inventory.inventory.get_host(host)
        for var in vars:
            assert inventory.ok['all']['children']['test']['hosts'][host][var] == h.get_vars()[var]

    # test groups
    test_101_info = inventory.inventory.get_host('test_101')
    test_102_info = inventory.inventory.get_host('test_102')
    test_103_info = inventory.inventory.get_host('test_103')
    g = inventory.inventory.groups['test']
    assert g.hosts == [test_101_info, test_102_info, test_103_info]