/*
  Copyright (c) 2013, 2014, Oracle and/or its affiliates. All rights reserved.

  The MySQL Connector/J is licensed under the terms of the GPLv2
  <http://www.gnu.org/licenses/old-licenses/gpl-2.0.html>, like most MySQL Connectors.
  There are special exceptions to the terms and conditions of the GPLv2 as it is applied to
  this software, see the FLOSS License Exception
  <http://www.mysql.com/about/legal/licensing/foss-exception.html>.

  This program 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; version 2
  of the License.

  This program 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 this
  program; if not, write to the Free Software Foundation, Inc., 51 Franklin St, Fifth
  Floor, Boston, MA 02110-1301  USA

 */

package com.mysql.fabric.xmlrpc;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.HttpURLConnection;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.HashMap;
import java.util.Map;

import javax.xml.parsers.ParserConfigurationException;
import javax.xml.parsers.SAXParser;
import javax.xml.parsers.SAXParserFactory;

import org.xml.sax.SAXException;

import com.mysql.fabric.xmlrpc.base.MethodCall;
import com.mysql.fabric.xmlrpc.base.MethodResponse;
import com.mysql.fabric.xmlrpc.base.ResponseParser;
import com.mysql.fabric.xmlrpc.exceptions.MySQLFabricException;

/**
 * XML-RPC client.
 */
public class Client {
	private URL url;
	private Map<String, String> headers = new HashMap<String, String>();

	public Client(String url) throws MalformedURLException {
		this.url = new URL(url);
	}

	public void setHeader(String name, String value) {
		headers.put(name, value);
	}

	public void clearHeader(String name) {
		headers.remove(name);
	}

	public MethodResponse execute(MethodCall methodCall) throws IOException,
			ParserConfigurationException, SAXException, MySQLFabricException {
		HttpURLConnection connection = null;
		try {
			connection = (HttpURLConnection) this.url.openConnection();
			connection.setRequestMethod("POST");
			connection.setRequestProperty("User-Agent", "MySQL XML-RPC");
			connection.setRequestProperty("Content-Type", "text/xml");
			connection.setUseCaches(false);
			connection.setDoInput(true);
			connection.setDoOutput(true);

			// apply headers
			for (Map.Entry<String, String> entry : headers.entrySet()) {
				connection.setRequestProperty(entry.getKey(), entry.getValue());
			}

			String out = methodCall.toString();

			// Send request
			OutputStream os = connection.getOutputStream();
			os.write(out.getBytes());
			os.flush();
			os.close();

			// Get Response
			InputStream is = connection.getInputStream();
			SAXParserFactory factory = SAXParserFactory.newInstance();
			SAXParser parser = factory.newSAXParser();
			ResponseParser saxp = new ResponseParser();

			parser.parse(is, saxp);

			is.close();

			MethodResponse resp = saxp.getMethodResponse();
			if (resp.getFault() != null) {
				throw new MySQLFabricException(resp.getFault()); 
			}
			
			return resp;

		} finally {
			if (connection != null) {
				connection.disconnect();
			}
		}

	}

}
