_ = unicode

from simplejson import dumps

from cubicweb.web._exceptions import *

def json_dumps(value):

    if isinstance(value, decimal.Decimal):

        value = float(value)

    elif isinstance(value, (date, datetime)):

        value = value.strftime('%Y-%m-%d %H:%M')

    elif isinstance(value, timedelta):

        value = (value.days * 24*60*60) + value.seconds

    try:

        return simplejson.dumps(value)

    except TypeError:

        return simplejson.dumps(repr(value))

def jsonize(function):

    def newfunc(*args, **kwargs):

        return json_dumps(function(*args, **kwargs))

    return newfunc

def ajax_replace_url(nodeid, rql, vid=None, swap=False, **extraparams):

    """builds a replacePageChunk-like url

    >>> ajax_replace_url('foo', 'Person P')

    "javascript: replacePageChunk('foo', 'Person%20P');"

    >>> ajax_replace_url('foo', 'Person P', 'oneline')

    "javascript: replacePageChunk('foo', 'Person%20P', 'oneline');"

    >>> ajax_replace_url('foo', 'Person P', 'oneline', name='bar', age=12)

    "javascript: replacePageChunk('foo', 'Person%20P', 'oneline', {'age':12, 'name':'bar'});"

    >>> ajax_replace_url('foo', 'Person P', name='bar', age=12)

    "javascript: replacePageChunk('foo', 'Person%20P', 'null', {'age':12, 'name':'bar'});"

    """

    params = [repr(nodeid), repr(urlquote(rql))]

    if extraparams and not vid:

        params.append("'null'")

    elif vid:

        params.append(repr(vid))

    if extraparams:

        params.append(json_dumps(extraparams))

    if swap:

        params.append('true')

    return "javascript: replacePageChunk(%s);" % ', '.join(params)