/*
    * SPDX-FileCopyrightText: none
    * SPDX-License-Identifier: CC0-1.0
    */
   
   package gov.nist.secauto.metaschema.databind;
   
   import gov.nist.secauto.metaschema.core.model.IBoundObject;
   import gov.nist.secauto.metaschema.core.model.IModule;
  import gov.nist.secauto.metaschema.core.model.IModuleLoader;
  import gov.nist.secauto.metaschema.core.util.CollectionUtil;
  import gov.nist.secauto.metaschema.databind.model.IBoundDefinitionModelComplex;
  
  import java.util.ArrayList;
  import java.util.HashSet;
  import java.util.List;
  import java.util.Set;
  
  import edu.umd.cs.findbugs.annotations.NonNull;
  
  class PostProcessingModuleLoaderStrategy
      extends SimpleModuleLoaderStrategy {
    @NonNull
    private final List<IModuleLoader.IModulePostProcessor> modulePostProcessors;
    private final Set<IModule> resolvedModules = new HashSet<>();
  
    protected PostProcessingModuleLoaderStrategy(
        @NonNull IBindingContext bindingContext,
        @NonNull List<IModuleLoader.IModulePostProcessor> modulePostProcessors) {
      super(bindingContext);
      this.modulePostProcessors = CollectionUtil.unmodifiableList(new ArrayList<>(modulePostProcessors));
    }
  
    @NonNull
    protected List<IModuleLoader.IModulePostProcessor> getModulePostProcessors() {
      return modulePostProcessors;
    }
  
    @Override
    public IBoundDefinitionModelComplex getBoundDefinitionForClass(@NonNull Class<? extends IBoundObject> clazz) {
      IBoundDefinitionModelComplex retval = super.getBoundDefinitionForClass(clazz);
      if (retval != null) {
        // force loading of metaschema information to apply constraints
        IModule module = retval.getContainingModule();
        synchronized (resolvedModules) {
          if (!resolvedModules.contains(module)) {
            // add first, to avoid loops
            resolvedModules.add(module);
            handleModule(module);
          }
        }
      }
      return retval;
    }
  
    private void handleModule(@NonNull IModule module) {
      for (IModuleLoader.IModulePostProcessor postProcessor : getModulePostProcessors()) {
        postProcessor.processModule(module);
      }
    }
  }