Skip to content

Commit

Permalink
concept cloud plugin generate code action
Browse files Browse the repository at this point in the history
  • Loading branch information
tanghanzheng committed Jun 2, 2023
1 parent 33d25eb commit 105c991
Show file tree
Hide file tree
Showing 8 changed files with 300 additions and 71 deletions.
13 changes: 13 additions & 0 deletions ...-intellij/src/main/java/com/github/linyuzai/cloud/plugin/intellij/GenerateCodeAction.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -47,6 +47,19 @@ public static class Context {
String userClassName;

Object model;

public void withParentPackage() {
if (psiPackage == null || path == null) {
return;
}
PsiPackage parentPackage = psiPackage.getParentPackage();
File parentFile = new File(path).getParentFile();
if (parentPackage == null || parentFile == null) {
return;
}
psiPackage = parentPackage;
path = parentFile.getPath();
}
}

@Override
Expand Down
12 changes: 11 additions & 1 deletion ...lij/src/main/java/com/github/linyuzai/cloud/plugin/intellij/GenerateModuleCodeAction.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -60,9 +60,12 @@ public DialogBuilder createDialogBuilder(Context context) {
String domainRepositoryClassName = domainRepositoryClass == null ? "" :
domainRepositoryClass.getQualifiedName();

context.withParentPackage();//to domain
context.withParentPackage();//to module

final ModuleModel model = new ModuleModel(context.userClassName,
loginAnnotationClassName == null ? "" : loginAnnotationClassName,
context.selectModule, context.fullPackage,
context.selectModule, getModulePackage(context.psiPackage),
domainObjectClassName == null ? "" : domainObjectClassName,
domainCollectionClassName == null ? "" : domainCollectionClassName,
domainServiceClassName == null ? "" : domainServiceClassName,
Expand All @@ -74,6 +77,13 @@ public DialogBuilder createDialogBuilder(Context context) {
return ModuleComponents.createGenerateModuleCodeDialog(context.project, model, getDialogTitle());
}

private String getModulePackage(PsiPackage psiPackage) {
if (psiPackage == null) {
return "";
}
return psiPackage.getQualifiedName();
}

private String getDomainDescriptionByComment(PsiClass psiClass) {
if (psiClass == null) {
return "";
Expand Down
94 changes: 65 additions & 29 deletions ...intellij/src/main/kotlin/com/github/linyuzai/cloud/plugin/intellij/builder/JavaBuilder.kt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -14,8 +14,12 @@ const val ANNOTATION_DELETE_MAPPING = "org.springframework.web.bind.annotation.D
const val ANNOTATION_GET_MAPPING = "org.springframework.web.bind.annotation.GetMapping"
const val ANNOTATION_PATH_VARIABLE = "org.springframework.web.bind.annotation.PathVariable"
const val ANNOTATION_SERVICE = "org.springframework.stereotype.Service"
const val ANNOTATION_REPOSITORY = "org.springframework.stereotype.Repository"
const val ANNOTATION_COMPONENT = "org.springframework.stereotype.Component"
const val ANNOTATION_AUTOWIRED = "org.springframework.beans.factory.annotation.Autowired"
const val ANNOTATION_CONFIGURATION = "org.springframework.context.annotation.Configuration"
const val ANNOTATION_BEAN = "org.springframework.context.annotation.Bean"
const val ANNOTATION_CONDITIONAL_ON_MISSING_BEAN = "org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean"
const val ANNOTATION_GETTER = "lombok.Getter"
const val ANNOTATION_DATA = "lombok.Data"
const val ANNOTATION_NO_ARGS_CONSTRUCTOR = "lombok.NoArgsConstructor"
Expand All @@ -25,10 +29,16 @@ const val ANNOTATION_TAG = "io.swagger.v3.oas.annotations.tags.Tag"
const val ANNOTATION_SCHEMA = "io.swagger.v3.oas.annotations.media.Schema"
const val ANNOTATION_OPERATION = "io.swagger.v3.oas.annotations.Operation"
const val ANNOTATION_PARAMETER = "io.swagger.v3.oas.annotations.Parameter"
const val ANNOTATION_TABLE_NAME = "com.baomidou.mybatisplus.annotation.TableName"
const val ANNOTATION_TABLE_ID = "com.baomidou.mybatisplus.annotation.TableId"
const val ANNOTATION_TABLE_LOGIC = "com.baomidou.mybatisplus.annotation.TableLogic"

const val TYPE_BOOLEAN = "java.lang.Boolean"
const val TYPE_STRING = "java.lang.String"
const val TYPE_DATE = "java.util.Date"
const val TYPE_LIST = "java.util.List"
const val TYPE_COLLECTORS = "java.util.stream.Collectors"
const val TYPE_IDENTIFIABLE = "com.github.linyuzai.domain.core.Identifiable"
const val TYPE_DOMAIN_VALUE = "com.github.linyuzai.domain.core.DomainValue"
const val TYPE_DOMAIN_ENTITY = "com.github.linyuzai.domain.core.DomainEntity"
const val TYPE_DOMAIN_COLLECTION = "com.github.linyuzai.domain.core.DomainCollection"
Expand All @@ -40,12 +50,16 @@ const val TYPE_PAGES = "com.github.linyuzai.domain.core.page.Pages"
const val TYPE_DOMAIN_CONDITIONS = "com.github.linyuzai.domain.core.condition.Conditions"
const val TYPE_DOMAIN_LAMBDA_CONDITIONS = "com.github.linyuzai.domain.core.condition.LambdaConditions"
const val TYPE_DOMAIN_NOT_FOUND_EXCEPTION = "com.github.linyuzai.domain.core.exception.DomainNotFoundException"
const val TYPE_MBP_DOMAIN_REPOSITORY = "com.github.linyuzai.domain.mbp.MBPDomainRepository"
const val TYPE_MBP_DOMAIN_ID_GENERATOR = "com.github.linyuzai.domain.mbp.MBPDomainIdGenerator"
const val TYPE_BASE_MAPPER = "com.baomidou.mybatisplus.core.mapper.BaseMapper"

const val PARAM_ID = "id"
const val PARAM_DESCRIPTION = "description"
const val PARAM_NAME = "name"
const val PARAM_SUMMARY = "summary"
val PARAM_ACCESS_PROTECTED = "access" to ("lombok.AccessLevel" to "PROTECTED")
val PARAM_ID_TYPE_INPUT = "type" to ("com.baomidou.mybatisplus.annotation.IdType" to "INPUT")
fun schemaDescription(desc: String): Pair<String, Pair<String, String>> {
return PARAM_DESCRIPTION to ("" to "\"$desc\"")
}
Expand All @@ -66,6 +80,8 @@ fun annotationValue(value: String): Pair<String, Pair<String, String>> {
return "" to ("" to "\"$value\"")
}

val IGNORE_CLASS_ARRAY = arrayOf("Pages.Args")

data class JavaBuilder(val _name: String) : ContentGenerator() {

private var _package = ""
Expand Down Expand Up @@ -102,10 +118,12 @@ data class JavaBuilder(val _name: String) : ContentGenerator() {
override fun content(): String {
return buildString {
append("package $_package;\n")
_imports.filterNot { it.startsWith("java.lang.") }
_imports.asSequence()
.filterNot { it.startsWith("java.lang.") }
.filterNot { IGNORE_CLASS_ARRAY.contains(it) }
.filter { it.isNotBlank() }
.filter { it.contains(".") }
.sorted()
.sorted().toList()
.forEach {
append("import $it;\n")
}
Expand Down Expand Up @@ -195,11 +213,17 @@ data class ClassBuilder(val _java: JavaBuilder, val _name: String) : ContentGene
fun _extends(_type: String, vararg _generic: String) {
this._extends = _type to _generic
this._java._import(_type)
_generic.forEach {
this._java._import(it)
}
}

fun _implements(_type: String, vararg _generic: String) {
_interfaces.add(_type to _generic)
_java._import(_type)
this._interfaces.add(_type to _generic)
this._java._import(_type)
_generic.forEach {
this._java._import(it)
}
}

fun _field(_name: String, _init: FieldBuilder.() -> Unit): FieldBuilder {
Expand Down Expand Up @@ -244,11 +268,7 @@ data class ClassBuilder(val _java: JavaBuilder, val _name: String) : ContentGene

if (_extends.first.isNotEmpty()) {
append("extends ")
append(_extends.first.toSampleName())
if (_extends.second.isNotEmpty()) {
append(_extends.second.joinToString(",", "<", ">")
{ s -> s.toSampleName() })
}
addGeneric(_extends)
append(" ")
}

Expand All @@ -267,7 +287,7 @@ data class MethodBuilder(private val _java: JavaBuilder, private val _name: Stri

private val _params = mutableListOf<FieldBuilder>()

private var _return = "void"
private var _return = Pair<String, Array<out String>>("void", emptyArray())

private val _annotations = mutableListOf<Pair<String, Array<out Pair<String, Pair<String, String>>>>>()

Expand All @@ -292,10 +312,11 @@ data class MethodBuilder(private val _java: JavaBuilder, private val _name: Stri
return builder
}

fun _return(_return: String, _import: Boolean = true) {
this._return = _return
if (_import) {
this._java._import(_return)
fun _return(_type: String, vararg _generic: String) {
this._return = _type to _generic
this._java._import(_type)
_generic.forEach {
this._java._import(it)
}
}

Expand Down Expand Up @@ -343,7 +364,8 @@ data class MethodBuilder(private val _java: JavaBuilder, private val _name: Stri

val args = _params.joinToString(",") { it._asParam() }

append("${_return.toSampleName()} $_name($args)")
addGeneric(_return)
append(" $_name($args)")

if (_hasBody) {
append("{\n")
Expand All @@ -362,12 +384,16 @@ data class FieldBuilder(private val _java: JavaBuilder, private val _name: Strin

private var _final = false

private var _type = ""
private var _type = Pair<String, Array<out String>>("", emptyArray())

private val _annotations = mutableListOf<Pair<String, Array<out Pair<String, Pair<String, String>>>>>()

private var _comment = ""

fun _private() {
this._access = "private"
}

fun _protected() {
this._access = "protected"
}
Expand All @@ -376,10 +402,11 @@ data class FieldBuilder(private val _java: JavaBuilder, private val _name: Strin
this._final = true
}

fun _type(_type: String, _import: Boolean = true) {
this._type = _type
if (_import) {
this._java._import(_type)
fun _type(_type: String, vararg _generic: String) {
this._type = _type to _generic
this._java._import(_type)
_generic.forEach {
this._java._import(it)
}
}

Expand All @@ -401,7 +428,8 @@ data class FieldBuilder(private val _java: JavaBuilder, private val _name: Strin

addAccess(_access)

append("${_type.toSampleName()} $_name")
addGeneric(_type)
append(" $_name")
}
}

Expand All @@ -417,7 +445,8 @@ data class FieldBuilder(private val _java: JavaBuilder, private val _name: Strin
append("final ")
}

append("${_type.toSampleName()} $_name;\n")
addGeneric(_type)
append(" $_name;\n")
}
}
}
Expand Down Expand Up @@ -447,9 +476,8 @@ inline fun _java(_name: String, init: JavaBuilder.() -> Unit): JavaBuilder {
return builder
}

val IGNORE_SAMPLE_NAME_ARRAY = arrayOf("Pages.Args")

fun String.toSampleName() = if (IGNORE_SAMPLE_NAME_ARRAY.contains(this)) {
fun String.toSampleName() = if (IGNORE_CLASS_ARRAY.contains(this)) {
this
} else {
this.substringAfterLast(".")
Expand All @@ -468,6 +496,10 @@ fun String.toGetter(ifIs: Boolean): String {
}
}

fun String.toUnderlineCase(): String {
return replace(Regex("[A-Z]"), "_\$0").lowercase()
}

fun StringBuilder.addComment(_comment: String) {
if (_comment.isNotBlank()) {
append("/**\n")
Expand Down Expand Up @@ -509,14 +541,18 @@ fun StringBuilder.addInterfaces(implementsOrExtends: String, _interfaces: List<P
if (_interfaces.isNotEmpty()) {
append("$implementsOrExtends ")
_interfaces.forEach {
append(it.first.toSampleName())
if (it.second.isNotEmpty()) {
append(it.second.joinToString(",", "<", ">")
{ s -> s.toSampleName() })
}
addGeneric(it)
}
append(" ")
}
}

fun StringBuilder.addGeneric(pair: Pair<String, Array<out String>>) {
append(pair.first.toSampleName())
if (pair.second.isNotEmpty()) {
append(pair.second.joinToString(",", "<", ">")
{ s -> s.toSampleName() })
}
}


Loading

0 comments on commit 105c991

Please sign in to comment.