/*
    *    Copyright 2009-2021 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
    *
    *       http://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.apache.ibatis.submitted.custom_collection_handling;
  
  import java.lang.reflect.Array;
  import java.lang.reflect.Constructor;
  import java.util.Collection;
  import java.util.HashSet;
  import java.util.LinkedHashMap;
  import java.util.LinkedList;
  import java.util.List;
  import java.util.Map;
  import java.util.Set;
  import java.util.SortedSet;
  import java.util.TreeSet;
  
  import org.apache.ibatis.reflection.ReflectionException;
  import org.apache.ibatis.reflection.factory.ObjectFactory;
  
  public class CustomObjectFactory implements ObjectFactory {
  
      @Override
      public <T> T create(Class<T> type) {
          return create(type, null, null);
      }
  
      @Override
      public <T> T create(Class<T> type, List<Class<?>> constructorArgTypes, List<Object> constructorArgs) {
          Class<?> classToCreate = resolveInterface(type);
          @SuppressWarnings("unchecked") // we know types are assignable
          T created = (T) instantiateClass(classToCreate, constructorArgTypes, constructorArgs);
          return created;
      }
  
      private <T> T instantiateClass(Class<T> type, List<Class<?>> constructorArgTypes, List<Object> constructorArgs) {
          try {
              Constructor<T> constructor;
              if (constructorArgTypes == null || constructorArgs == null) {
                  constructor = type.getDeclaredConstructor();
                  if (!constructor.isAccessible()) {
                      constructor.setAccessible(true);
                  }
                  return constructor.newInstance();
              }
              constructor = type.getDeclaredConstructor(constructorArgTypes.toArray(new Class[constructorArgTypes.size()]));
              if (!constructor.isAccessible()) {
                  constructor.setAccessible(true);
              }
              return constructor.newInstance(constructorArgs.toArray(new Object[constructorArgs.size()]));
          } catch (Exception e) {
              StringBuilder argTypes = new StringBuilder();
              if (constructorArgTypes != null) {
                  for (Class<?> argType : constructorArgTypes) {
                      argTypes.append(argType.getSimpleName());
                      argTypes.append(",");
                  }
              }
              StringBuilder argValues = new StringBuilder();
              if (constructorArgs != null) {
                  for (Object argValue : constructorArgs) {
                      argValues.append(String.valueOf(argValue));
                      argValues.append(",");
                  }
              }
              throw new ReflectionException("Error instantiating " + type + " with invalid types (" + argTypes + ") or values (" + argValues + "). Cause: " + e, e);
          }
      }
  
      private Class<?> resolveInterface(Class<?> type) {
          Class<?> classToCreate;
          if (type == List.class || type == Collection.class) {
              classToCreate = LinkedList.class;
          } else if (type == Map.class) {
              classToCreate = LinkedHashMap.class;
          } else if (type == SortedSet.class) { // issue #510 Collections Support
              classToCreate = TreeSet.class;
          } else if (type == Set.class) {
              classToCreate = HashSet.class;
          } else {
              classToCreate = type;
          }
          return classToCreate;
      }
  
      @Override
      public <T> boolean isCollection(Class<T> type) {
       return CustomCollection.class.isAssignableFrom(type);
     }
 
     @SuppressWarnings("unchecked")
     public <T> T[] createArray(Class<T> type, int size) {
       return (T[]) Array.newInstance(type, size);
     }
 
 }