no.digipost.api.representations.Dokumentpakke Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sdp-api-commons Show documentation
Felles kode for API-relatert funksjonalitet
The newest version!
package no.digipost.api.representations;


import jakarta.activation.DataSource;
import org.bouncycastle.jcajce.provider.digest.SHA256;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.security.DigestOutputStream;
import java.security.MessageDigest;

public class Dokumentpakke implements DataSource {

    public static final String CONTENT_TYPE_KRYPTERT_DOKUMENTPAKKE = "application/cms";
    private final InputStream asicStream;
    private byte[] asicBytes;

    public Dokumentpakke(final InputStream asicStream) {
        this.asicStream = asicStream;
        this.asicBytes = null;
    }

    public Dokumentpakke(final byte[] asicBytes) {
        this.asicBytes = asicBytes;
        this.asicStream = new ByteArrayInputStream(asicBytes);
    }

    /**
     * Based on the {@code private} {@link java.nio.file.Files#copy(InputStream, OutputStream)}. This
     * implementation is included here to avoid depending on Apache Commons IO just for this simple case.
     */
    private static void copy(InputStream source, OutputStream sink) throws IOException {
        byte[] buf = new byte[8192];
        int n;
        while ((n = source.read(buf)) > 0) {
            sink.write(buf, 0, n);
        }
    }

    public byte[] getSHA256() throws IOException {
        MessageDigest digest = getDigest();
        if (asicBytes != null) {
            return digest.digest(asicBytes);
        }
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        try (InputStream asicToRead = asicStream; DigestOutputStream digestStream = new DigestOutputStream(baos, digest)) {
            copy(asicToRead, digestStream);
        }
        asicBytes = baos.toByteArray();
        return digest.digest();
    }

    protected MessageDigest getDigest() {
        return new SHA256.Digest();
    }

    @Override
    public String getContentType() {
        return Dokumentpakke.CONTENT_TYPE_KRYPTERT_DOKUMENTPAKKE;
    }

    @Override
    public InputStream getInputStream() throws IOException {
        if (asicBytes != null) {
            return new ByteArrayInputStream(asicBytes);
        }
        return asicStream;
    }

    @Override
    public String getName() {
        return "asic.cms";
    }

    @Override
    public OutputStream getOutputStream() throws IOException {
        throw new UnsupportedOperationException("Not supported by handler");
    }
}