/*
 * JBoss, Home of Professional Open Source.
 * Copyright 2011, Red Hat, Inc., and individual contributors
 * as indicated by the @author tags. See the copyright.txt file in the
 * distribution for a full listing of individual contributors.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation; either version 2.1 of
 * the License, or (at your option) any later version.
 *
 * This software is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this software; if not, write to the Free
 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */
package org.jboss.as.test.integration.management.util;

import io.undertow.util.StatusCodes;
import java.net.URL;

import org.apache.http.HttpResponse;
import org.apache.http.auth.AuthScope;
import org.apache.http.auth.UsernamePasswordCredentials;
import org.apache.http.client.CredentialsProvider;
import org.apache.http.client.HttpClient;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.BasicCredentialsProvider;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.protocol.BasicHttpContext;
import org.apache.http.protocol.HttpContext;
import org.apache.http.util.EntityUtils;
import org.jboss.as.test.http.Authentication;
import org.jboss.dmr.ModelNode;

/**
 *
 * @author Dominik Pospisil <dpospisi@redhat.com>
 */
public class HttpMgmtProxy {

    private static final String APPLICATION_JSON = "application/json";
    private URL url;
    private HttpClient httpClient;
    private HttpContext httpContext = new BasicHttpContext();

    public HttpMgmtProxy(URL mgmtURL) {
        this.url = mgmtURL;
        CredentialsProvider credsProvider = new BasicCredentialsProvider();
                credsProvider.setCredentials(new AuthScope(url.getHost(), url.getPort()), new UsernamePasswordCredentials(Authentication.USERNAME, Authentication.PASSWORD));

        this.httpClient = HttpClients.custom()
                .setDefaultCredentialsProvider(credsProvider)
                .build();
    }

    public ModelNode sendGetCommand(String cmd) throws Exception {
        return ModelNode.fromJSONString(sendGetCommandJson(cmd));
    }

    public String sendGetCommandJson(String cmd) throws Exception {
        HttpGet get = new HttpGet(url.toURI().toString() + cmd);

        HttpResponse response = httpClient.execute(get, httpContext);
        return EntityUtils.toString(response.getEntity());
    }

    public ModelNode sendPostCommand(String address, String operation) throws Exception {
        return sendPostCommand(getOpNode(address, operation));
    }

    public ModelNode sendPostCommand(ModelNode cmd) throws Exception {
        return sendPostCommand(cmd, false);
    }

    public ModelNode sendPostCommand(ModelNode cmd, boolean ignoreFailure) throws Exception {
        String cmdStr = cmd.toJSONString(true);
        HttpPost post = new HttpPost(url.toURI());
        StringEntity entity = new StringEntity(cmdStr);
        entity.setContentType(APPLICATION_JSON);
        post.setEntity(entity);

        HttpResponse response = httpClient.execute(post, httpContext);
        String str = EntityUtils.toString(response.getEntity());
        if (response.getStatusLine().getStatusCode() == StatusCodes.OK || ignoreFailure) {
            return ModelNode.fromJSONString(str);
        }
        throw new Exception("Could not execute command: " + str);
    }

    public static ModelNode getOpNode(String address, String operation) {
        ModelNode op = new ModelNode();

        // set address
        String[] pathSegments = address.split("/");
        ModelNode list = op.get("address").setEmptyList();
        for (String segment : pathSegments) {
            String[] elements = segment.split("=");
            list.add(elements[0], elements[1]);
        }
        op.get("operation").set(operation);
        return op;
    }
}
