// Copyright (c) 2021, the R8 project authors. Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.

package com.android.tools.r8.regress.b191296688;

import static com.android.tools.r8.ToolHelper.getKotlinAnnotationJar;
import static com.android.tools.r8.ToolHelper.getKotlinStdlibJar;
import static com.android.tools.r8.utils.codeinspector.CodeMatchers.isInvokeWithTarget;
import static com.android.tools.r8.utils.codeinspector.Matchers.isPresent;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.junit.Assert.assertTrue;

import com.android.tools.r8.KotlinCompilerTool.KotlinCompilerVersion;
import com.android.tools.r8.KotlinTestBase;
import com.android.tools.r8.KotlinTestParameters;
import com.android.tools.r8.TestParameters;
import com.android.tools.r8.TestRuntime;
import com.android.tools.r8.TestRuntime.CfRuntime;
import com.android.tools.r8.ToolHelper.KotlinTargetVersion;
import com.android.tools.r8.utils.DescriptorUtils;
import com.android.tools.r8.utils.codeinspector.ClassSubject;
import com.android.tools.r8.utils.codeinspector.CodeInspector;
import com.android.tools.r8.utils.codeinspector.InstructionSubject;
import com.android.tools.r8.utils.codeinspector.MethodSubject;
import java.nio.file.Path;
import java.util.Collection;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.Parameterized;

@RunWith(Parameterized.class)
public class Regress191296688 extends KotlinTestBase {

  private static final String PKG = Regress191296688.class.getPackage().getName();

  private final TestParameters parameters;

  @Parameterized.Parameters(name = "{0}, {1}")
  public static Collection<Object[]> data() {
    return buildParameters(
        getTestParameters().withDexRuntimes().withAllApiLevels().build(),
        getKotlinTestParameters()
            .withCompiler(KotlinCompilerVersion.KOTLINC_1_5_0)
            .withTargetVersion(KotlinTargetVersion.JAVA_8)
            .build());
  }

  public Regress191296688(TestParameters parameters, KotlinTestParameters kotlinParameters) {
    super(kotlinParameters);
    this.parameters = parameters;
  }

  @Test
  public void testRegress191296688() throws Exception {
    Path aLib = temp.newFolder().toPath().resolve("alib.jar");
    writeClassesToJar(aLib, A.class);
    String folder = DescriptorUtils.getBinaryNameFromJavaType(PKG);
    CfRuntime cfRuntime = TestRuntime.getCheckedInJdk9();
    Path ktClasses =
        kotlinc(cfRuntime, kotlinc, targetVersion)
            .addSourceFiles(getKotlinFileInTest(folder, "B"))
            .addClasspathFiles(aLib)
            .compile();
    Path desugaredJar =
        testForD8(Backend.CF)
            .addLibraryFiles(getKotlinStdlibJar(kotlinc), getKotlinAnnotationJar(kotlinc))
            .addProgramFiles(ktClasses)
            .addProgramClasses(A.class)
            .setMinApi(parameters.getApiLevel())
            .compile()
            .inspect(this::verifyVirtualCallToPrivate)
            .writeToZip();
    testForD8()
        .addProgramFiles(desugaredJar)
        .setMinApi(parameters.getApiLevel())
        .disableDesugaring()
        .run(parameters.getRuntime(), PKG + ".BKt")
        .assertSuccessWithOutputLines("hep");
  }

  private void verifyVirtualCallToPrivate(CodeInspector inspector) {
    ClassSubject bClassSubject = inspector.clazz(PKG + ".B");
    MethodSubject proceedMethodSubject = bClassSubject.uniqueMethodWithName("proceed");
    assertThat(proceedMethodSubject, isPresent());
    assertTrue(
        bClassSubject.allMethods().stream()
            .anyMatch(
                method ->
                    method
                        .streamInstructions()
                        .filter(InstructionSubject::isInvokeVirtual)
                        .anyMatch(isInvokeWithTarget(proceedMethodSubject))));
  }
}