import com.intellij.database.model.DasTable
import com.intellij.database.util.Case
import com.intellij.database.util.DasUtil

import java.time.LocalDate

/*
 * Available context bindings:
 * SELECTION Iterable<DasObject>
 * PROJECT  project
 * FILES  files helper
 */

// 此處指定包路徑，路徑需要自行維護(hù);
packageName = "com.qgy.web.entity;"
// 此處指定對應(yīng)的類型映射，可按需修改，目前tinyint如果要映射到自定義枚舉類型，只能手動修改
typeMapping = [
  (~/(?i)bigint/)     : "Long",
  (~/(?i)int/)      : "Integer",
  (~/(?i)tinyint/)     : "Boolean",
  (~/(?i)float|double|decimal|real/): "BigDecimal",
  (~/(?i)time|datetime|timestamp/) : "LocalDateTime",
  (~/(?i)date/)      : "LocalDate",
  (~/(?i)/)       : "String"
]

// 上面用到類和它的導(dǎo)入路徑的之間的映射
importMap = [
  "BigDecimal" : "java.math.BigDecimal",
  "LocalDate" : "java.time.LocalDate",
  "LocalDateTime": "java.time.LocalDateTime",
]

// 導(dǎo)入路徑列表，下面引用的時候會去重，也可以直接聲明成一個 HashSet
importList = []

// 彈出選擇文件的對話框
FILES.chooseDirectoryAndSave("Choose directory", "Choose where to store generated files") { dir ->
 SELECTION.filter { it instanceof DasTable }.each { generate(it, dir) }
}

def generate(table, dir) {
 def className = javaName(table.getName(), true) + "Entity"
 def fields = calcFields(table)
 new PrintWriter(new OutputStreamWriter(new FileOutputStream(new File(dir, className + ".java")), "utf-8")).withPrintWriter { out -> generate(out, className, fields, table) }
}

// 從這里開始，拼實體類的具體邏輯代碼
def generate(out, className, fields, table) {
 out.println "package $packageName"
 out.println ""
 // 引入所需的包
 out.println "import lombok.Data;"
 out.println "import lombok.EqualsAndHashCode;"
 out.println "import lombok.experimental.Accessors;"
 out.println "import com.baomidou.mybatisplus.annotation.*;"
 out.println "import java.io.Serializable;"
 // 去重后導(dǎo)入列表
 importList.unique().each() { pkg ->
  out.println "import " + pkg + ";"
 }
 out.println ""
 // 添加類注釋
 out.println "/**"
 // 如果添加了表注釋，會加到類注釋上
 if (isNotEmpty(table.getComment())) {
  out.println " * " + table.getComment()
 }
 out.println " *"
 out.println " * @author 輸入作者"
 out.println " * @date " + LocalDate.now()
 out.println " */"
 // 添加類注解
 out.println "@Data"
 out.println "@EqualsAndHashCode(callSuper = false)"
 out.println "@Accessors(chain = true)"
 out.println "@TableName(\"${table.getName()}\")"
 out.println "public class $className implements Serializable {"
 out.println ""
 out.println genSerialID()
 boolean isId = true
 // 遍歷字段，按下面的規(guī)則生成
 fields.each() {
  // 輸出注釋
  if (isNotEmpty(it.comment)) {
   out.println "\t/**"
   out.println "\t * ${it.comment}"
   out.println "\t */"
  }
  // 這邊默認(rèn)第一個字段為主鍵，實際情況大多數(shù)如此，遇到特殊情況可能需要手動修改
  if (isId) {
   out.println "\t@TableId(type = IdType.AUTO)"
   isId = false
  }
  if ((it.annos + "").indexOf("[@Id]") >= 0) out.println "\t@Id"

  if (it.annos != "") out.println " ${it.annos.replace("[@Id]", "")}"

  out.println "\tprivate ${it.type} ${it.name};"
  out.println ""
 }
 out.println ""
 out.println "}"
}

def calcFields(table) {
 DasUtil.getColumns(table).reduce([]) { fields, col ->
  def spec = Case.LOWER.apply(col.getDataType().getSpecification())
  def typeStr = typeMapping.find { p, t -> p.matcher(spec).find() }.value
  if (importMap.containsKey(typeStr)) {
   importList.add(importMap.get(typeStr))
  }
  fields += [[
       name : javaName(col.getName(), false),
       type : typeStr,
       comment: col.getComment(),
       annos : "\t@TableField(\"" + col.getName() + "\" )"
     ]]
 }
}

def isNotEmpty(content) {
 return content != null && content.toString().trim().length() > 0
}

def javaName(str, capitalize) {
 def s = com.intellij.psi.codeStyle.NameUtil.splitNameIntoWords(str)
   .collect { Case.LOWER.apply(it).capitalize() }
   .join("")
   .replaceAll(/[^\p{javaJavaIdentifierPart}[_]]/, "_")
 capitalize || s.length() == 1 ? s : Case.LOWER.apply(s[0]) + s[1..-1]
}

static String genSerialID() {
 return "\tprivate static final long serialVersionUID = " + Math.abs(new Random().nextLong()) + "L;"
}

package com.qgy.web.entity;

import lombok.Data;
import lombok.EqualsAndHashCode;
import lombok.experimental.Accessors;
import com.baomidou.mybatisplus.annotation.*;
import java.io.Serializable;
import java.time.LocalDateTime;

/**
 * 超級管理員賬號信息表
 *
 * @author rog
 * @date 2020-07-20
 */
@Data
@EqualsAndHashCode(callSuper = false)
@Accessors(chain = true)
@TableName("sys_super")
public class SysSuperEntity implements Serializable {

	private static final long serialVersionUID = 5132404354445122973L;
	/**
	 * 主鍵id標(biāo)識
	 */
	@TableId(type = IdType.AUTO)
 	@TableField("id" )
	private Integer id;

	/**
	 * 管理員id唯一標(biāo)識
	 */
 	@TableField("account_id" )
	private String accountId;

	/**
	 * 管理員賬號
	 */
 	@TableField("account" )
	private String account;

	/**
	 * 管理員密碼
	 */
 	@TableField("account_password" )
	private String accountPassword;

	/**
	 * 上一次登錄ip
	 */
 	@TableField("account_ip_last" )
	private String accountIpLast;

	/**
	 * 當(dāng)前登錄ip
	 */
 	@TableField("account_ip_now" )
	private String accountIpNow;

	/**
	 * 最近一次登錄時間
	 */
 	@TableField("login_time" )
	private LocalDateTime loginTime;

	/**
	 * 是否啟用
	 */
 	@TableField("isEnable" )
	private Integer isEnable;
}