from __future__ import unicode_literals
from coreapi.codecs.base import BaseCodec
from coreapi.document import Document, Link, Array, Object, Error


def _to_repr(node):
    if isinstance(node, Document):
        content = ', '.join([
            '%s: %s' % (repr(key), _to_repr(value))
            for key, value in node.items()
        ])
        return 'Document(url=%s, title=%s, content={%s})' % (
            repr(node.url), repr(node.title), content
        )

    elif isinstance(node, Object):
        return '{%s}' % ', '.join([
            '%s: %s' % (repr(key), _to_repr(value))
            for key, value in node.items()
        ])

    elif isinstance(node, Array):
        return '[%s]' % ', '.join([
            _to_repr(value) for value in node
        ])

    elif isinstance(node, Link):
        args = "url=%s" % repr(node.url)
        if node.action:
            args += ", action=%s" % repr(node.action)
        if node.inplace is not None:
            args += ", inplace=%s" % repr(node.inplace)
        if node.fields:
            fields_repr = ', '.join([
                'required(%s)' % repr(field.name)
                if field.required else
                repr(field.name)
                for field in node.fields
            ])
            args += ", fields=[%s]" % fields_repr
        return "Link(%s)" % args

    elif isinstance(node, Error):
        return 'Error(%s)' % repr(list(node.messages))

    return repr(node)


class PythonCodec(BaseCodec):
    """
    A Python representation of a Document, for use with '__repr__'.
    """
    media_type = 'text/python'

    def dump(self, node, **kwargs):
        # Object and Array only have the class name wrapper if they
        # are the outermost element.
        if isinstance(node, Object):
            return 'Object(%s)' % _to_repr(node)
        elif isinstance(node, Array):
            return 'Array(%s)' % _to_repr(node)
        return _to_repr(node)