diff --git a/initializer-generator/src/main/java/com/alibaba/initializer/generation/condition/ConditionalOnRequestedModule.java b/initializer-generator/src/main/java/com/alibaba/initializer/generation/condition/ConditionalOnRequestedModule.java new file mode 100644 index 0000000..e7c8f28 --- /dev/null +++ b/initializer-generator/src/main/java/com/alibaba/initializer/generation/condition/ConditionalOnRequestedModule.java @@ -0,0 +1,42 @@ +/* + * Copyright 2022 the original author or authors. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * https://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package com.alibaba.initializer.generation.condition; + +import java.lang.annotation.Documented; +import java.lang.annotation.ElementType; +import java.lang.annotation.Retention; +import java.lang.annotation.RetentionPolicy; +import java.lang.annotation.Target; + +import org.springframework.context.annotation.Conditional; + +/** + * @author Weix Sun + */ +@Retention(RetentionPolicy.RUNTIME) +@Target({ ElementType.TYPE, ElementType.METHOD }) +@Documented +@Conditional(OnRequestedModuleCondition.class) +public @interface ConditionalOnRequestedModule { + + /** + * The identifier of the module. + * @return the module name + */ + String value(); + +} diff --git a/initializer-generator/src/main/java/com/alibaba/initializer/generation/condition/OnRequestedModuleCondition.java b/initializer-generator/src/main/java/com/alibaba/initializer/generation/condition/OnRequestedModuleCondition.java new file mode 100644 index 0000000..073321d --- /dev/null +++ b/initializer-generator/src/main/java/com/alibaba/initializer/generation/condition/OnRequestedModuleCondition.java @@ -0,0 +1,41 @@ +/* + * Copyright 2022 the original author or authors. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * https://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package com.alibaba.initializer.generation.condition; + +import com.alibaba.initializer.metadata.Module; +import io.spring.initializr.generator.condition.ProjectGenerationCondition; +import io.spring.initializr.generator.project.ProjectDescription; +import org.apache.commons.lang3.StringUtils; + +import org.springframework.context.annotation.ConditionContext; +import org.springframework.core.type.AnnotatedTypeMetadata; + +/** + * @author Weix Sun + */ +public class OnRequestedModuleCondition extends ProjectGenerationCondition { + + @Override + protected boolean matches(ProjectDescription description, ConditionContext context, + AnnotatedTypeMetadata metadata) { + Module module = context.getBeanFactory().getBean(Module.class); + String requestModuleName = module.getName() == null && module.isRoot() ? "root" : module.getName(); + String annotatedModuleName = (String) metadata.getAnnotationAttributes( + ConditionalOnRequestedModule.class.getName()).get("value"); + return StringUtils.equals(requestModuleName, annotatedModuleName); + } +}