# pushkey.py - dispatching for pushing and pulling keys

#

# Copyright 2010 Matt Mackall <mpm@selenic.com>

#

# This software may be used and distributed according to the terms of the

# GNU General Public License version 2 or any later version.

from __future__ import absolute_import

from . import (

    bookmarks,

    encoding,

    obsolete,

    phases,

)

def _nslist(repo):

    n = {}

    for k in _namespaces:

        n[k] = b""

    if not obsolete.isenabled(repo, obsolete.exchangeopt):

        n.pop(b'obsolete')

    return n

_namespaces = {

    b"namespaces": (lambda *x: False, _nslist),

    b"bookmarks": (bookmarks.pushbookmark, bookmarks.listbookmarks),

    b"phases": (phases.pushphase, phases.listphases),

    b"obsolete": (obsolete.pushmarker, obsolete.listmarkers),

}

def register(namespace, pushkey, listkeys):

    _namespaces[namespace] = (pushkey, listkeys)

def _get(namespace):

    return _namespaces.get(namespace, (lambda *x: False, lambda *x: {}))

def push(repo, namespace, key, old, new):

    '''should succeed iff value was old'''

    pk = _get(namespace)[0]

    return pk(repo, key, old, new)

def list(repo, namespace):

    '''return a dict'''

    lk = _get(namespace)[1]

    return lk(repo)

encode = encoding.fromlocal

decode = encoding.tolocal

def encodekeys(keys):

    """encode the content of a pushkey namespace for exchange over the wire"""

    return b'\n'.join([b'%s\t%s' % (encode(k), encode(v)) for k, v in keys])

def decodekeys(data):

    """decode the content of a pushkey namespace from exchange over the wire"""

    result = {}

    for l in data.splitlines():

        k, v = l.split(b'\t')

        result[decode(k)] = decode(v)

    return result