com.xebialabs.overthere.cifs.winrm.connector.JdkHttpConnector.java Source code

Introduction

Here is the source code for com.xebialabs.overthere.cifs.winrm.connector.JdkHttpConnector.java

Source

/* License added by: GRADLE-LICENSE-PLUGIN
 *
 * Copyright 2008-2012 XebiaLabs
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *     http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.xebialabs.overthere.cifs.winrm.connector;

import com.google.common.io.Closeables;
import com.xebialabs.overthere.cifs.winrm.HttpConnector;
import com.xebialabs.overthere.cifs.winrm.SoapAction;
import com.xebialabs.overthere.cifs.winrm.TokenGenerator;
import com.xebialabs.overthere.cifs.winrm.exception.BlankValueRuntimeException;
import com.xebialabs.overthere.cifs.winrm.exception.InvalidFilePathRuntimeException;
import com.xebialabs.overthere.cifs.winrm.exception.WinRMRuntimeIOException;
import org.dom4j.Document;
import org.dom4j.DocumentHelper;
import org.dom4j.io.OutputFormat;
import org.dom4j.io.XMLWriter;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;
import java.net.URLConnection;

/**
 */
public class JdkHttpConnector implements HttpConnector {

    private final URL targetURL;

    private final TokenGenerator tokenGenerator;

    public JdkHttpConnector(URL targetURL, TokenGenerator tokenGenerator) {
        this.targetURL = targetURL;
        this.tokenGenerator = tokenGenerator;
    }

    @Override
    public Document sendMessage(Document requestDocument, SoapAction soapAction) {
        try {
            final URLConnection urlConnection = targetURL.openConnection();
            HttpURLConnection con = (HttpURLConnection) urlConnection;

            con.setDoInput(true);
            con.setDoOutput(true);
            con.setRequestMethod("POST");
            con.setRequestProperty("Content-Type", "application/soap+xml; charset=UTF-8");

            final String authToken = tokenGenerator.generateToken();
            if (authToken != null)
                con.addRequestProperty("Authorization", authToken);

            if (soapAction != null) {
                con.setRequestProperty("SOAPAction", soapAction.getValue());
            }

            final String requestDocAsString = toString(requestDocument);
            logger.trace("Sending request to {}", targetURL);
            logger.trace("Request body: {} {}", targetURL, requestDocAsString);
            BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(con.getOutputStream()));
            try {
                bw.write(requestDocAsString, 0, requestDocAsString.length());
            } finally {
                Closeables.closeQuietly(bw);
            }

            InputStream is = null;
            if (con.getResponseCode() >= 400) {
                is = con.getErrorStream();
            }
            if (is == null) {
                is = con.getInputStream();
            }

            Writer writer = new StringWriter();
            Reader reader = new BufferedReader(new InputStreamReader(is, "UTF-8"));
            try {
                int n;
                char[] buffer = new char[1024];
                while ((n = reader.read(buffer)) != -1) {
                    writer.write(buffer, 0, n);
                }
            } finally {
                Closeables.closeQuietly(reader);
                Closeables.closeQuietly(is);
            }

            if (logger.isDebugEnabled()) {
                for (int i = 0; i < con.getHeaderFields().size(); i++) {
                    logger.trace("Header {}: {}", con.getHeaderFieldKey(i), con.getHeaderField(i));
                }
            }

            final String text = writer.toString();
            logger.trace("Response body: {}", text);

            return DocumentHelper.parseText(text);
        } catch (BlankValueRuntimeException bvrte) {
            throw bvrte;
        } catch (InvalidFilePathRuntimeException ifprte) {
            throw ifprte;
        } catch (Exception e) {
            throw new WinRMRuntimeIOException("Send message on " + targetURL + " error ", requestDocument, null, e);
        }
    }

    private String toString(Document doc) {
        StringWriter stringWriter = new StringWriter();
        XMLWriter xmlWriter = new XMLWriter(stringWriter, OutputFormat.createPrettyPrint());
        try {
            xmlWriter.write(doc);
            xmlWriter.close();
        } catch (IOException e) {
            throw new WinRMRuntimeIOException("Cannnot convert XML to String ", e);
        }
        return stringWriter.toString();
    }

    private static Logger logger = LoggerFactory.getLogger(JdkHttpConnector.class);
}