package com.kidgrow.generator.controller;

import com.kidgrow.common.model.PageResult;
import com.kidgrow.generator.service.SysGeneratorService;
import io.swagger.annotations.Api;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.io.IOUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 石家庄喜高科技有限责任公司 版权所有 © Copyright 2020<br>
 *
 * @Description: 代码生成器<br>
 * @Project: <br>
 * @CreateDate: Created in 2020/2/24 09:40 <br>
 * @Author: <a href="4345453@kidgrow.com">liuke</a>
 */
@RestController
@Api(tags = "代码生成器")
@RequestMapping("/generator")
@Slf4j
public class SysGeneratorController {
    @Autowired
    private SysGeneratorService sysGeneratorService;

    /**
     * 列表
     */
    @ResponseBody
    @GetMapping("/list")
    public PageResult getTableList(@RequestParam Map<String, Object> params) {
        return sysGeneratorService.queryList(params);
    }


    @GetMapping("/codeAll")
    public void makeCodeAll(HttpServletResponse response) throws IOException {

        Map<String, Object> map = new HashMap<String, Object>();
        map.put("page",1);
        map.put("limit",200);

        PageResult pageResult = sysGeneratorService.queryList(map);
        List<HashMap<String,String>> tableEntityList = pageResult.getData();
        String[] tableNames = new String[tableEntityList.size()];
        for(int i=0;i<tableEntityList.size();i++){
            log.info(tableEntityList.get(i).get("tableName"));
            tableNames[i]=tableEntityList.get(i).get("tableName");
        }

        byte[] data = sysGeneratorService.generatorCode(tableNames);
        response.reset();
        response.setHeader("Content-Disposition", "attachment; filename=\"generator.zip\"");
        response.addHeader("Content-Length", "" + data.length);
        response.setContentType("application/octet-stream; charset=UTF-8");
        IOUtils.write(data, response.getOutputStream());
    }

    /**
     * 生成代码FileUtil
     */
    @GetMapping("/code")
    public void makeCode(String tables, HttpServletResponse response) throws IOException {
        byte[] data = sysGeneratorService.generatorCode(tables.split(","));
        response.reset();
        response.setHeader("Content-Disposition", "attachment; filename=\"generator.zip\"");
        response.addHeader("Content-Length", "" + data.length);
        response.setContentType("application/octet-stream; charset=UTF-8");
        IOUtils.write(data, response.getOutputStream());
    }
}