// Copyright 2020 The Nomulus Authors. All Rights Reserved.
//
// 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 google.registry.persistence;

import static com.google.common.collect.ImmutableSet.toImmutableSet;
import static com.google.common.truth.Truth.assertThat;
import static com.google.common.truth.Truth.assertWithMessage;
import static google.registry.persistence.transaction.TransactionManagerFactory.tm;
import static google.registry.testing.DatabaseHelper.persistResource;

import com.google.common.collect.ImmutableSet;
import google.registry.model.ImmutableObject;
import google.registry.persistence.EntityCallbacksListener.RecursivePostLoad;
import google.registry.persistence.EntityCallbacksListener.RecursivePostPersist;
import google.registry.persistence.EntityCallbacksListener.RecursivePostRemove;
import google.registry.persistence.EntityCallbacksListener.RecursivePostUpdate;
import google.registry.persistence.EntityCallbacksListener.RecursivePrePersist;
import google.registry.persistence.EntityCallbacksListener.RecursivePreRemove;
import google.registry.persistence.EntityCallbacksListener.RecursivePreUpdate;
import google.registry.persistence.transaction.JpaTestExtensions;
import google.registry.persistence.transaction.JpaTestExtensions.JpaUnitTestExtension;
import jakarta.persistence.Embeddable;
import jakarta.persistence.Embedded;
import jakarta.persistence.Entity;
import jakarta.persistence.Id;
import jakarta.persistence.MappedSuperclass;
import jakarta.persistence.Transient;
import java.lang.reflect.Method;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.RegisterExtension;

/** Unit tests for {@link EntityCallbacksListener}. */
class EntityCallbacksListenerTest {

  @RegisterExtension
  public final JpaUnitTestExtension jpaExtension =
      new JpaTestExtensions.Builder().withEntityClass(TestEntity.class).buildUnitTestExtension();

  @Test
  void verifyAllCallbacks_executedExpectedTimes() {
    TestEntity testPersist = new TestEntity();
    tm().transact(() -> tm().insert(testPersist));
    checkAll(testPersist, 1, 0, 0, 0);

    TestEntity testUpdate = new TestEntity();
    TestEntity updated =
        tm().transact(
                () -> {
                  TestEntity merged = tm().getEntityManager().merge(testUpdate);
                  merged.nonTransientField++;
                  tm().getEntityManager().flush();
                  return merged;
                });
    // Note that when we get the merged entity, its @PostLoad callbacks are also invoked
    checkAll(updated, 0, 1, 0, 1);

    TestEntity testLoad = tm().transact(() -> tm().loadByKey(VKey.create(TestEntity.class, "id")));
    checkAll(testLoad, 0, 0, 0, 1);

    TestEntity testRemove =
        tm().transact(
                () -> {
                  TestEntity removed = tm().loadByKey(VKey.create(TestEntity.class, "id"));
                  return tm().delete(removed);
                });
    checkAll(testRemove, 0, 0, 1, 1);
  }

  @Test
  void verifyAllManagedEntities_haveNoMethodWithEmbedded() {
    ImmutableSet<Class<?>> violations =
        PersistenceXmlUtility.getManagedClasses().stream()
            .filter(clazz -> clazz.isAnnotationPresent(Entity.class))
            .filter(EntityCallbacksListenerTest::hasMethodAnnotatedWithEmbedded)
            .collect(toImmutableSet());
    assertWithMessage(
            "Found entity classes having methods annotated with @Embedded. EntityCallbacksListener"
                + " only supports annotating fields with @Embedded.")
        .that(violations)
        .isEmpty();
  }

  @Test
  void verifyHasMethodAnnotatedWithEmbedded_work() {
    assertThat(hasMethodAnnotatedWithEmbedded(ViolationEntity.class)).isTrue();
  }

  @Test
  void verifyCallbacksNotCalledOnCommit() {
    persistResource(new TestEntity());

    TestEntity testLoad = tm().transact(() -> tm().loadByKey(VKey.create(TestEntity.class, "id")));
    assertThat(testLoad.entityPreUpdate).isEqualTo(0);

    testLoad = tm().transact(() -> tm().loadByKey(VKey.create(TestEntity.class, "id")));

    // Verify that post-load happened but pre-update didn't.
    assertThat(testLoad.entityPostLoad).isEqualTo(1);
    assertThat(testLoad.entityPreUpdate).isEqualTo(0);
    // since we didn't save the non-transient field, should only be 1
    assertThat(testLoad.nonTransientField).isEqualTo(1);
  }

  private static boolean hasMethodAnnotatedWithEmbedded(Class<?> entityType) {
    boolean result = false;
    Class<?> parentType = entityType.getSuperclass();
    if (parentType != null && parentType.isAnnotationPresent(MappedSuperclass.class)) {
      result = hasMethodAnnotatedWithEmbedded(parentType);
    }
    for (Method method : entityType.getDeclaredMethods()) {
      if (method.isAnnotationPresent(Embedded.class)) {
        result = true;
        break;
      }
    }
    return result;
  }

  private static void checkAll(
      TestEntity testEntity,
      int expectedPersist,
      int expectedUpdate,
      int expectedRemove,
      int expectedLoad) {
    assertThat(testEntity.entityEmbedded.entityEmbeddedNested.entityEmbeddedNestedPostPersist)
        .isEqualTo(expectedPersist);
    assertThat(testEntity.entityEmbedded.entityEmbeddedNested.entityEmbeddedNestedPrePersist)
        .isEqualTo(expectedPersist);

    assertThat(testEntity.entityEmbedded.entityEmbeddedNested.entityEmbeddedNestedPreUpdate)
        .isEqualTo(expectedUpdate);
    assertThat(testEntity.entityEmbedded.entityEmbeddedNested.entityEmbeddedNestedPostUpdate)
        .isEqualTo(expectedUpdate);

    assertThat(testEntity.entityEmbedded.entityEmbeddedNested.entityEmbeddedNestedPreRemove)
        .isEqualTo(expectedRemove);
    assertThat(testEntity.entityEmbedded.entityEmbeddedNested.entityEmbeddedNestedPostRemove)
        .isEqualTo(expectedRemove);

    assertThat(testEntity.entityPostLoad).isEqualTo(expectedLoad);
    assertThat(testEntity.entityEmbedded.entityEmbeddedPostLoad).isEqualTo(expectedLoad);
    assertThat(testEntity.entityEmbedded.entityEmbeddedNested.entityEmbeddedNestedPostLoad)
        .isEqualTo(expectedLoad);
    assertThat(testEntity.entityEmbedded.entityEmbeddedParentPostLoad).isEqualTo(expectedLoad);

    assertThat(testEntity.parentPostLoad).isEqualTo(expectedLoad);
    assertThat(testEntity.parentPrePersist).isEqualTo(expectedPersist);
    assertThat(testEntity.parentEmbedded.parentEmbeddedPostLoad).isEqualTo(expectedLoad);
    assertThat(testEntity.parentEmbedded.parentEmbeddedNested.parentEmbeddedNestedPostLoad)
        .isEqualTo(expectedLoad);
    assertThat(testEntity.parentEmbedded.parentEmbeddedParentPostLoad).isEqualTo(expectedLoad);
  }

  @Entity(name = "TestEntity")
  private static class TestEntity extends ParentEntity {
    @Id String name = "id";
    int nonTransientField = 0;

    @Transient int entityPostLoad = 0;
    @Transient int entityPreUpdate = 0;

    @Embedded EntityEmbedded entityEmbedded = new EntityEmbedded();

    @RecursivePostLoad
    void entityPostLoad() {
      entityPostLoad++;
      nonTransientField++;
    }

    @RecursivePreUpdate
    void entityPreUpdate() {
      entityPreUpdate++;
    }
  }

  @Embeddable
  private static class EntityEmbedded extends EntityEmbeddedParent {
    @Embedded EntityEmbeddedNested entityEmbeddedNested = new EntityEmbeddedNested();

    @Transient int entityEmbeddedPostLoad = 0;

    String entityEmbedded = "placeholder";

    @RecursivePostLoad
    void entityEmbeddedPostLoad() {
      entityEmbeddedPostLoad++;
    }
  }

  @MappedSuperclass
  private static class EntityEmbeddedParent {
    @Transient int entityEmbeddedParentPostLoad = 0;

    String entityEmbeddedParent = "placeholder";

    @RecursivePostLoad
    void entityEmbeddedParentPostLoad() {
      entityEmbeddedParentPostLoad++;
    }
  }

  @Embeddable
  private static class EntityEmbeddedNested {
    @Transient int entityEmbeddedNestedPrePersist = 0;
    @Transient int entityEmbeddedNestedPreRemove = 0;
    @Transient int entityEmbeddedNestedPostPersist = 0;
    @Transient int entityEmbeddedNestedPostRemove = 0;
    @Transient int entityEmbeddedNestedPreUpdate = 0;
    @Transient int entityEmbeddedNestedPostUpdate = 0;
    @Transient int entityEmbeddedNestedPostLoad = 0;

    String entityEmbeddedNested = "placeholder";

    @RecursivePrePersist
    void entityEmbeddedNestedPrePersist() {
      entityEmbeddedNestedPrePersist++;
    }

    @RecursivePreRemove
    void entityEmbeddedNestedPreRemove() {
      entityEmbeddedNestedPreRemove++;
    }

    @RecursivePostPersist
    void entityEmbeddedNestedPostPersist() {
      entityEmbeddedNestedPostPersist++;
    }

    @RecursivePostRemove
    void entityEmbeddedNestedPostRemove() {
      entityEmbeddedNestedPostRemove++;
    }

    @RecursivePreUpdate
    void entityEmbeddedNestedPreUpdate() {
      entityEmbeddedNestedPreUpdate++;
    }

    @RecursivePostUpdate
    void entityEmbeddedNestedPostUpdate() {
      entityEmbeddedNestedPostUpdate++;
    }

    @RecursivePostLoad
    void entityEmbeddedNestedPostLoad() {
      entityEmbeddedNestedPostLoad++;
    }
  }

  @MappedSuperclass
  private static class ParentEntity extends ImmutableObject {
    @Embedded ParentEmbedded parentEmbedded = new ParentEmbedded();
    @Transient int parentPostLoad = 0;
    @Transient int parentPrePersist = 0;

    String parentEntity = "placeholder";

    @RecursivePostLoad
    void parentPostLoad() {
      parentPostLoad++;
    }

    @RecursivePrePersist
    void parentPrePersist() {
      parentPrePersist++;
    }
  }

  @Embeddable
  private static class ParentEmbedded extends ParentEmbeddedParent {
    @Transient int parentEmbeddedPostLoad = 0;

    String parentEmbedded = "placeholder";

    @Embedded ParentEmbeddedNested parentEmbeddedNested = new ParentEmbeddedNested();

    @RecursivePostLoad
    void parentEmbeddedPostLoad() {
      parentEmbeddedPostLoad++;
    }
  }

  @Embeddable
  private static class ParentEmbeddedNested {
    @Transient int parentEmbeddedNestedPostLoad = 0;

    String parentEmbeddedNested = "placeholder";

    @RecursivePostLoad
    void parentEmbeddedNestedPostLoad() {
      parentEmbeddedNestedPostLoad++;
    }
  }

  @MappedSuperclass
  private static class ParentEmbeddedParent {
    @Transient int parentEmbeddedParentPostLoad = 0;

    String parentEmbeddedParent = "placeholder";

    @RecursivePostLoad
    void parentEmbeddedParentPostLoad() {
      parentEmbeddedParentPostLoad++;
    }
  }

  @Entity
  private static class ViolationEntity {

    @Embedded
    EntityEmbedded getEntityEmbedded() {
      return new EntityEmbedded();
    }
  }
}
