import lilliput
from lilliput.constants import NONCE_BYTES as NPUBBYTES, TAG_BYTES

# Import KEYBYTES to expose it to genkat_aead.
# Import MODE to provide it to lilliput.
from parameters import KEYBYTES, MODE


def encrypt(m, ad, npub, k):
    c, tag = lilliput.encrypt(m, ad, k, npub, MODE)
    return c+tag


def decrypt(c, ad, npub, k):
    clen = len(c)-TAG_BYTES
    ctext = c[:clen]
    tag = c[clen:]
    return lilliput.decrypt(ctext, tag, ad, k, npub, MODE)