// Copyright 2000-2025 JetBrains s.r.o. and contributors. Use of this source code is governed by the Apache 2.0 license.
package com.intellij.codeInspection.dataFlow;

import com.intellij.codeInsight.Nullability;
import com.intellij.codeInspection.dataFlow.jvm.SpecialField;
import com.intellij.codeInspection.dataFlow.memory.DfaMemoryState;
import com.intellij.codeInspection.dataFlow.rangeSet.LongRangeBinOp;
import com.intellij.codeInspection.dataFlow.rangeSet.LongRangeSet;
import com.intellij.codeInspection.dataFlow.rangeSet.LongRangeType;
import com.intellij.codeInspection.dataFlow.types.DfIntType;
import com.intellij.codeInspection.dataFlow.types.DfLongType;
import com.intellij.codeInspection.dataFlow.types.DfType;
import com.intellij.codeInspection.dataFlow.value.DfaBinOpValue;
import com.intellij.codeInspection.dataFlow.value.DfaValue;
import com.intellij.codeInspection.dataFlow.value.DfaValueFactory;
import com.intellij.codeInspection.dataFlow.value.RelationType;
import com.intellij.codeInspection.util.ChronoUtil;
import com.intellij.codeInspection.util.OptionalUtil;
import com.intellij.psi.*;
import com.intellij.psi.util.*;
import com.intellij.util.ArrayUtil;
import com.intellij.util.ReflectionUtil;
import com.siyeh.ig.callMatcher.CallMapper;
import com.siyeh.ig.callMatcher.CallMatcher;
import com.siyeh.ig.psiutils.TypeUtils;
import org.jetbrains.annotations.Contract;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.time.temporal.ChronoField;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.List;
import java.util.Locale;

import static com.intellij.codeInspection.dataFlow.jvm.SpecialField.*;
import static com.intellij.codeInspection.dataFlow.types.DfTypes.*;
import static com.intellij.psi.CommonClassNames.*;
import static com.siyeh.ig.callMatcher.CallMatcher.*;

public final class CustomMethodHandlers {
  private static final CallMatcher CONSTANT_CALLS = anyOf(
    exactInstanceCall(JAVA_LANG_STRING, "contains", "indexOf", "startsWith", "endsWith", "lastIndexOf", "length", "trim",
                      "substring", "equals", "equalsIgnoreCase", "charAt", "codePointAt", "compareTo", "replace", "strip",
                      "stripLeading", "stripTrailing", "isBlank"),
    staticCall(JAVA_LANG_STRING, "valueOf").parameterCount(1),
    staticCall(JAVA_LANG_MATH, "abs", "sqrt", "min", "max", "addExact", "absExact", "subtractExact", "multiplyExact",
               "incrementExact", "decrementExact", "toIntExact", "negateExact", "sin", "cos", "tan", "asin", "acos", "atan", "cbrt",
               "hypot", "exp", "pow", "log", "log10"),
    staticCall(JAVA_LANG_STRICT_MATH, "abs", "sqrt", "min", "max", "addExact", "absExact", "subtractExact", "multiplyExact",
               "incrementExact", "decrementExact", "toIntExact", "negateExact", "sin", "cos", "tan", "asin", "acos", "atan", "cbrt",
               "hypot", "exp", "pow", "log", "log10"),
    staticCall(JAVA_LANG_INTEGER, "toString", "toBinaryString", "toHexString", "toOctalString", "toUnsignedString").parameterTypes("int"),
    staticCall(JAVA_LANG_LONG, "toString", "toBinaryString", "toHexString", "toOctalString", "toUnsignedString").parameterTypes("long"),
    staticCall(JAVA_LANG_DOUBLE, "toString", "toHexString").parameterTypes("double"),
    staticCall(JAVA_LANG_FLOAT, "toString", "toHexString").parameterTypes("float"),
    staticCall(JAVA_LANG_BYTE, "toString").parameterTypes("byte"),
    staticCall(JAVA_LANG_SHORT, "toString").parameterTypes("short"),
    staticCall(JAVA_LANG_BOOLEAN, "parseBoolean").parameterTypes("java.lang.String"),
    staticCall(JAVA_LANG_INTEGER, "parseInt").parameterTypes("java.lang.String"),
    staticCall(JAVA_LANG_LONG, "parseLong").parameterTypes("java.lang.String"),
    staticCall(JAVA_LANG_DOUBLE, "parseDouble").parameterTypes("java.lang.String"),
    staticCall(JAVA_LANG_FLOAT, "parseFloat").parameterTypes("java.lang.String"),
    staticCall(JAVA_LANG_BYTE, "parseByte").parameterTypes("java.lang.String"),
    staticCall(JAVA_LANG_SHORT, "parseShort").parameterTypes("java.lang.String"),
    staticCall(JAVA_LANG_INTEGER, "compare", "compareUnsigned").parameterTypes("int", "int"),
    staticCall(JAVA_LANG_LONG, "compare", "compareUnsigned").parameterTypes("long", "long"),
    staticCall(JAVA_LANG_DOUBLE, "compare").parameterTypes("double", "double"),
    staticCall(JAVA_LANG_FLOAT, "compare").parameterTypes("float", "float"),
    staticCall(JAVA_LANG_BYTE, "compare", "compareUnsigned").parameterTypes("byte", "byte"),
    staticCall(JAVA_LANG_SHORT, "compare", "compareUnsigned").parameterTypes("short", "short"),
    staticCall(JAVA_LANG_BOOLEAN, "compare").parameterTypes("boolean", "boolean"),
    exactInstanceCall(JAVA_LANG_INTEGER, "toString").parameterCount(0),
    exactInstanceCall(JAVA_LANG_LONG, "toString").parameterCount(0),
    exactInstanceCall(JAVA_LANG_DOUBLE, "toString").parameterCount(0),
    exactInstanceCall(JAVA_LANG_FLOAT, "toString").parameterCount(0),
    exactInstanceCall(JAVA_LANG_BYTE, "toString").parameterCount(0),
    exactInstanceCall(JAVA_LANG_SHORT, "toString").parameterCount(0),
    exactInstanceCall(JAVA_LANG_CHARACTER, "toString").parameterCount(0),
    exactInstanceCall(JAVA_LANG_BOOLEAN, "toString").parameterCount(0)
  );
  public static final int MAX_STRING_CONSTANT_LENGTH_TO_TRACK = 256;

  public interface CustomMethodHandler {

    @Nullable
    DfaValue getMethodResultValue(DfaCallArguments callArguments,
                                  DfaMemoryState memState,
                                  DfaValueFactory factory,
                                  PsiMethod method);

    default CustomMethodHandler compose(CustomMethodHandler other) {
      if (other == null) return this;
      return (args, memState, factory, method) -> {
        DfaValue result = this.getMethodResultValue(args, memState, factory, method);
        return result == null ? other.getMethodResultValue(args, memState, factory, method) : result;
      };
    }
  }

  interface DfTypeCustomMethodHandler extends CustomMethodHandler {
    @NotNull
    DfType getMethodResult(DfaCallArguments callArguments,
                           DfaMemoryState memState,
                           DfaValueFactory factory,
                           PsiMethod method);

    @Override
    default @Nullable DfaValue getMethodResultValue(DfaCallArguments callArguments,
                                                    DfaMemoryState memState,
                                                    DfaValueFactory factory,
                                                    PsiMethod method) {
      DfType dfType = getMethodResult(callArguments, memState, factory, method);
      return dfType == DfType.TOP ? null : factory.fromDfType(dfType);
    }
  }

  private static CustomMethodHandler toValue(DfTypeCustomMethodHandler handler) {
    return handler;
  }

  private static final CallMapper<CustomMethodHandler> CUSTOM_METHOD_HANDLERS = new CallMapper<CustomMethodHandler>()
    .register(instanceCall(JAVA_LANG_STRING, "indexOf", "lastIndexOf"),
              toValue((args, memState, factory, method) -> indexOf(args.myQualifier, memState, factory, STRING_LENGTH)))
    .register(instanceCall(JAVA_LANG_STRING, "trim", "strip", "stripLeading", "stripTrailing"),
              toValue((args, memState, factory, method) -> shorterOrEqualString(args.myQualifier, memState, factory)))
    .register(instanceCall(JAVA_UTIL_LIST, "indexOf", "lastIndexOf"),
              toValue((args, memState, factory, method) -> indexOf(args.myQualifier, memState, factory, COLLECTION_SIZE)))
    .register(staticCall(JAVA_LANG_MATH, "abs").parameterTypes("int"),
              toValue((args, memState, factory, method) -> mathAbs(args.myArguments, memState, false)))
    .register(staticCall(JAVA_LANG_MATH, "abs").parameterTypes("long"),
              toValue((args, memState, factory, method) -> mathAbs(args.myArguments, memState, true)))
    .register(exactInstanceCall(JAVA_LANG_STRING, "substring"),
              (args, memState, factory, method) -> substring(args, memState, factory, method.getReturnType()))
    .register(OptionalUtil.OPTIONAL_OF_NULLABLE,
              toValue((args, memState, factory, method) -> OPTIONAL_VALUE.asDfType(memState.getDfType(args.myArguments[0]))))
    .register(instanceCall(JAVA_UTIL_CALENDAR, "get").parameterTypes("int"),
              toValue((args, memState, factory, method) -> calendarGet(args, memState, factory)))
    .register(anyOf(instanceCall(JAVA_TIME_LOCAL_DATE, "get", "getLong").parameterTypes("java.time.temporal.TemporalField"),
                    instanceCall(JAVA_TIME_LOCAL_TIME, "get", "getLong").parameterTypes("java.time.temporal.TemporalField"),
                    instanceCall(JAVA_TIME_LOCAL_DATE_TIME, "get", "getLong").parameterTypes("java.time.temporal.TemporalField"),
                    instanceCall(JAVA_TIME_OFFSET_TIME, "get", "getLong").parameterTypes("java.time.temporal.TemporalField"),
                    instanceCall(JAVA_TIME_OFFSET_DATE_TIME, "get", "getLong").parameterTypes("java.time.temporal.TemporalField"),
                    instanceCall(JAVA_TIME_ZONED_DATE_TIME, "get", "getLong").parameterTypes("java.time.temporal.TemporalField")
              ),
              toValue((args, memState, factory, method) -> chronoGet(args, memState, method)))
    .register(anyOf(instanceCall("java.io.InputStream", "skip").parameterTypes("long"),
                    instanceCall("java.io.Reader", "skip").parameterTypes("long")),
              toValue((args, memState, factory, method) -> skip(args.myArguments, memState)))
    .register(anyOf(
      staticCall(JAVA_LANG_INTEGER, "toString").parameterCount(1),
      staticCall(JAVA_LANG_LONG, "toString").parameterCount(1),
      staticCall(JAVA_LANG_STRING, "valueOf").parameterTypes("int"),
      staticCall(JAVA_LANG_STRING, "valueOf").parameterTypes("long")
    ), toValue((args, memState, factory, method) -> numberAsDecimalString(args, memState)))
    .register(staticCall(JAVA_LANG_INTEGER, "toHexString").parameterCount(1),
              toValue((args, memState, factory, method) -> numberAsString(args, memState, 4, Integer.SIZE)))
    .register(staticCall(JAVA_LANG_INTEGER, "toOctalString").parameterCount(1),
              toValue((args, memState, factory, method) -> numberAsString(args, memState, 3, Integer.SIZE)))
    .register(staticCall(JAVA_LANG_INTEGER, "toBinaryString").parameterCount(1),
              toValue((args, memState, factory, method) -> numberAsString(args, memState, 1, Integer.SIZE)))
    .register(staticCall(JAVA_LANG_LONG, "toHexString").parameterCount(1),
              toValue((args, memState, factory, method) -> numberAsString(args, memState, 4, Long.SIZE)))
    .register(staticCall(JAVA_LANG_LONG, "toOctalString").parameterCount(1),
              toValue((args, memState, factory, method) -> numberAsString(args, memState, 3, Long.SIZE)))
    .register(staticCall(JAVA_LANG_LONG, "toBinaryString").parameterCount(1),
              toValue((args, memState, factory, method) -> numberAsString(args, memState, 1, Long.SIZE)))
    .register(instanceCall(JAVA_LANG_ENUM, "name").parameterCount(0),
              toValue((args, memState, factory, method) -> enumName(args.myQualifier, memState, method.getReturnType())))
    .register(staticCall(JAVA_UTIL_COLLECTIONS, "emptyList", "emptySet", "emptyMap").parameterCount(0),
              toValue((args, memState, factory, method) -> getEmptyCollectionConstant(method)))
    .register(exactInstanceCall(JAVA_LANG_CLASS, "getName", "getSimpleName", "getCanonicalName").parameterCount(0),
              toValue((args, memState, factory, method) -> className(memState, args.myQualifier, method.getName(), method.getReturnType())))
    .register(anyOf(
      staticCall(JAVA_UTIL_COLLECTIONS, "singleton", "singletonList", "singletonMap"),
      staticCall(JAVA_UTIL_LIST, "of"),
      staticCall(JAVA_UTIL_SET, "of"),
      staticCall(JAVA_UTIL_MAP, "of", "ofEntries"),
      staticCall(JAVA_UTIL_ARRAYS, "asList")), CustomMethodHandlers::collectionFactory)
    .register(anyOf(
      staticCall(JAVA_LANG_INTEGER, "compare").parameterTypes("int", "int"),
      staticCall(JAVA_LANG_LONG, "compare").parameterTypes("long", "long"),
      staticCall(JAVA_LANG_BYTE, "compare").parameterTypes("byte", "byte"),
      staticCall(JAVA_LANG_SHORT, "compare").parameterTypes("short", "short")),
              toValue((args, state, factory, method) -> compareInteger(args, state)))
    .register(anyOf(
      instanceCall("java.util.Random", "nextInt").parameterTypes("int"),
      instanceCall("java.util.SplittableRandom", "nextInt").parameterTypes("int"),
      instanceCall("java.util.SplittableRandom", "nextInt").parameterTypes("int", "int")), toValue(CustomMethodHandlers::randomNextInt))
    .register(staticCall(JAVA_UTIL_ARRAYS, "copyOf"), (arguments, state, factory, method) -> copyOfArray(arguments, factory, method))
    .register(staticCall(JAVA_UTIL_COLLECTIONS, "unmodifiableCollection", "unmodifiableList", "unmodifiableSet", "unmodifiableMap",
                         "unmodifiableSortedSet", "unmodifiableSortedMap", "unmodifiableNavigableSet", "unmodifiableNavigableMap")
                .parameterCount(1), (arguments, state, factory, method) -> unmodifiableView(arguments, factory, method))
    .register(staticCall(JAVA_UTIL_COLLECTIONS, "checkedCollection", "checkedList", "checkedSet", "checkedMap",
                         "checkedSortedSet", "checkedSortedMap", "checkedNavigableSet", "checkedNavigableMap", "checkedQueue",
                         "synchronizedCollection", "synchronizedList", "synchronizedSet", "synchronizedMap",
                         "synchronizedSortedSet", "synchronizedSortedMap", "synchronizedNavigableSet", "synchronizedNavigableMap"),
              (arguments, state, factory, method) -> collectionView(ArrayUtil.getFirstElement(arguments.myArguments), factory, method))
    .register(anyOf(instanceCall(JAVA_UTIL_MAP, "keySet", "values", "entrySet").parameterCount(0),
                    instanceCall("java.util.NavigableSet", "descendingSet").parameterCount(0),
                    instanceCall("java.util.NavigableMap", "descendingMap").parameterCount(0)),
              (arguments, state, factory, method) -> collectionView(arguments.myQualifier, factory, method))
    .register(instanceCall(JAVA_UTIL_COLLECTION, "toArray").parameterTypes("T[]"), CustomMethodHandlers::collectionToArray)
    .register(instanceCall(JAVA_UTIL_COLLECTION, "toArray").parameterCount(0), CustomMethodHandlers::collectionToArray)
    .register(instanceCall(JAVA_LANG_STRING, "toCharArray").parameterCount(0), CustomMethodHandlers::stringToCharArray)
    .register(exactInstanceCall(JAVA_LANG_OBJECT, "getClass").parameterCount(0), toValue(CustomMethodHandlers::objectGetClass))
    .register(anyOf(staticCall(JAVA_LANG_MATH, "random").parameterCount(0),
                    instanceCall("java.util.Random", "nextDouble").parameterCount(0),
                    instanceCall("java.util.SplittableRandom", "nextDouble").parameterCount(0)), 
              toValue((arguments, state, factory, method) -> doubleRange(0.0, Math.nextDown(1.0))))
    .register(anyOf(staticCall(JAVA_LANG_MATH, "sin", "cos").parameterCount(1),
                    staticCall(JAVA_LANG_STRICT_MATH, "sin", "cos").parameterCount(1)), 
              toValue((arguments, state, factory, method) -> doubleRange(-1.0, 1.0).join(DOUBLE_NAN)))
    .register(anyOf(staticCall(JAVA_LANG_MATH, "hypot", "sqrt", "exp").parameterCount(1),
                    staticCall(JAVA_LANG_STRICT_MATH, "hypot", "sqrt", "exp").parameterCount(1)), 
              toValue((arguments, state, factory, method) -> doubleRange(0.0, Double.POSITIVE_INFINITY).join(DOUBLE_NAN)))
    .register(instanceCall("java.util.Random", "nextFloat").parameterCount(0), 
              toValue((arguments, state, factory, method) -> floatRange(0.0f, Math.nextDown(1.0f))))
    .register(staticCall(JAVA_LANG_DOUBLE, "isNaN").parameterTypes("double"),
              toValue((arguments, state, factory, method) -> isNaN(arguments, state, DOUBLE_NAN)))
    .register(staticCall(JAVA_LANG_FLOAT, "isNaN").parameterTypes("float"),
              toValue((arguments, state, factory, method) -> isNaN(arguments, state, FLOAT_NAN)))
    .register(anyOf(
                staticCall("com.google.common.collect.Lists", "newArrayList", "newLinkedList", "newCopyOnWriteArrayList").parameterCount(0),
                staticCall("com.google.common.collect.Sets", "newHashSet", "newLinkedHashSet", "newIdentityHashSet",
                           "newCopyOnWriteArraySet", "newConcurrentHashSet", "newTreeSet").parameterCount(0),
                staticCall("com.google.common.collect.Maps", "newHashMap", "newLinkedHashMap", "newIdentityHashMap",
                           "newConcurrentHashMap", "newTreeMap").parameterCount(0)),
              toValue((arguments, state, factory, method) -> COLLECTION_SIZE.asDfType(intValue(0)).meet(LOCAL_OBJECT)));

  public static CustomMethodHandler find(PsiMethod method) {
    CustomMethodHandler handler = null;
    if (isConstantCall(method)) {
      handler = toValue((arguments, state, factory, m) -> handleConstantCall(arguments, state, m));
    }
    CustomMethodHandler handler2 = CUSTOM_METHOD_HANDLERS.mapFirst(method);
    return handler == null ? handler2 : handler.compose(handler2);
  }

  /**
   * @param method method to check
   * @return true if method will be evaluated to constant when all arguments are constant
   */
  @Contract("null -> false")
  public static boolean isConstantCall(PsiMethod method) {
    return CONSTANT_CALLS.methodMatches(method);
  }

  private static @NotNull DfType handleConstantCall(DfaCallArguments arguments, DfaMemoryState state, PsiMethod method) {
    PsiType returnType = method.getReturnType();
    if (returnType == null) return DfType.TOP;
    List<Object> args = new ArrayList<>();
    Object qualifierValue = null;
    if (!method.hasModifierProperty(PsiModifier.STATIC)) {
      qualifierValue = getConstantValue(state, arguments.myQualifier);
      if (qualifierValue == null) return DfType.TOP;
    }
    for (DfaValue argument : arguments.myArguments) {
      Object argumentValue = getConstantValue(state, argument);
      if (argumentValue == null) return DfType.TOP;
      if (argumentValue instanceof Long) {
        long longValue = ((Long)argumentValue).longValue();
        if (longValue >= Integer.MIN_VALUE && longValue <= Integer.MAX_VALUE) {
          argumentValue = (int)longValue;
        }
      }
      args.add(argumentValue);
    }
    Method jvmMethod = toJvmMethod(method);
    if (jvmMethod == null) return DfType.TOP;
    Object result;
    try {
      result = jvmMethod.invoke(qualifierValue, args.toArray());
    }
    catch (InvocationTargetException e) {
      if (e.getCause() instanceof NumberFormatException || e.getCause() instanceof ArithmeticException) {
        return DfType.FAIL;
      }
      return DfType.TOP;
    }
    catch (IllegalAccessException | IllegalArgumentException e) {
      return DfType.TOP;
    }
    return constant(result, returnType);
  }

  private static Method toJvmMethod(PsiMethod method) {
    return CachedValuesManager.getCachedValue(method, new CachedValueProvider<>() {
      @Override
      public @NotNull Result<Method> compute() {
        Method reflection = getMethod();
        return Result.create(reflection, method);
      }

      private static Class<?> toJvmType(PsiType type) {
        if (TypeUtils.isJavaLangString(type)) {
          return String.class;
        }
        if (TypeUtils.isJavaLangObject(type)) {
          return Object.class;
        }
        if (TypeUtils.typeEquals(JAVA_LANG_CHAR_SEQUENCE, type)) {
          return CharSequence.class;
        }
        if (PsiTypes.intType().equals(type)) {
          return int.class;
        }
        if (PsiTypes.booleanType().equals(type)) {
          return boolean.class;
        }
        if (PsiTypes.byteType().equals(type)) {
          return byte.class;
        }
        if (PsiTypes.shortType().equals(type)) {
          return short.class;
        }
        if (PsiTypes.charType().equals(type)) {
          return char.class;
        }
        if (PsiTypes.longType().equals(type)) {
          return long.class;
        }
        if (PsiTypes.floatType().equals(type)) {
          return float.class;
        }
        if (PsiTypes.doubleType().equals(type)) {
          return double.class;
        }
        return null;
      }

      private @Nullable Method getMethod() {
        PsiClass aClass = method.getContainingClass();
        Class<?> containingClass;
        if (aClass == null) return null;
        try {
          containingClass = Class.forName(aClass.getQualifiedName());
        }
        catch (ClassNotFoundException ignored) {
          return null;
        }
        PsiParameter[] parameters = method.getParameterList().getParameters();
        Class<?>[] parameterTypes = new Class[parameters.length];
        for (int i = 0; i < parameters.length; i++) {
          PsiParameter parameter = parameters[i];
          PsiType type = parameter.getType();
          Class<?> jvmType = toJvmType(type);
          if (jvmType == null) return null;
          parameterTypes[i] = jvmType;
        }
        return ReflectionUtil.getMethod(containingClass, method.getName(), parameterTypes);
      }
    });
  }

  private static @NotNull DfType indexOf(DfaValue qualifier,
                                         DfaMemoryState memState,
                                         DfaValueFactory factory,
                                         SpecialField specialField) {
    DfaValue length = specialField.createValue(factory, qualifier);
    LongRangeSet range = DfIntType.extractRange(memState.getDfType(length));
    return intRange(LongRangeSet.range(-1, range.max() - 1));
  }

  private static @NotNull DfType shorterOrEqualString(DfaValue qualifier,
                                                      DfaMemoryState memState,
                                                      DfaValueFactory factory) {
    DfaValue length = STRING_LENGTH.createValue(factory, qualifier);
    LongRangeSet range = DfIntType.extractRange(memState.getDfType(length));
    return STRING_LENGTH.asDfType(intRange(LongRangeSet.range(0, range.max())));
  }

  private static @Nullable DfaValue copyOfArray(DfaCallArguments arguments,
                                                DfaValueFactory factory,
                                                PsiMethod method) {
    if (arguments.myArguments.length < 2) return null;
    return factory.getWrapperFactory().createWrapper(typedObject(method.getReturnType(), Nullability.NOT_NULL).meet(LOCAL_OBJECT),
                                                     ARRAY_LENGTH, arguments.myArguments[1]);
  }

  @Contract("null, _, _ -> null; !null, _, _ -> !null")
  private static DfaValue collectionView(@Nullable DfaValue orig,
                                         @NotNull DfaValueFactory factory,
                                         @NotNull PsiMethod method) {
    if (orig == null) return null;
    return factory.getWrapperFactory().createWrapper(typedObject(method.getReturnType(), Nullability.NOT_NULL),
                                                     COLLECTION_SIZE, COLLECTION_SIZE.createValue(factory, orig));
  }

  private static @Nullable DfaValue unmodifiableView(@NotNull DfaCallArguments arguments,
                                                     @NotNull DfaValueFactory factory,
                                                     @NotNull PsiMethod method) {
    if (arguments.myArguments.length != 1) return null;
    return factory.getWrapperFactory().createWrapper(
      typedObject(method.getReturnType(), Nullability.NOT_NULL).meet(Mutability.UNMODIFIABLE_VIEW.asDfType()),
      COLLECTION_SIZE, COLLECTION_SIZE.createValue(factory, arguments.myArguments[0]));
  }

  private static @Nullable DfaValue collectionFactory(DfaCallArguments args,
                                                      DfaMemoryState memState, DfaValueFactory factory,
                                                      PsiMethod method) {
    PsiType type = method.getReturnType();
    if (!(type instanceof PsiClassType)) return null;
    int factor = PsiTypesUtil.classNameEquals(type, JAVA_UTIL_MAP) ? 2 : 1;
    DfaValue size;
    if (method.isVarArgs()) {
      size = ARRAY_LENGTH.createValue(factory, args.myArguments[0]);
    }
    else {
      size = factory.fromDfType(intValue(args.myArguments.length / factor));
    }
    boolean asList = method.getName().equals("asList");
    Mutability mutability = asList ? Mutability.MUTABLE : Mutability.UNMODIFIABLE;
    DfType result = typedObject(type, Nullability.NOT_NULL).meet(mutability.asDfType());
    if (asList) {
      result = result.meet(LOCAL_OBJECT);
    }
    return factory.getWrapperFactory().createWrapper(result, COLLECTION_SIZE, size);
  }

  private static DfType getEmptyCollectionConstant(PsiMethod method) {
    String fieldName = "EMPTY_" + method.getName().substring("empty".length()).toUpperCase(Locale.ROOT);
    PsiClass collectionsClass = method.getContainingClass();
    if (collectionsClass == null) return DfType.TOP;
    PsiField field = collectionsClass.findFieldByName(fieldName, false);
    if (field == null) return DfType.TOP;
    return referenceConstant(field, field.getType());
  }

  private static @Nullable DfaValue substring(DfaCallArguments args, DfaMemoryState state, DfaValueFactory factory, PsiType stringType) {
    if (stringType == null || !stringType.equalsToText(JAVA_LANG_STRING)) return null;
    DfaValue qualifier = args.myQualifier;
    DfaValue[] arguments = args.myArguments;
    if (arguments.length < 1 || arguments.length > 2 || arguments[0] == null) return null;
    DfaValue from = arguments[0];
    DfaValue lenVal = STRING_LENGTH.createValue(factory, qualifier);
    DfaValue to = arguments.length == 1 ? lenVal : arguments[1];
    DfaValue resultLen = factory.getBinOpFactory().create(to, from, state, INT, LongRangeBinOp.MINUS);
    if (resultLen instanceof DfaBinOpValue) {
      resultLen = factory.fromDfType(state.getDfType(resultLen));
    }
    return factory.getWrapperFactory().createWrapper(typedObject(stringType, Nullability.NOT_NULL), STRING_LENGTH, resultLen);
  }

  private static @NotNull DfType mathAbs(DfaValue[] args, DfaMemoryState memState, boolean isLong) {
    DfaValue arg = ArrayUtil.getFirstElement(args);
    if (arg == null) return DfType.TOP;
    DfType type = memState.getDfType(arg);
    LongRangeSet range = isLong ? DfLongType.extractRange(type) : DfIntType.extractRange(type);
    return isLong ? longRange(range.abs(LongRangeType.INT64)) : intRange(range.abs(LongRangeType.INT32));
  }


  private static @NotNull DfType chronoGet(DfaCallArguments arguments, DfaMemoryState state, PsiMethod method) {
    DfaValue[] myArguments = arguments.myArguments;
    if (myArguments.length != 1 || myArguments[0] == null) {
      return DfType.TOP;
    }
    DfType myArgument = state.getDfType(myArguments[0]);
    PsiEnumConstant enumConstant = myArgument.getConstantOfType(PsiEnumConstant.class);
    if (enumConstant == null) {
      return DfType.TOP;
    }
    PsiType enumType = enumConstant.getType();
    if (!enumType.equalsToText("java.time.temporal.ChronoField")) {
      return DfType.TOP;
    }
    PsiClass containingClass = method.getContainingClass();
    if (containingClass == null) {
      return DfType.TOP;
    }
    ChronoField myChronoField = ChronoUtil.getChronoField(enumConstant.getName());
    if (myChronoField == null || !ChronoUtil.isAnyGetSupported(method, myChronoField)) {
      return DfType.TOP;
    }
    LongRangeSet range = LongRangeSet.range(myChronoField.range().getMinimum(), myChronoField.range().getMaximum());
    String methodName = method.getName();
    if ("get".equals(methodName)) {
      return intRange(range);
    }
    else if ("getLong".equals(methodName)) {
      return longRange(range);
    }
    else {
      return DfType.TOP;
    }
  }

  private static @NotNull DfType calendarGet(DfaCallArguments arguments, DfaMemoryState state, DfaValueFactory factory) {
    if (arguments.myArguments.length != 1) return DfType.TOP;
    Integer val = state.getDfType(arguments.myArguments[0]).getConstantOfType(Integer.class);
    if (val == null) return DfType.TOP;
    LongRangeSet range = switch (val) {
      case Calendar.DATE -> LongRangeSet.range(1, 31);
      case Calendar.MONTH -> {
        PsiType type = TypeConstraint.fromDfType(state.getDfType(arguments.myQualifier)).getPsiType(factory.getProject());
        if (TypeUtils.typeEquals("java.util.GregorianCalendar", type)) {
          yield LongRangeSet.range(0, 11);
        }
        else {
          // Could be lunar calendar
          yield LongRangeSet.range(0, 12);
        }
      }
      case Calendar.AM_PM -> LongRangeSet.range(0, 1);
      case Calendar.DAY_OF_YEAR -> LongRangeSet.range(1, 366);
      case Calendar.HOUR -> LongRangeSet.range(0, 11);
      case Calendar.HOUR_OF_DAY -> LongRangeSet.range(0, 23);
      case Calendar.MINUTE, Calendar.SECOND -> LongRangeSet.range(0, 59);
      case Calendar.MILLISECOND -> LongRangeSet.range(0, 999);
      default -> null;
    };
    return range == null ? DfType.TOP : intRange(range);
  }

  private static @NotNull DfType skip(DfaValue[] arguments, DfaMemoryState state) {
    if (arguments.length != 1) return DfType.TOP;
    LongRangeSet range = DfLongType.extractRange(state.getDfType(arguments[0]));
    return longRange(LongRangeSet.range(0, Math.max(0, range.max())));
  }

  private static @NotNull DfType numberAsDecimalString(DfaCallArguments args, DfaMemoryState state) {
    DfaValue arg = args.myArguments[0];
    if (arg == null) return DfType.TOP;
    return numberAsDecimalString(state, arg);
  }

  public static @NotNull DfType numberAsDecimalString(@NotNull DfaMemoryState state, @NotNull DfaValue arg) {
    LongRangeSet range = DfLongType.extractRange(state.getDfType(arg));
    if (range.isEmpty()) return DfType.TOP;
    long min = range.min();
    long max = range.max();
    DfType length;
    if (min >= 0) {
      length = intRange(LongRangeSet.range(String.valueOf(min).length(), String.valueOf(max).length()));
    }
    else if (max <= 0) {
      length = intRange(LongRangeSet.range(String.valueOf(max).length(), String.valueOf(min).length()));
    }
    else {
      length = intRange(LongRangeSet.range(1, Math.max(String.valueOf(max).length(), String.valueOf(min).length())));
    }
    return STRING_LENGTH.asDfType(length);
  }

  private static @NotNull DfType numberAsString(DfaCallArguments args, DfaMemoryState state, int bitsPerChar, int maxBits) {
    DfaValue arg = args.myArguments[0];
    if (arg == null) return DfType.TOP;
    LongRangeSet range = DfLongType.extractRange(state.getDfType(arg));
    int usedBits = range.min() >= 0 ? Long.SIZE - Long.numberOfLeadingZeros(range.max()) : maxBits;
    int max = Math.max(1, (usedBits - 1) / bitsPerChar + 1);
    return STRING_LENGTH.asDfType(intRange(LongRangeSet.range(1, max)));
  }

  private static @NotNull DfType enumName(DfaValue qualifier, DfaMemoryState state, PsiType type) {
    DfType dfType = state.getDfType(qualifier);
    PsiEnumConstant value = dfType.getConstantOfType(PsiEnumConstant.class);
    if (value != null) {
      return referenceConstant(value.getName(), type);
    }
    return DfType.TOP;
  }

  private static @NotNull DfType objectGetClass(DfaCallArguments arguments, DfaMemoryState state, DfaValueFactory factory, PsiMethod method) {
    DfaValue qualifier = arguments.myQualifier;
    TypeConstraint fact = TypeConstraint.fromDfType(state.getDfType(qualifier));
    if (fact instanceof TypeConstraint.Exact && !(fact instanceof TypeConstraints.ExactSubclass)) {
      PsiType qualifierType = fact.getPsiType(factory.getProject());
      PsiType classType = method.getReturnType();
      if (classType != null && qualifierType != null) {
        return referenceConstant(qualifierType, classType);
      }
    }
    return INSTANTIABLE_CLASS.asDfType(TRUE);
  }

  private static Object getConstantValue(DfaMemoryState memoryState, DfaValue value) {
    DfType type = DfaUtil.getUnboxedDfType(memoryState, value);
    Object constant = type.getConstantOfType(Object.class);
    if (constant instanceof String && ((String)constant).length() > MAX_STRING_CONSTANT_LENGTH_TO_TRACK) return null;
    return constant;
  }

