import argparse
import regex as re

from bitstring import BitArray


def decode(string):
    sym = {'ПУК': '00',
           'СРЕНЬК': '01',
           'ХРЮК': '10',
           'УИИИ': '11'}

    return BitArray(''.join(['0b'] + [sym[s]*int(c and c or 1) for (c, s) in re.findall(r'(\d*)(\w+)-?', string)])).bytes.decode()


def encode(string):
    sym = {'00': 'ПУК',
           '01': 'СРЕНЬК',
           '10': 'ХРЮК',
           '11': 'УИИИ'}

    def join(l):
        result = [l[0]]
        last = None

        for e in l[1:]:
            if last == e:
                result[-1] += e
            else:
                result.append(e)

            last = e

        return result

    return '-'.join([f'{len(g)//2 > 1 and len(g)//2 or ""}{sym[g[:2]]}' for g in join(re.findall(r'([01]{2})', BitArray(string.encode()).bin))])


if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='Encode and decode 2ch.hk fart code.')

    parser.add_argument('string', metavar='string', type=str,
                        help='string for process')

    function = parser.add_mutually_exclusive_group(required=True)

    function.add_argument('-d', '--decode', dest='function', action='store_const',
                          const=decode, help='decode string')

    function.add_argument('-e', '--encode', dest='function', action='store_const',
                          const=encode, help='encode string')

    args = parser.parse_args()

    print(args.function(args.string))