package com.ui.ctrl;

import com.ui.http.HttpRestClient;
import com.ui.model.domain.BuildMessage;
import com.ui.model.req.BuildRequest;
import com.ui.model.req.User;
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.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;


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

    @Autowired
    private HttpRestClient httpRestClient;

    private String URL = "http://172.31.16.167:8883/node/";

    @RequestMapping("/toProject")
    public ModelAndView toProject(Model model) {
        return new ModelAndView("project/node_project");
    }

    /**
     * 获取所有项目
     *
     * @return
     */
    @RequestMapping(value = "getProjects", method = RequestMethod.POST)
    @ResponseBody
    public String getProjects() {
        try {
            return httpRestClient.get(URL + "getProjects", String.class, null);
        } catch (Exception ex) {
            return "failed";
        }
    }

    /**
     * 获取所有项目
     *
     * @return
     */
    @RequestMapping(value = "getProject", method = RequestMethod.POST)
    @ResponseBody
    public String getProjects(String name) {
        try {
            return httpRestClient.get(URL + "getProject?name="+name , String.class, null);
        } catch (Exception ex) {
            return "failed";
        }
    }


    /**
     * 执行任务
     *
     * @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 hosts_name, Model model, HttpSession session) {
        try {
            User user = (User) session.getAttribute("user");
            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.setHost(hosts_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] + ",";
                    }
                    for (String project : newArray) {
                        buildRequest.setProject(project.substring(0, project.length() - 1));
                        BuildMessage buildMessage = httpRestClient.post(URL + "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(URL + "build", buildRequest, BuildMessage.class);
                        if (buildMessage != null) {
                            list.add(buildMessage);
                            messageids = messageids + buildMessage.getId() + ",";
                        }
                    }
                }


            } else {
                buildRequest.setProject(project_name);
                BuildMessage buildMessage = httpRestClient.post(URL + "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);
            model.addAttribute("hosts_name", hosts_name);

        } catch (Exception ex) {
        }
        return new ModelAndView("project/node_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(URL + "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(URL + "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(URL + "rollbackList", String.class, map);
        } catch (Exception ex) {
            return "failed";
        }
    }

    @RequestMapping(value = "refreshGit", method = RequestMethod.POST)
    @ResponseBody
    public String refreshGit() {
        try {
            return httpRestClient.get(URL + "refreshGit", String.class, null);
        } catch (Exception ex) {
            return "failed";
        }
    }


}