（四）——Request

package my.tomcat2;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.UnsupportedEncodingException;
import java.net.Socket;
import java.net.URLDecoder;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 用来接收客户端发来的头信息，并且解析头信息的第一句，也就是 method 是 Get 还是 POST， URL， param（参数）
 */
public class Request {
    private static final String CRLF = "\r\n";
    //将参数存入 Map 中去，因为有可能会有要读取 CheckBox 复选框的参数值，他们属于同一个 name，所以要将他们存入到一个 List 中去
    private Map<String, List<String>> paramMap;
    //读取客户端的 头信息
    private BufferedReader bw;
    //将读取到的头信息append到一个StringBuilder中
    private StringBuilder requestInfo;
    //URL
    private String url;
    //Get 或者 POST
    private String method;
    //参数
    private String param = "";
    
    //初始化 字段
    public Request() {
        paramMap = new HashMap<>();
    }

    public Request(Socket client) {
        try {
            paramMap = new HashMap<>();
            bw = new BufferedReader(new InputStreamReader(client.getInputStream()));
            requestInfo = new StringBuilder();
            String msg;
            //这里有两个问题，线程是会循环的，所以会读到什么还请自己动手 BUG 一番，第一次循环最后会读到“”，第二次会首先读到 null
            String hi = bw.readLine();
            while (hi != null && !((msg = hi).equals(""))) {
                hi = bw.readLine();
                requestInfo.append(msg).append(CRLF);
            }
        } catch (IOException e) {
            CloseUtil.closeSocket(client);
        }
        //将读到的 头信息 解析
        if(!(requestInfo.toString().equals(""))) {
            parseRequestInfo(requestInfo.toString());
        }
    }
    
    //将所有 substring（）获得的字符串调用 trim（），去掉空白字符，不然会出问题
    public void parseRequestInfo(String info) {
        String firstInfo = info.substring(0, info.indexOf(CRLF)).trim();
        method = firstInfo.substring(0, firstInfo.indexOf("/")).trim();
        String shortInfo = firstInfo.substring(firstInfo.indexOf("/"), firstInfo.indexOf("HTTP/")).trim();

        if (method.equalsIgnoreCase("get")) {
            if (shortInfo.contains("?")) {
                String[] str = shortInfo.split("[?]");
                this.url = str[0];
                this.param = str[1];
            } else {
                this.url = shortInfo;
            }
        } else if (method.equalsIgnoreCase("post")) {
            this.url = shortInfo;
            this.param = info.substring(info.lastIndexOf(CRLF)).trim();
        }

        if (param.equals("")) {
            return;
        }
        parseParam(param);
    }
    
    //解析 参数
    public void parseParam(String param) {
        String[] values = param.split("&");
        String[][] pair = new String[values.length][2];

        for (int i = 0; i < pair.length; i++) {
            String[] str = values[i].split("=");
            for (int j = 0; j < 2; j++) {
                if (1 == str.length) {
                    pair[i][j] = str[j];
                    pair[i][j + 1] = null;
                    break;
                } else {
                    pair[i][j] = str[j];
                }
            }
        }

        storeParam(pair);
    }
    
    //将解析完的参数，存入到 Map 中去
    public void storeParam(String[][] pair) {
        for (int i = 0; i < pair.length; i++) {
            if (paramMap.containsKey(pair[i][0])) {
                paramMap.get(pair[i][0]).add(decode(pair[i][1], "utf8"));
            }else{
                paramMap.put(pair[i][0], new ArrayList<>());
                paramMap.get(pair[i][0]).add(decode(pair[i][1], "utf8"));
            }
        }
    }
    
    //获取多个参数，例如复选框之类的
    public String[] getParams(String key){
        if(paramMap.containsKey(key)){
            return paramMap.get(key).toArray(new String[0]);
        }else {
            return null;
        }
    }
    
    //这个是我用来在页面上显示复选框的多个属性值的，暂且搁浅
    public String splitParams(String key){
        String[] str = getParams(key);
        StringBuilder sb = new StringBuilder();
        for(String temp : str){
            sb.append(temp);
        }
        return sb.toString();
    }
    //这里如果是没有get参数的话，是会返回null, getParams(key)[0] 也就是 null[0]，当然会报错。应该是getParams(key)，这个问题缠绕了我一些时间。。。
    public String getParam(String key){
//        return getParams(key)[0] == null ? "null" : getParams(key)[0];
        return getParams(key) == null ? "null" : getParams(key)[0];
    }

    //解决中文参数，如果没有这个方法的话，你的参数值为中文的话，在页面上显示的是乱码
    public String decode(String value, String code) {
        try {
            return URLDecoder.decode(value, code);
        } catch (UnsupportedEncodingException e) {
            e.printStackTrace();
        }
        return null;
    }
    
    //获取URL
    public String getUrl(){
        return this.url;
    }
}