refactor generateo2o

Author: bigcoder84

object-helper.jar

@@ -27,7 +27,7 @@
   </extensions>
 
   <actions>
-    <action id="ObjectHelper" class="cn.bigcoder.plugin.objecthelper.ObjectHelper" text="generateo2o">
+    <action id="ObjectHelper" class="cn.bigcoder.plugin.objecthelper.GenerateO2O" text="generateo2o">
       <add-to-group group-id="GenerateGroup" anchor="last"/>
     </action>
   </actions>

resources/META-INF/plugin.xml

@@ -0,0 +1,62 @@
+package cn.bigcoder.plugin.objecthelper;
+
+import cn.bigcoder.plugin.objecthelper.generator.Generator;
+import cn.bigcoder.plugin.objecthelper.generator.method.ObjectCopyMethodGenerator;
+import com.intellij.openapi.actionSystem.AnAction;
+import com.intellij.openapi.actionSystem.AnActionEvent;
+import com.intellij.openapi.actionSystem.LangDataKeys;
+import com.intellij.openapi.actionSystem.PlatformDataKeys;
+import com.intellij.openapi.command.WriteCommandAction;
+import com.intellij.openapi.editor.Editor;
+import com.intellij.psi.*;
+import com.intellij.psi.util.PsiTreeUtil;
+
+public class GenerateO2O extends AnAction {
+
+    @Override
+    public void actionPerformed(AnActionEvent e) {
+        PsiMethod method = getPsiMethodFromContext(e);
+        generateO2OMethod(method);
+    }
+
+    /**
+     * 启动写线程
+     *
+     * @param psiMethod
+     */
+    private void generateO2OMethod(PsiMethod psiMethod) {
+        new WriteCommandAction.Simple(psiMethod.getProject(), psiMethod.getContainingFile()) {
+            @Override
+            protected void run() throws Throwable {
+                generateO2O(psiMethod);
+            }
+        }.execute();
+    }
+
+    private void generateO2O(PsiMethod psiMethod) {
+        Generator generator = ObjectCopyMethodGenerator.getInstance(psiMethod);
+        PsiElementFactory elementFactory = JavaPsiFacade.getElementFactory(psiMethod.getProject());
+        PsiMethod toMethod = elementFactory.createMethodFromText(generator.generate(), psiMethod);
+        psiMethod.replace(toMethod);
+    }
+
+    private PsiMethod getPsiMethodFromContext(AnActionEvent e) {
+        PsiElement elementAt = getPsiElement(e);
+        if (elementAt == null) {
+            return null;
+        }
+        return PsiTreeUtil.getParentOfType(elementAt, PsiMethod.class);
+    }
+
+    private PsiElement getPsiElement(AnActionEvent e) {
+        PsiFile psiFile = e.getData(LangDataKeys.PSI_FILE);
+        Editor editor = e.getData(PlatformDataKeys.EDITOR);
+        if (psiFile == null || editor == null) {
+            e.getPresentation().setEnabled(false);
+            return null;
+        }
+        //用来获取当前光标处的PsiElement
+        int offset = editor.getCaretModel().getOffset();
+        return psiFile.findElementAt(offset);
+    }
+}

src/cn/bigcoder/plugin/objecthelper/GenerateO2O.java

@@ -0,0 +1,37 @@
+package cn.bigcoder.plugin.objecthelper.common.enums;
+
+/**
+ * @author: Jindong.Tian
+ * @date: 2021-01-09
+ **/
+public enum JavaModify {
+    PUBLIC("public", 1),
+    PROTECTED("protected", 1),
+    PRIVATE("private", 1),
+    STATIC("static", 2),
+    FINAL("final", 3),
+    ;
+    private String name;
+    private Integer priority;
+
+    JavaModify(String name, Integer priority) {
+        this.name = name;
+        this.priority = priority;
+    }
+
+    public String getName() {
+        return name;
+    }
+
+    public JavaModify nameOf(String modify){
+        if (modify == null){
+            return null;
+        }
+        for (JavaModify item : values()) {
+            if (modify.equals(item.getName())){
+                return item;
+            }
+        }
+        return null;
+    }
+}

src/cn/bigcoder/plugin/objecthelper/ObjectHelper.java

@@ -0,0 +1,61 @@
+package cn.bigcoder.plugin.objecthelper.common.util;
+
+import java.util.Collection;
+import java.util.Locale;
+import java.util.function.Function;
+
+/**
+ * @author: Jindong.Tian
+ * @date: 2021-01-09
+ **/
+public class StringUtils {
+
+    public static String join(Collection<String> items, String separator) {
+        return join(items, Function.identity(), separator);
+    }
+
+    public static <T> String join(Collection<T> items, Function<T, String> handle, String separator) {
+        if (items == null || items.isEmpty()) {
+            return "";
+        }
+        StringBuilder result = new StringBuilder();
+        int index = 0;
+        for (T item : items) {
+            result.append(handle.apply(item));
+            if (index < items.size() - 1) {
+                result.append(separator);
+            }
+            index++;
+        }
+        return result.toString();
+    }
+
+    public static boolean isEmpty(String str) {
+        if (str == null || " ".equals(str)) {
+            return true;
+        }
+        return false;
+    }
+
+    public static boolean isNotEmpty(String str) {
+        return !isEmpty(str);
+    }
+
+    /**
+     * 将首字母转为大写
+     * @param oldStr
+     * @return
+     */
+    public static String firstUpperCase(String oldStr) {
+        return oldStr.substring(0, 1).toUpperCase() + oldStr.substring(1);
+    }
+
+    /**
+     * 将首字母转为小写
+     * @param oldStr
+     * @return
+     */
+    public static String firstLowerCase(String oldStr) {
+        return oldStr.substring(0, 1).toLowerCase(Locale.ROOT) + oldStr.substring(1);
+    }
+}

src/cn/bigcoder/plugin/objecthelper/common/enums/JavaModify.java

@@ -0,0 +1,10 @@
+package cn.bigcoder.plugin.objecthelper.generator;
+
+/**
+ * @author: Jindong.Tian
+ * @date: 2021-01-09
+ * @description:
+ **/
+public interface Generator {
+    String generate();
+}

src/cn/bigcoder/plugin/objecthelper/common/util/StringUtils.java

@@ -0,0 +1,71 @@
+package cn.bigcoder.plugin.objecthelper.generator.method;
+
+import cn.bigcoder.plugin.objecthelper.common.enums.JavaModify;
+import cn.bigcoder.plugin.objecthelper.generator.Generator;
+import cn.bigcoder.plugin.objecthelper.common.util.StringUtils;
+import com.intellij.psi.PsiClass;
+import com.intellij.psi.PsiParameter;
+import org.apache.commons.collections.CollectionUtils;
+
+import java.util.List;
+
+/**
+ * @author: Jindong.Tian
+ * @date: 2021-01-09
+ **/
+public abstract class AbstractMethodGenerator implements Generator {
+
+    /**
+     * 方法修饰符
+     */
+    protected List<JavaModify> methodModifies;
+    /**
+     * 方法返回类型名称
+     */
+    protected String returnClassName;
+    /**
+     * 方法名称
+     */
+    protected String methodName;
+    /**
+     * 方法参数
+     */
+    protected List<PsiParameter> parameters;
+    /**
+     * 方法参数
+     */
+    protected List<PsiClass> parameterClass;
+
+
+    public String generate() {
+        StringBuilder result = generateMethodFirstLine()
+                .append(generateMethodBody())
+                .append("}\n");
+        return result.toString();
+    }
+
+    protected StringBuilder generateMethodFirstLine() {
+        StringBuilder builder = new StringBuilder();
+        methodModifies.forEach(e -> builder.append(e.getName() + " "));
+        builder.append(returnClassName + " ")
+                .append(methodName)
+                .append("(")
+                .append(StringUtils.join(parameters, PsiParameter::getText, " ,"))
+                .append(") {\n");
+        return builder;
+    }
+
+    protected boolean hasModifierProperty(String modify) {
+        if (CollectionUtils.isEmpty(methodModifies)) {
+            return false;
+        }
+        for (JavaModify methodModify : methodModifies) {
+            if (methodModify.getName().equals(modify)) {
+                return true;
+            }
+        }
+        return false;
+    }
+
+    abstract String generateMethodBody();
+}