root/xmppony/jep0106.py @ 1:0603290d9eb7

# -*- 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')