package com.terra.proxy.controller;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.alibaba.fastjson.serializer.SerializerFeature;
import com.terra.proxy.bean.ServletBean;
import com.terra.proxy.mapper.ProxyServletMapper;
import com.terra.proxy.properties.TerraProperties;
import com.terra.proxy.util.CrpUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.CrossOrigin;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.ResponseBody;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.function.Consumer;


/**
 * 该类用于配置代理servlet
 *
 * @author zsx
 */
@Controller
@RequestMapping("/serv")
@CrossOrigin
public class ProxyServletConfigController {

    @Autowired
    private TerraProperties properties;

    @Resource
    ProxyServletMapper mapper;

    @RequestMapping(value = "/addServlet", method = {RequestMethod.GET, RequestMethod.POST})
    @ResponseBody
    public String addArcgisServer(HttpServletRequest request, String url, Integer enable, String remark, Integer ispublic) {
        ServletBean bean = new ServletBean();
        bean.setEnable(enable);
        bean.setTime(new Date());
        bean.setRemark(remark);
        bean.setUrl(url);
        bean.setIspublic(ispublic);
        mapper.addServlet(bean);
        String proxyurl = new StringBuilder(properties.getProxy().getHost()).append("/servlet/").append(CrpUtil.encode(bean.getId()))
                .toString();
        JSONObject json = new JSONObject();
        json.put("code", 200);
        json.put("data", proxyurl);
        json.put("message", "获取成功");
        return json.toJSONString(0);
    }


    @RequestMapping(value = "/batchImportFromPath", method = {RequestMethod.GET, RequestMethod.POST})
    @ResponseBody
    public String test(HttpServletRequest request, String path, Integer enable) {

        String json = readFile(path);
        List<Map> t = JSON.parseArray(json, Map.class);
        t.forEach(new Consumer<Map>() {

            @Override
            public void accept(Map b) {
                ServletBean bean = new ServletBean();
                bean.setEnable(enable);
                bean.setTime(new Date());
                bean.setRemark(b.get("serverurl").toString());
                bean.setUrl(b.get("targetUri").toString());
                mapper.addServlet(bean);

            }
        });

        return "";
    }

    public static String readFile(String json) {
        //读取servletjson文件
        File file = new File(json);
        byte[] buffer = new byte[1024];
        StringBuilder sb = new StringBuilder();
        String s = null;
        BufferedReader br;
        try {
            br = new BufferedReader(new FileReader(file));
            while ((s = br.readLine()) != null) {
                sb.append(s).append("\n");
            }
            br.close();
        } catch (IOException e1) {
            e1.printStackTrace();
        }
        return sb.toString();
    }


    @RequestMapping(value = "/updateServlet", method = {RequestMethod.GET, RequestMethod.POST})
    @ResponseBody
    public String updateServer(HttpServletRequest request, String url, Integer enable, String remark, Integer ispublic, Integer id) {
        ServletBean bean = new ServletBean();
        bean.setEnable(enable);
        bean.setRemark(remark);
        bean.setUrl(url);
        bean.setIspublic(ispublic);
        bean.setId(id);
        mapper.updateServer(bean);
        return "ok";
    }

    @RequestMapping(value = "/delServlet", method = {RequestMethod.GET, RequestMethod.POST})
    @ResponseBody
    public String delServlet(HttpServletRequest request, Integer id) {

        mapper.delServer(id);
        return "ok";
    }

    @RequestMapping(value = "/queryServlet", method = {RequestMethod.GET, RequestMethod.POST})
    @ResponseBody
    public String queryServlet(HttpServletRequest request, Integer id, String url, Integer enable,
                        String remark, Integer page, Integer limit, Date starttime, Date endtime) {
        Map<String, Object> params = new HashMap<>();
        Map<String, Object> resultmap = new HashMap<>();
        params.put("url", url);
        params.put("enable", enable);
        params.put("remark", remark);
        params.put("id", id);
        params.put("starttime", starttime);
        params.put("endtime", endtime);
        params.put("offSet", (page - 1) * limit);
        params.put("limit", limit);
        List<Map> list = mapper.queryServer(params);
        int total = mapper.queryTotalServer(params);
        resultmap.put("data", list);
        resultmap.put("total", total);
        String str = JSONObject.toJSONStringWithDateFormat(resultmap, "yyyy-MM-dd HH:mm:ss", SerializerFeature.PrettyFormat);
        return str;
    }
}