AbstractIndexingVisitor.java

/*
 * SPDX-FileCopyrightText: none
 * SPDX-License-Identifier: CC0-1.0
 */

package gov.nist.secauto.oscal.lib.profile.resolver.support;

import gov.nist.secauto.metaschema.core.metapath.item.node.IAssemblyNodeItem;
import gov.nist.secauto.metaschema.core.metapath.item.node.IRootAssemblyNodeItem;
import gov.nist.secauto.metaschema.core.util.ObjectUtils;

import java.util.EnumSet;
import java.util.Set;

import edu.umd.cs.findbugs.annotations.NonNull;

public abstract class AbstractIndexingVisitor<T, R>
    extends AbstractCatalogEntityVisitor<T, R> {

  public AbstractIndexingVisitor() {
    this(ObjectUtils.notNull(EnumSet.allOf(IEntityItem.ItemType.class)));
  }

  public AbstractIndexingVisitor(@NonNull Set<IEntityItem.ItemType> itemTypesToIndex) {
    super(itemTypesToIndex);
  }

  @NonNull
  protected abstract IIndexer getIndexer(T state);

  @Override
  public R visitGroup(IAssemblyNodeItem item, R childResult, T state) {
    getIndexer(state).addGroup(item);
    return childResult;
  }

  @Override
  public R visitControl(IAssemblyNodeItem item, R childResult, T state) {
    getIndexer(state).addControl(item);
    return childResult;
  }

  @Override
  protected R visitParameter(@NonNull IAssemblyNodeItem parameterItem,
      @NonNull IAssemblyNodeItem catalogOrGroupOrControl, T state) {
    getIndexer(state).addParameter(parameterItem);
    return newDefaultResult(state);
  }

  @Override
  protected void visitPart(@NonNull IAssemblyNodeItem partItem,
      @NonNull IAssemblyNodeItem catalogOrGroupOrControl, T state) {
    getIndexer(state).addPart(partItem);
  }

  @Override
  protected void visitRole(IAssemblyNodeItem roleItem, IAssemblyNodeItem metadataItem, T state) {
    getIndexer(state).addRole(roleItem);
  }

  @Override
  protected void visitLocation(IAssemblyNodeItem locationItem, IAssemblyNodeItem metadataItem,
      T state) {
    getIndexer(state).addLocation(locationItem);
  }

  @Override
  protected void visitParty(IAssemblyNodeItem partyItem, IAssemblyNodeItem metadataItem, T state) {
    getIndexer(state).addParty(partyItem);
  }

  @Override
  protected void visitResource(IAssemblyNodeItem resourceItem, IRootAssemblyNodeItem rootItem, T state) {
    getIndexer(state).addResource(resourceItem);
  }
}