// SPDX-FileCopyrightText: 2022 Paul Schaub <vanitasvitae@fsfe.org>
//
// SPDX-License-Identifier: Apache-2.0

package org.pgpainless.cert_d;

import static org.junit.jupiter.api.Assertions.assertArrayEquals;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertFalse;
import static org.junit.jupiter.api.Assertions.assertNotNull;
import static org.junit.jupiter.api.Assertions.assertNull;
import static org.junit.jupiter.api.Assertions.assertThrows;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Set;

import org.bouncycastle.util.encoders.Hex;
import org.bouncycastle.util.io.Streams;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.pgpainless.certificate_store.KeyReader;
import org.pgpainless.certificate_store.SharedPGPCertificateDirectoryAdapter;
import pgp.cert_d.InMemorySubkeyLookup;
import pgp.cert_d.SharedPGPCertificateDirectoryImpl;
import pgp.certificate_store.CertificateStore;
import pgp.certificate_store.exception.BadDataException;
import pgp.certificate_store.exception.BadNameException;
import pgp.certificate_store.exception.NotAStoreException;
import pgp.certificate_store.Certificate;

public class SharedPGPCertificateDirectoryAdapterTest {

    private static final String testCertificate = "98330462069cc616092b06010401da470f010107400db5906b09f701ab1f7f96087eedab6ba44c02fcbd2470137cfeacac5a2d032db405416c696365888f0413160a0041050262069cc609906f054e826378552516a104505b134a7e62f0f154ec3d036f054e8263785525029e01029b01059602030100048b09080705950a09080b0299010000a12600fd117925c0f2192ef5b2a44e3d3038e2a7ce5ba0343fc2dfb661a3a46d1276fb380100bf2872e7e36b63f61ae3556464c4a04344e7d36e0d7313e623effb0290ce0b0fb8380462069cc6120a2b06010401975501050101074034ffd523242385fe92034a5e326a82f4edff614516cc1028ca91fb653557f25b0301080788750418160a001d050262069cc6029e01029b0c059602030100048b09080705950a09080b000a09106f054e8263785525391400ff4eb85df8ddfc15e94c9cf28bc0aa9d0426b571ca64c5421be5889d5410d8632f00fd1ac5e9aed683e711282489d8980222d2ceff15c5ce0499fcb36716d850749406b8330462069cc616092b06010401da470f0101074058f296fb7ce456039856144db677f14018963a8bfd281c84aaeebe7e14df8f1c88d50418160a007d050262069cc6029e01029b02059602030100048b09080705950a09080b5f200419160a0006050262069cc6000a09108119c86e0a4c6dc73a7600ff5e25427da84d824cc3f8890bc6bd037f423f610006e1249b1aad3d7f70ac47a100fc08e67a6a945c1feec301df9dc27e7ea4e61d107d0720e814eea1dc4f1da20a08000a09106f054e8263785525359700ff4ce78cf267c261468322de906118d4f003ceefa72fa3b86119e26f99be3727fc00fe3895207c4aac814549f0189d2f494f5b1fcee7f6da344e63a0c32743b216b406";
    private static final String testCertFingerprint = "505b134a7e62f0f154ec3d036f054e8263785525";
    private static final long testCertificateSubkey1 = 7999886635015099685L;
    private static final long testCertificateSubkey2 = -5375724347241457298L;
    private static final long testCertificateSubkey3 = -9144057193454342713L;

    private SharedPGPCertificateDirectoryAdapter adapter;
    private CertificateStore store;

    @BeforeEach
    public void setupInstance() throws IOException, NotAStoreException {
        adapter = new SharedPGPCertificateDirectoryAdapter(
                new SharedPGPCertificateDirectoryImpl(tempDir(), new KeyReader()),
                new InMemorySubkeyLookup());
        store = adapter;
    }

    private static File tempDir() throws IOException {
        File tempDir = Files.createTempDirectory("pgp.cert.d-").toFile();
        tempDir.deleteOnExit();
        return tempDir;
    }

    @Test
    public void getNonExistentCertIsNull() throws IOException, BadDataException, BadNameException {
        assertNull(store.getCertificate("eb85bb5fa33a75e15e944e63f231550c4f47e38e"));
    }

    @Test
    public void getInvalidIdentifierThrows() {
        assertThrows(BadNameException.class, () -> store.getCertificate("invalid"));
    }

    @Test
    public void insertAndGet() throws IOException, InterruptedException, BadDataException, BadNameException {
        byte[] bytes = Hex.decode(testCertificate);
        ByteArrayInputStream byteIn = new ByteArrayInputStream(bytes);
        String fingerprint = testCertFingerprint;

        Certificate certificate = store.insertCertificate(byteIn, (data, existing) -> data);

        assertEquals(fingerprint, certificate.getFingerprint());
        Set<Long> expectedSubkeys = new HashSet<>(Arrays.asList(testCertificateSubkey1, testCertificateSubkey2, testCertificateSubkey3));
        Set<Long> subkeys = certificate.getSubkeyIds();
        assertEquals(expectedSubkeys, subkeys);
        for (long subkey : subkeys) {
            assertEquals(Collections.singleton(fingerprint), store.getCertificateFingerprintsForSubkeyId(subkey));
        }

        Certificate retrieved = store.getCertificate(fingerprint);
        assertNotNull(retrieved);
        ByteArrayOutputStream retrievedOut = new ByteArrayOutputStream();
        Streams.pipeAll(retrieved.getInputStream(), retrievedOut);

        assertArrayEquals(bytes, retrievedOut.toByteArray());
    }


    @Test
    public void tryInsertAndGet() throws IOException, BadDataException, BadNameException {
        byte[] bytes = Hex.decode(testCertificate);
        ByteArrayInputStream byteIn = new ByteArrayInputStream(bytes);
        String fingerprint = testCertFingerprint;

        Certificate certificate = store.tryInsertCertificate(byteIn, (data, existing) -> data);

        assertEquals(fingerprint, certificate.getFingerprint());
        Set<Long> subkeys = certificate.getSubkeyIds();
        assertEquals(3, subkeys.size());
        for (long subkey : subkeys) {
            assertEquals(Collections.singleton(fingerprint), store.getCertificateFingerprintsForSubkeyId(subkey));
        }

        Certificate retrieved = store.getCertificate(fingerprint);
        assertNotNull(retrieved);
        ByteArrayOutputStream retrievedOut = new ByteArrayOutputStream();
        Streams.pipeAll(retrieved.getInputStream(), retrievedOut);

        assertArrayEquals(bytes, retrievedOut.toByteArray());
    }


    @Test
    public void insertAndGetIfChanged() throws IOException, InterruptedException, BadDataException, BadNameException {
        byte[] bytes = Hex.decode(testCertificate);
        ByteArrayInputStream byteIn = new ByteArrayInputStream(bytes);
        String fingerprint = testCertFingerprint;

        Certificate certificate = store.insertCertificate(byteIn, (data, existing) -> data);
        String tag = certificate.getTag();

        assertNull(store.getCertificateIfChanged(fingerprint, tag));
        assertNotNull(store.getCertificateIfChanged(fingerprint, "invalid"));
    }

    @Test
    public void insertBySpecialNameAndGet() throws IOException, InterruptedException, BadDataException, BadNameException {
        byte[] bytes = Hex.decode(testCertificate);
        ByteArrayInputStream byteIn = new ByteArrayInputStream(bytes);
        String fingerprint = testCertFingerprint;
        String identifier = "trust-root";

        Certificate certificate = store.insertCertificateBySpecialName(identifier, byteIn, (data, existing) -> data);

        assertEquals(fingerprint, certificate.getFingerprint());

        Certificate retrieved = store.getCertificate(identifier);
        assertNotNull(retrieved);
        ByteArrayOutputStream retrievedOut = new ByteArrayOutputStream();
        Streams.pipeAll(retrieved.getInputStream(), retrievedOut);

        assertArrayEquals(bytes, retrievedOut.toByteArray());
    }

    @Test
    public void tryInsertBySpecialNameAndGet() throws IOException, BadDataException, BadNameException {
        byte[] bytes = Hex.decode(testCertificate);
        ByteArrayInputStream byteIn = new ByteArrayInputStream(bytes);
        String fingerprint = testCertFingerprint;
        String identifier = "trust-root";

        Certificate certificate = store.tryInsertCertificateBySpecialName(identifier, byteIn, (data, existing) -> data);

        assertEquals(fingerprint, certificate.getFingerprint());

        Certificate retrieved = store.getCertificate(identifier);
        assertNotNull(retrieved);
        ByteArrayOutputStream retrievedOut = new ByteArrayOutputStream();
        Streams.pipeAll(retrieved.getInputStream(), retrievedOut);

        assertArrayEquals(bytes, retrievedOut.toByteArray());
    }

    @Test
    public void insertBySpecialNameAndGetIfChanged() throws IOException, InterruptedException, BadDataException, BadNameException {
        byte[] bytes = Hex.decode(testCertificate);
        ByteArrayInputStream byteIn = new ByteArrayInputStream(bytes);
        String fingerprint = testCertFingerprint;
        String identifier = "trust-root";

        Certificate certificate = store.insertCertificateBySpecialName(identifier, byteIn, (data, existing) -> data);
        String tag = certificate.getTag();

        certificate = store.getCertificateIfChanged(identifier, tag);
        assertNull(certificate);
        certificate = store.getCertificateIfChanged(identifier, "invalid");
        assertNotNull(certificate);
        assertEquals(fingerprint, certificate.getFingerprint());
    }

    @Test
    public void getItemsAndFingerprints() throws IOException, InterruptedException, BadDataException, BadNameException {
        byte[] bytes1 = Hex.decode(testCertificate);
        ByteArrayInputStream byteIn1 = new ByteArrayInputStream(bytes1);
        Certificate firstCert = store.insertCertificate(byteIn1, (data, existing) -> data);

        byte[] bytes2 = Hex.decode("9833046206a37516092b06010401da470f010107409f55baab1599044096ba901d69854cf5307b84b0542871b15db3dd4c62664f37b403426f62888f0413160a004105026206a3750990ba01b5a9eea7e76716a104f1d47fb85ad74549a37974f3ba01b5a9eea7e767029e01029b01059602030100048b09080705950a09080b0299010000e6170100e08374a6fd32d0b4be2d3f7c75d3f6c13cb47b1b73589aa452a1b2a16b888b5000fe274e6565ab9faa34338cf4d805663f8775fdee4ec6a0fdf1ec2cf84b72907f05b838046206a375120a2b0601040197550105010107405641e74d2dda92003ce200422c3ab6f3562fc49a8ecc67ea02593988442b23780301080788750418160a001d05026206a375029e01029b0c059602030100048b09080705950a09080b000a0910ba01b5a9eea7e76732850100910a6049779773f455226cd91645884842b91017796287a634104ab5364a0c0d00fe20b5febb17de271394f31128f709c307c0bbca4f9502570744bd54e6dc9c2209b833046206a37516092b06010401da470f0101074059f008928cb69b48bed07a639f03f43a48808aade67109cd658f54bddefa5ec288d50418160a007d05026206a375029e01029b02059602030100048b09080705950a09080b5f200419160a000605026206a375000a0910dcdb34f4068368c0dffb010095fb1f6daac239bf3221d9d2ecc81b6cb258c2b058a300a7e103f7f36a58bf1900fe273a9eaaa03b613236df22bebcbbd69d7c02caf1b7af4fa29320c8d96d32310f000a0910ba01b5a9eea7e7671de20100a5044d24a9d860f9af7e8b9a095d4eac8820fad8b045e70be1ae5607fa4d6b4f010097b53d1527f3b3e3d3b78367c8269c999ee37575a51ffc582f73d2cba4df080f");
        ByteArrayInputStream byteIn2 = new ByteArrayInputStream(bytes2);
        Certificate secondCert = store.insertCertificate(byteIn2, ((data, existing) -> data));

        String trustRootHex = "9833046206a57e16092b06010401da470f010107401ad7351d9766843bf11a8414f68790df0649fad8b01c244323f47e4ebc87fc35b40a74727573742d726f6f74888f0413160a004105026206a57f09907c619691ddee5fc216a10489e1e05cb458758d0729eb0c7c619691ddee5fc2029e01029b01059602030100048b09080705950a09080b029901000080c100ff45d97dda133895e337416266f1ff2c38ff3947ecfbfe21328d51bc877ccba367010096698a5fbac9444b7b28b96389c66ca405821f04871f1bbbf5b5bf8b800f9104b838046206a57f120a2b06010401975501050101074074ff41705c50e8f27b18df40a53aded6cacd2ce4f88b471c7130036010ca60240301080788750418160a001d05026206a57f029e01029b0c059602030100048b09080705950a09080b000a09107c619691ddee5fc27b3c0100fba12230adf80a6a7a376b9568481ab4ae86628274db67412074cb4a846011a200ff437e4047bbafec42b41594b296f8be93fc03482b2d35ac92e87ce632b86bc900b833046206a57f16092b06010401da470f01010740ce99f97d1f0b5aa2f4e6f2a7a2aa231da8c2a2f489a593b747983a750f3928ae88d50418160a007d05026206a57f029e01029b02059602030100048b09080705950a09080b5f200419160a000605026206a57f000a0910b905cb706dec67e3f6050100a7ae51ea07f3d0d493fd1fdfbcbbe112c19de8dbbd29e03ba5e755345444402300fe2663252eeca21772012c5dc4eb9efa4e01566dffbb44e7d1536181eb3f8b420e000a09107c619691ddee5fc2a4190100fdbedf9defd5d30bad77937a5589441ef336028613a6fcfc4a959bee51de134e00fd128628567b66fa03ef099d6936324f7593e2060608b433828d336dda552e2c04";
        byte[] trustRootBytes = Hex.decode(trustRootHex);
        ByteArrayInputStream trustRootIn = new ByteArrayInputStream(trustRootBytes);
        Certificate trustRoot = store.insertCertificateBySpecialName("trust-root", trustRootIn, (data, existing) -> data);

        Set<String> expectedFingerprints = new HashSet<>();
        expectedFingerprints.add(firstCert.getFingerprint());
        expectedFingerprints.add(secondCert.getFingerprint());

        Iterator<Certificate> certificateIterator = store.getCertificates();
        Set<String> actualFingerprints = new HashSet<>();
        Certificate c = certificateIterator.next();
        actualFingerprints.add(c.getFingerprint());
        c = certificateIterator.next();
        actualFingerprints.add(c.getFingerprint());
        assertFalse(certificateIterator.hasNext());

        assertEquals(expectedFingerprints, actualFingerprints);
        assertFalse(actualFingerprints.contains(trustRoot.getFingerprint()));

        Iterator<String> fingerprintIterator = store.getFingerprints();
        actualFingerprints = new HashSet<>();
        actualFingerprints.add(fingerprintIterator.next());
        actualFingerprints.add(fingerprintIterator.next());
        assertFalse(fingerprintIterator.hasNext());

        assertEquals(expectedFingerprints, actualFingerprints);
        assertFalse(actualFingerprints.contains(trustRoot.getFingerprint()));
    }
}