wkd-java/wkd-java-cli/src/test/java/pgp/wkd/cli/test_suite/DirectoryBasedCertificateFetcher.java

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

package pgp.wkd.cli.test_suite;

import pgp.wkd.discovery.DiscoveryMethod;
import pgp.wkd.WKDAddress;
import pgp.wkd.discovery.CertificateFetcher;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.net.URI;
import java.nio.file.Path;

public class DirectoryBasedCertificateFetcher implements CertificateFetcher {

    // The directory containing the .well-known subdirectory
    private final Path rootPath;

    public DirectoryBasedCertificateFetcher(Path rootPath) {
        this.rootPath = rootPath;
    }

    @Override
    public InputStream fetchCertificate(WKDAddress address, DiscoveryMethod method) throws IOException {
        return inputStreamFromFile(address, method);
    }

    @Override
    public InputStream fetchPolicy(WKDAddress address, DiscoveryMethod method) throws IOException {
        return inputStreamFromFile(address, method);
    }

    private InputStream inputStreamFromFile(WKDAddress address, DiscoveryMethod method) throws FileNotFoundException {
        URI uri = address.getUri(method);
        String path = uri.getPath().substring(1); // get rid of leading slash at start of path
        File file = rootPath.resolve(path).toFile();
        FileInputStream fileIn = new FileInputStream(file);
        return fileIn;
    }
}