#-------------------------------------------------------------------------------
# Name:           Crypto
# Purpose:       Vernam cipher with hex output module
#
# Author:       Michael M. Yazdani
#
# Created:       03/05/2013
# Copyright:   (c) Michael M. Yazdani 2013
# Licence:       N/A
#-------------------------------------------------------------------------------
#!/usr/bin/env python
import random
random.seed()

def csr(x):
    if x % 2 == 1:
        return (x >> 1) + 64
    elif x % 2 == 0:
        return x >> 1

def csl(x):
    if x < 64:
        return x << 1
    else:
        return ((x - 64) << 1) + 1

class Crypto:
    def __init__(self, name, msg):
        self.decoded = ""
        self.encoded = ""
        self.alphas = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz,.?\"'()[]{}1234567890<>;:!@#$%^&*`~/\\-=_+ "
        self.name = name
        self.msg = msg
        if not(self.name == 'gen'):
            stream = open("keys/"+self.name)
            if (stream):
                self.key = stream.read()
                stream.close()
            else:
                print("Error, key not found, make sure its in the key folder and you spelled it right")
                stream.close()

    def update(self, msg):
        self.msg = msg

    def get_key(self):
        stream = open("keys/"+self.name)
        if (stream):
            self.key = stream.read()
            stream.close()
            return self.key
        else:
            print("Error, key not found, make sure its in the key folder and you spelled it right")
            stream.close()

    def In(self):
        self.encode = self.msg
        self.encoded = ""
        for x in range(0, len(self.encode)):
            self.encoded += hex(csl(csr(ord(self.key[x % len(self.key)]))^ord(self.encode[x])))+" "
        return self.encoded

    def Out(self):
        self.encoded = self.msg.split(' ')
        self.decoded = ""
        del self.encoded[len(self.encoded)-1]
        for x in range(0, len(self.encoded)):
            self.decoded += chr(csr(int(self.encoded[x], base=16))^csr(ord(self.key[x % len(self.key)])))
        return self.decoded

    def gen_key(self, name):
        self.key = ""
        used = [False]
        check = []
        for i in range(len(self.alphas)):
                used.append(False)
        for i in range(len(self.alphas)):
            var = True
            while var:
                num = random.randint(0, len(self.alphas)-1)
                if (used[num] == False):
                    check.append(num)
                    var = False
            used[num] = True
            self.key += self.alphas[num]
        check = sorted(check)
        print(check)
        print("New key: " + self.key)
        stream = open("keys/"+name, mode='w')
        stream.write(self.key)
        stream.close()

class CryptIO:

    def __init__(self, fin, fout, key):
        self.file = open(fin)
        self.file2 = open(fout, 'w')
        Crypto.__init__(self, key, '')

    def close(self):
        self.file.close()
        self.file2.close()

    def In(self):
        for line in self.file:
            Crypto.update(self, line)
            fin = Crypto.In(self)
            print(line)
            print(fin)
            self.file2.write(fin+'\n')
        self.close()

    def Out(self):
        for line in self.file:
            Crypto.update(self, line)
            fout = Crypto.Out(self)
            print(line)
            print(fout)
            self.file2.write(fout)
        self.close()