POJO-jpa-lombok-controller.groovy

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

import java.text.SimpleDateFormat

packageName = “com.example.demo”

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)
new File(dir, className + “Controller.java”).withPrintWriter(“utf-8”) { out -> generate(out, table, className) }
}

def generate(out, table, className) {
out.println “package ${packageName}.controller;”
out.println “”
out.println “import $packageName.service.${className}Service;”
out.println “import lombok.extern.slf4j.Slf4j;”
out.println “import org.springframework.beans.factory.annotation.Autowired;”
out.println “import org.springframework.web.bind.annotation.*;”
out.println “”
out.println “/**”
out.println " * @ClassName ${className}Controller"
out.println " * @Description Automatic script generation"
out.println " * @Author"
out.println " * @Date KaTeX parse error: Can't use function '\"' in math mode at position 215: …RequestMapping(\̲"̲{className}")"
out.println “public class ${className}Controller {”
out.println “”
out.println “\t@Autowired”
out.println “\tprivate ${className}Service service;”
out.println “”
out.println “\t@RequestMapping(“get”)”
out.println “\tpublic Object get() {”
out.println “”
out.println “\t\treturn null;”
out.println “\t}”
out.println “”
out.println “\t@RequestMapping(“list”)”
out.println “\tpublic Object list() {”
out.println “”
out.println “\t\treturn null;”
out.println “\t}”
out.println “”
out.println “}”

}

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]
}