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

/*
 * This file is part of WinRM.
 *
 * WinRM is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * WinRM 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with WinRM.  If not, see <http://www.gnu.org/licenses/>.
 */
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) {
        Document responseDocument = null;
        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.debug("send message to {}:request {}", targetURL, requestDocAsString);
            BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(con.getOutputStream()));
            bw.write(requestDocAsString, 0, requestDocAsString.length());
            bw.flush();
            bw.close();

            InputStream is;
            if (con.getResponseCode() >= 400) {
                /* Read error response */
                is = httpConn.getErrorStream();
            } else {
                is = httpConn.getInputStream();
            }

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

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

            final String text = writer.toString();
            logger.debug("send message:response {}", text);
            responseDocument = DocumentHelper.parseText(text);
            return responseDocument;

        } catch (BlankValueRuntimeException bvrte) {
            throw bvrte;
        } catch (InvalidFilePathRuntimeException ifprte) {
            throw ifprte;
        } catch (Exception e) {
            throw new WinRMRuntimeIOException("send message on " + targetURL + " error ", requestDocument,
                    responseDocument, 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("error ", e);
        }
        return stringWriter.toString();
    }

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