/**
 * Copyright (c) 2018 人人开源 All rights reserved.
 * <p>
 * https://www.renren.io
 * <p>
 * 版权所有，侵权必究！
 */

package com.zt.generator.controller;

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.zt.common.constant.Constant;
import com.zt.common.db.constant.OP;
import com.zt.common.exception.RenException;
import com.zt.common.servlet.Result;
import com.zt.generator.model.Config;
import com.zt.generator.data.DBUtil;
import com.zt.generator.data.DataTable;
import com.zt.generator.utils.GenUtils;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.Data;
import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.*;

import java.io.File;
import java.io.IOException;
import java.io.Serializable;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.Properties;

/**
 * 代码生成器
 *
 * @author xhb
 */

@RestController
@RequestMapping("/sys/generator")
@Api(tags = "代码生成")
public class SysGeneratorController {

    @GetMapping("properties")
    //@RequiresPermissions(Constant.Permissions.SUPER_ADMIN)
    public Result<JSONObject> queryProperties() {
        JSONObject jsonObject = new JSONObject();
        try {
            File directory = new File("");// 参数为空
            String courseFile = directory.getCanonicalPath();
            File root = new File(courseFile);
            JSONArray array = new JSONArray();
            getSubFiles(root, array);
            jsonObject.put("modules", array);
        } catch (IOException e) {
            e.printStackTrace();
        }
        JSONArray eqList = new JSONArray();
        for (OP.Type value : OP.Type.values()) {
            JSONObject json = new JSONObject();
            json.put("title", value.getTitle());
            json.put("op", value.getOp());
            eqList.add(json);
        }
        jsonObject.put("eqList", eqList);

        DbProperties jdbc = queryJdbcProperties();

        DataTable table = null;
        Connection conn = null;
        try {
            // 注册数据库驱动
            Class.forName(jdbc.getDriverClass());
            Properties props = new Properties();
            props.put("remarksReporting", "true");// 为了得到说明
            props.setProperty("user", jdbc.getUserName());
            props.setProperty("password", jdbc.getPassWord());
            // 获取连接
            conn = DriverManager.getConnection(jdbc.getUrl(), props);
            table = DBUtil.getTableList(conn, false);

        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (conn != null) {
                try {
                    conn.close();
                } catch (SQLException e) {
                    e.printStackTrace();
                }
            }
        }

        jsonObject.put("tableList", table.toJSONArray());
        return Result.ok(jsonObject);
    }

    private void getSubFiles(File file, JSONArray array) {
        if (file.isDirectory()) {
            // 获取当前目录下的所有子项
            File[] subs = file.listFiles();
            for (File sub : subs) {
                if (sub.isDirectory() && new File(sub.getPath() + "/pom.xml").exists()) {
                    if (new File(sub.getPath() + "/src").exists()) {
                        JSONObject object = new JSONObject();
                        object.put("filePath", sub.getPath());
                        object.put("fileName", sub.getName());
                        array.add(object);
                    } else {
                        getSubFiles(sub, array);
                    }
                }
                if (sub.isDirectory() && new File(sub.getPath() + "/package.json").exists()) {
                    JSONObject object = new JSONObject();
                    object.put("filePath", sub.getPath());
                    object.put("fileName", sub.getName());
                    array.add(object);
                }
            }
        }
    }

    @GetMapping("getTable")
    //@RequiresPermissions(Constant.Permissions.SUPER_ADMIN)
    public Result<JSONArray> tableNameVerification(String tableName) {
        DbProperties jdbc = queryJdbcProperties();

        DataTable table = null;
        Connection conn = null;
        try {
            // 注册数据库驱动
            Class.forName(jdbc.getDriverClass());
            Properties props = new Properties();
            props.put("remarksReporting", "true");// 为了得到说明
            props.setProperty("user", jdbc.getUserName());
            props.setProperty("password", jdbc.getPassWord());
            // 获取连接
            conn = DriverManager.getConnection(jdbc.getUrl(), props);

            if (DBUtil.findTable(conn, tableName)) {
                table = DBUtil.getColumnInfo(conn, tableName, false);
            }

        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (conn != null) {
                try {
                    conn.close();
                } catch (SQLException e) {
                    e.printStackTrace();
                }
            }
        }

        if (table != null) {
            JSONArray array = table.toJSONArray();
            for (int i = 0; i < array.size(); i++) {
                JSONObject json = array.getJSONObject(i);
                String columnName = json.getString("columnName");
                if ("is_delete,dept_id,company_id,tenant_id,creator,create_date,updater,update_date"
                        .indexOf(columnName.toLowerCase()) >= 0) {
                    json.put("isTableColumn", "false");
                }
            }
            return Result.ok(array);
        } else {
            throw new RenException("表不存在!");
        }
    }

    @Value("${spring.datasource.druid.driver-class-name}")
    private String dbDriverName;
    @Value("${spring.datasource.druid.url}")
    private String dbDriverUrl;
    @Value("${spring.datasource.druid.username}")
    private String dbDriverUserName;
    @Value("${spring.datasource.druid.password}")
    private String dbDriverPassword;

    private DbProperties queryJdbcProperties() {
        DbProperties jdbc = new DbProperties();
        String driver = dbDriverName;
        jdbc.setDbName(driver.equals("oracle.jdbc.driver.OracleDriver") ? "ORACLE" : "MYSQL");
        jdbc.setDriverClass(driver);
        jdbc.setUrl(dbDriverUrl);
        jdbc.setUserName(dbDriverUserName);
        jdbc.setPassWord(dbDriverPassword);

        return jdbc;
    }

    @Data
    class DbProperties implements Serializable {
        private String dbName;
        private String driverClass;
        private String url;
        private String userName;
        private String passWord;
    }

    /**
     * 生成代码
     */
    @PostMapping
    @ApiOperation("保存")
    //@RequiresPermissions(Constant.Permissions.SUPER_ADMIN)
    public Result code(@RequestBody Config dto) {
        String isGenService = dto.getChecked();

        GenUtils.generatorCode(dto.getTableName(), isGenService, dto.getClassName(), dto.getIsPageFlag(),
                dto.getIsExport(), JSONArray.parseArray(dto.getTableData()), dto.getPackName(), dto.getJavaFilePath(),
                dto.getVueFilePath());
        return Result.ok();
    }

}