import hashlib
import uuid
from random import randint
from secrets import choice
from string import ascii_letters, digits
from cryptography.fernet import Fernet, InvalidToken

from configuration import Config


class Password:
    def __init__(self):
        self.encryption_key = Config().get_secret()
        if self.encryption_key is None:
            self.set_encryption_key()
        else:  # just take what's given
            self.cipher = Fernet(self.encryption_key)

    def get_encryption_key(self):
        return self.encryption_key

    def set_encryption_key(self):
        self.encryption_key = Fernet.generate_key()
        Config().set_secret(self.encryption_key.decode())  # store as string
        # Don't forget to update the cipher!!!
        self.cipher = Fernet(self.encryption_key)

    def encrypt(self, plain_password):
        return self.cipher.encrypt(plain_password.encode())

    def decrypt(self, encrypted_password):
        try:
            return self.cipher.decrypt(encrypted_password).decode()
        except InvalidToken:
            return "ERROR: Invalid Encryption Key"

    @staticmethod
    def generate(mn=16, mx=64):
        return ''.join(
            choice(ascii_letters + digits) for _ in range(randint(mn, mx))
        )

    # @staticmethod
    # def hash_password(password):
    #     salt = uuid.uuid4().hex
    #     return hashlib.sha512(
    #         salt.encode() + password.encode()).hexdigest() + ':' + salt
    #
    # @staticmethod
    # def check_password(hashed_password, user_password):
    #     password, salt = hashed_password.split(':')
    #     return password == hashlib.sha512(
    #         salt.encode() + user_password.encode()).hexdigest()
    #
    # @staticmethod
    # def dsa_encode(password):
    #     hash_object = hashlib.new('DSA')
    #     hash_object.update(password)
    #     return hash_object.h