package com.terra.proxy.servlet;

import com.terra.proxy.bean.VistorBean;
import com.terra.proxy.controller.ZiyuanRightManager;
import org.apache.commons.lang3.StringUtils;
import org.apache.http.client.utils.URIUtils;

import javax.servlet.http.HttpServletRequest;
import java.net.URI;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;


public class RerequestServlet extends GisProxyServlet {

    @Override
    public String rewriteUrlFromRequest(HttpServletRequest servletRequest) {
        URI errorurl = URI.create("http://" + "localhost" + ":" + servletRequest.getServerPort() + "/404.html");

        StringBuilder uri = new StringBuilder(1000);
        StringBuilder queryString2 = new StringBuilder(1000);
        System.out.print(servletRequest.getHeader("Content-Length") + servletRequest.getHeader("Transfer-Encoding"));

        if (servletRequest.getPathInfo() != null) {// ex: /my/path.html
            System.out.println(servletRequest.getPathInfo());
            uri.append(encodeUriQuery(servletRequest.getPathInfo()));
        }

        // Handle the query string & fragment
        String queryString = servletRequest.getQueryString();// ex:(following
        //阻止请求为JSON的代理服务
        Boolean jsonFlag = false;

        if (jsonFlag) {
            servletRequest.setAttribute(ATTR_TARGET_HOST, URIUtils.extractHost(errorurl));
            return "http://" + "localhost" + ":" + servletRequest.getServerPort() + "/404.html";
        }                                                        // '?'):

        String fragment = null;
        // 根据zyid和suffix后缀查询对应的真实服务地址,判断是否公开, 公开不需要token拼接参数返回

        List<Map> serverlist = new ArrayList<>();
        String zyid = "";
        String path = servletRequest.getPathInfo();
        String[] splitpath = path.split("/");

        zyid = splitpath[1];
        String suffix = null;
        if (splitpath.length > 3 && splitpath[3] != null) {
            suffix = isdigit(splitpath[2]) ? splitpath[2] : null;
        }

        Map<String, Object> param = new HashMap();
        if (!isInteger(zyid)) {
            servletRequest.setAttribute(ATTR_TARGET_HOST, URIUtils.extractHost(errorurl));
            return "http://" + "localhost" + ":" + servletRequest.getServerPort() + "/404.html";
        }
        param.put("resourceid", Integer.valueOf(zyid));
        param.put("suffix", suffix != null ? Integer.valueOf(suffix) : null);
        param.put("enable", 1);
        param.put("fromsys", "HYJGPT");

        serverlist = service.queryServer(param);
        param.put("SERVERURL", serverlist.get(0).get("SERVERURL").toString());
        serverlist.add(param);
        System.out.print(serverlist.toString());

        String url = serverlist.get(0).get("SERVERURL").toString();
        String substringAfter = StringUtils.substringAfter(url, "?");
        //代理地址是否包含layers 字段
        int layers = url.indexOf("layers");
        int server = path.indexOf("Server");
        String originalParam = "";

        int paraIndex = url.indexOf("?");
        if (server != -1) {
            if (paraIndex != -1) {
                originalParam = StringUtils.split(url, "?")[1];
                url = StringUtils.split(url, "?")[0];
            }

            url = url + StringUtils.substringAfter(path, "Server");

        }
        Integer ispublic = Integer.valueOf(serverlist.get(0).get("ISPUBLIC").toString());
        URI hostpath = URI.create(url);
        servletRequest.setAttribute(ATTR_TARGET_HOST, URIUtils.extractHost(hostpath));

        VistorBean vb = new VistorBean();
        vb.setRequestip(servletRequest.getRemoteAddr());
        vb.setResourceid(Integer.valueOf(zyid));
        vb.setVisitfromsys("HYJGPT");
        vb.setServerurl(serverlist.get(0).get("SERVERURL").toString());
        if (queryString != null) {
            // 针对请求地址的参数做操作
            // 判断是否带了token
            if (queryString.indexOf("token") == -1 && ispublic != 1) {
                saveActionResult(servletRequest, vb, errorurl, "fail");
            }
            String par[] = queryString.split("&");
            for (int i = 0; i < par.length; i++) {
                String a = par[i].split("=")[0];
                if ("token".equals(a)) {
                    String v = par[i].split("=")[1];
                    //这里使用lantool的token,把子资源id也申请token
                    result = ZiyuanRightManager.checkRight(v, servletRequest.getHeader("referer"), servletRequest.getRemoteAddr(), zyid, ispublic.toString(), servletRequest.getPathInfo());
                    // 判断内存里有咩有匹配的token,咩有去redis库里查询，库里没有，则在内存和库中存储，库里有，存入内存
                    if (!result.get("code").equals(0)) {
                        return saveActionResult(servletRequest, vb, errorurl, "fail");
                    }
                } else {
                    boolean layers1 = par[i].contains("layers");
                    if (layers1 && layers != -1) {
                        //除去多余字段
                        continue;
                    }
                    int index = queryString2.indexOf("=");
                    if (index > -1) {
                        queryString2.append("&").append(par[i]);
                    } else {
                        queryString2.append(par[i]);
                    }
                }

            }

            int fragIdx = queryString.indexOf('#');
            if (fragIdx >= 0) {
                fragment = queryString.substring(fragIdx + 1);
                queryString = queryString.substring(0, fragIdx);
            }
        } else {
            if (ispublic != 1) {
                return saveActionResult(servletRequest, vb, errorurl, "fail");
            }
        }

        if (!serverlist.isEmpty()) {
            System.out.print(serverlist.get(0).get("SERVERURL").toString());
            StringBuilder returnuri = new StringBuilder(url);

            if (queryString2 != null && queryString2.length() > 0) {
                if (!(returnuri.indexOf("?") > -1)) {
                    returnuri.append('?');
                }
                returnuri.append(encodeUriQuery(queryString2));
            }

            if (doSendUrlFragment && fragment != null) {
                returnuri.append('#');
                returnuri.append(encodeUriQuery(fragment));
            }
            vb.setResult("success");
            visitlogsaveinredis(vb);
            String resultStr = returnuri.toString();
            if (resultStr.contains("?")) {
                resultStr += "&" + originalParam;
            } else {
                resultStr += "?" + originalParam;
            }

            return resultStr;
        }
        return null;
    }


    public static void main(String[] args) {
        String string = "where=1%3D1&text=&objectIds=&time=&geometry=&geometryType=esriGeometryEnvelope&inSR=&spatialRel=esriSpatialRelIntersects&relationParam=&outFields=&returnGeometry=true&maxAllowableOffset=&geometryPrecision=&outSR=&returnIdsOnly=false&returnCountOnly=false&orderByFields=&groupByFieldsForStatistics=&outStatistics=&returnZ=false&returnM=false&gdbVersion=&returnDistinctValues=false&returnTrueCurves=false&resultOffset=&resultRecordCount=&f=JSON";
        StringBuilder stringBuilder = new StringBuilder();
        String[] split = string.split("&");
        for (String str : split) {
            String[] strings = str.split("=");
            if (!(strings.length == 1)) {
                stringBuilder.append(strings[0]);
                stringBuilder.append("=");
                stringBuilder.append(strings[1]);
                stringBuilder.append("&");
            }
        }

        System.out.println(stringBuilder.toString());

    }


}