// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.

package labapi;

import javax.net.ssl.HttpsURLConnection;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.UnsupportedEncodingException;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLEncoder;
import java.util.Map;

class HttpClientHelper {
    static String sendRequestToLab(String url, Map<String, String> queryMap, String accessToken)
            throws IOException {
        return sendRequestToLab(buildUrl(url, queryMap), accessToken);
    }

    static String sendRequestToLab(URL labUrl, String accessToken) throws IOException {
        HttpsURLConnection conn = (HttpsURLConnection)labUrl.openConnection();

        conn.setRequestProperty("Authorization", "Bearer " + accessToken);

        conn.setReadTimeout(20000);
        conn.setConnectTimeout(20000);

        StringBuilder content;
        try (BufferedReader in = new BufferedReader(new InputStreamReader(conn.getInputStream()))) {
            String inputLine;
            content = new StringBuilder();
            while ((inputLine = in.readLine()) != null) {
                content.append(inputLine);
            }
        }
        conn.disconnect();
        return content.toString();
    }

    private static URL buildUrl(String url, Map<String, String> queryMap)
            throws MalformedURLException, UnsupportedOperationException {
        String queryParameters;
        queryParameters = queryMap.entrySet()
                                  .stream()
                                  .map(p -> encodeUTF8(p.getKey()) + "=" + encodeUTF8(p.getValue()))
                                  .reduce((p1, p2) -> p1 + "&" + p2)
                                  .orElse("");

        String urlString = url + "?" + queryParameters;
        return new URL(urlString);
    }

    private static String encodeUTF8(String s) {
        try {
            return URLEncoder.encode(s, "UTF-8");
        } catch (UnsupportedEncodingException e) {
            throw new IllegalArgumentException("Error: cannot encode query parameter " + s);
        }
    }
}