package com.terra.proxy.servlet;

import com.alibaba.fastjson.JSONObject;
import com.alibaba.fastjson.serializer.SerializerFeature;
import com.terra.proxy.bean.ResActionRecord;
import com.terra.proxy.bean.VistorBean;
import com.terra.proxy.service.Impl.ServerRegisterServiceImpl;
import com.terra.proxy.util.JedisUtils;
import com.terra.proxy.util.Result;
import org.apache.http.*;
import org.apache.http.client.utils.URIUtils;
import org.apache.http.entity.BasicHttpEntity;
import org.mitre.dsmiley.httpproxy.ProxyServlet;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.context.WebApplicationContext;
import org.springframework.web.context.support.WebApplicationContextUtils;
import redis.clients.jedis.Jedis;

import javax.annotation.PostConstruct;
import javax.servlet.ServletException;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.net.URI;
import java.net.URLEncoder;
import java.sql.Timestamp;
import java.util.Date;
import java.util.regex.Pattern;
import java.util.zip.GZIPInputStream;
import java.util.zip.GZIPOutputStream;

public abstract class GisProxyServlet extends ProxyServlet {
    private Logger log = LoggerFactory.getLogger(RerequestServlet.class);
    @Autowired
    JedisUtils jedisUtils;
    protected Result result;
    ServerRegisterServiceImpl service;


    @PostConstruct
    public void init() {
        try {
            super.init();
        } catch (ServletException e) {

            e.printStackTrace();
        }
        WebApplicationContext wac = WebApplicationContextUtils.getRequiredWebApplicationContext(getServletContext());
        service = wac.getBean(ServerRegisterServiceImpl.class);
    }

    public static boolean isdigit(String str) {

        return str.matches("[0-9]+");
    }

    public abstract String rewriteUrlFromRequest(HttpServletRequest servletRequest);

    public static boolean isInteger(String str) {
        Pattern pattern = Pattern.compile("^[1-9]\\d*|0$");
        return pattern.matcher(str).matches();
    }

    protected void copyResponseEntity(HttpResponse proxyResponse, HttpServletResponse servletResponse)
            throws IOException {
        HttpEntity entity = proxyResponse.getEntity();
        BufferedReader reader;
        InputStream is = entity.getContent();
        if (result != null && !result.get("code").equals(0)) {
            servletResponse.setStatus((int) result.get("code"));
        }
        ServletOutputStream outputStream = servletResponse.getOutputStream();
        Boolean imageFlag = false;
        Boolean htmlFlag = false;
        HeaderElement[] elements = entity.getContentType().getElements();
        for (HeaderElement element : elements) {
            if (element.getName().contains("image")) {
                imageFlag = true;
            }
        }
        if (!imageFlag && isgzip(entity)) {

            reader = new BufferedReader(new InputStreamReader(new GZIPInputStream(is)));
            StringBuffer sb = new StringBuffer();
            String s = null;
            while ((s = reader.readLine()) != null) {
                sb.append(s);
            }
            String result = sb.toString();
            result = result.replaceAll("<table[^>]*>((?!table).)*</table>", "");
            String repx = "<a[^>]+>([^<]+</a>)";//a标签过滤
            result = result.replaceAll(repx, "");

            result = result.replaceAll("(http|https)://[^/]+/", "/");

            result = result.replaceAll("/arcgis/rest/services", "");
            BasicHttpEntity entity2 = new BasicHttpEntity();
            byte[] bt;
            bt = compress(result);
            ByteArrayInputStream btinput = new ByteArrayInputStream(bt);
            entity2.setContent(btinput);
            entity2.setContentEncoding("gzip");
            entity2.setContentType(entity.getContentType());
            entity2.setChunked(false);
            //entity2.setContentType("image/png");
            if (entity2 != null) {
                servletResponse.addHeader("Content-Length", String.valueOf(bt.length));
                OutputStream servletOutputStream = servletResponse.getOutputStream();
                entity2.writeTo(servletOutputStream);
            }

        } else {
            if (entity != null) {
                OutputStream servletOutputStream = servletResponse.getOutputStream();
                entity.writeTo(servletOutputStream);
            }
        }
    }

    /*
     * 统计不同ip调用服务次数,并将其存入入redis
     */
    protected void visitlogsaveinredis(VistorBean vb) {
        String fromip = vb.getRequestip();
        vb.setDate(new Date());
        Jedis jedis = null;
        int count = 0;
        try {
            jedis = JedisUtils.getJedis();
            jedis.sadd("visitlog", JSONObject.toJSONStringWithDateFormat(vb, "yyyy-MM-dd HH:mm:ss", SerializerFeature.PrettyFormat));
        } catch (Exception e) {

            log.error("redis包错");
        } finally {
            jedis.close();
        }

    }


    /*
     * 统计不同ip调用服务次数,并将其存入入redis
     */
    protected void saveResActionRecord(ResActionRecord vb) {

        vb.setActiontime(new Timestamp(new Date().getTime()));
        Jedis jedis = null;
        int count = 0;
        try {
            jedis = JedisUtils.getJedis();
            jedis.sadd("TerraResActionRecord", JSONObject.toJSONStringWithDateFormat(vb, "yyyy-MM-dd HH:mm:ss", SerializerFeature.PrettyFormat));
        } catch (Exception e) {

            log.error("redis包错");
        } finally {
            jedis.close();
        }

    }

    public static byte[] compress(String str) {
        ByteArrayOutputStream out = new ByteArrayOutputStream();

        GZIPOutputStream gzip = null;
        try {
            gzip = new GZIPOutputStream(out);
            gzip.write(str.getBytes());
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if (gzip != null) {
                try {
                    gzip.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
        return out.toByteArray();
    }

    public boolean isgzip(HttpEntity entity) {
        return entity.getContentEncoding() != null
                && "Content-Encoding: gzip".equals(entity.getContentEncoding().toString()) ? true : false;
    }

    public boolean ishtml(HttpEntity entity) {
        String temp = entity.getContentType().toString();
        return temp.contains("text/html") ? true : false;
    }

    protected void setXForwardedForHeader(HttpServletRequest servletRequest, HttpRequest proxyRequest) {
        String headerName = "X-Forwarded-For";
        if (this.doForwardIP) {
            String newHeader = servletRequest.getRemoteAddr();
            String existingHeader = servletRequest.getHeader(headerName);
            if (existingHeader != null) {
                newHeader = existingHeader + ", " + newHeader;
            }
            proxyRequest.setHeader(headerName, newHeader);
        }
    }

    protected void copyResponseHeaders(HttpResponse proxyResponse, HttpServletRequest servletRequest,
                                       HttpServletResponse servletResponse) {
        for (Header header : proxyResponse.getAllHeaders()) {
            if (!hopByHopHeaders.containsHeader(header.getName())) {
                if ((header.getName().equalsIgnoreCase("Set-Cookie"))
                        || (header.getName().equalsIgnoreCase("Set-Cookie2"))) {
                    copyProxyCookie(servletRequest, servletResponse, header);
                } else {
                    servletResponse.addHeader(header.getName(), header.getValue());
                    System.out.print(header.getName() + header.getValue());
                }
            }
        }
        System.out.print(servletResponse.containsHeader("Content-Length"));
    }

    protected String saveActionResult(HttpServletRequest servletRequest, VistorBean vb, URI errorurl, String sucessflag) {
        servletRequest.setAttribute(ATTR_TARGET_HOST, URIUtils.extractHost(errorurl));
        vb.setResult(sucessflag);
        visitlogsaveinredis(vb);
        if (result != null && !result.get("code").equals(0)) {
            URI str = URI.create("http://" + "localhost" + ":" + servletRequest.getLocalPort() + "/log/error?result=" + URLEncoder.encode(JSONObject.toJSONString(result)));
            return str.toString();
        } else {
            return "http://" + "localhost" + ":" + servletRequest.getLocalPort() + "/404.html";
        }
    }


    public static void main(String[] args) {

        String str = "<a herf='eee'>1111</a>";
        String repx = "<a[^>]+>([^<]+</a>)";
        String s = str.replaceAll(repx, "");
        System.out.println("s = " + s);
        String result = "http://18.1.1.231:8088/2222/cas/login.js";
        result = result.replaceAll("(http|https)://[^/]*\\.js$", "/");
        System.out.println("result = " + result);
        System.out.println("result = " + result);
    }
}