package com.bosshand.virgo.controller;

import com.bosshand.virgo.core.model.MgrAgent;
import com.bosshand.virgo.core.response.Response;
import com.bosshand.virgo.core.utils.ContextUtils;
import com.bosshand.virgo.service.AgentService;
import io.swagger.annotations.ApiOperation;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

@RestController
@RequestMapping("agent")
public class AgentController {

    @Autowired
    AgentService agentService;

    @ApiOperation("获取")
    @RequestMapping(value = "/{currPage}/{pageSize}", method = RequestMethod.POST)
    public Response list(@RequestBody MgrAgent mgrAgent, @PathVariable int currPage, @PathVariable int pageSize) {
        long id = ContextUtils.getCurrentUser().getId();
        mgrAgent.setUserId(id);
        int totalCount = agentService.getTotalCount(mgrAgent);
        List<MgrAgent> dataList = agentService.getLimit(mgrAgent, currPage, pageSize);
        Map<String, Object> result = new HashMap<>();
        result.put("dataList", dataList);
        result.put("totalCount", totalCount);
        return Response.ok(result);
    }

    @ApiOperation("保存")
    @RequestMapping(value = "", method = RequestMethod.POST)
    public Response insert(@RequestBody MgrAgent mgrAgent) {
        long id = ContextUtils.getCurrentUser().getId();
        mgrAgent.setUserId(id);
        agentService.insert(mgrAgent);
        return Response.ok();
    }

    @ApiOperation("更新")
    @RequestMapping(value = "/update", method = RequestMethod.PUT)
    public Response update(@RequestBody MgrAgent mgrAgent) {
        agentService.update(mgrAgent);
        return Response.ok();
    }

    @ApiOperation("删除")
    @RequestMapping(value = "/delete/{id}", method = RequestMethod.DELETE)
    public Response delete(@PathVariable long id) {
        agentService.delete(id);
        return Response.ok();
    }

    @ApiOperation("详情")
    @RequestMapping(value = "/{id}", method = RequestMethod.GET)
    public Response get(@PathVariable long id) {
        return Response.ok(agentService.get(id));
    }

}