/*
    *    Copyright 2006-2025 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 org.mybatis.generator.api.dom.java;
  
  import static org.mybatis.generator.internal.util.StringUtility.stringHasValue;
  import static org.mybatis.generator.internal.util.messages.Messages.getString;
  
  import java.util.ArrayList;
  import java.util.List;
  import java.util.Objects;
  import java.util.StringTokenizer;
  import java.util.stream.Collectors;
  
  public class FullyQualifiedJavaType implements Comparable<FullyQualifiedJavaType> {
  
      private static final String JAVA_LANG = "java.lang"; //$NON-NLS-1$
  
      private static FullyQualifiedJavaType intInstance = null;
  
      private static FullyQualifiedJavaType stringInstance = null;
  
      private static FullyQualifiedJavaType booleanPrimitiveInstance = null;
  
      private static FullyQualifiedJavaType objectInstance = null;
  
      private static FullyQualifiedJavaType dateInstance = null;
  
      private static FullyQualifiedJavaType criteriaInstance = null;
  
      private static FullyQualifiedJavaType generatedCriteriaInstance = null;
  
      /** The short name without any generic arguments. */
      private String baseShortName;
  
      /** The fully qualified name without any generic arguments. */
      private String baseQualifiedName;
  
      private boolean explicitlyImported;
  
      private String packageName;
  
      private boolean primitive;
  
      private boolean isArray;
  
      private PrimitiveTypeWrapper primitiveTypeWrapper;
  
      private final List<FullyQualifiedJavaType> typeArguments;
  
      // the following three values are used for dealing with wildcard types
      private boolean wildcardType;
  
      private boolean boundedWildcard;
  
      private boolean extendsBoundedWildcard;
  
      /**
       * Use this constructor to construct a generic type with the specified type parameters.
       *
       * @param fullTypeSpecification
       *            the full type specification
       */
      public FullyQualifiedJavaType(String fullTypeSpecification) {
          super();
          typeArguments = new ArrayList<>();
          parse(fullTypeSpecification);
      }
  
      public boolean isExplicitlyImported() {
          return explicitlyImported;
      }
  
      /**
       * Returns the fully qualified name - including any generic type parameters.
       *
       * @return Returns the fullyQualifiedName.
       */
      public String getFullyQualifiedName() {
          String s = getFullyQualifiedNameWithoutTypeParameters();
  
          if (typeArguments.isEmpty()) {
              return s;
          } else {
              return typeArguments.stream()
                      .map(FullyQualifiedJavaType::getFullyQualifiedName)
                      .collect(Collectors.joining(", ", s + "<", ">")); //$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$
         }
     }
 
     public String getFullyQualifiedNameWithoutTypeParameters() {
         return calculateBaseType(baseQualifiedName);
     }
 
     /**
      * The name (fully qualified) that should be imported.
      *
      * @return the fully qualified name that should be imported. Does not include the wildcard bounds.
      */
     public String getImportName() {
         return baseQualifiedName;
     }
 
     /**
      * Returns a list of Strings that are the fully qualified names of this type, and any generic type argument
      * associated with this type.
      *
      * @return the import list
      */
     public List<String> getImportList() {
         List<String> answer = new ArrayList<>();
         if (isExplicitlyImported()) {
             int index = baseShortName.indexOf('.');
             if (index == -1) {
                 answer.add(calculateActualImport(baseQualifiedName));
             } else {
                 // an inner class is specified, only import the top
                 // level class
                 String sb = packageName + '.' + calculateActualImport(baseShortName.substring(0, index));
                 answer.add(sb);
             }
         }
 
         typeArguments.forEach(t -> answer.addAll(t.getImportList()));
 
         return answer;
     }
 
     private String calculateActualImport(String name) {
         String answer = name;
         if (this.isArray()) {
             int index = name.indexOf('[');
             if (index != -1) {
                 answer = name.substring(0, index);
             }
         }
         return answer;
     }
 
     public String getPackageName() {
         return packageName;
     }
 
     public String getShortName() {
         String s = getShortNameWithoutTypeArguments();
 
         if (typeArguments.isEmpty()) {
             return s;
         } else {
             return typeArguments.stream()
                     .map(FullyQualifiedJavaType::getShortName)
                     .collect(Collectors.joining(", ", s + "<", ">")); //$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$
         }
     }
 
     public String getShortNameWithoutTypeArguments() {
         return calculateBaseType(baseShortName);
     }
 
     private String calculateBaseType(String name) {
         StringBuilder sb = new StringBuilder();
         if (wildcardType) {
             sb.append('?');
             if (boundedWildcard) {
                 if (extendsBoundedWildcard) {
                     sb.append(" extends "); //$NON-NLS-1$
                 } else {
                     sb.append(" super "); //$NON-NLS-1$
                 }
 
                 sb.append(name);
             }
         } else {
             sb.append(name);
         }
         return sb.toString();
     }
 
     @Override
     public boolean equals(Object obj) {
         if (this == obj) {
             return true;
         }
 
         if (!(obj instanceof FullyQualifiedJavaType other)) {
             return false;
         }
 
         return getFullyQualifiedName().equals(other.getFullyQualifiedName());
     }
 
     @Override
     public int hashCode() {
         return getFullyQualifiedName().hashCode();
     }
 
     @Override
     public String toString() {
         return getFullyQualifiedName();
     }
 
     public boolean isPrimitive() {
         return primitive;
     }
 
     public PrimitiveTypeWrapper getPrimitiveTypeWrapper() {
         return primitiveTypeWrapper;
     }
 
     public static FullyQualifiedJavaType getIntInstance() {
         if (intInstance == null) {
             intInstance = new FullyQualifiedJavaType("int"); //$NON-NLS-1$
         }
 
         return intInstance;
     }
 
     public static FullyQualifiedJavaType getNewListInstance() {
         // always return a new instance because the type may be parameterized
         return new FullyQualifiedJavaType("java.util.List"); //$NON-NLS-1$
     }
 
     public static FullyQualifiedJavaType getNewHashMapInstance() {
         // always return a new instance because the type may be parameterized
         return new FullyQualifiedJavaType("java.util.HashMap"); //$NON-NLS-1$
     }
 
     public static FullyQualifiedJavaType getNewArrayListInstance() {
         // always return a new instance because the type may be parameterized
         return new FullyQualifiedJavaType("java.util.ArrayList"); //$NON-NLS-1$
     }
 
     public static FullyQualifiedJavaType getNewIteratorInstance() {
         // always return a new instance because the type may be parameterized
         return new FullyQualifiedJavaType("java.util.Iterator"); //$NON-NLS-1$
     }
 
     public static FullyQualifiedJavaType getStringInstance() {
         stringInstance = Objects.requireNonNullElseGet(stringInstance,
                 () -> new FullyQualifiedJavaType("java.lang.String")); //$NON-NLS-1$
         return stringInstance;
     }
 
     public static FullyQualifiedJavaType getBooleanPrimitiveInstance() {
         booleanPrimitiveInstance = Objects.requireNonNullElseGet(booleanPrimitiveInstance,
                 () -> new FullyQualifiedJavaType("boolean")); //$NON-NLS-1$
         return booleanPrimitiveInstance;
     }
 
     public static FullyQualifiedJavaType getObjectInstance() {
         objectInstance = Objects.requireNonNullElseGet(objectInstance,
                 () -> new FullyQualifiedJavaType("java.lang.Object")); //$NON-NLS-1$
         return objectInstance;
     }
 
     public static FullyQualifiedJavaType getDateInstance() {
         dateInstance = Objects.requireNonNullElseGet(dateInstance,
                 () -> new FullyQualifiedJavaType("java.util.Date")); //$NON-NLS-1$
         return dateInstance;
     }
 
     public static FullyQualifiedJavaType getCriteriaInstance() {
         criteriaInstance = Objects.requireNonNullElseGet(criteriaInstance,
                 () -> new FullyQualifiedJavaType("Criteria")); //$NON-NLS-1$
         return criteriaInstance;
     }
 
     public static FullyQualifiedJavaType getGeneratedCriteriaInstance() {
         generatedCriteriaInstance = Objects.requireNonNullElseGet(generatedCriteriaInstance,
                 () -> new FullyQualifiedJavaType("GeneratedCriteria")); //$NON-NLS-1$
         return generatedCriteriaInstance;
     }
 
     @Override
     public int compareTo(FullyQualifiedJavaType other) {
         return getFullyQualifiedName().compareTo(other.getFullyQualifiedName());
     }
 
     public void addTypeArgument(FullyQualifiedJavaType type) {
         typeArguments.add(type);
     }
 
     private void parse(String fullTypeSpecification) {
         String spec = fullTypeSpecification.trim();
 
         if (spec.startsWith("?")) { //$NON-NLS-1$
             wildcardType = true;
             spec = spec.substring(1).trim();
             if (spec.startsWith("extends ")) { //$NON-NLS-1$
                 boundedWildcard = true;
                 extendsBoundedWildcard = true;
                 spec = spec.substring(8); // "extends ".length()
             } else if (spec.startsWith("super ")) { //$NON-NLS-1$
                 boundedWildcard = true;
                 extendsBoundedWildcard = false;
                 spec = spec.substring(6); // "super ".length()
             } else {
                 boundedWildcard = false;
             }
             parse(spec);
         } else {
             int index = fullTypeSpecification.indexOf('<');
             if (index == -1) {
                 simpleParse(fullTypeSpecification);
             } else {
                 simpleParse(fullTypeSpecification.substring(0, index));
                 int endIndex = fullTypeSpecification.lastIndexOf('>');
                 if (endIndex == -1) {
                     throw new RuntimeException(getString(
                             "RuntimeError.22", fullTypeSpecification)); //$NON-NLS-1$
                 }
                 genericParse(fullTypeSpecification.substring(index, endIndex + 1));
             }
 
             // this is far from a perfect test for detecting arrays, but is close
             // enough for most cases. It will not detect an improperly specified
             // array type like byte], but it will detect byte[] and byte[ ]
             // which are both valid
             isArray = fullTypeSpecification.endsWith("]"); //$NON-NLS-1$
         }
     }
 
     private void simpleParse(String typeSpecification) {
         baseQualifiedName = typeSpecification.trim();
         if (baseQualifiedName.contains(".")) { //$NON-NLS-1$
             packageName = getPackage(baseQualifiedName);
             baseShortName = baseQualifiedName.substring(packageName.length() + 1);
             int index = baseShortName.lastIndexOf('.');
             if (index != -1) {
                 baseShortName = baseShortName.substring(index + 1);
             }
 
             //$NON-NLS-1$
             explicitlyImported = !JAVA_LANG.equals(packageName);
         } else {
             baseShortName = baseQualifiedName;
             explicitlyImported = false;
             packageName = ""; //$NON-NLS-1$
 
             switch (baseQualifiedName) {
             case "byte":  //$NON-NLS-1$
                 primitive = true;
                 primitiveTypeWrapper = PrimitiveTypeWrapper.getByteInstance();
                 break;
             case "short":  //$NON-NLS-1$
                 primitive = true;
                 primitiveTypeWrapper = PrimitiveTypeWrapper.getShortInstance();
                 break;
             case "int":  //$NON-NLS-1$
                 primitive = true;
                 primitiveTypeWrapper = PrimitiveTypeWrapper.getIntegerInstance();
                 break;
             case "long":  //$NON-NLS-1$
                 primitive = true;
                 primitiveTypeWrapper = PrimitiveTypeWrapper.getLongInstance();
                 break;
             case "char":  //$NON-NLS-1$
                 primitive = true;
                 primitiveTypeWrapper = PrimitiveTypeWrapper.getCharacterInstance();
                 break;
             case "float":  //$NON-NLS-1$
                 primitive = true;
                 primitiveTypeWrapper = PrimitiveTypeWrapper.getFloatInstance();
                 break;
             case "double":  //$NON-NLS-1$
                 primitive = true;
                 primitiveTypeWrapper = PrimitiveTypeWrapper.getDoubleInstance();
                 break;
             case "boolean":  //$NON-NLS-1$
                 primitive = true;
                 primitiveTypeWrapper = PrimitiveTypeWrapper.getBooleanInstance();
                 break;
             default:
                 primitive = false;
                 primitiveTypeWrapper = null;
                 break;
             }
         }
     }
 
     private void genericParse(String genericSpecification) {
         int lastIndex = genericSpecification.lastIndexOf('>');
         if (lastIndex == -1) {
             // shouldn't happen - should be caught already, but just in case...
             throw new RuntimeException(getString(
                     "RuntimeError.22", genericSpecification)); //$NON-NLS-1$
         }
         String argumentString = genericSpecification.substring(1, lastIndex);
         // need to find "," outside a <> bounds
         StringTokenizer st = new StringTokenizer(argumentString, ",<>", true); //$NON-NLS-1$
         int openCount = 0;
         StringBuilder sb = new StringBuilder();
         while (st.hasMoreTokens()) {
             String token = st.nextToken();
             if ("<".equals(token)) { //$NON-NLS-1$
                 sb.append(token);
                 openCount++;
             } else if (">".equals(token)) { //$NON-NLS-1$
                 sb.append(token);
                 openCount--;
             } else if (",".equals(token)) { //$NON-NLS-1$
                 if (openCount == 0) {
                     typeArguments
                             .add(new FullyQualifiedJavaType(sb.toString()));
                     sb.setLength(0);
                 } else {
                     sb.append(token);
                 }
             } else {
                 sb.append(token);
             }
         }
 
         if (openCount != 0) {
             throw new RuntimeException(getString(
                     "RuntimeError.22", genericSpecification)); //$NON-NLS-1$
         }
 
         String finalType = sb.toString();
         if (stringHasValue(finalType)) {
             typeArguments.add(new FullyQualifiedJavaType(finalType));
         }
     }
 
     /**
      * Returns the package name of a fully qualified type.
      *
      * <p>This method calculates the package as the part of the fully qualified name up to, but not including, the last
      * element. Therefore, it does not support fully qualified inner classes. Not totally foolproof, but correct in
      * most instances.
      *
      * @param baseQualifiedName
      *            the base qualified name
      *
      * @return the package
      */
     private static String getPackage(String baseQualifiedName) {
         int index = baseQualifiedName.lastIndexOf('.');
         return baseQualifiedName.substring(0, index);
     }
 
     public boolean isArray() {
         return isArray;
     }
 
     public List<FullyQualifiedJavaType> getTypeArguments() {
         return typeArguments;
     }
 }