package com.ruoyi.inspectiontask.controller;

import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.ruoyi.common.utils.poi.ExcelUtil;
import com.ruoyi.framework.aspectj.lang.annotation.Log;
import com.ruoyi.framework.aspectj.lang.enums.BusinessType;
import com.ruoyi.framework.web.controller.BaseController;
import com.ruoyi.framework.web.domain.R;
import com.ruoyi.inspectiontask.dto.InspectionTaskDto;
import com.ruoyi.inspectiontask.dto.TimingTaskDto;
import com.ruoyi.inspectiontask.pojo.TimingTask;
import com.ruoyi.inspectiontask.service.InspectionTaskService;
import com.ruoyi.inspectiontask.service.TimingTaskService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.quartz.SchedulerException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.HttpServletResponse;
import java.util.List;

@RestController
@Api(tags = "Inspection Timing Task")
@RequestMapping("/timingTask")
public class TimingTaskController extends BaseController {

    @Autowired
    private TimingTaskService timingTaskService;

    @Autowired
    private InspectionTaskService inspectionTaskService;

    @GetMapping("/list")
    @ApiOperation(value = "Timing task list")
    public R<IPage<TimingTaskDto>> list(Page<TimingTask> page, TimingTask timingTask) {
        IPage<TimingTaskDto> list = timingTaskService.selectTimingTaskList(page, timingTask);
        return R.ok(list);
    }

    @GetMapping("/recordList/{timingId}")
    @ApiOperation(value = "Inspection records by timing task id")
    public R<List<InspectionTaskDto>> recordList(@PathVariable Long timingId) {
        return R.ok(inspectionTaskService.selectInspectionTaskRecordListByTimingId(timingId));
    }

    @PostMapping("/export")
    @ApiOperation(value = "Export timing tasks")
    public void export(HttpServletResponse response) {
        Page page = new Page<>(-1, -1);
        TimingTaskDto timingTask = new TimingTaskDto();
        IPage<TimingTaskDto> list = timingTaskService.selectTimingTaskList(page, timingTask);
        ExcelUtil<TimingTaskDto> util = new ExcelUtil<>(TimingTaskDto.class);
        util.exportExcel(response, list.getRecords(), "timing_task_export");
    }

    @PostMapping("/addOrEditTimingTask")
    @ApiOperation(value = "Create or update timing task")
    @Log(title = "Timing Task", businessType = BusinessType.INSERT)
    public R addOrEditTimingTask(@RequestBody TimingTaskDto timingTaskDto) throws SchedulerException {
        return R.ok(timingTaskService.addOrEditTimingTask(timingTaskDto));
    }

    @PostMapping("/changeEnable")
    @ApiOperation(value = "Change enable status")
    @Log(title = "Timing Task", businessType = BusinessType.UPDATE)
    public R changeEnable(@RequestBody TimingTask timingTask) throws SchedulerException {
        if (timingTask.getId() == null || timingTask.getIsEnabled() == null) {
            return R.fail("id and isEnabled are required");
        }
        return R.ok(timingTaskService.changeEnable(timingTask.getId(), timingTask.getIsEnabled()));
    }

    @DeleteMapping("/delTimingTask")
    @ApiOperation(value = "Delete timing tasks")
    @Log(title = "Timing Task", businessType = BusinessType.DELETE)
    public R remove(@RequestBody Long[] ids) {
        return R.ok(timingTaskService.delByIds(ids));
    }
}