/*
    * MIT License
    * Copyright (c) 2006-2025 JMockit developers
    * See LICENSE file for full license text.
    */
   package mockit.internal.faking;
   
   import static mockit.internal.util.Utilities.getClassType;
   
  import edu.umd.cs.findbugs.annotations.NonNull;
  import edu.umd.cs.findbugs.annotations.Nullable;
  
  import java.lang.reflect.Type;
  
  import mockit.MockUp;
  import mockit.asm.classes.ClassReader;
  import mockit.internal.BaseClassModifier;
  import mockit.internal.capturing.CaptureOfImplementations;
  
  public final class CaptureOfFakedImplementations extends CaptureOfImplementations<Void> {
      private final FakeClassSetup fakeClassSetup;
  
      public CaptureOfFakedImplementations(@NonNull MockUp<?> fake, @NonNull Type baseType) {
          Class<?> baseClassType = getClassType(baseType);
          fakeClassSetup = new FakeClassSetup(baseClassType, baseType, fake, null);
      }
  
      @NonNull
      @Override
      protected BaseClassModifier createModifier(@Nullable ClassLoader cl, @NonNull ClassReader cr,
              @NonNull Class<?> baseType, Void typeMetadata) {
          return fakeClassSetup.createClassModifier(cr);
      }
  
      @Override
      protected void redefineClass(@NonNull Class<?> realClass, @NonNull byte[] modifiedClass) {
          fakeClassSetup.applyClassModifications(realClass, modifiedClass);
      }
  
      @Nullable
      public <T> Class<T> apply() {
          @SuppressWarnings("unchecked")
          Class<T> baseType = (Class<T>) fakeClassSetup.realClass;
          Class<T> baseClassType = baseType;
          Class<T> fakedClass = null;
  
          if (baseType.isInterface()) {
              fakedClass = new FakedImplementationClass<T>(fakeClassSetup.fake).createImplementation(baseType);
              baseClassType = fakedClass;
          }
  
          if (baseClassType != Object.class) {
              redefineClass(baseClassType, baseType, null);
          }
  
          makeSureAllSubtypesAreModified(baseType, false, null);
          return fakedClass;
      }
  }