content: /*
   *  Licensed to the Apache Software Foundation (ASF) under one
   *  or more contributor license agreements.  See the NOTICE file
   *  distributed with this work for additional information
   *  regarding copyright ownership.  The ASF licenses this file
   *  to you 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 org.apache.causeway.persistence.jdo.datanucleus.entities;
  
  import java.lang.reflect.Method;
  import java.util.Set;
  
  import org.datanucleus.enhancement.Persistable;
  
  import org.jspecify.annotations.Nullable;
  import org.springframework.stereotype.Component;
  
  import org.apache.causeway.applib.services.repository.EntityState;
  import org.apache.causeway.commons.internal.base._NullSafe;
  import org.apache.causeway.commons.internal.collections._Sets;
  import org.apache.causeway.core.metamodel.facetapi.FacetHolder;
  import org.apache.causeway.core.metamodel.facets.object.entity.EntityFacet;
  import org.apache.causeway.persistence.jdo.datanucleus.metamodel.facets.entity.JdoEntityFacet;
  import org.apache.causeway.persistence.jdo.provider.entities.JdoFacetContext;
  
  @Component
  public class DnEntityStateProvider implements JdoFacetContext {
  
      @Override
      public EntityState getEntityState(final Object pojo) {
          return entityState(pojo);
      }
  
      @Override
      public boolean isPersistenceEnhanced(final @Nullable Class<?> cls) {
          if(cls==null) {
              return false;
          }
          return org.datanucleus.enhancement.Persistable.class.isAssignableFrom(cls);
      }
  
      @Override
      public boolean isMethodProvidedByEnhancement(final @Nullable Method method) {
          if(method==null) {
              return false;
          }
          ensureInit();
          return /*methodStartsWith(method, "jdo") || */
                  jdoMethodsProvidedByEnhancement.contains(method.toString());
      }
  
      public static EntityState entityState(final Object pojo) {
          if(pojo==null
                  || !(pojo instanceof Persistable)) {
              return EntityState.NOT_PERSISTABLE;
          }
  
          var persistable = (Persistable) pojo;
          var isDeleted = persistable.dnIsDeleted();
          if(isDeleted) {
              return EntityState.REMOVED;
          }
          var isPersistent = persistable.dnIsPersistent();
          if(isPersistent) {
              var oid = persistable.dnGetObjectId();
              return oid!=null
                      ? EntityState.ATTACHED
                      : EntityState.ATTACHED_NO_OID;
          }
          return EntityState.HOLLOW;
      }
  
      // -- HELPER
  
      private static final Set<String> jdoMethodsProvidedByEnhancement = _Sets.newHashSet();
  
      private static Method[] getMethodsProvidedByEnhancement() {
          return org.datanucleus.enhancement.Persistable.class.getDeclaredMethods();
      }
  
      private static void ensureInit() {
          if(jdoMethodsProvidedByEnhancement.isEmpty()) {
              _NullSafe.stream(getMethodsProvidedByEnhancement())
              .map(Method::toString)
              .forEach(jdoMethodsProvidedByEnhancement::add);
          }
      }
  
      @Override
      public EntityFacet createEntityFacet(final FacetHolder facetHolder, final Class<?> entityClass) {
          return new JdoEntityFacet(facetHolder, entityClass);
      }
  
  }
  