DzmitrySha/python-project-lvl2

"""Stylish module - apply stylish view to diff (by default)"""


def get_indent(depth: int) -> str:
    return " " * (depth * 4 - 2)


def to_string(value, depth: int) -> str:
    if isinstance(value, bool):
        return 'true' if value else 'false'

    if value is None:
        return 'null'

    if isinstance(value, dict):
        indent = get_indent(depth)
        current_indent = indent + (" " * 6)
        lines = []
        for k, v in value.items():
            lines.append(f"{current_indent}{k}: {to_string(v, depth + 1)}")
        result = "\n".join(lines)
        return f'{{\n{result}\n  {indent}}}'

    return value


def iter_(node: dict, depth=0) -> str:
    children = node.get('children')
    indent = get_indent(depth)
    formatted_value = to_string(node.get('value'), depth)
    formatted_value1 = to_string(node.get('old_value'), depth)
    formatted_value2 = to_string(node.get('new_value'), depth)

    if node['type'] == 'root':
        lines = map(lambda child: iter_(child, depth + 1), children)
        result = '\n'.join(lines)
        return f'{{\n{result}\n}}'

    if node['type'] == 'nested':
        lines = map(lambda child: iter_(child, depth + 1), children)
        result = '\n'.join(lines)
        return f"{indent}  {node['key']}: {{\n{result}\n  {indent}}}"

    if node['type'] == 'changed':
        line1 = f"{indent}- {node['key']}: {formatted_value1}\n"
        line2 = f"{indent}+ {node['key']}: {formatted_value2}"
        result = line1 + line2
        return result

    if node['type'] == 'unchanged':
        return f"{indent}  {node['key']}: {formatted_value}"

    if node['type'] == 'removed':
        return f"{indent}- {node['key']}: {formatted_value}"

    if node['type'] == 'added':
        return f"{indent}+ {node['key']}: {formatted_value}"


def format_(node: dict):
    return iter_(node)