import com.intellij.database.model.DasTable
import com.intellij.database.model.ObjectKind
import com.intellij.database.util.Case
import com.intellij.database.util.DasUtil
import com.intellij.database.psi.DbColumn;
FILES.chooseDirectoryAndSave("Choose directory", "Choose where to store generated files") { dir ->
def schemaMap = new LinkedHashMap>();
SELECTION.filter { (it.getKind() == ObjectKind.TABLE || it.getKind() == ObjectKind.SCHEMA) }.each {
def schemaName = DasUtil.getSchema(it);
if(!schemaMap.containsKey(schemaName)){
schemaMap.put(schemaName,new LinkedHashMap());
}
LOG.print("SELECTION type:"+it.getKind());
if(it.getKind() == ObjectKind.TABLE){
def tableName = it.getName();
if(!schemaMap.get(schemaName).containsKey(tableName)){
schemaMap.get(schemaName).put(tableName,generateTableHTMLStyle(it));
}
}else if(it.getKind() == ObjectKind.SCHEMA){
LOG.print("DasChildren:"+it.getDasChildren(ObjectKind.TABLE));
it.getDasChildren(ObjectKind.TABLE).each { table ->
def tableName = table.getName();
if(!schemaMap.get(schemaName).containsKey(tableName)){
schemaMap.get(schemaName).put(tableName,generateTableHTMLStyle(table));
}
}
}
}
def outlineHtml = '';
def tableHtml = '';
LOG.print("schemaMap:"+schemaMap);
schemaMap.each { schema ->
outlineHtml += '';
outlineHtml += '
'+schema.key+'';
outlineHtml += '
';
tableHtml += '
'+schema.key+'
';
schema.value.each { t ->
outlineHtml += '
'+(schema.key+'.'+t.key)+'';
tableHtml += (t.value);
}
outlineHtml += '
';
outlineHtml += '
'+tableHtml+'
';
def outlineBody = ''+outlineHtml+'
';
def body = ''+docBody + outlineBody +'';
def doc = '\n'+head+body+'';
new File(dir, "test.html").withPrintWriter { out -> out.println doc }
}
def generateTableHTMLStyle(table){
def schema = DasUtil.getSchema(table);
def tableName = table.getName();
def tableComment = table.getComment();
def tr = '';
DasUtil.getColumns(table).reduce([]) { fields, col ->
def columnComment = (col.getComment() == null ? '' : col.getComment());
def columnName = javaName(col.getName(),false);
def columnType = col.getDataType().getSpecification();
def isNotNullHtml = (col.isNotNull() ? '是' : '否');
def defaultVal = (col.getDefault() == null ? '' : col.getDefault());
def isPrimaryHtml = (DasUtil.isPrimary(col) ? '是' : '否');
def expandHtml = '' +
(DasUtil.isAutoGenerated(col) ? '- 自增
' : '') +
(DasUtil.isForeign(col) ? '- 外键
' : '') +
(col.getDataType().getSpecification().indexOf("unsigned") != -1 ? '- 无符号
' : '') +
(col.getDataType().getSpecification().indexOf("zerofill") != -1 ? '- 零填充
' : '') +
'
';
tr += '| '+columnName+' | '+columnType+' | '+columnComment+' |
';
}
// def thead = '| 字段名 | 类型 | 说明 |
|---|
';
def thead = '| 字段名 | 类型 | 说明 |
|---|
';
def tbody = ''+tr+'';
def tableHtml = '';
return tableHtml;
}
def javaName(str, capitalize) {
def s = com.intellij.psi.codeStyle.NameUtil.splitNameIntoWords(str)
.collect { Case.LOWER.apply(it).capitalize() }
.join("")
.replaceAll(/[^\p{javaJavaIdentifierPart}[_]]/, "_")
name = capitalize || s.length() == 1 ? s : Case.LOWER.apply(s[0]) + s[1..-1]
}
