package com.cmy.urlcon;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.net.HttpURLConnection;
import java.net.InetSocketAddress;
import java.net.Proxy;
import java.net.URL;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;

public class HttpRequestor {

    private String charset = "utf-8";
    private Integer connectTimeout = null;
    private Integer socketTimeout = null;
    private String proxyHost = null;
    private Integer proxyPort = null;

    public String doGet(String urlPath) throws Exception {

        URL url = new URL(urlPath);
        HttpURLConnection connection = (HttpURLConnection) url.openConnection();
        connection.setRequestProperty("Accept-Charset", charset);
        connection.setRequestProperty("Content-Type",
                "application/x-www-form-urlencoded");

        InputStream inputStream = null;
        InputStreamReader inputStreamReader = null;
        BufferedReader bfReader = null;
        StringBuffer sb = new StringBuffer();
        String readLine = null;

        if (connection.getResponseCode() >= 300) {
            throw new Exception("HTTP Request is failue, Response code is "
                    + connection.getResponseCode());
        }

        try {
            inputStream = connection.getInputStream();
            inputStreamReader = new InputStreamReader(inputStream);
            bfReader = new BufferedReader(inputStreamReader);
            while ((readLine = bfReader.readLine()) != null) {
                sb.append(readLine);
            }

        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            bfReader.close();
            inputStreamReader.close();
            inputStream.close();
        }

        return sb.toString();

    }

    public static String doPost(String urlPath, Map paramMap) throws Exception {
        URL url = new URL(urlPath);
        HttpURLConnection con = (HttpURLConnection) url.openConnection();
        StringBuffer paramBuffer = new StringBuffer();

        if (paramMap != null) {
            Iterator iterator = paramMap.keySet().iterator();
            String key = null;
            String value = null;
            while (iterator.hasNext()) {
                key = (String) iterator.next();

                if ((value = http://www.mamicode.com/(String) paramMap.get(key)) != null) {
                    paramBuffer.append(value);
                } else {
                    value = "";
                }
                paramBuffer.append(key).append("=").append(value);
                if (iterator.hasNext()) {
                    paramBuffer.append("&");
                }
            }
        }
        System.out.println("POST parameter: " + paramBuffer.toString());

        con.setRequestProperty("Accept-Charset", "utf-8");
        con.setRequestProperty("Content-Type",
                "application/x-www-form-urlencoded");
        con.setConnectTimeout(3000);
        con.setDoInput(true);
        con.setDoOutput(true);
        con.setUseCaches(false);
        String cookie0 = con.getHeaderField("Set-Cookie");

        OutputStream outputStream = null;
        OutputStreamWriter outputStreamWriter = null;
        InputStream inputStream = null;
        InputStreamReader inputStreamReader = null;
        BufferedReader bfReader = null;
        StringBuffer sb = new StringBuffer();
        String readLine = null;

        try {
            outputStream = con.getOutputStream();
            outputStreamWriter = new OutputStreamWriter(outputStream);
            outputStreamWriter.write(paramBuffer.toString());
            outputStreamWriter.flush();

        } catch (Exception e) {
            e.printStackTrace();
        }

        if (con.getResponseCode() >= 300) {
            throw new Exception("HTTP Request is failue, Response code is "
                    + con.getResponseCode());
        }

        try {
            inputStream = con.getInputStream();
            inputStreamReader = new InputStreamReader(inputStream);
            bfReader = new BufferedReader(inputStreamReader);
            while ((readLine = bfReader.readLine()) != null) {
                sb.append(readLine);
            }

        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            outputStreamWriter.close();
            outputStream.close();
            bfReader.close();
            inputStreamReader.close();
            inputStream.close();
        }

        return sb.toString();
    }

    private HttpURLConnection openConnection(URL localURL) throws IOException {
        HttpURLConnection connection;
        if (proxyHost != null && proxyPort != null) {
            Proxy proxy = new Proxy(Proxy.Type.HTTP, new InetSocketAddress(proxyHost, proxyPort));
            connection = (HttpURLConnection) localURL.openConnection(proxy);
        } else {
            connection = (HttpURLConnection) localURL.openConnection();
        }
        return connection;
    }

    private void renderRequest(HttpURLConnection connection) {

        if (connectTimeout != null) {
            connection.setConnectTimeout(connectTimeout);
        }

        if (socketTimeout != null) {
            connection.setReadTimeout(socketTimeout);
        }

    }
    

    public String getCharset() {
        return charset;
    }

    public void setCharset(String charset) {
        this.charset = charset;
    }

    public Integer getConnectTimeout() {
        return connectTimeout;
    }

    public void setConnectTimeout(Integer connectTimeout) {
        this.connectTimeout = connectTimeout;
    }

    public Integer getSocketTimeout() {
        return socketTimeout;
    }

    public void setSocketTimeout(Integer socketTimeout) {
        this.socketTimeout = socketTimeout;
    }

    public String getProxyHost() {
        return proxyHost;
    }

    public void setProxyHost(String proxyHost) {
        this.proxyHost = proxyHost;
    }

    public Integer getProxyPort() {
        return proxyPort;
    }

    public void setProxyPort(Integer proxyPort) {
        this.proxyPort = proxyPort;
    }
    
    /**
     * test case 
     * @param args
     * @throws Exception
     */
    public static void main(String[] args) throws Exception {
        /* Post Request */
        Map dataMap = new HashMap();
        dataMap.put("username", "Nick Huang");
        dataMap.put("blog", "IT");
        System.out.println(new HttpRequestor().doPost("http://localhost:8080/OneHttpServer/", dataMap));
        
        /* Get Request */
        System.out.println(new HttpRequestor().doGet("http://localhost:8080/OneHttpServer/"));

    }

}