Lab 023

使用方法: Database -> 右键数据库表 -> Scripted Extensions -> vuejs.groovy

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

/*
 * Available context bindings:
 *   SELECTION   Iterable<DasObject>
 *   PROJECT     project
 *   FILES       files helper
 *
 * Last Modify: 2018-03-20
 */

typeMapping = [
        (~/(?i)int/)                      : "Long",
        (~/(?i)float|double|decimal|real/): "Double",
        (~/(?i)datetime|ts/)              : "Timestamp",
        (~/(?i)date/)                     : "Date",
        (~/(?i)time/)                     : "Timestamp",
        (~/(?i)/)                         : "String"
]

FILES.chooseDirectoryAndSave("Choose directory", "Choose where to store generated files") { dir ->
    SELECTION.filter { it instanceof DasTable && it.getKind() == ObjectKind.TABLE }.each { generate(it, dir) }
}

def generate(table, dir) {
    def className = javaName2(table.getName(), true)
    className = Case.LOWER.apply(className[0]) + className[1..-1]
    def fields = calcFields(table)
    new File(dir, className + ".js").withPrintWriter { out -> generate(table, out, className, fields) }
}

def generate(table, out, className, fields) {
    // +js
    getJsObject(out, fields)
    getJsArray(out, fields)
    getJsEntity(out, fields)
    // -js
}

// 字段内容
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
        def colName = col.getName() + ""

        // PK字段的开头必须是 idt_
        if (colName ==~ /(^idt$|^idt_.*)/) {
            fields += [[
                               name : colName,
                               type : typeStr,
                               annos: "  @Id" + (typeStr == "Long" ? "\n    @GeneratedValue(strategy = GenerationType.IDENTITY)" : "")]]
        } else {
            fields += [[
                               name : colName,
                               type : typeStr,
                               annos: ""]]
        }
    }
}

// 表名转类名
def javaName(str, capitalize) {
    def s = str.split(/[^\p{Alnum}]/).collect { def s = Case.LOWER.apply(it).capitalize() }.join("")
    capitalize ? s : Case.LOWER.apply(s[0]) + s[1..-1]
}

def javaName2(str, capitalize) {
    def s = str.replaceAll("^t_", "").split(/[^\p{Alnum}]/).collect {
        def s = Case.LOWER.apply(it).capitalize()
    }.join("")
    capitalize ? s : Case.LOWER.apply(s[0]) + s[1..-1]

}

// 驼峰命令
static String toCamelCase(String text, boolean capitalized = false) {
    text = text.replaceAll("(_)([A-Za-z0-9])", { Object[] it -> it[2].toUpperCase() })
    return capitalized ? capitalize(text) : text
}

// 蛇形命名
static String toSnakeCase(String text) {
    text.replaceAll(/([A-Z])/, /_$1/).toLowerCase().replaceAll(/^_/, '')
}

// 首字母大写
def firstUpper(str) {
    Case.UPPER.apply(str[0]) + str[1..-1]
}

/* JS对象 */

def getJsObject(out, fields) {
    out.println("")
    out.println("// JS Object")

    fs = ""
    fs += "export function keyObject() {\n"
    fs += "  return {\n"
    fields.each() {
        fs += "    ${toCamelCase(it.name)}: '',\n"
    }
    fs = fs.replaceAll(',\\n$', "")
    fs += "\n  }\n"
    fs += "}\n"
    out.println(fs)
}

/* JS数组 */

def getJsArray(out, fields) {
    out.println("// JS array")

    fs = ""
    fs += "export function keyArray() {\n"
    fs += "  return ["
    fields.each() {
        fs += "'${toCamelCase(it.name)}', "
    }
    fs = fs.replaceAll(', $', "")
    fs += "]\n"
    fs += "}\n"
    out.println(fs)
}

/* JS复杂对象 */
// { display_name: '序号', key: 'idt', canSearch: true, canSort: true, showInExtend: true, type: 'String', cantNull: true }
def getJsEntity(out, fields) {
    out.println("// JS Entity")

    fs = ""
    fs += "export function entity() {\n"
    fs += "  return [\n"
    fields.each() {
        fs += "    { display_name: '_', key: '${toCamelCase(it.name)}', exclude: false, canSearch: ${it.type != "Timestamp" && it.type != "Double"}, canSort: true, showInExtend: ${it.type == "Timestamp"}, type: '${it.type}', canNull: ${it.type != "String"}, canEdit: ${!it.name.contains("idt")} },\n"
    }
    fs = fs.replaceAll(',\\n$', "")
    fs += "\n  ]\n"
    fs += "}\n"
    out.println(fs)
}

导出示例

// JS Object
export function keyObject() {
  return {
    idt: '',
    tBusinessUserIdt: '',
    mobileNumber: '',
    sendTime: '',
    sendFlag: '',
    ip: '',
    appToken: '',
    appId: '',
    appVersion: '',
    appChannel: '',
    appOs: '',
    appOsVersion: '',
    appTerminalVendor: '',
    appTerminalId: '',
    appTerminalModel: '',
    respCode: '',
    respMessage: '',
    respBizId: '',
    smsType: '',
    smsContent: '',
    addTime: ''
  }
}

// JS array
export function keyArray() {
  return ['idt', 'tBusinessUserIdt', 'mobileNumber', 'sendTime', 'sendFlag', 'ip', 'appToken', 'appId', 'appVersion', 'appChannel', 'appOs', 'appOsVersion', 'appTerminalVendor', 'appTerminalId', 'appTerminalModel', 'respCode', 'respMessage', 'respBizId', 'smsType', 'smsContent', 'addTime']
}

// JS Entity
export function entity() {
  return [
    { display_name: '_', key: 'idt', exclude: false, canSearch: true, canSort: true, showInExtend: false, type: 'String', canNull: false, canEdit: false },
    { display_name: '_', key: 'tBusinessUserIdt', exclude: false, canSearch: true, canSort: true, showInExtend: false, type: 'String', canNull: false, canEdit: false },
    { display_name: '_', key: 'mobileNumber', exclude: false, canSearch: true, canSort: true, showInExtend: false, type: 'String', canNull: false, canEdit: true },
    { display_name: '_', key: 'sendTime', exclude: false, canSearch: false, canSort: true, showInExtend: true, type: 'Timestamp', canNull: true, canEdit: true },
    { display_name: '_', key: 'sendFlag', exclude: false, canSearch: true, canSort: true, showInExtend: false, type: 'Long', canNull: true, canEdit: true },
    { display_name: '_', key: 'ip', exclude: false, canSearch: true, canSort: true, showInExtend: false, type: 'String', canNull: false, canEdit: true },
    { display_name: '_', key: 'appToken', exclude: false, canSearch: true, canSort: true, showInExtend: false, type: 'String', canNull: false, canEdit: true },
    { display_name: '_', key: 'appId', exclude: false, canSearch: true, canSort: true, showInExtend: false, type: 'String', canNull: false, canEdit: true },
    { display_name: '_', key: 'appVersion', exclude: false, canSearch: true, canSort: true, showInExtend: false, type: 'String', canNull: false, canEdit: true },
    { display_name: '_', key: 'appChannel', exclude: false, canSearch: true, canSort: true, showInExtend: false, type: 'String', canNull: false, canEdit: true },
    { display_name: '_', key: 'appOs', exclude: false, canSearch: true, canSort: true, showInExtend: false, type: 'String', canNull: false, canEdit: true },
    { display_name: '_', key: 'appOsVersion', exclude: false, canSearch: true, canSort: true, showInExtend: false, type: 'String', canNull: false, canEdit: true },
    { display_name: '_', key: 'appTerminalVendor', exclude: false, canSearch: true, canSort: true, showInExtend: false, type: 'String', canNull: false, canEdit: true },
    { display_name: '_', key: 'appTerminalId', exclude: false, canSearch: true, canSort: true, showInExtend: false, type: 'String', canNull: false, canEdit: true },
    { display_name: '_', key: 'appTerminalModel', exclude: false, canSearch: true, canSort: true, showInExtend: false, type: 'String', canNull: false, canEdit: true },
    { display_name: '_', key: 'respCode', exclude: false, canSearch: true, canSort: true, showInExtend: false, type: 'String', canNull: false, canEdit: true },
    { display_name: '_', key: 'respMessage', exclude: false, canSearch: true, canSort: true, showInExtend: false, type: 'String', canNull: false, canEdit: true },
    { display_name: '_', key: 'respBizId', exclude: false, canSearch: true, canSort: true, showInExtend: false, type: 'String', canNull: false, canEdit: true },
    { display_name: '_', key: 'smsType', exclude: false, canSearch: true, canSort: true, showInExtend: false, type: 'String', canNull: false, canEdit: true },
    { display_name: '_', key: 'smsContent', exclude: false, canSearch: true, canSort: true, showInExtend: false, type: 'String', canNull: false, canEdit: true },
    { display_name: '_', key: 'addTime', exclude: false, canSearch: false, canSort: true, showInExtend: true, type: 'Timestamp', canNull: true, canEdit: true }
  ]
}