package com.terra.proxy.config;

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.terra.proxy.properties.TerraProperties;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.web.servlet.ServletContextInitializer;
import org.springframework.stereotype.Component;

import javax.servlet.Servlet;
import javax.servlet.ServletContext;
import javax.servlet.ServletException;
import javax.servlet.ServletRegistration;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;


@Component
public class ServletConfig implements ServletContextInitializer {

    private Logger log = LoggerFactory.getLogger(ServletConfig.class);

    @Autowired
    private TerraProperties te;

    @Value("${servletJsonPath}")
    private String servletJsonPath;

    @Override
    public void onStartup(ServletContext servletContext) throws ServletException {
        System.out.println(te.getProxy());
        // 本地
        // JSONArray jb = JSONObject.parseArray(readjson(te.getProxy().getServletmapjson()));
        // 服务器
        JSONArray jb = JSONObject.parseArray(readjson(servletJsonPath));

        ServletRegistration.Dynamic initservlet;
        for (int i = 0; i < jb.size(); i++) {
            JSONObject t = jb.getJSONObject(i);
            Servlet sv;
            try {
                sv = (Servlet) Class.forName(t.getString("servletclass")).newInstance();
                initservlet = servletContext.addServlet("servlet" + i, sv);

                initservlet.addMapping(t.getString("serverurl"));
                Map<String, String> map = new HashMap<String, String>();
                map.put("targetUri", t.getString("targetUri"));
                initservlet.setInitParameters(map);

                System.out.println(servletContext.getServletRegistrations());
            } catch (InstantiationException | IllegalAccessException | ClassNotFoundException e) {
                log.error("servlet注册异常");
                e.printStackTrace();
            }

        }
    }


    public static String readjson(String json) {
        BufferedReader br;
        try {
            // 本地
            // File file = new ClassPathResource(json).getFile();
            // 服务器
            File file =new File(ServletConfig.class.getResource("/").getPath()+json);
            byte[] buffer = new byte[1024];
            StringBuilder sb = new StringBuilder();
            String s = null;
            br = new BufferedReader(new FileReader(file));
            while ((s = br.readLine()) != null) {
                sb.append(s);
            }
            br.close();
            return sb.toString();
        } catch (IOException e1) {
            e1.printStackTrace();
        }
        return null;
    }


}