from typing import Generator
import traceback

from nacl.signing import VerifyKey
import nacl.exceptions

from logger import log as logging
import blockdb

from wot.identity import Identity, identities
from wot.exceptions import IdentitySerializationError
from wot.getbykey import get_identity_by_key
from wot.identityprocessing import processtrustsignature
import wot.wotcommand


def load_identity_from_block(block) -> Identity:
    return Identity.deserialize(block.data)


def load_identities_from_blocks() -> Generator[Identity, None, None]:
    for block in blockdb.get_blocks_by_type(b'wotb'):
        try:
            yield load_identity_from_block(block)
        except IdentitySerializationError:
            pass


def load_signatures_from_blocks() -> None:
    for block in blockdb.get_blocks_by_type(wot.wotcommand.block_type_map.trust):
        try:
            # If good signature,
            # it adds the signature to the signed identity's trust set
            # noop if already signed
            processtrustsignature.process_trust_signature(block.data)
        except nacl.exceptions.BadSignatureError:
            logging.warn('Bad signature in block:')
            logging.warn(traceback.format_exc())