package com.ngtesting.platform.action.client;

import com.alibaba.fastjson.JSONObject;
import com.ngtesting.platform.action.BaseAction;
import com.ngtesting.platform.config.Constant;
import com.ngtesting.platform.dao.ProjectDao;
import com.ngtesting.platform.model.TstCase;
import com.ngtesting.platform.model.TstProject;
import com.ngtesting.platform.model.TstUser;
import com.ngtesting.platform.service.intf.*;
import com.ngtesting.platform.servlet.PrivPrj;
import org.apache.shiro.SecurityUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletRequest;
import java.util.HashMap;
import java.util.List;
import java.util.Map;


@RestController
@RequestMapping(Constant.API_PATH_CLIENT + "case/")
public class CaseAction extends BaseAction {
	@Autowired
    ProjectService projectService;

	@Autowired
    CaseService caseService;
    @Autowired
    CaseExportService caseExportService;
    @Autowired
    CaseTypeService caseTypeService;
    @Autowired
    CasePriorityService casePriorityService;

    @Autowired
    ProjectDao projectDao;

	@RequestMapping(value = "query", method = RequestMethod.POST)
    @PrivPrj(perms = {"test_case:view"})
	public Map<String, Object> query(HttpServletRequest request, @RequestBody JSONObject json) {
		Map<String, Object> ret = new HashMap<String, Object>();

		TstUser user = (TstUser) SecurityUtils.getSubject().getPrincipal();

        Integer orgId = user.getDefaultOrgId();
		Integer prjId = user.getDefaultPrjId();

		List<TstCase> ls = caseService.query(prjId);
        ret.put("data", ls);
		ret.put("code", Constant.RespCode.SUCCESS.getCode());
		return ret;
	}

	@RequestMapping(value = "queryForSuiteSelection", method = RequestMethod.POST)
    @PrivPrj(perms = {"test_case:view"})
	public Map<String, Object> queryForSuiteSelection(HttpServletRequest request, @RequestBody JSONObject json) {
		Map<String, Object> ret = new HashMap<String, Object>();
        TstUser user = (TstUser) SecurityUtils.getSubject().getPrincipal();
		Integer prjId = user.getDefaultPrjId();

        Integer caseProjectId = json.getInteger("caseProjectId");
		Integer suiteId = json.getInteger("suiteId");

        Integer projectId = caseProjectId == null? prjId: caseProjectId;

        List<TstCase> vos = caseService.queryForSuiteSelection(projectId, suiteId);
		List<TstProject> projects = projectDao.listBrothers(prjId);

		ret.put("data", vos);
		ret.put("brotherProjects", projects);
		ret.put("code", Constant.RespCode.SUCCESS.getCode());
		return ret;
	}

	@RequestMapping(value = "queryForTaskSelection", method = RequestMethod.POST)
    @PrivPrj(perms = {"test_case:view"})
	public Map<String, Object> queryForTaskSelection(HttpServletRequest request, @RequestBody JSONObject json) {
		Map<String, Object> ret = new HashMap<String, Object>();
        TstUser user = (TstUser) SecurityUtils.getSubject().getPrincipal();
		Integer prjId = user.getDefaultPrjId();

        Integer caseProjectId = json.getInteger("caseProjectId");
		Integer taskId = json.getInteger("taskId");

        Integer projectId = caseProjectId == null? prjId: caseProjectId;

		List<TstCase> vos = caseService.queryForTaskSelection(projectId, taskId);
		List<TstProject> projects = projectDao.listBrothers(prjId);

		ret.put("data", vos);
		ret.put("brotherProjects", projects);
		ret.put("code", Constant.RespCode.SUCCESS.getCode());
		return ret;
	}

    @RequestMapping(value = "get", method = RequestMethod.POST)
    @PrivPrj(perms = {"test_case:view", "test_case:maintain"})
    public Map<String, Object> get(HttpServletRequest request, @RequestBody JSONObject json) {
        Map<String, Object> ret = new HashMap<String, Object>();
        TstUser user = (TstUser) SecurityUtils.getSubject().getPrincipal();

        Integer prjId = user.getDefaultPrjId();
        Integer caseId = json.getInteger("id");

        TstCase testCase = caseService.getDetail(caseId, prjId);

        ret.put("data", testCase);
        ret.put("code", Constant.RespCode.SUCCESS.getCode());
        return ret;
    }

    @RequestMapping(value = "rename", method = RequestMethod.POST)
    @PrivPrj(perms = {"test_case:maintain"})
    public Map<String, Object> rename(HttpServletRequest request, @RequestBody JSONObject json) {
        Map<String, Object> ret = new HashMap<String, Object>();

		TstUser user = (TstUser) SecurityUtils.getSubject().getPrincipal();

        TstCase testCase = caseService.rename(json, user);
        if (testCase == null) {
            return authorFail();
        }

        ret.put("data", testCase);
        ret.put("code", Constant.RespCode.SUCCESS.getCode());
        return ret;
    }

	@RequestMapping(value = "move", method = RequestMethod.POST)
    @PrivPrj(perms = {"test_case:maintain"})
	public Map<String, Object> move(HttpServletRequest request, @RequestBody JSONObject json) {
		Map<String, Object> ret = new HashMap<String, Object>();

		TstUser user = (TstUser) SecurityUtils.getSubject().getPrincipal();
		TstCase testCase = caseService.move(json, user);
        if (testCase == null) {
            return authorFail();
        }

		ret.put("data", testCase);
		ret.put("code", Constant.RespCode.SUCCESS.getCode());
		return ret;
	}

	@RequestMapping(value = "delete", method = RequestMethod.POST)
    @PrivPrj(perms = {"test_case:delete"})
	public Map<String, Object> delete(HttpServletRequest request, @RequestBody JSONObject json) {
		Map<String, Object> ret = new HashMap<String, Object>();

		Integer id = json.getInteger("id");

		TstUser user = (TstUser) SecurityUtils.getSubject().getPrincipal();

		Integer count = caseService.delete(id, user);
        if (count == 0) {
            return authorFail();
        }

		ret.put("code", Constant.RespCode.SUCCESS.getCode());
		return ret;
	}

    @RequestMapping(value = "update", method = RequestMethod.POST)
    @PrivPrj(perms = {"test_case:maintain"})
    public Map<String, Object> update(HttpServletRequest request, @RequestBody JSONObject json) {
        Map<String, Object> ret = new HashMap<String, Object>();
		TstUser user = (TstUser) SecurityUtils.getSubject().getPrincipal();

        TstCase testCase = caseService.update(json, user);
        if (testCase == null) {
            return authorFail();
        }

        ret.put("data", testCase);
        ret.put("code", Constant.RespCode.SUCCESS.getCode());
        return ret;
    }

    @PrivPrj(perms = {"test_case:maintain"})
	@RequestMapping(value = "saveField", method = RequestMethod.POST)
	public Map<String, Object> saveField(HttpServletRequest request, @RequestBody JSONObject json) {
		Map<String, Object> ret = new HashMap<String, Object>();
		TstUser user = (TstUser) SecurityUtils.getSubject().getPrincipal();

		TstCase testCase = caseService.saveField(json, user);
        if (testCase == null) {
            return authorFail();
        }

        ret.put("data", testCase);
		ret.put("code", Constant.RespCode.SUCCESS.getCode());
		return ret;
	}

	@RequestMapping(value = "changeContentType", method = RequestMethod.POST)
    @PrivPrj(perms = {"test_case:maintain"})
	public Map<String, Object> changeContentType(HttpServletRequest request, @RequestBody JSONObject json) {
		Map<String, Object> ret = new HashMap<String, Object>();
        TstUser user = (TstUser) SecurityUtils.getSubject().getPrincipal();

		Integer id = json.getInteger("id");
        String contentType = json.getString("contentType");

        TstCase testCase = caseService.changeContentType(id, contentType, user);
        if (testCase == null) {
            return authorFail();
        }

		ret.put("code", Constant.RespCode.SUCCESS.getCode());
		return ret;
	}

	@RequestMapping(value = "reviewResult", method = RequestMethod.POST)
    @PrivPrj(perms = {"test_case:review"})
	public Map<String, Object> reviewResult(HttpServletRequest request, @RequestBody JSONObject json) {
		Map<String, Object> ret = new HashMap<String, Object>();
        TstUser user = (TstUser) SecurityUtils.getSubject().getPrincipal();

		Integer id = json.getInteger("id");
		Boolean result = json.getBoolean("result");
        Integer nextId = json.getInteger("nextId");

		TstCase testCase = caseService.reviewResult(id, result, nextId, user);
        if (testCase == null) {
            return authorFail();
        }

        ret.put("data", testCase);
		ret.put("code", Constant.RespCode.SUCCESS.getCode());
		return ret;
	}

    @RequestMapping(value = "exportAll", method = RequestMethod.POST)
    @PrivPrj(perms = {"test_case:view"})
    public Map<String, Object> exportAll(HttpServletRequest request, @RequestBody JSONObject json) {
        Map<String, Object> ret = new HashMap<String, Object>();
        TstUser user = (TstUser) SecurityUtils.getSubject().getPrincipal();
        Integer prjId = user.getDefaultPrjId();

		String excelPath = caseExportService.export(prjId);
		ret.put("code", Constant.RespCode.SUCCESS.getCode());
		ret.put("excelPath", excelPath);

        return ret;
    }

}