package com.ui.ctrl;

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.ui.contants.AlarmGroupContants;
import com.ui.contants.HttpUriContants;
import com.ui.http.HttpRestClient;
import com.ui.model.BaseResponse;
import com.ui.model.domain.BuildMessage;
import com.ui.model.req.*;
import com.ui.project.ProjectEnvironment;
import com.ui.project.ProjectOnline;
import org.apache.commons.lang.RandomStringUtils;
import org.apache.commons.lang.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpSession;
import java.text.SimpleDateFormat;
import java.util.*;

@RestController
@RequestMapping("project")
public class ProjectBuildCtrl {
    Logger log = LoggerFactory.getLogger(ProjectBuildCtrl.class);

    @Autowired
    private HttpRestClient httpRestClient;

    @RequestMapping("/toOm")
    public ModelAndView toOm() {
        return new ModelAndView("project/om");
    }

    @RequestMapping("/toNewOm")
    public ModelAndView toOmNew() {
        return new ModelAndView("project/newOm");
    }

    @RequestMapping("/toNode")
    public ModelAndView toNode() {
        return new ModelAndView("project/node");
    }

    @RequestMapping("/toProject")
    public ModelAndView toProject(Model model) {
        model.addAttribute("environments", ProjectEnvironment.getEnviroments());
        return new ModelAndView("project/project");
    }

    @RequestMapping("/toHistory")
    public ModelAndView toHistory(Model model) {
        model.addAttribute("environments", ProjectEnvironment.getEnviroments());

        model.addAttribute("projects", ProjectOnline.getJavaList());

        return new ModelAndView("project/history");
    }

    /**
     * 获取所有项目
     *
     * @return
     */
    @RequestMapping(value = "getProjects", method = RequestMethod.POST)
    @ResponseBody
    public String getProjects() {
        try {
            return JSONArray.toJSON(ProjectOnline.getJavaList()).toString();
        } catch (Exception ex) {
            return "failed";
        }
    }

    /**
     * 校验tag
     *
     * @return
     */
    @RequestMapping(value = "checkTag", method = RequestMethod.POST)
    @ResponseBody
    public String checkTag(String projects, String tag, String enviromment) {
        Map<String, String> map = new HashMap<>();
        map.put("projects", projects);
        map.put("tag", tag);
        return httpRestClient.get(ProjectEnvironment.getUrl(enviromment) + "checkTag", String.class, map);

    }

    /**
     * 发布前校验
     *
     * @return
     */
    @RequestMapping(value = "check")
    @ResponseBody
    public String check(String projects, String branch, String environment,String workid,HttpSession httpSession) {

        User user = (User) httpSession.getAttribute("user");
        Map<String,String> map = new HashMap<>();
        if (!StringUtils.equals("release0",workid)){//查询用户是否有工单处理权限
            map.put("id",workid);
            map.put("user",user.getName());
            BaseResponse baseResponse = httpRestClient.defaultGet(HttpUriContants.CHECK_JOB_BY_OPERATOR,BaseResponse.class,map);
            if (baseResponse.getCode() != 200){
                return baseResponse.getMessage();
            }
        }

        if (!("master".equals(branch) || "test".equals(branch) || "dev".equals(branch))) {
            map.clear();
            map.put("projects", projects);
            map.put("branch", branch);
            return httpRestClient.get(ProjectEnvironment.getUrl(environment) + "checkBranch", String.class, map);
        }
        return "1";


    }

    @RequestMapping(value = "tobuildpage")
    @ResponseBody
    public ModelAndView tobuildpage(String messageid, String project, String environment, String operate, String branch, String rollbackfile, Model model) {
        model.addAttribute("messageid", messageid);
        model.addAttribute("project", project);
        model.addAttribute("environment", environment);
        model.addAttribute("operate", operate);
        model.addAttribute("branch", branch);
        model.addAttribute("rollbackfile", rollbackfile);
        return new ModelAndView("project/single_project_build");
    }

    /**
     * 执行任务
     * 最终执行的方法是 project-boot 工程里的JavaBuildCtrl(web) -> build.do
     * 执行的ip是http://172.31.16.167:8883/web/
     * @return
     */
    @RequestMapping(value = "build", method = RequestMethod.POST)
    public ModelAndView build(String project_name, String environment_name, String operate_name, String branch_name, String rollbackfile_name,String workid_name, Model model, HttpSession session) {
        try {
            User user = (User) session.getAttribute("user");
            Map<String,String> map = new HashMap<>();
            if (StringUtils.equals("Deploy",operate_name) && !StringUtils.equals("release0",workid_name)){//查询用户是否有工单处理权限
                map.put("id",workid_name);
                map.put("user",user.getName());
                BaseResponse baseResponse = httpRestClient.defaultGet(HttpUriContants.UPDATE_JOB_BY_OPERATOR,BaseResponse.class,map);
                if (baseResponse.getCode() != 200){
                    model.addAttribute("environment_name", baseResponse.getMessage());
                    return new ModelAndView("project/project_build");
                }
            }

            List<BuildMessage> list = new ArrayList<>();
            BuildRequest buildRequest = new BuildRequest();
            //分支名称
            buildRequest.setBranch(branch_name);
            //登陆用户
            buildRequest.setUser(user.getName());
            //发布环境
            buildRequest.setEnvironment(environment_name);
            //回滚文件名称
            buildRequest.setRollbackfile(rollbackfile_name);
            //操作类型
            buildRequest.setOperate(operate_name);
            //关联工单
            buildRequest.setWorkid(workid_name);
            String messageids = "";
            if ("Deploy".equals(operate_name)) {
                String[] array = project_name.split(",");
                if (array.length > 4) {
                    String[] newArray = new String[]{"", "", "", ""};
                    for (int i = 0; i < array.length; i++) {
                        newArray[i % 4] = newArray[i % 4] + array[i] + ",";
                    }
                    //如上下所述，只支持长度为4。。。。。不明白
                    for (String project : newArray) {
                        buildRequest.setProject(project.substring(0, project.length() - 1));
                        BuildMessage buildMessage = httpRestClient.post(ProjectEnvironment.getUrl(environment_name) + "build", buildRequest, BuildMessage.class);
                        if (buildMessage != null) {
                            list.add(buildMessage);
                            messageids = messageids + buildMessage.getId() + ",";
                        }
                    }

                } else {
                    for (String project : array) {
                        buildRequest.setProject(project);
                        BuildMessage buildMessage = httpRestClient.post(ProjectEnvironment.getUrl(environment_name) + "build", buildRequest, BuildMessage.class);
                        if (buildMessage != null) {
                            list.add(buildMessage);
                            messageids = messageids + buildMessage.getId() + ",";
                        }
                    }
                }


            } else {
                buildRequest.setProject(project_name);
                BuildMessage buildMessage = httpRestClient.post(ProjectEnvironment.getUrl(environment_name) + "build", buildRequest, BuildMessage.class);
                if (buildMessage != null) {
                    list.add(buildMessage);
                    messageids = messageids + buildMessage.getId() + ",";
                }
            }
            model.addAttribute("environment_name", environment_name);
            model.addAttribute("operate_name", operate_name);
            model.addAttribute("branch_name", branch_name);
            model.addAttribute("rollbackfile_name", rollbackfile_name);
            model.addAttribute("messageids", messageids);
            model.addAttribute("messageList", list);
        } catch (Exception ex) {
        }
        return new ModelAndView("project/project_build");
    }

    /**
     * 获取执行结果
     *
     * @param messageid
     * @return
     */
    @RequestMapping(value = "getbuildmsg", method = RequestMethod.POST)
    @ResponseBody
    public String getbuildmsg(String messageid, String project) {
        try {
            Map<String, String> map = new HashMap<>();
            map.put("messageid", messageid);
            map.put("project", project);

            return httpRestClient.get(ProjectEnvironment.getUrl(messageid.split("_")[0]) + "getbuildmsg", String.class, map);

        } catch (Exception ex) {
            return "failed";
        }
    }

    @RequestMapping(value = "cancelBuild")
    @ResponseBody
    public String cancelBuild(String id) {
        try {
            Map<String, String> map = new HashMap<>();
            map.put("id", id);
            return httpRestClient.get(ProjectEnvironment.getUrl(id.split("_")[0]) + "cancelBuild", String.class, map);
        } catch (Exception ex) {
            return "failed";
        }
    }

    @RequestMapping(value = "rollbackList", method = RequestMethod.POST)
    @ResponseBody
    public String rollbackList(String project, String environment) {
        try {
            Map<String, String> map = new HashMap<>();
            map.put("project", project);
            map.put("environment", environment);
            return httpRestClient.get(ProjectEnvironment.getUrl(environment) + "rollbackList", String.class, map);
        } catch (Exception ex) {
            return "failed";
        }
    }

    @RequestMapping("/getHistory")
    @ResponseBody
    public BaseResponse getHistory(ProjectBuildReq req) throws Exception {
        if ("all".equals(req.getEnvironment())) {
            req.setEnvironment(null);
        }
        if ("all".equals(req.getCurrentProject())) {
            req.setCurrentProject(null);
        }
        BaseResponse response = httpRestClient.defaultPost("/projectBuild/getHistory", req, BaseResponse.class);
        return response;
    }

    @RequestMapping("/toRollbackByTime")
    @ResponseBody
    public ModelAndView toRollbackByTime() {
        return new ModelAndView("project/project_rollbackByTime");
    }

    @RequestMapping("/rollbackByTime")
    @ResponseBody
    public String rollbackByTime(String time, String environment) {
        Map<String, String> map = new HashMap<>();
        Calendar calendar = Calendar.getInstance();
        calendar.set(Calendar.HOUR_OF_DAY,1);
        calendar.set(Calendar.MINUTE,0);
        calendar.set(Calendar.SECOND,0);

        if (StringUtils.equals("yestoday",time)){
            time = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(calendar.getTime());
        }else if (StringUtils.equals("beforeYestoday",time)){
            calendar.add(Calendar.DAY_OF_YEAR,-1);
            time = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(calendar.getTime());
        }
        map.put("time", time);
        map.put("environment", environment);
        return httpRestClient.get(ProjectEnvironment.getUrl(environment) + "rollbackByTime", String.class, map);
    }

    @RequestMapping("/doRollbackByTime")
    @ResponseBody
    public ModelAndView doRollbackByTime(String data, String environment, HttpSession session, Model model) {
        try {
            JSONArray jsonArray = JSONArray.parseArray(data);
            User user = (User) session.getAttribute("user");
            List<BuildMessage> list = new ArrayList<>();
            BuildRequest buildRequest = new BuildRequest();
            buildRequest.setUser(user.getName());
            buildRequest.setEnvironment(environment);
            buildRequest.setOperate("Rollback");

            String messageids = "";
            for (int i = 0; i < jsonArray.size(); i++) {
                JSONObject jsonObject = jsonArray.getJSONObject(i);
                String project = jsonObject.getString("project");
                buildRequest.setProject(project);
                buildRequest.setRollbackfile(jsonObject.getString("rollbackFile"));
                BuildMessage buildMessage = httpRestClient.post(ProjectEnvironment.getUrl(environment) + "build", buildRequest, BuildMessage.class);
                if (buildMessage != null) {
                    list.add(buildMessage);
                    messageids = messageids + buildMessage.getId() + ",";
                }
            }
            model.addAttribute("environment_name", environment);
            model.addAttribute("operate_name", "Rollback");
            model.addAttribute("messageids", messageids);
            model.addAttribute("messageList", list);
        } catch (Exception ex) {
        }
        return new ModelAndView("project/project_build");


    }


    @RequestMapping("/newOm/query")
    public BaseResponse queryProjects(ProjectInfoReq req) {

        BaseResponse response = httpRestClient.defaultPost("/projectManager/query", req, BaseResponse.class);

        return response;
    }


    @RequestMapping("/newOm/save")
    public BaseResponse addProject(ProjectInfoReq req) {

        BaseResponse response = httpRestClient.defaultPost("/projectManager/add", req, BaseResponse.class);
        return response;
    }

    @RequestMapping("/newOm/del")
    public BaseResponse delProject(ProjectInfoReq req) {

        BaseResponse response = httpRestClient.defaultPost("/projectManager/del", req, BaseResponse.class);
        return response;
    }

    @RequestMapping("/newOm/build")
    public BaseResponse buildOmProject(ProjectBuildInfoReq req) {
        BaseResponse response = httpRestClient.defaultPost("/projectManager/build", req, BaseResponse.class);
        return response;
    }

    @RequestMapping("/aototest/autoTestService")
    @ResponseBody
    public BaseResponse autoTestService(String models, String branch, HttpSession session) {
        User user = (User) session.getAttribute("user");
        AutoTestReq autoTestReq = new AutoTestReq();
        String id = new SimpleDateFormat("yyyyMMddHHmmss").format(new Date()) + RandomStringUtils.randomNumeric(6);
        autoTestReq.setTaskname(id);
        autoTestReq.setEnv("5");
        autoTestReq.setModels(models.substring(0, models.length() - 1));
        autoTestReq.setBranch(branch);
        autoTestReq.setProId(0);
        autoTestReq.setVersionId(0);
        autoTestReq.setMailto(user.getEmail());
        BaseResponse baseResponse = httpRestClient.post("http://qmc.yohops.com:9999/autotest/autoTestService", autoTestReq, BaseResponse.class);
        System.out.println(id);
        return baseResponse;
    }

    @RequestMapping("/aototest/getAutoTestResult")
    @ResponseBody
    public BaseResponse getAutoTestResult(String id) {
        AutoTestReq autoTestReq = new AutoTestReq();
        autoTestReq.setTaskname(id);
        BaseResponse baseResponse = httpRestClient.post("http://qmc.yohops.com:9999/autotest/getAutoTestResult", autoTestReq, BaseResponse.class);
        return baseResponse;
    }

    @RequestMapping("/aototest/toAutoTestResult")
    public ModelAndView toAutoTestResult(String id, Model model) {
        model.addAttribute("id", id);
        return new ModelAndView("project/autotestresult");
    }

    @RequestMapping("/testexecute")
    @ResponseBody
    public BaseResponse testexecute(String env) {
        AutoTestReq autoTestReq = new AutoTestReq();
        autoTestReq.setEnv(env);
        BaseResponse emailResponse = httpRestClient.defaultPost(HttpUriContants.GET_EMAIL_BY_ALARM_GROUP+"?groupName=" + AlarmGroupContants.GROUP_NAME_YUNWEI, null, BaseResponse.class);
        if(emailResponse.getData() != null){
            autoTestReq.setMailto((String)emailResponse.getData());
        }
        BaseResponse baseResponse = httpRestClient.post("http://qmc.yohops.com:9999/autoTask/executeApiAutoTask", autoTestReq, BaseResponse.class);
        return baseResponse;
    }

    @RequestMapping("/testget")
    public BaseResponse testget(String id) {
        AutoTestReq autoTestReq = new AutoTestReq();
        autoTestReq.setId(id);
        BaseResponse baseResponse = httpRestClient.post("http://qmc.yohops.com:9999/autoTask/getAutoTestExecuteResult", autoTestReq, BaseResponse.class);
        return baseResponse;
    }
}