630: Added default values support for all types of plugins

Author: bohdan-harniuk

src/com/magento/idea/magento2plugin/actions/generation/generator/code/util/ConvertPluginParamsToString.java

@@ -66,7 +66,19 @@ public static String execute(
             if (element instanceof Parameter) {
                 String parameterText = PhpCodeUtil.paramToString(element);
 
-                if (parameterText.contains(Package.fqnSeparator)) {
+                // Parameter has default value.
+                if (parameterText.contains("=")) {
+                    final String[] paramParts = parameterText.split("=");
+                    parameterText = paramParts[0];
+                    parameterText += " = ";
+                    String defaultValue = paramParts[1];
+
+                    if (defaultValue.contains(Package.fqnSeparator)) {
+                        final String[] fqnArray = defaultValue.split("\\\\");
+                        defaultValue = fqnArray[fqnArray.length - 1];
+                    }
+                    parameterText += defaultValue;
+                } else if (parameterText.contains(Package.fqnSeparator)) {
                     final String[] fqnArray = parameterText.split("\\\\");
                     parameterText = fqnArray[fqnArray.length - 1];
                 }

src/com/magento/idea/magento2plugin/actions/generation/references/PhpClassReferenceExtractor.java

@@ -9,6 +9,9 @@
 import com.jetbrains.php.lang.psi.elements.Parameter;
 import com.jetbrains.php.lang.psi.elements.PhpClass;
 import com.jetbrains.php.lang.psi.elements.PhpReturnType;
+import com.jetbrains.php.lang.psi.elements.impl.ClassConstImpl;
+import com.jetbrains.php.lang.psi.elements.impl.ClassConstantReferenceImpl;
+import com.jetbrains.php.lang.psi.elements.impl.ParameterImpl;
 import com.jetbrains.php.lang.psi.resolve.types.PhpType;
 import com.jetbrains.php.lang.psi.visitors.PhpElementVisitor;
 import java.util.Collection;
@@ -39,6 +42,7 @@ public void visitPhpClass(final PhpClass phpClass) {
         @Override
         public void visitPhpParameter(final Parameter parameter) {
             PhpClassReferenceExtractor.this.processParameterReference(parameter);
+            PhpClassReferenceExtractor.this.processUsedForDefaultValueTypeReference(parameter);
         }
 
         /**
@@ -77,6 +81,39 @@ public void processParameterReference(final @NotNull Parameter parameter) {
         this.processReference(getNameFromFqn(complexType), complexType, parameter);
     }
 
+    /**
+     * Process reference for complex type if it is exists in the default parameter value.
+     *
+     * @param parameter Parameter
+     */
+    public void processUsedForDefaultValueTypeReference(final @NotNull Parameter parameter) {
+        if (!(parameter instanceof ParameterImpl)) {
+            return;
+        }
+        final PsiElement defaultValue = parameter.getDefaultValue();
+
+        if (!(defaultValue instanceof ClassConstantReferenceImpl)) {
+            return;
+        }
+        final PsiElement constant = ((ClassConstantReferenceImpl) defaultValue).resolve();
+
+        if (!(constant instanceof ClassConstImpl)) {
+            return;
+        }
+        final PhpClass usedTypeClass = ((ClassConstImpl) constant).getContainingClass();
+
+        if (usedTypeClass == null) {
+            return;
+        }
+        final String complexType = extractComplexType(usedTypeClass.getType());
+
+        if (complexType == null) {
+            return;
+        }
+        
+        this.processReference(getNameFromFqn(complexType), complexType, parameter);
+    }
+
     /**
      * Process reference for non primitive return type.
      *

src/com/magento/idea/magento2plugin/actions/generation/references/PhpClassReferenceResolver.java

@@ -142,22 +142,22 @@ private Map<String, String> importWithConflictResolve(
                 referencesToReplace.put(name, originalFqn);
             } else {
                 final String importedFqn = aliases.get(name);
-                if (!PhpLangUtil.equalsClassNames(importedFqn, originalFqn)) {
-                    if (importedFqn != null) {
-                        final String originalName = PhpLangUtil.toShortName(originalFqn);
-                        final String fqnForOriginalName = aliases.get(originalName);
-                        if (fqnForOriginalName != null
-                                && !PhpLangUtil.equalsClassNames(fqnForOriginalName, originalFqn)) {
-                            referencesToReplace.put(name, originalFqn);
-                        } else {
-                            referencesToReplace.put(name, originalName);
-                            if (fqnForOriginalName == null) {
-                                insertUseStatement(scopeHolder, originalName, originalFqn);
-                            }
-                        }
+
+                if (importedFqn != null
+                        && !PhpLangUtil.equalsClassNames(importedFqn, originalFqn)) {
+                    final String originalName = PhpLangUtil.toShortName(originalFqn);
+                    final String fqnForOriginalName = aliases.get(originalName);
+                    if (fqnForOriginalName != null
+                            && !PhpLangUtil.equalsClassNames(fqnForOriginalName, originalFqn)) {
+                        referencesToReplace.put(name, originalFqn);
                     } else {
-                        insertUseStatement(scopeHolder, name, originalFqn);
+                        referencesToReplace.put(name, originalName);
+                        if (fqnForOriginalName == null) {
+                            insertUseStatement(scopeHolder, originalName, originalFqn);
+                        }
                     }
+                } else {
+                    insertUseStatement(scopeHolder, name, originalFqn);
                 }
             }
         }