package com.tievd.jyz.controller;

import cn.hutool.core.io.FileUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.tievd.cube.commons.annotations.AutoLog;
import com.tievd.cube.commons.annotations.DictApi;
import com.tievd.cube.commons.base.CubeController;
import com.tievd.cube.commons.base.Result;
import com.tievd.cube.commons.mybatisplus.QueryGenerator;
import com.tievd.jyz.entity.SysInfo;
import com.tievd.jyz.service.ISysInfoService;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

/**
 * SysInfo
 *
 * @author cube
 * @version V2.0.0
 * @since 2023-02-27
 */
@Slf4j
@DictApi
@RestController
@RequestMapping("/jyz/sysInfo")
public class SysInfoController extends CubeController<SysInfo, ISysInfoService> {

    /**
     * 路径名
     */
    private static final String LOGO_PATH = "/sysLogo/";
    
    //预改标记
    private static final String prefix = "_pre";
    
    private static final Map<Integer, String> pathMap = new HashMap(){{
        put(1, "logo");
        put(2, "login");
        put(3, "big_data");
    }};

    @Value("${cube.resources.logo-path:/usr/local/uploadFile}")
    private String uploadFilePath;

    @Autowired
    private ISysInfoService sysInfoService;

    /**
     * 分页列表查询
     */
    @GetMapping("/list")
    public Result<?> queryPageList(SysInfo sysInfo,
                                   @RequestParam(defaultValue = "1") Integer pageNo,
                                   @RequestParam(defaultValue = "10") Integer pageSize,
                                   HttpServletRequest req) {
        QueryWrapper<SysInfo> queryWrapper = QueryGenerator.initQueryWrapper(sysInfo, req.getParameterMap());
        Page<SysInfo> page = new Page<>(pageNo, pageSize);
        IPage<SysInfo> pageList = sysInfoService.page(page, queryWrapper);
        return Result.ok(pageList);
    }

    /**
     * 添加
     */
    @AutoLog("SysInfo-添加")
    @PostMapping("/add")
    public Result<?> add(@RequestBody SysInfo sysInfo) {
        sysInfoService.save(sysInfo);
        return Result.ok();
    }

    private String saveLogo(String filePath, int type) {
        File file = new File(uploadFilePath + filePath);
        if (!file.getParentFile().exists()) {
            file.getParentFile().mkdirs();
        }
        String name = file.getName();
        String suffixName = name.substring(name.lastIndexOf("."));
        String logoName = pathMap.get(type) + suffixName;
        FileUtil.copy(file, FileUtil.file(uploadFilePath, LOGO_PATH, logoName), true);
        return LOGO_PATH + logoName;
    }

    /**
     * 编辑
     */
    @AutoLog("SysInfo-编辑")
    @PutMapping("/edit")
    public Result<?> edit(@RequestBody SysInfo sysInfo) {
        String logoPath = sysInfo.getLogoPath();
        String backgroundPath = sysInfo.getBackground();
        if (StringUtils.isNotBlank(logoPath) && logoPath.contains(prefix)) {
            sysInfo.setLogoPath(saveLogo(logoPath, 1));
        }
        if (StringUtils.isNotBlank(backgroundPath) && backgroundPath.contains(prefix)) {
            sysInfo.setBackground(saveLogo(backgroundPath, 2));
        }
        if (StringUtils.isNotBlank(sysInfo.getBigDataPath()) && sysInfo.getBigDataPath().contains(prefix)) {
            sysInfo.setBigDataPath(saveLogo(sysInfo.getBigDataPath(), 3));
        }
        sysInfoService.updateById(sysInfo);
        return Result.ok();
    }

    /**
     * 通过id删除
     */
    @AutoLog("SysInfo-通过id删除")
    @DeleteMapping("/delete")
    public Result<?> delete(@RequestParam String id) {
        sysInfoService.removeById(id);
        return Result.ok();
    }

    /**
     * 批量删除
     */
    @AutoLog("SysInfo-批量删除")
    @DeleteMapping("/deleteBatch")
    public Result<?> deleteBatch(@RequestParam String ids) {
        this.sysInfoService.removeByIds(Arrays.asList(ids.split(",")));
        return Result.ok();
    }

    /**
     * 通过id查询
     */
    @GetMapping("/queryById")
    public Result<?> queryById(@RequestParam String id) {
        SysInfo sysInfo = sysInfoService.getById(id);
        return Result.ok(sysInfo);
    }

    /**
     * 导出excel
     */
    @RequestMapping("/exportXls")
    public void exportXls(HttpServletRequest request, HttpServletResponse response, SysInfo sysInfo) throws IOException {
        super.exportXls(request, response, sysInfo, "SysInfo");
    }

    /**
     * 通过excel导入数据
     */
    @PostMapping("/importExcel")
    public Result<?> importExcel(HttpServletRequest request) throws Exception {
        return super.importExcel(request, SysInfo.class);
    }

    @GetMapping("/localInfo")
    public Result<?> getLocalInfo() {
        SysInfo info = service.getOne(new QueryWrapper<>(), false);
        return Result.ok(info);
    }

    @RequestMapping(value = "/uploadLogo", method = RequestMethod.POST)
    public Result importLogo(MultipartFile file, Integer type) throws IOException {
        String filename = file.getOriginalFilename();
        if (StringUtils.isBlank(filename)) {
            return Result.error("上传失败，请选择文件");
        }
        String suffixName = filename.substring(filename.lastIndexOf("."));
        if (!".png".equals(suffixName) && !".jpg".equals(suffixName) && !".gif".equals(suffixName)) {
            return Result.error("文件格式错误，请选择PNG,JPG,GIF文件");
        }
        if (file.isEmpty()) {
            return Result.error("上传失败，请选择文件");
        }
        return Result.ok("上传成功", getLogoUrl(file, suffixName, type));
    }

    private String getLogoUrl(MultipartFile file, String suffixName, int type) throws IOException {
        String logoName = StringUtils.EMPTY;
        logoName = pathMap.get(type) + prefix + suffixName;
        if (StringUtils.isBlank(logoName)) {
            return logoName;
        }
        InputStream inputStream;
        try {
            inputStream = file.getInputStream();
            FileUtil.writeFromStream(inputStream, FileUtil.file(uploadFilePath, LOGO_PATH, logoName));
        } catch (IOException e) {
            e.printStackTrace();
        }
        return LOGO_PATH + logoName;
    }
}