package com.sequenceiq.cloudbreak.cloud.azure.task.interactivelogin;

import static com.sequenceiq.cloudbreak.cloud.azure.task.interactivelogin.AzureInteractiveLoginStatusCheckerTask.GRAPH_API_VERSION;
import static com.sequenceiq.cloudbreak.cloud.azure.task.interactivelogin.AzureInteractiveLoginStatusCheckerTask.GRAPH_WINDOWS;

import java.io.IOException;
import java.time.LocalDateTime;
import java.util.Date;
import java.util.UUID;

import javax.ws.rs.client.Client;
import javax.ws.rs.client.ClientBuilder;
import javax.ws.rs.client.Entity;
import javax.ws.rs.client.Invocation;
import javax.ws.rs.client.WebTarget;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Service;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.google.gson.JsonArray;
import com.google.gson.JsonObject;
import com.google.gson.JsonPrimitive;

/**
 * Created by perdos on 10/18/16.
 */
@Service
public class ApplicationCreator {

    public static final int CREDENTIAL_END_YEAR = 3;

    private static final Logger LOGGER = LoggerFactory.getLogger(ApplicationCreator.class);

    public String createApplication(String accessToken, String tenantId) throws InteractiveLoginException {
        Response response = createApplicationWithGraph(accessToken, tenantId);

        if (response.getStatusInfo().getFamily() == Response.Status.Family.SUCCESSFUL) {
            String application = response.readEntity(String.class);
            try {
                JsonNode applicationJson = new ObjectMapper().readTree(application);
                String appId = applicationJson.get("appId").asText();
                LOGGER.info("Application created with appId: " + appId);
                return appId;
            } catch (IOException e) {
                throw new IllegalStateException(e);
            }
        } else {
            String errorResponse = response.readEntity(String.class);
            try {
                String errorMessage = new ObjectMapper().readTree(errorResponse).get("odata.error").get("message").get("value").asText();
                throw new InteractiveLoginException("AD Application creation error: " + errorMessage);
            } catch (IOException e) {
                throw new IllegalStateException(e);
            }
        }

    }

    private Response createApplicationWithGraph(String accessToken, String tenantId) {
        Client client = ClientBuilder.newClient();
        WebTarget resource = client.target(GRAPH_WINDOWS + tenantId);
        Invocation.Builder request = resource.path("/applications").queryParam("api-version", GRAPH_API_VERSION).request();
        request.accept(MediaType.APPLICATION_JSON);

        long timeStamp = new Date().getTime();

        JsonObject jsonObject = new JsonObject();
        jsonObject.addProperty("availableToOtherTenants", false);
        jsonObject.addProperty("displayName", "hwx-cloud-" + timeStamp);
        jsonObject.addProperty("homepage", "http://hwx-cloud-" + timeStamp);

        JsonArray identifierUris = new JsonArray();
        identifierUris.add(new JsonPrimitive("http://hwx-cloud-" + timeStamp));
        jsonObject.add("identifierUris", identifierUris);

        JsonArray passwordCredentials = new JsonArray();
        JsonObject password = new JsonObject();
        password.addProperty("keyId", UUID.randomUUID().toString());
        password.addProperty("value", "cloudbreak");
        password.addProperty("startDate", LocalDateTime.now().minusDays(1).toString());
        password.addProperty("endDate", LocalDateTime.now().plusYears(CREDENTIAL_END_YEAR).toString());
        passwordCredentials.add(password);

        jsonObject.add("passwordCredentials", passwordCredentials);

        request.header("Authorization", "Bearer " + accessToken);
        return request.post(Entity.entity(jsonObject.toString(), MediaType.APPLICATION_JSON));
    }
}