From 8934d8c0e360364bca37270a5638f31152503e89 Mon Sep 17 00:00:00 2001
From: liuzhihang <liuzhihangs@qq.com>
Date: Thu, 19 Aug 2021 20:42:52 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E6=B3=9B=E5=9E=8B=E6=95=B0?=
 =?UTF-8?q?=E6=8D=AE=E8=8E=B7=E5=8F=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 build.gradle                                  |  2 +-
 parts/changeNotes.html                        |  6 +-
 .../doc/view/utils/ParamPsiUtils.java         | 85 ++++++++++++++++---
 3 files changed, 77 insertions(+), 16 deletions(-)

diff --git a/build.gradle b/build.gradle
index 2e19a83..9b2e591 100644
--- a/build.gradle
+++ b/build.gradle
@@ -4,7 +4,7 @@ plugins {
 }
 
 group 'com.liuzhihang'
-version '1.1.10'
+version '1.1.11'
 
 JavaVersion.VERSION_11
 
diff --git a/parts/changeNotes.html b/parts/changeNotes.html
index cc0dec7..c768761 100644
--- a/parts/changeNotes.html
+++ b/parts/changeNotes.html
@@ -3,7 +3,7 @@
 
 <h4>English introduction</h4>
 <ul>
-    <li>1.1.10
+    <li>1.1.11
         <ol>
             <li>Optimize generic data acquisition</li>
         </ol>
@@ -12,9 +12,9 @@ <h4>English introduction</h4>
 </ul>
 <h4>中文介绍</h4>
 <ul>
-    <li>1.1.10
+    <li>1.1.11
         <ol>
-            <li>优化泛型数据获取</li>
+            <li>优化泛型嵌套数据获取</li>
             <li>备注:当前版本使用 swingx-core 依赖生成 TreeTable, 导致插件包占用空间变大, 需要后期调整, 具体可参考:
                 https://youtrack.jetbrains.com/issue/IDEA-275249 ,如果小伙伴有其他解决方式, 希望可以联系我,或者提 Issue.
             </li>
diff --git a/src/main/java/com/liuzhihang/doc/view/utils/ParamPsiUtils.java b/src/main/java/com/liuzhihang/doc/view/utils/ParamPsiUtils.java
index 05f92f9..c7da769 100644
--- a/src/main/java/com/liuzhihang/doc/view/utils/ParamPsiUtils.java
+++ b/src/main/java/com/liuzhihang/doc/view/utils/ParamPsiUtils.java
@@ -1,6 +1,7 @@
 package com.liuzhihang.doc.view.utils;
 
 import com.intellij.psi.*;
+import com.intellij.psi.impl.source.PsiClassReferenceType;
 import com.intellij.psi.util.InheritanceUtil;
 import com.intellij.psi.util.PsiTypesUtil;
 import com.intellij.psi.util.PsiUtil;
@@ -30,11 +31,6 @@ public static void buildBodyParam(PsiField field, Map<String, PsiType> genericsM
 
         PsiType type = field.getType();
 
-        // 如果是泛型, 且泛型字段是当前字段, 将当前字段类型替换为泛型类型
-        if (genericsMap != null && genericsMap.containsKey(type.getPresentableText())) {
-            type = genericsMap.get(type.getPresentableText());
-        }
-
         Body body = new Body();
         body.setRequired(DocViewUtils.isRequired(field));
         body.setName(field.getName());
@@ -47,6 +43,12 @@ public static void buildBodyParam(PsiField field, Map<String, PsiType> genericsM
         if (type instanceof PsiPrimitiveType || FieldTypeConstant.FIELD_TYPE.containsKey(type.getPresentableText())) {
             return;
         }
+
+        // 提前替换字段比如 T -> UserDTO   List<T> -> List<UserDTO>
+        // 如果是泛型, 且泛型字段是当前字段, 将当前字段类型替换为泛型类型, 替换完之后重新设置 body 的 type
+        type = replaceFieldType(genericsMap, type);
+        body.setType(type.getPresentableText());
+
         // 剩下都是 PsiClass 类型处理
         PsiClass fieldClass = PsiUtil.resolveClassInClassTypeOnly(type);
 
@@ -79,7 +81,7 @@ public static void buildBodyParam(PsiField field, Map<String, PsiType> genericsM
             }
             // 集合参数构建, 集合就一个参数, 泛型 E
             fieldGenericsMap = CustomPsiUtils.getGenericsMap((PsiClassType) iterableType);
-            parentBody = buildFieldGenericsBody("E", childClass, body);
+            parentBody = buildFieldGenericsBody("element", childClass, body);
 
 
         } else if (InheritanceUtil.isInheritor(type, CommonClassNames.JAVA_UTIL_MAP)) {
@@ -131,6 +133,68 @@ public static void buildBodyParam(PsiField field, Map<String, PsiType> genericsM
 
     }
 
+    /**
+     * 判断当前字段是否含有泛型, 从泛型映射表中替换字段类型
+     * <p>
+     * 替换字段比如 T -> UserDTO   List<T> -> List<UserDTO>
+     *
+     * @param genericsMap 含有泛型的 Map
+     * @param type        当前字段的类型
+     * @return
+     */
+    private static PsiType replaceFieldType(Map<String, PsiType> genericsMap, PsiType type) {
+
+        if (genericsMap == null || genericsMap.isEmpty()) {
+            return type;
+        }
+
+        if (!(type instanceof PsiClassType)) {
+            return type;
+        }
+
+        if (!(type instanceof PsiClassReferenceType)) {
+            return type;
+        }
+
+        // 当前字段的类型就是泛型 private T data;
+        if (genericsMap.containsKey(type.getPresentableText())) {
+            return genericsMap.get(type.getPresentableText());
+        }
+
+        // 当前字段含有泛型 List<T> -> List<UserDTO>
+        PsiClass fieldClass = PsiUtil.resolveClassInClassTypeOnly(type);
+
+        if (fieldClass == null || !fieldClass.hasTypeParameters()) {
+            return type;
+        }
+
+        PsiType[] parameters = ((PsiClassType) type).getParameters();
+
+        List<PsiType> psiTypeList = new ArrayList<>();
+        // 替换泛型为实际类型
+        for (PsiType psiType : parameters) {
+
+            if (genericsMap.get(psiType.getPresentableText()) == null) {
+                psiTypeList.add(psiType);
+            } else {
+                psiTypeList.add(genericsMap.get(psiType.getPresentableText()));
+            }
+        }
+
+        if (!psiTypeList.isEmpty()) {
+
+            PsiType[] psiTypes = new PsiType[psiTypeList.size()];
+
+            for (int i = 0; i < psiTypeList.size(); i++) {
+                psiTypes[i] = psiTypeList.get(i);
+            }
+
+            return PsiElementFactory.getInstance(fieldClass.getProject()).createType(fieldClass, psiTypes);
+        }
+
+        return type;
+    }
+
     @NotNull
     private static Body buildFieldGenericsBody(String name, PsiClass GenericsClass, Body parent) {
         Body listBody = new Body();
@@ -201,18 +265,15 @@ public static Map<String, Object> getFieldsAndDefaultValue(PsiClass psiClass, Ma
             PsiType type = field.getType();
             String name = field.getName();
 
-            // 如果是泛型, 且泛型字段是当前字段, 将当前字段类型替换为泛型类型
-            if (genericMap != null && genericMap.containsKey(type.getPresentableText())) {
-                type = genericMap.get(type.getPresentableText());
-            }
-
-
             if (type instanceof PsiPrimitiveType) {
                 // 基本类型
                 fieldMap.put(name, PsiTypesUtil.getDefaultValue(type));
                 continue;
             }
 
+            // 如果是泛型, 且泛型字段是当前字段, 将当前字段类型替换为泛型类型
+            type = replaceFieldType(genericMap, type);
+
             // 引用类型
             String fieldTypeName = type.getPresentableText();
             // 指定的类型