import logging
	import os
	from unittest import mock
	
	from lxml import etree
	
	from pcs import settings
	from pcs.lib.external import CommandRunner
	
	from pcs_test.tools.assertions import AssertPcsMixin
	from pcs_test.tools.bin_mock import get_mock_settings
	from pcs_test.tools.custom_mock import MockLibraryReportProcessor
	from pcs_test.tools.misc import (
	    get_test_resource,
	    get_tmp_file,
	    write_file_to_tmpfile,
	)
	from pcs_test.tools.pcs_runner import PcsRunner
	from pcs_test.tools.xml import etree_to_str
	
	
	class CachedCibFixture(AssertPcsMixin):
	    def __init__(self, cache_name, empty_cib_path):
	        self._empty_cib_path = empty_cib_path
	        self._cache_name = cache_name
	        self._cache_path = None
	        self._pcs_runner = None
	
	    def _setup_cib(self):
	        raise NotImplementedError()
	
	    def set_up(self):
	        fixture_dir = get_test_resource("temp_fixtures")
	        os.makedirs(fixture_dir, exist_ok=True)
	        self._cache_path = os.path.join(fixture_dir, self._cache_name)
	        self._pcs_runner = PcsRunner(self._cache_path)
	        self._pcs_runner.mock_settings = get_mock_settings()
	
	        with (
	            open(self._empty_cib_path, "r") as template_file,
	            open(self.cache_path, "w") as cache_file,
	        ):
	            cache_file.write(template_file.read())
	        self._setup_cib()
	
	    def clean_up(self):
	        if os.path.isfile(self.cache_path):
	            os.unlink(self.cache_path)
	
	    @property
	    def cache_path(self):
	        if self._cache_path is None:
	            raise AssertionError("Cache has not been initialized")
	        return self._cache_path
	
	    # methods for supporting assert_pcs_success
	    @property
	    def pcs_runner(self):
	        if self._pcs_runner is None:
	            raise AssertionError("Cache has not been initialized")
	        return self._pcs_runner
	
	    def assertEqual(self, first, second, msg=None):
	        if first != second:
	            raise AssertionError(
	                f"{msg}\n{first} != {second}" if msg else f"{first} != {second}"
	            )
	
	
	def wrap_element_by_master(cib_file, resource_id, master_id=None):
	    cib_file.seek(0)

	    cib_tree = etree.parse(cib_file, etree.XMLParser(huge_tree=True)).getroot()
	    element = cib_tree.find(f'.//*[@id="{resource_id}"]')
	    final_master_id = (
	        master_id if master_id is not None else f"{resource_id}-master"
	    )
	    master_element = _xml_to_element(
	        f"""
	        <master id="{final_master_id}">
	        </master>
	    """
	    )
	    element.getparent().append(master_element)
	    master_element.append(element)
	    final_xml = etree_to_str(cib_tree)
	
	    environ = dict(os.environ)
	    environ["CIB_file"] = cib_file.name
	    runner = CommandRunner(
	        mock.MagicMock(logging.Logger), MockLibraryReportProcessor(), environ
	    )
	    stdout, stderr, retval = runner.run(
	        [
	            settings.cibadmin_exec,
	            "--replace",
	            "--scope",
	            "resources",
	            "--xml-pipe",
	        ],
	        stdin_string=final_xml,
	    )
	    assert retval == 0, (
	        "Error running wrap_element_by_master:\n" + stderr + "\n" + stdout
	    )
	
	
	def wrap_element_by_master_file(filepath, resource_id, master_id=None):
	    cib_tmp = get_tmp_file("wrap_by_master")
	    write_file_to_tmpfile(filepath, cib_tmp)
	    wrap_element_by_master(cib_tmp, resource_id, master_id=master_id)
	    cib_tmp.seek(0)
	    with open(filepath, "w") as target:
	        target.write(cib_tmp.read())
	    cib_tmp.close()
	
	
	def fixture_master_xml(name, all_ops=True, meta_dict=None):
	    default_ops = f"""
	            <op id="{name}-notify-interval-0s" interval="0s" name="notify"
	                timeout="5"
	            />
	            <op id="{name}-start-interval-0s" interval="0s" name="start"
	                timeout="20"
	            />
	            <op id="{name}-stop-interval-0s" interval="0s" name="stop"
	                timeout="20"
	            />
	    """
	    meta_xml = ""
	    if meta_dict:
	        meta_lines = (
	            [f'<meta_attributes id="{name}-master-meta_attributes">']
	            + [
	                f'<nvpair id="{name}-master-meta_attributes-{key}" name="{key}" value="{val}"/>'
	                for key, val in meta_dict.items()
	            ]
	            + ["</meta_attributes>"]
	        )
	        meta_xml = "\n".join(meta_lines)
	    master = f"""
	      <master id="{name}-master">
	        <primitive class="ocf" id="{name}" provider="pcsmock" type="stateful">
	          <operations>
	            <op id="{name}-monitor-interval-10" interval="10" name="monitor"
	                role="Master" timeout="20"
	            />
	            <op id="{name}-monitor-interval-11" interval="11" name="monitor"
	                role="Slave" timeout="20"
	            />
	    """
	    if all_ops:
	        master += default_ops
	    master += f"""
	          </operations>
	        </primitive>
	        {meta_xml}
	      </master>
	    """
	    return master
	
	
	def fixture_to_cib(cib_file, xml):
	    environ = dict(os.environ)
	    environ["CIB_file"] = cib_file
	    runner = CommandRunner(
	        mock.MagicMock(logging.Logger), MockLibraryReportProcessor(), environ
	    )
	    stdout, stderr, retval = runner.run(
	        [
	            settings.cibadmin_exec,
	            "--create",
	            "--scope",
	            "resources",
	            "--xml-text",
	            xml,
	        ]
	    )
	    assert retval == 0, (
	        "Error running fixture_to_cib:\n" + stderr + "\n" + stdout
	    )
	
	
	def _replace(element_to_replace, new_element):
	    parent = element_to_replace.getparent()
	    for child in parent:
	        if element_to_replace == child:
	            index = list(parent).index(child)
	            parent.remove(child)
	            parent.insert(index, new_element)
	            return
	
	
	def _xml_to_element(xml):
	    try:
	        new_element = etree.fromstring(xml)
	    except etree.XMLSyntaxError as e:
	        raise AssertionError(
	            "Cannot put to the cib a non-xml fragment:\n'{0}'".format(xml)
	        ) from e
	    return new_element
	
	
	def _find_all_in(cib_tree, element_xpath):
	    element_list = cib_tree.xpath(element_xpath)
	    if not element_list:
	        raise AssertionError(
	            "Cannot find '{0}' in given cib:\n{1}".format(
	                element_xpath, etree_to_str(cib_tree)
	            )
	        )
	    return element_list
	
	
	def _find_in(cib_tree, element_xpath):
	    element_list = _find_all_in(cib_tree, element_xpath)
	    if len(element_list) > 1:
	        raise AssertionError(
	            "Found more than one '{0}' in given cib:\n{1}".format(
	                element_xpath, etree_to_str(cib_tree)
	            )
	        )
	    return element_list[0]
	
	
	def remove(element_xpath):
	    def _remove(cib_tree):
	        xpath_list = (
	            [element_xpath] if isinstance(element_xpath, str) else element_xpath
	        )
	        for xpath in xpath_list:
	            for element_to_remove in _find_all_in(cib_tree, xpath):
	                element_to_remove.getparent().remove(element_to_remove)
	
	    return _remove
	
	
	def put_or_replace(parent_xpath, new_content):
	    # This transformation makes sense in "configuration" section only. In this
	    # section there are sub-tags (optional or mandatory) that can occur max 1x.
	    #
	    # In other sections it is possible to have more occurrences of sub-tags. For
	    # such cases it is better to use `replace_all` - the difference is that in
	    # `replace_all` the element to be replaced is specified by full xpath
	    # whilst in `put_or_replace` the xpath to the parent element is specified.
	    def replace_optional(cib_tree):
	        element = _xml_to_element(new_content)
	        parent = _find_in(cib_tree, parent_xpath)
	        current_elements = parent.findall(element.tag)
	
	        if len(current_elements) > 1:
	            raise _cannot_multireplace(element.tag, parent_xpath, cib_tree)
	
	        if current_elements:
	            _replace(current_elements[0], element)
	        else:
	            parent.append(element)
	
	    return replace_optional
	
	
	def replace_all(replacements):
	    """
	    Return a function that replace more elements (defined by replacement_dict)
	    in the cib_tree with new_content.
	
	    dict replacemens -- contains more replacements:
	        key is xpath - its destination must be one element: replacement is
	        applied only on the first occurrence
	        value is new content -contains a content that have to be placed instead
	        of an element found by element_xpath
	    """
	
	    def replace(cib_tree):
	        for xpath, new_content in replacements.items():
	            _replace(_find_in(cib_tree, xpath), _xml_to_element(new_content))
	
	    return replace
	
	
	def append_all(append_map):
	    """
	    Return a function that appends more elements after specified (xpath) element
	    dict append_map -- a key is an xpath pointing to a target element (for
	        appending), value is appended content
	    """
	
	    def append(cib_tree):
	        for xpath, new_content in append_map.items():
	            _find_in(cib_tree, xpath).append(_xml_to_element(new_content))
	
	    return append
	
	
	# Possible modifier shortcuts are defined here.
	# Keep in mind that every key will be named parameter in config function
	# (see modifier_shortcuts param in some of pcs_test.tools.command_env.config_*
	# modules)
	#
	# DO NOT USE CONFLICTING KEYS HERE!
	# 1) args of pcs_test.tools.command_env.calls#CallListBuilder.place:
	#  name, before, instead
	# 2) args of pcs_test.tools.command_env.mock_runner#Call.__init__
	#  command, stdout, stderr, returncode, check_stdin
	# 3) special args of pcs_test.tools.command_env.config_*
	#  modifiers, filename, load_key, wait, exception
	# It would be not applied. Not even mention that the majority of these names do
	# not make sense for a cib modifying ;)
	MODIFIER_GENERATORS = {
	    "remove": remove,
	    "replace": replace_all,
	    "append": append_all,
	    "resources": lambda xml: replace_all({"./configuration/resources": xml}),
	    "nodes": lambda xml: replace_all({"./configuration/nodes": xml}),
	    "constraints": lambda xml: replace_all(
	        {"./configuration/constraints": xml}
	    ),
	    "crm_config": lambda xml: replace_all({"./configuration/crm_config": xml}),
	    "fencing_topology": lambda xml: put_or_replace("./configuration", xml),
	    "status": lambda xml: put_or_replace(".", xml),
	    "tags": lambda xml: put_or_replace("./configuration", xml),
	    "acls": lambda xml: put_or_replace("./configuration", xml),
	    "optional_in_conf": lambda xml: put_or_replace("./configuration", xml),
	    # common modifier `put_or_replace` makes not sense - see explanation inside
	    # this function - all occurrences should be satisfied by `optional_in_conf`
	}
	
	
	def create_modifiers(**modifier_shortcuts):
	    """
	    Return list of modifiers: list of functions that transform cib
	
	    dict modifier_shortcuts -- a new modifier is generated from each modifier
	        shortcut.
	        As key there can be keys of MODIFIER_GENERATORS.
	        Value is passed into appropriate generator from MODIFIER_GENERATORS.
	
	    """
	    unknown_shortcuts = set(modifier_shortcuts.keys()) - set(
	        MODIFIER_GENERATORS.keys()
	    )
	    if unknown_shortcuts:
	        raise AssertionError(
	            "Unknown modifier shortcuts '{0}', available are: '{1}'".format(
	                "', '".join(list(unknown_shortcuts)),
	                "', '".join(MODIFIER_GENERATORS.keys()),
	            )
	        )
	
	    return [
	        MODIFIER_GENERATORS[name](param)
	        for name, param in modifier_shortcuts.items()
	    ]
	
	
	def modify_cib(cib_xml, modifiers=None, **modifier_shortcuts):
	    """
	    Apply modifiers to cib_xml and return the result cib_xml
	
	    string cib_xml -- initial cib
	    list of callable modifiers -- each takes cib (etree.Element)
	    dict modifier_shortcuts -- a new modifier is generated from each modifier
	        shortcut.
	        As key there can be keys of MODIFIER_GENERATORS.
	        Value is passed into appropriate generator from MODIFIER_GENERATORS.
	    """
	    modifiers = modifiers if modifiers else []
	    all_modifiers = modifiers + create_modifiers(**modifier_shortcuts)
	
	    if not all_modifiers:
	        return cib_xml
	
	    cib_tree = etree.fromstring(cib_xml)
	    for modify in all_modifiers:
	        modify(cib_tree)
	
	    return etree_to_str(cib_tree)
	
	
	def modify_cib_file(file_path, **modifiers_shortcuts):
	    with open(file_path, "r") as file:
	        return modify_cib(file.read(), **modifiers_shortcuts)
	
	
	def _cannot_multireplace(tag, parent_xpath, cib_tree):
	    return AssertionError(
	        (
	            "Cannot replace '{element}' in '{parent}' because '{parent}'"
	            " contains more than one '{element}' in given cib:\n{cib}"
	        ).format(element=tag, parent=parent_xpath, cib=etree_to_str(cib_tree))
	    )