com.github.nbaars.pwnedpasswords4j.client.PwnedPasswordClient Maven / Gradle / Ivy

Go to download
package com.github.nbaars.pwnedpasswords4j.client;

import okhttp3.*;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutionException;

import static java.lang.System.getProperty;

public class PwnedPasswordClient {

    private final OkHttpClient httpClient;
    private final String url;
    private final String userAgent;

    public PwnedPasswordClient(OkHttpClient httpClient, String url, String userAgent) {
        this.httpClient = httpClient;
        this.url = url;
        this.userAgent = userAgent;
    }

    CompletableFuture> fetchHashesAsync(Hex hashedPassword) {
        Call call = httpClient.newCall(buildRequest(hashedPassword));
        CompletableFuture> future = new CompletableFuture<>();

        call.enqueue(new Callback() {

            @Override
            public void onFailure(Call call, IOException e) {
                future.completeExceptionally(e);
            }

            @Override
            public void onResponse(Call call, Response response) {
                try {
                    future.complete(parseResponse(hashedPassword.firstFive(), response));
                } catch (IOException e) {
                    future.completeExceptionally(e);
                }
            }
        });
        return future;
    }

    private Request buildRequest(Hex hashedPassword) {
        Hex range = hashedPassword.firstFive();
        return new Request.Builder()
                .url(url + "/" + range)
                .header("User-Agent", userAgent)
                .build();
    }

    private List parseResponse(Hex range, Response response) throws IOException {
        try {
            if (!response.isSuccessful()) {
                throw new RuntimeException("Call failed to " + url + " got " + response);
            }
            List hashes = new ArrayList<>();
            String[] lines = response.body().string().split(getProperty("line.separator"));

            for (String line : lines) {
                hashes.add(Hex.from(range + line.substring(0, line.lastIndexOf(":"))));
            }
            return hashes;
        } finally {
            response.close();
        }
    }

    public static void main(String[] args) throws ExecutionException, InterruptedException {
        List hashes = new PwnedPasswordClient(
                new OkHttpClient(), "https://api.pwnedpasswords.com/range/", "PwnedPasswordClient for Java"
        ).fetchHashesAsync(Hex.from("5baa6")).get();

        hashes.forEach(System.out::println);
    }
}