from encodings import utf_8
import encodings
import codecs
import io

def cpreprocessor_decode(input, errors='strict'):
    print('!!!')
    stream = io.StringIO(bytes(input).decode('utf-8'))
    output = ''.join(stream.readlines())
    output = preprocess(output)
    return output, len(input)


def preprocess(code):
    print('!!!')
    new_code = ''
    for line in code.split('\n'):
        position = line.find('#!')
        if position == -1:
            new_code += line + '\n'
        else:
            new_code += line[:position-1] + '\n'
            new_code += line[position+2] + '\n'
        
    return new_code


def search_function(encoding):
    print('!!!')
    if encoding == 'cpreprocessor':
        utf8 = encodings.search_function('utf8')
        return codecs.CodecInfo(
            name='cpreprocessor',
            encode=utf8.encode,
            decode=cpreprocessor_decode,
            incrementalencoder=utf8.incrementalencoder,
            incrementaldecoder=utf_8.IncrementalDecoder,
            streamreader=utf_8.StreamReader,
            streamwriter=utf8.streamwriter)


codecs.register(search_function)