# -*- coding: utf-8 -*-

#  xmppony
#  jep106.py

#  Copyright (c) 2009 Anaël Verrier
#  Copyright (c) 2003-2009 Alexey "Snake" Nezhdanov, Norman Rasmussen

#  This program is free software; you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation; version 3 only.

#  This program is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.

#  You should have received a copy of the GNU General Public License
#  along with this program; if not, write to the Free Software
#  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

"""This file is the XEP-0106 commands.

Implemented commands as follows:

4.2 Encode : Encoding Transformation
4.3 Decode : Decoding Transformation


"""

xep0106mapping = [
    [' ', '20'],
    ['"', '22'],
    ['&', '26'],
    ['\'', '27'],
    ['/', '2f'],
    [':', '3a'],
    ['<', '3c'],
    ['>', '3e'],
    ['@', '40']]


def JIDEncode(str_):
    str_ = str_.replace('\\5c', '\\5c5c')
    for each in xep0106mapping:
        str_ = str_.replace('\\' + each[1], '\\5c' + each[1])
    for each in xep0106mapping:
        str_ = str_.replace(each[0], '\\' + each[1])
    return str_


def JIDDecode(str_):
    for each in xep0106mapping:
        str_ = str_.replace('\\' + each[1], each[0])
    return str_.replace('\\5c', '\\')

if __name__ == "__main__":
    def test(before, valid):
        during = JIDEncode(before)
        after = JIDDecode(during)
        if during == valid and after == before:
            print 'PASS Before: ' + before
            print 'PASS During: ' + during
        else:
            print 'FAIL Before: ' + before
            print 'FAIL During: ' + during
            print 'FAIL After : ' + after
        print

    test('jid escaping', r'jid\20escaping')
    test(r'\3and\2is\5@example.com', r'\5c3and\2is\5\40example.com')
    test(r'\3catsand\2catsis\5cats@example.com',
         r'\5c3catsand\2catsis\5c5cats\40example.com')
    test(r'\2plus\2is\4', r'\2plus\2is\4')
    test(r'foo\bar', r'foo\bar')
    test(r'foob\41r', r'foob\41r')
    test('here\'s_a wild_&_/cr%zy/_address@example.com',
         r'here\27s_a\20wild_\26_\2fcr%zy\2f_address\40example.com')