AbstractEntityItem.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.IModelNodeItem;
import gov.nist.secauto.metaschema.core.util.ObjectUtils;

import java.net.URI;

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

public abstract class AbstractEntityItem implements IEntityItem {

  @NonNull
  private final String originalIdentifier;
  @NonNull
  private IModelNodeItem<?, ?> instance;
  @NonNull
  private final ItemType itemType;
  @NonNull
  private final URI source;
  private int referenceCount; // 0 by default
  // private boolean resolved; // false by default

  protected AbstractEntityItem(@NonNull Builder builder) {
    this.itemType = ObjectUtils.requireNonNull(builder.itemType, "itemType");
    this.originalIdentifier = ObjectUtils.requireNonNull(builder.originalIdentifier, "originalIdentifier");
    this.instance = ObjectUtils.requireNonNull(builder.instance, "instance");
    this.source = ObjectUtils.requireNonNull(builder.source, "source");
  }

  @Override
  @NonNull
  public String getOriginalIdentifier() {
    return originalIdentifier;
  }

  @Override
  @NonNull
  public abstract String getIdentifier();

  // @NonNull
  // public String getIdentifier() {
  // final String checkedReassignedIdentifier = reassignedIdentifier;
  // return checkedReassignedIdentifier == null ? originalIdentifier :
  // checkedReassignedIdentifier;
  // }

  @Override
  @NonNull
  public IModelNodeItem<?, ?> getInstance() {
    return instance;
  }

  @Override
  public void setInstance(IModelNodeItem<?, ?> item) {
    instance = item;
  }

  @Override
  @NonNull
  @SuppressWarnings("unchecked")
  public <T> T getInstanceValue() {
    return ObjectUtils.notNull((T) getInstance().getValue());
  }

  @Override
  @NonNull
  public ItemType getItemType() {
    return itemType;
  }

  @Override
  @NonNull
  public URI getSource() {
    return source;
  }

  @Override
  public int getReferenceCount() {
    return referenceCount;
  }

  // public boolean isResolved() {
  // return resolved;
  // }
  //
  // public void markResolved() {
  // resolved = true;
  // }

  @Override
  public void incrementReferenceCount() {
    referenceCount += 1;
  }

  @Override
  public int resetReferenceCount() {
    int retval = referenceCount;
    referenceCount = 0;
    return retval;
  }

  static final class Builder {
    private String originalIdentifier;
    private String reassignedIdentifier;
    private IModelNodeItem<?, ?> instance;
    private ItemType itemType;
    private URI source;

    @NonNull
    public Builder instance(@NonNull IModelNodeItem<?, ?> item, @NonNull ItemType itemType) {
      this.instance = item;
      this.itemType = itemType;
      return this;
    }

    // @NonNull
    // public Builder reassignedIdentifier(@NonNull UUID identifier) {
    // // no need to normalize, since UUIDs are formatted lower case
    // return reassignedIdentifier(identifier.toString());
    // }

    @NonNull
    public Builder reassignedIdentifier(@NonNull String identifier) {
      this.reassignedIdentifier = identifier;
      return this;
    }
    //
    // @NonNull
    // public Builder originalIdentifier(@NonNull UUID identifier) {
    // // no need to normalize, since UUIDs are formatted lower case
    // return originalIdentifier(identifier.toString());
    // }

    @NonNull
    public Builder originalIdentifier(@NonNull String identifier) {
      this.originalIdentifier = identifier;
      return this;
    }

    @NonNull
    public Builder source(@NonNull URI source) {
      this.source = source;
      return this;
    }

    @NonNull
    public IEntityItem build() {
      return reassignedIdentifier == null ? new OriginalEntityItem(this) : new ReassignedEntityItem(this);
    }
  }

  static final class OriginalEntityItem
      extends AbstractEntityItem {

    protected OriginalEntityItem(@NonNull Builder builder) {
      super(builder);
    }

    @Override
    public String getIdentifier() {
      return getOriginalIdentifier();
    }

    @Override
    public boolean isIdentifierReassigned() {
      return false;
    }
  }

  static final class ReassignedEntityItem
      extends AbstractEntityItem {
    @NonNull
    private final String reassignedIdentifier;

    protected ReassignedEntityItem(@NonNull Builder builder) {
      super(builder);
      this.reassignedIdentifier = ObjectUtils.requireNonNull(builder.reassignedIdentifier);
    }

    @Override
    public String getIdentifier() {
      return reassignedIdentifier;
    }

    @Override
    public boolean isIdentifierReassigned() {
      return true;
    }
  }
}