import json
from collections import OrderedDict


def _preencode(data, magic_marker, in_coords=False, in_groups=False):
    if isinstance(data, dict):
        data = data.copy()
        for name, value in tuple(data.items()):
            if name in ('bounds', 'point', 'locations') and isinstance(value, (tuple, list)):
                data[name] = magic_marker+json.dumps(value)+magic_marker
            else:
                data[name] = _preencode(value, magic_marker,
                                        in_coords=(name == 'coordinates'), in_groups=(name == 'groups'))
        return data
    elif isinstance(data, (tuple, list)):
        if (in_coords and data and isinstance(data[0], (int, float))) or in_groups:
            return magic_marker+json.dumps(data)+magic_marker
        else:
            return tuple(_preencode(value, magic_marker, in_coords) for value in data)
    else:
        return data


def json_encoder_reindent(method, data, *args, **kwargs):
    magic_marker = '***JSON_MAGIC_MARKER***'
    test_encode = json.dumps(data)
    while magic_marker in test_encode:
        magic_marker += '*'
    result = method(_preencode(data, magic_marker), *args, **kwargs)
    if type(result) == str:
        return result.replace('"'+magic_marker, '').replace(magic_marker+'"', '')
    else:
        magic_marker = magic_marker.encode()
        return result.replace(b'"'+magic_marker, b'').replace(magic_marker+b'"', b'')


def format_geojson(data, round=True):
    coordinates = data.get('coordinates', None)
    if coordinates is not None:
        return OrderedDict((
            ('type', data['type']),
            ('coordinates', round_coordinates(data['coordinates']) if round else data['coordinates']),
        ))
    return OrderedDict((
        ('type', data['type']),
        ('geometries', [format_geojson(geometry, round=round) for geometry in data['geometries']]),
    ))


def round_coordinates(data):
    if isinstance(data, (list, tuple)):
        return tuple(round_coordinates(item) for item in data)
    else:
        return round(data, 2)