package com.kidgrow.oprationcenter.controller;

import com.kidgrow.common.constant.SecurityConstants;
import com.kidgrow.common.controller.BaseController;
import com.kidgrow.common.model.PageResult;
import com.kidgrow.common.model.ResultBody;
import com.kidgrow.common.utils.EncryptUtils;
import com.kidgrow.oprationcenter.model.PartnerDockingInfo;
import com.kidgrow.oprationcenter.service.IPartnerDockingInfoService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.validation.BindingResult;
import org.springframework.validation.ObjectError;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletResponse;
import javax.validation.Valid;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;


/**
 * 石家庄喜高科技有限责任公司 版权所有 © Copyright 2020<br>
 *
 * @Description: 合作商客户对接信息表
 * @Project: 运营中心
 * @CreateDate: Created in 2020-12-08 14:35:33 <br>
 * @Author: <a href="1024030301@kidgrow.com">LuLiQiang</a>
 * @version: 1.0
 */
@Slf4j
@RestController
@RequestMapping("/partnerdockinginfo")
@Api(tags = "合作商客户对接信息表")
public class PartnerDockingInfoController extends BaseController {
    @Autowired
    private IPartnerDockingInfoService partnerDockingInfoService;

    /**
     * 列表
     */
    @ApiOperation(value = "查询列表")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "page", value = "分页起始位置", required = true, dataType = "Integer"),
            @ApiImplicitParam(name = "limit", value = "分页结束位置", required = true, dataType = "Integer")
    })
    @GetMapping
    public PageResult list(@RequestParam Map<String, Object> params) {
        if (params.size() == 0) {
            params.put("page", 1);
            params.put("limit", 10);
        }
        return partnerDockingInfoService.findList(params);
    }

    /**
     * 查询
     */
    @ApiOperation(value = "查询")
    @GetMapping("/{id}")
    public ResultBody findById(@PathVariable Long id) {
        PartnerDockingInfo model = partnerDockingInfoService.getById(id);
        return ResultBody.ok().data(model).msg("查询成功");
    }

    /**
     * 根据PartnerDockingInfo当做查询条件进行查询
     */
    @ApiOperation(value = "根据PartnerDockingInfo当做查询条件进行查询")
    @PostMapping("/query")
    public ResultBody findByObject(@RequestBody PartnerDockingInfo partnerDockingInfo) {
        PartnerDockingInfo model = partnerDockingInfoService.findByObject(partnerDockingInfo);
        return ResultBody.ok().data(model).msg("查询成功");
    }

    /**
     * 新增or更新
     */
    @ApiOperation(value = "保存")
    @PostMapping
    public ResultBody save(@Valid @RequestBody PartnerDockingInfo partnerDockingInfo, BindingResult bindingResult) {
        List<String> errMsg = new ArrayList<>();
        if (bindingResult.hasErrors()) {
            for (ObjectError error : bindingResult.getAllErrors()) {
                errMsg.add(error.getDefaultMessage());
            }
            return ResultBody.failed().msg(errMsg.toString());
        } else {
            // 为产品制作服务器识别码
            String accessKey = "";
            // 若服务器机器码不为空，则根据合作商、医院、科室、服务器机器码生成识别码。
            if ((partnerDockingInfo.getServerMachineCode() != null)  && (!partnerDockingInfo.getServerMachineCode().isEmpty())){
                accessKey = EncryptUtils.encryptAES((partnerDockingInfo.getPartnerId().toString() + "|" + partnerDockingInfo.getHospitalId().toString() + "|" + partnerDockingInfo.getDepartmentId().toString() + "|" + partnerDockingInfo.getServerMachineCode()), SecurityConstants.AES_KEY);

            }
            // 若服务器机器码为空，则根据合作商、医院、科室生成识别码。
            else {
                accessKey = EncryptUtils.encryptAES((partnerDockingInfo.getPartnerId().toString() + "|" + partnerDockingInfo.getHospitalId().toString() + "|" + partnerDockingInfo.getDepartmentId().toString()), SecurityConstants.AES_KEY);
            }
            partnerDockingInfo.setAccessKey(accessKey);
            boolean v = partnerDockingInfoService.saveOrUpdate(partnerDockingInfo);
            if (v) {
                return ResultBody.ok().data(partnerDockingInfo).msg("保存成功");
            } else {
                return ResultBody.failed().msg("保存失败");
            }
        }
    }

    /**
     * 更新状态
     */
    @ApiOperation(value = "更新状态")
    @PostMapping("/updatestatus")
    public ResultBody updateStatus(@RequestBody PartnerDockingInfo partnerDockingInfo) {
        boolean v = partnerDockingInfoService.saveOrUpdate(partnerDockingInfo);
        if (v) {
            return ResultBody.ok().data(partnerDockingInfo).msg("状态更新成功");
        } else {
            return ResultBody.failed().data("状态更新失败");
        }
    }

    /**
     * 删除
     */
    @ApiOperation(value = "删除")
    @DeleteMapping("/{id}")
    public ResultBody delete(@PathVariable Long id) {
        boolean v = partnerDockingInfoService.removeById(id);
        if (v) {
            return ResultBody.ok().msg("删除成功");
        } else {
            return ResultBody.failed().msg("删除失败");
        }
    }

    /**
     * 创建该客户所需的私有云数据库脚本
     *
     * @param partnerDockingInfo
     * @return
     */
    @ApiOperation(value = "创建并下载私有云数据库脚本")
    @PostMapping("/downloaddb")
    public void downLoadDataBase(@RequestBody PartnerDockingInfo partnerDockingInfo, HttpServletResponse response) {
        String dbSql = partnerDockingInfoService.createSQL(partnerDockingInfo);

        if ((dbSql  == null) ||(dbSql.isEmpty())) {
            log.error("创建私有云数据库脚本出错！");
        } else {

            OutputStream out = null;
            InputStream fileInputStream = null;
            try {
                //BASE64Decoder dencoder = new BASE64Decoder();
                byte[] arr = dbSql.getBytes();

                // 设置输出的格式
                response.reset();
                response.setContentType("application/x-msdownload");
                response.addHeader("Content-Length", "" + arr.length);
                response.addHeader("Content-Disposition", "attachment; filename=\"" + partnerDockingInfo.getHospitalName() + partnerDockingInfo.getDepartmentName() + ".sql\"");


                out = response.getOutputStream();
                fileInputStream = new ByteArrayInputStream(arr);
                byte[] buffer = new byte[1024 * 10];
                int len = 0;
                while ((len = fileInputStream.read(buffer)) > 0) {
                    out.write(buffer, 0, len);
                }

            } catch (Exception e) {
                e.printStackTrace();

            } finally {

                if (fileInputStream != null) {
                    try {
                        fileInputStream.close();
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                }
            }
        }
    }
}