import {test, expect, describe} from 'bun:test'

import {signature} from '../index'
import {derive_keypair, gen_keypair, sign, verify} from "../src/signature";

test('base case', async () => {
    const keypair = await gen_keypair()
    const data = new TextEncoder().encode("Message 123 !")

    const sig = await sign(data, keypair.privateKey)
    const verification = await verify(data, keypair.publicKey, sig)

    expect(verification).toBe(true)
})

test('inverted keys', async () => {
    const keypair = await signature.gen_keypair()
    const data = new TextEncoder().encode("Message 123 !")

    expect(async () => await sign(data, keypair.publicKey)).toThrow()
})

test('tampered message', async () => {
    const keypair = await signature.gen_keypair()
    const data1 = new TextEncoder().encode("Message 123 !")
    const data2 = new TextEncoder().encode("Message 321 !")
    expect(data1).not.toEqual(data2)

    const sig = await sign(data1, keypair.privateKey)
    const verification = await verify(data2, keypair.publicKey, sig)

    expect(verification).toBe(false)
})

test('different keypair', async () => {
    const keypair = await signature.gen_keypair()
    const keypair2 = await signature.gen_keypair()
    const data = new TextEncoder().encode("Message 123 !")

    const sig = await sign(data, keypair.privateKey)
    const verification = await verify(data, keypair2.publicKey, sig)

    expect(verification).toBe(false)
})

test('tampered signature', async () => {
    const keypair = await signature.gen_keypair()
    const data = new TextEncoder().encode("Message 123 !")

    const sig = await sign(data, keypair.privateKey)
    sig[0] ^= 1
    const verification = await verify(data, keypair.publicKey, sig)

    expect(verification).toBe(false)
})

describe('Derive array', () => {
    test('Case derive key and use it', async () => {
        const data = new Uint8Array([1])
        const kp = await derive_keypair(data)

        const message = new Uint8Array(12)
        const sig = await sign(message, kp[0])
        const verif = await verify(message, kp[1], sig)

        expect(verif).toBeTrue()
    })

    test('Same derive, same key', async () => {
        const data = new Uint8Array([1,2,3])

        const k1 = await derive_keypair(data)
        const k2 = await derive_keypair(data)

        const message = new Uint8Array(12)
        const sig = await sign(message, k1[0])
        const verif = await verify(message, k2[1], sig)

        expect(verif).toBeTrue()
    })

    test('Different input, different key', async () => {
        const d1 = new Uint8Array([1,2,3])
        const d2 = new Uint8Array([3,2,1])

        const k1 = await derive_keypair(d1)
        const k2 = await derive_keypair(d2)

        const message = new Uint8Array(12)
        const sig = await sign(message, k1[0])
        const verif = await verify(message, k2[1], sig)

        expect(verif).toBeFalse()
    })
})