package com.qianwen.smartman.modules.develop.controller;

import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.IdWorker;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
import com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import javax.validation.Valid;
import org.apache.commons.lang3.StringUtils;
import com.qianwen.smartman.common.cache.RegionCache;
import com.qianwen.smartman.common.utils.ZipFileTree;
import com.qianwen.core.boot.ctrl.BladeController;
import com.qianwen.core.mp.base.BaseEntity;
import com.qianwen.core.mp.support.Condition;
import com.qianwen.core.mp.support.Query;
import com.qianwen.core.tenant.annotation.NonDS;
import com.qianwen.core.tool.api.R;
import com.qianwen.core.tool.utils.Func;
import com.qianwen.core.util.CodeUtil;
import org.springblade.generator.action.config.GeneratorConfig;
import org.springblade.generator.action.engine.CustomTemplateEngine;
import org.springblade.generator.action.engine.WebPageTemplateEngine;
import org.springblade.generator.action.model.BladeCodeEntity;
import org.springblade.generator.action.model.BladeWebPageEntity;
import org.springblade.generator.action.model.TemplateEntity;
import org.springblade.generator.entity.TabEntity;
import com.qianwen.smartman.modules.develop.entity.Code;
import com.qianwen.smartman.modules.develop.service.ICodeService;
import com.qianwen.smartman.modules.develop.service.IDatasourceService;
import com.qianwen.smartman.modules.develop.vo.CodeTemplateVo;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.GetMapping;
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.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import springfox.documentation.annotations.ApiIgnore;

@RequestMapping({"blade-develop/code"})
@Api(value = "代码生成", tags = {"代码生成"})
@ApiIgnore
@RestController
@NonDS
/* loaded from: blade-api.jar:BOOT-INF/classes/org/springblade/modules/develop/controller/CodeController.class */
public class CodeController extends BladeController {
    private final ICodeService codeService;
    private final IDatasourceService datasourceService;
    @Value("${spring.datasource.dynamic.datasource.master.url}")
    private String url;
    @Value("${spring.datasource.dynamic.datasource.master.username}")
    private String user;
    @Value("${spring.datasource.dynamic.datasource.master.password}")
    private String password;

    public CodeController(ICodeService codeService, IDatasourceService datasourceService) {
        this.codeService = codeService;
        this.datasourceService = datasourceService;
    }

    @ApiOperationSupport(order = 1)
    @GetMapping({"/detail"})
    @ApiOperation(value = "详情", notes = "传入code")
    public R<Code> detail(Code code) {
        Code detail = (Code) this.codeService.getOne(Condition.getQueryWrapper(code));
        return R.data(detail);
    }

    @ApiOperationSupport(order = 2)
    @ApiImplicitParams({@ApiImplicitParam(name = "codeName", value = "模块名", paramType = "query", dataType = "string"), @ApiImplicitParam(name = "tableName", value = "表名", paramType = "query", dataType = "string"), @ApiImplicitParam(name = "modelName", value = "实体名", paramType = "query", dataType = "string")})
    @ApiOperation(value = "分页", notes = "传入code")
    @GetMapping({"/list"})
    public R<IPage<Code>> list(@RequestParam @ApiIgnore Map<String, Object> code, Query query) {
        IPage<Code> pages = this.codeService.page(Condition.getPage(query), Condition.getQueryWrapper(code, Code.class));
        return R.data(pages);
    }

    @PostMapping({"/submit"})
    @ApiOperationSupport(order = 3)
    @ApiOperation(value = "新增或修改", notes = "传入code")
    public R submit(@Valid @RequestBody Code code) {
        return R.status(this.codeService.submit(code));
    }

    @PostMapping({"/remove"})
    @ApiOperationSupport(order = 4)
    @ApiOperation(value = "删除", notes = "传入ids")
    public R remove(@RequestParam @ApiParam(value = "主键集合", required = true) String ids) {
        return R.status(this.codeService.removeByIds(Func.toLongList(ids)));
    }

    @PostMapping({"/copy"})
    @ApiOperationSupport(order = RegionCache.VILLAGE_LEVEL)
    @ApiOperation(value = "复制", notes = "传入id")
    public R copy(@RequestParam @ApiParam(value = "主键", required = true) Long id) {
        Code code = (Code) this.codeService.getById(id);
        code.setId(null);
        code.setCodeName(code.getCodeName() + "-copy");
        return R.status(this.codeService.save(code));
    }

    @PostMapping({"/gen-code-new"})
    @ApiOperationSupport(order = 6)
    @ApiOperation(value = "新版本代码生成,生成后端方法", notes = "传入ids")
    public R genCodeNew(@RequestParam @ApiParam(value = "主键集合", required = true) String ids) throws IOException {
        Collection<Code> codes = this.codeService.listByIds(Func.toLongList(ids));
        String millisecond = IdWorker.getMillisecond();
        File dir = new File("static/codefile");
        if (!dir.exists()) {
            dir.mkdir();
        }
        String uploadPath = "static/codefile/" + millisecond;
        String bladeBootPath = uploadPath + "/Bladex-Boot";
        codes.forEach(code -> {
            BladeCodeEntity codeEntity = new BladeCodeEntity("", code.getCodeName(), code.getTableName(), code.getTablePrefix(), code.getServiceName(), code.getPackageName(), bladeBootPath + "/src/main/java", BaseEntity.class);
            codeEntity.setServiceName(code.getServiceName());
            AutoGenerator autoGenerator = new GeneratorConfig(new DataSourceConfig.Builder(this.url, this.user, this.password).build()).build(codeEntity);
            autoGenerator.execute(new CustomTemplateEngine());
        });
        new ZipFileTree().zipFile(uploadPath);
        R<Object> res = R.success("代码生成成功");
        res.setData(millisecond);
        return res;
    }

    @PostMapping({"/template"})
    @ApiOperationSupport(order = RegionCache.VILLAGE_LEVEL)
    @ApiOperation(value = "前端代码生成，选择相关模板生成,只生成前端代码", notes = "传入模板对象")
    public R template(@ApiParam(value = "模板生成", required = true) @RequestBody CodeTemplateVo codeTemplate) throws IOException {
        String millisecond = IdWorker.getMillisecond();
        String uploadPath = "static/codefile/" + millisecond;
        String saberPath = uploadPath + "/Saber";
        if (!codeTemplate.isTabPage()) {
            Code code = (Code) this.codeService.getById(codeTemplate.getTabs().get(0).getId());
            createWebPageEntity(codeTemplate, saberPath, code, null);
        } else {
            List<TabEntity> tabEntities = new ArrayList<>();
            codeTemplate.getTabs().forEach(codeVo -> {
                Code code2 = (Code) this.codeService.getById(codeVo.getId());
                createWebPageEntity(codeTemplate, saberPath, code2, codeTemplate.getFileName());
                String fileName = CodeUtil.lineToHump(code2.getTableName().replace(code2.getTablePrefix(), ""));
                tabEntities.add(new TabEntity(fileName));
            });
            TemplateEntity codeEntity = new TemplateEntity("", codeTemplate.getModulePath(), "src", saberPath);
            codeEntity.setTabs(tabEntities);
            if (StringUtils.isNotEmpty(codeTemplate.getFileName())) {
                codeEntity.setFileName(codeTemplate.getFileName());
            }
            AutoGenerator autoGenerator = new GeneratorConfig(new DataSourceConfig.Builder(this.url, this.user, this.password).build()).build(codeEntity);
            autoGenerator.execute(new WebPageTemplateEngine());
        }
        new ZipFileTree().zipFile(uploadPath);
        R<Object> res = R.success("代码生成成功");
        res.setData(millisecond);
        return res;
    }

    private void createWebPageEntity(CodeTemplateVo codeTemplate, String saberPath, Code code, String tabName) {
        BladeWebPageEntity codeEntity = new BladeWebPageEntity("", codeTemplate.getModulePath(), code.getTableName(), code.getTablePrefix(), "src", saberPath);
        codeEntity.setPageTemplate(codeTemplate.pageTemplate());
        codeEntity.setTabName(tabName);
        AutoGenerator autoGenerator = new GeneratorConfig(new DataSourceConfig.Builder(this.url, this.user, this.password).build()).build(codeEntity);
        autoGenerator.execute(new WebPageTemplateEngine());
    }
}