// Copyright (c) 2018, 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.shaking.keepclassmembers.b115867670;

import static com.android.tools.r8.utils.codeinspector.Matchers.isPresent;
import static com.android.tools.r8.utils.codeinspector.Matchers.isRenamed;
import static org.hamcrest.CoreMatchers.not;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotEquals;
import static org.junit.Assert.assertThat;

import com.android.tools.r8.TestBase;
import com.android.tools.r8.graph.invokesuper.Consumer;
import com.android.tools.r8.utils.AndroidApp;
import com.android.tools.r8.utils.codeinspector.ClassSubject;
import com.android.tools.r8.utils.codeinspector.CodeInspector;
import com.google.common.collect.ImmutableList;
import org.junit.Ignore;
import org.junit.Test;

@interface JsonClass {}

@JsonClass
class Foo {

  final Interaction[] interactions;

  public Foo(Interaction[] interactions) {
    this.interactions = interactions;
  }

  @JsonClass
  static class Interaction {
    final Request request;

    Interaction(Request request) {
      this.request = request;
    }
  }

  @JsonClass
  static class Request {
    final String path;

    Request(String path) {
      this.path = path;
    }
  }
}

class Main {

  public static void main(String[] args) {
    System.out.println(Foo.class);
  }
}

public class B115867670 extends TestBase {
  String pkg = getClass().getPackage().getName();

  public void runTest(String additionalKeepRules, Consumer<CodeInspector> inspection)
      throws Exception {
    AndroidApp input =
        readClasses(
            JsonClass.class, Foo.class, Foo.Interaction.class, Foo.Request.class, Main.class);
    ImmutableList.Builder<String> builder = ImmutableList.builder();
    builder
        .add("-keep @interface **.JsonClass")
        .add("-keep class **.Main {")
        .add("  public static void main(java.lang.String[]);")
        .add("}")
        .add(additionalKeepRules);
    String config = String.join(System.lineSeparator(), builder.build());
    CodeInspector inspector = new CodeInspector(compileWithR8(input, config));
    inspection.accept(inspector);
  }

  private void checkNoKeepClassMembers(CodeInspector inspector) {
    assertThat(inspector.clazz(Main.class), isPresent());
    assertThat(inspector.clazz(Foo.class), isPresent());
    assertThat(inspector.clazz(Foo.Interaction.class), not(isPresent()));
    assertThat(inspector.clazz(Foo.Request.class), not(isPresent()));
  }

  private void checkKeepClassMembers(CodeInspector inspector) {
    assertThat(inspector.clazz(Main.class), isPresent());
    for (Class clazz : new Class[] {Foo.class, Foo.Interaction.class, Foo.Request.class}) {
      ClassSubject cls = inspector.clazz(clazz);
      assertThat(cls, isPresent());
      assertEquals(1, cls.asFoundClassSubject().allFields().size());
      // TODD(116079696): This is a hack!
      cls.forAllFields(field -> assertNotEquals(1, field.getFinalName().length()));
    }
  }

  private void checkKeepClassMembersRenamed(CodeInspector inspector) {
    assertThat(inspector.clazz(Main.class), isPresent());
    for (Class clazz : new Class[] {Foo.class, Foo.Interaction.class, Foo.Request.class}) {
      ClassSubject cls = inspector.clazz(clazz);
      assertThat(cls, isPresent());
      assertThat(cls, isRenamed());
      assertEquals(1, cls.asFoundClassSubject().allFields().size());
      // TODD(116079696): This is a hack!
      cls.forAllFields(field -> assertEquals(1, field.getFinalName().length()));
    }
  }

  @Test
  public void testNoDependentRules() throws Exception {
    runTest("", this::checkNoKeepClassMembers);
  }

  @Test
  public void testDenendentWithKeepClassMembers() throws Exception {
    runTest(
        "-keepclassmembers @" + pkg + ".JsonClass class ** { <fields>; }",
        this::checkKeepClassMembers);
  }

  @Test
  public void testDependentWithKeepClassMembersAllowObfuscation() throws Exception {
    runTest(
        "-keepclassmembers,allowobfuscation @" + pkg + ".JsonClass class ** { <fields>; }",
        this::checkKeepClassMembersRenamed);
  }

  @Test
  @Ignore("b/116092333")
  public void testDenendentWithIfKeepClassMembers() throws Exception {
    runTest(
        "-if @" + pkg + ".JsonClass class * -keepclassmembers class <1> { <fields>; }",
        this::checkKeepClassMembers);
  }

  @Test
  @Ignore("b/116092333")
  public void testDependentWithIfKeepClassMembersAllowObfuscation() throws Exception {
    runTest(
        "-if @"
            + pkg
            + ".JsonClass class * -keepclassmembers,allowobfuscation class <1> { <fields>; }",
        this::checkKeepClassMembersRenamed);
  }

  @Test
  public void testDependentWithIfKeep() throws Exception {
    runTest(
        "-if @" + pkg + ".JsonClass class * -keep class <1> { <fields>; }",
        this::checkKeepClassMembers);
  }

  @Test
  public void testDenendentWithIfKeepAllowObfuscation() throws Exception {
    runTest(
        "-if @" + pkg + ".JsonClass class * -keep,allowobfuscation class <1> { <fields>; }",
        this::checkKeepClassMembersRenamed);
  }
}