package com.moon.server.controller.data;

import com.moon.server.annotation.SysLog;
import com.moon.server.controller.all.BaseController;
import com.moon.server.entity.all.ResponseMsg;
import com.moon.server.entity.data.DirEntity;
import com.moon.server.entity.sys.UserEntity;
import com.moon.server.helper.WebHelper;
import com.moon.server.service.data.DirService;
import com.moon.server.service.sys.TokenService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

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

/**
 * 目录管理
 * @author sws
 * @date   2022-09-22
 */
@Api(tags = "数据管理\\目录管理")
@RestController
@RequestMapping("/dir")
public class DirController extends BaseController {
    @Autowired
    DirService dirService;

    @Autowired
    TokenService tokenService;

    @SysLog()
    @ApiOperation(value = "插入一条")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "entity", value = "实体类", dataType = "com.lf.server.entity.data.DirEntity", paramType = "body", example = "")
    })
    @PostMapping(value = "/insertDir", produces = "application/json; charset=UTF-8")
    public ResponseMsg<Integer> insertDir(@RequestBody DirEntity entity, HttpServletRequest req) {
        try {
            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue != null) {
                entity.setCreateUser(ue.getId());
            }

            dirService.insert(entity);

            return success(entity.getId());
        } catch (Exception ex) {
            return fail(ex, -1);
        }
    }

    @SysLog()
    @ApiOperation(value = "插入多条")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "list", value = "实体类集合", dataType = "com.lf.server.entity.data.DirEntity", paramType = "body", example = "")
    })
    @PostMapping(value = "/insertDirs", produces = "application/json; charset=UTF-8")
    public ResponseMsg<Integer> insertDirs(@RequestBody List<DirEntity> list, HttpServletRequest req) {
        try {
            UserEntity ue = tokenService.getCurrentUser(req);
            for (DirEntity entity : list) {
                entity.setCreateUser(ue.getId());
            }

            int count = dirService.inserts(list);

            return success(count);
        } catch (Exception ex) {
            return fail(ex, -1);
        }
    }

    @SysLog()
    @ApiOperation(value = "删除一条")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "id", value = "ID", dataType = "Integer", paramType = "query", example = "1")
    })
    @GetMapping(value = "/deleteDir")
    public ResponseMsg<Integer> deleteDir(int id) {
        try {
            int count = dirService.deleteDir(id);

            return success(count);
        } catch (Exception ex) {
            return fail(ex, -1);
        }
    }

    @SysLog()
    @ApiOperation(value = "删除多条")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "ids", value = "ID数组", dataType = "com.lf.server.entity.data.DirEntity", paramType = "body", example = "")
    })
    @GetMapping(value = "/deleteDirs")
    public ResponseMsg<Integer> deleteDirs(@RequestParam List<Integer> ids) {
        try {
            if (ids == null || ids.isEmpty()) {
                return fail("id数组不能为空", -1);
            }

            int count = dirService.deleteDirs(ids);

            return success(count);
        } catch (Exception ex) {
            return fail(ex, -1);
        }
    }

    @SysLog()
    @ApiOperation(value = "更新一条")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "entity", value = "实体类", dataType = "DictEntity", paramType = "body", example = "")
    })
    @ResponseBody
    @PostMapping(value = "/updateDir", produces = "application/json; charset=UTF-8")
    public ResponseMsg<Integer> updateDir(@RequestBody DirEntity entity, HttpServletRequest req) {
        try {
            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue != null) {
                entity.setUpdateUser(ue.getId());
            }

            int count = dirService.update(entity);

            return success(count);
        } catch (Exception ex) {
            return fail(ex, -1);
        }
    }

    @SysLog()
    @ApiOperation(value = "更新多条")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "list", value = "实体类集合", dataType = "DirEntity", paramType = "body", example = "")
    })
    @ResponseBody
    @PostMapping(value = "/updateDirs", produces = "application/json; charset=UTF-8")
    public ResponseMsg<Integer> updateDirs(@RequestBody List<DirEntity> list, HttpServletRequest req) {
        try {
            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue != null) {
                for (DirEntity entity : list) {
                    entity.setUpdateUser(ue.getId());
                }
            }

            int count = dirService.updates(list);

            return success(count);
        } catch (Exception ex) {
            return fail(ex, -1);
        }
    }

    @SysLog()
    @ApiOperation(value = "根据ID查询数据")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "id", value = "ID", dataType = "Integer", paramType = "query", example = "1")
    })
    @GetMapping(value = "/selectDir")
    public ResponseMsg<DirEntity> selectDir(int id) {
        try {
            DirEntity dirEntity = dirService.selectDir(id);

            return success(dirEntity);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }

    @SysLog()
    @ApiOperation(value = "查询所有数据")
    @GetMapping(value = "/selectDirAll")
    public ResponseMsg<List<DirEntity>> selectDirAll() {
        try {
            List<DirEntity> list = dirService.selectDirAll();
            return success(list);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }

    @SysLog()
    @ApiOperation(value = "下载目录结构")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "id", value = "目录ID", dataType = "Integer", paramType = "query", example = "1")
    })
    @GetMapping(value = "/downloadDir")
    public void downloadDir(int id, HttpServletResponse res) {
        try {
            DirEntity entity = dirService.selectDir(id);
            if (null == entity || entity.getPid() > 0) {
                return;
            }

            String zipFile = dirService.createDirs(id);
            if (null == zipFile) {
                return;
            }

            WebHelper.download(zipFile, entity.getName() + ".zip", res);
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
        }
    }
}