import { ChaCha20Poly1305_RFC8439, Nonce } from '../dist/salty-crypto.mjs';
import { it, expect } from 'vitest';

it('section 2.8.2 from rfc 8439', () => {
    const sunscreen_str = "Ladies and Gentlemen of the class of '99: If I could offer you only one tip for the future, sunscreen would be it.";
    const sunscreen = new TextEncoder().encode(sunscreen_str);

    const associated_data = new Uint8Array([
        0x50, 0x51, 0x52, 0x53, 0xc0, 0xc1, 0xc2, 0xc3,
        0xc4, 0xc5, 0xc6, 0xc7,
    ]);

    const key = new DataView(new Uint8Array([
        0x80, 0x81, 0x82, 0x83, 0x84, 0x85, 0x86, 0x87,
        0x88, 0x89, 0x8a, 0x8b, 0x8c, 0x8d, 0x8e, 0x8f,
        0x90, 0x91, 0x92, 0x93, 0x94, 0x95, 0x96, 0x97,
        0x98, 0x99, 0x9a, 0x9b, 0x9c, 0x9d, 0x9e, 0x9f,
    ]).buffer);

    const nonce = new Nonce(0x43424140, 0x47464544, 0x7);

    const expectedEncrypted = new Uint8Array([
        0xd3, 0x1a, 0x8d, 0x34, 0x64, 0x8e, 0x60, 0xdb, 0x7b, 0x86, 0xaf, 0xbc, 0x53, 0xef, 0x7e, 0xc2,
        0xa4, 0xad, 0xed, 0x51, 0x29, 0x6e, 0x08, 0xfe, 0xa9, 0xe2, 0xb5, 0xa7, 0x36, 0xee, 0x62, 0xd6,
        0x3d, 0xbe, 0xa4, 0x5e, 0x8c, 0xa9, 0x67, 0x12, 0x82, 0xfa, 0xfb, 0x69, 0xda, 0x92, 0x72, 0x8b,
        0x1a, 0x71, 0xde, 0x0a, 0x9e, 0x06, 0x0b, 0x29, 0x05, 0xd6, 0xa5, 0xb6, 0x7e, 0xcd, 0x3b, 0x36,
        0x92, 0xdd, 0xbd, 0x7f, 0x2d, 0x77, 0x8b, 0x8c, 0x98, 0x03, 0xae, 0xe3, 0x28, 0x09, 0x1b, 0x58,
        0xfa, 0xb3, 0x24, 0xe4, 0xfa, 0xd6, 0x75, 0x94, 0x55, 0x85, 0x80, 0x8b, 0x48, 0x31, 0xd7, 0xbc,
        0x3f, 0xf4, 0xde, 0xf0, 0x8e, 0x4b, 0x7a, 0x9d, 0xe5, 0x76, 0xd2, 0x65, 0x86, 0xce, 0xc6, 0x4b,
        0x61, 0x16,
    ]);

    const tag = new Uint8Array(ChaCha20Poly1305_RFC8439.TAGBYTES);
    ChaCha20Poly1305_RFC8439.encrypt_detached(
        sunscreen, sunscreen, sunscreen.byteLength, tag, key, nonce, associated_data);
    expect(sunscreen).toEqual(expectedEncrypted);
    expect(tag).toEqual(new Uint8Array([
        0x1a, 0xe1, 0x0b, 0x59, 0x4f, 0x09, 0xe2, 0x6a,
        0x7e, 0x90, 0x2e, 0xcb, 0xd0, 0x60, 0x06, 0x91,
    ]));

    const sunscreen2 = Uint8Array.from(sunscreen);
    expect(ChaCha20Poly1305_RFC8439.decrypt_detached(
        sunscreen2, sunscreen2, sunscreen2.byteLength, tag, key, nonce, associated_data))
        .toBe(true);
    expect(new TextDecoder().decode(sunscreen2)).toBe(sunscreen_str);

    tag[0]++;
    expect(ChaCha20Poly1305_RFC8439.decrypt_detached(
        sunscreen, sunscreen, sunscreen.byteLength, tag, key, nonce, associated_data)).toBe(false);
    expect(sunscreen).toEqual(expectedEncrypted);
});