import subprocess

import serial

from mupq import mupq


class M4Settings(mupq.PlatformSettings):
    #: Specify folders to include
    scheme_folders = [  # mupq.PlatformSettings.scheme_folders + [
        ('pqm4', 'crypto_kem', ''),
        ('pqm4', 'crypto_sign', ''),
        ('mupq', 'mupq/crypto_kem', ''),
        ('mupq', 'mupq/crypto_sign', ''),
        ('pqclean', 'mupq/pqclean/crypto_kem', "PQCLEAN"),
        ('pqclean', 'mupq/pqclean/crypto_sign', "PQCLEAN"),
    ]

    #: List of dicts, in each dict specify (Scheme class) attributes of the
    #: scheme with values, if all attributes match the scheme is skipped.
    skip_list = (
        {'scheme': 'falcon-1024-tree', 'implementation': 'opt-leaktime'},
        {'scheme': 'falcon-1024-tree', 'implementation': 'opt-ct'},
        {'scheme': 'frodokem640aes', 'implementation': 'clean'},
        {'scheme': 'frodokem640aes', 'implementation': 'opt'},
        {'scheme': 'frodokem976aes', 'implementation': 'clean'},
        {'scheme': 'frodokem976aes', 'implementation': 'opt'},
        {'scheme': 'frodokem1344aes', 'implementation': 'clean'},
        {'scheme': 'frodokem1344aes', 'implementation': 'opt'},
        {'scheme': 'frodokem640shake', 'implementation': 'clean'},
        {'scheme': 'frodokem976shake', 'implementation': 'clean'},
        {'scheme': 'frodokem976shake', 'implementation': 'opt'},
        {'scheme': 'frodokem1344shake', 'implementation': 'clean'},
        {'scheme': 'frodokem1344shake', 'implementation': 'opt'},
        {'scheme': 'mqdss-48', 'implementation': 'clean'},
        {'scheme': 'mqdss-64', 'implementation': 'clean'},
        {'scheme': 'ledakemlt12', 'implementation': 'leaktime'},
        {'scheme': 'ledakemlt32', 'implementation': 'leaktime'},
        {'scheme': 'ledakemlt52', 'implementation': 'leaktime'},
        {'scheme': 'luov-79-76-341-chacha', 'implementation': 'ref'},
        {'scheme': 'luov-79-76-341-keccak', 'implementation': 'ref'},
        {'scheme': 'luov-7-83-283-chacha', 'implementation': 'ref'},
        {'scheme': 'luov-7-83-283-keccak', 'implementation': 'ref'},
        {'scheme': 'luov-7-110-374-chacha', 'implementation': 'ref'},
        {'scheme': 'luov-7-110-374-keccak', 'implementation': 'ref'},
        {'scheme': 'rainbowIa-classic', 'implementation': 'clean'},
        {'scheme': 'rainbowIa-cyclic', 'implementation': 'clean'},
        {'scheme': 'rainbowIa-cyclic-compressed', 'implementation': 'clean'},
        {'scheme': 'rainbowIIIc-classic', 'implementation': 'clean'},
        {'scheme': 'rainbowIIIc-cyclic', 'implementation': 'clean'},
        {'scheme': 'rainbowIIIc-cyclic-compressed', 'implementation': 'clean'},
        {'scheme': 'rainbowVc-classic', 'implementation': 'clean'},
        {'scheme': 'rainbowVc-cyclic', 'implementation': 'clean'},
        {'scheme': 'rainbowVc-cyclic-compressed', 'implementation': 'clean'},
        {'scheme': 'qtesla-p-I', 'implementation': 'clean'},
        {'scheme': 'qtesla-p-III', 'implementation': 'clean'},
        {'scheme': 'mceliece348864', 'implementation': 'clean'},
        {'scheme': 'mceliece348864f', 'implementation': 'clean'},
        {'scheme': 'mceliece460896', 'implementation': 'clean'},
        {'scheme': 'mceliece460896f', 'implementation': 'clean'},
        {'scheme': 'mceliece6688128', 'implementation': 'clean'},
        {'scheme': 'mceliece6688128f', 'implementation': 'clean'},
        {'scheme': 'mceliece6960119', 'implementation': 'clean'},
        {'scheme': 'mceliece6960119f', 'implementation': 'clean'},
        {'scheme': 'mceliece8192128', 'implementation': 'clean'},
        {'scheme': 'mceliece8192128f', 'implementation': 'clean'},
        {'scheme': 'mceliece348864', 'implementation': 'vec'},
        {'scheme': 'mceliece348864f', 'implementation': 'vec'},
        {'scheme': 'mceliece460896', 'implementation': 'vec'},
        {'scheme': 'mceliece460896f', 'implementation': 'vec'},
        {'scheme': 'mceliece6688128', 'implementation': 'vec'},
        {'scheme': 'mceliece6688128f', 'implementation': 'vec'},
        {'scheme': 'mceliece6960119', 'implementation': 'vec'},
        {'scheme': 'mceliece6960119f', 'implementation': 'vec'},
        {'scheme': 'mceliece8192128', 'implementation': 'vec'},
        {'scheme': 'mceliece8192128f', 'implementation': 'vec'},
        {'scheme': 'hqc-128-1-cca2', 'implementation': 'leaktime'},
        {'scheme': 'hqc-192-1-cca2', 'implementation': 'leaktime'},
        {'scheme': 'hqc-192-2-cca2', 'implementation': 'leaktime'},
        {'scheme': 'hqc-256-1-cca2', 'implementation': 'leaktime'},
        {'scheme': 'hqc-256-2-cca2', 'implementation': 'leaktime'},
        {'scheme': 'hqc-256-3-cca2', 'implementation': 'leaktime'},
        {'scheme': 'rqc256', 'implementation': 'ref'},
    )


class M4(mupq.Platform):

    def __enter__(self):
        self._dev = serial.Serial("/dev/ttyUSB0", 115200, timeout=10)
        return super().__enter__()

    def __exit__(self,*args, **kwargs):
        self._dev.close()
        return super().__exit__(*args, **kwargs)

    def device(self):
        return self._dev

    def flash(self, binary_path):
        super().flash(binary_path)
        subprocess.check_call(
            ["st-flash", "--reset", "write", binary_path, "0x8000000"],
            stdout=subprocess.DEVNULL,
            stderr=subprocess.DEVNULL
        )