1 files changed, 107 insertions, 0 deletions
diff --git a/plugins/org.eclipse.osee.ats.rest/src/org/eclipse/osee/ats/rest/internal/util/ArtifactResolverImpl.java b/plugins/org.eclipse.osee.ats.rest/src/org/eclipse/osee/ats/rest/internal/util/ArtifactResolverImpl.java
new file mode 100644
index 00000000000..cdcb36b8404
--- /dev/null
+++ b/plugins/org.eclipse.osee.ats.rest/src/org/eclipse/osee/ats/rest/internal/util/ArtifactResolverImpl.java
@@ -0,0 +1,107 @@
+/*******************************************************************************
+ * Copyright (c) 2012 Boeing.
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the Eclipse Public License v1.0
+ * which accompanies this distribution, and is available at
+ * http://www.eclipse.org/legal/epl-v10.html
+ *
+ * Contributors:
+ *     Boeing - initial API and implementation
+ *******************************************************************************/
+package org.eclipse.osee.ats.rest.internal.util;
+
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.List;
+import org.eclipse.core.runtime.Assert;
+import org.eclipse.osee.ats.api.IAtsObject;
+import org.eclipse.osee.ats.api.IAtsServices;
+import org.eclipse.osee.ats.api.IAtsWorkItem;
+import org.eclipse.osee.ats.api.util.IArtifactResolver;
+import org.eclipse.osee.ats.core.util.AtsUtilCore;
+import org.eclipse.osee.framework.core.data.ArtifactId;
+import org.eclipse.osee.framework.core.data.IArtifactType;
+import org.eclipse.osee.framework.jdk.core.type.OseeCoreException;
+import org.eclipse.osee.orcs.OrcsApi;
+import org.eclipse.osee.orcs.data.ArtifactReadable;
+
+/**
+ * @author Donald G. Dunne
+ */
+public class ArtifactResolverImpl implements IArtifactResolver {
+
+   private final IAtsServices services;
+   private final OrcsApi orcsApi;
+
+   public ArtifactResolverImpl(IAtsServices services, OrcsApi orcsApi) {
+      this.services = services;
+      this.orcsApi = orcsApi;
+   }
+
+   @Override
+   public ArtifactId get(IAtsObject atsObject) throws OseeCoreException {
+      if (atsObject instanceof ArtifactReadable) {
+         return (ArtifactReadable) atsObject;
+      }
+      ArtifactReadable artifact = orcsApi.getQueryFactory().fromBranch(AtsUtilCore.getAtsBranch()).andUuid(
+         atsObject.getUuid()).getResults().getAtMostOneOrNull();
+      return artifact;
+   }
+
+   @Override
+   @SuppressWarnings("unchecked")
+   public <A extends ArtifactId> A get(IAtsWorkItem workItem, Class<?> clazz) throws OseeCoreException {
+      Assert.isNotNull(workItem, "Work Item can not be null");
+      ArtifactId artifact = get(workItem);
+      if (clazz.isInstance(artifact)) {
+         return (A) artifact;
+      }
+      return null;
+   }
+
+   @Override
+   @SuppressWarnings("unchecked")
+   public <A extends ArtifactId> List<A> get(Collection<? extends IAtsWorkItem> workItems, Class<?> clazz) throws OseeCoreException {
+      Assert.isNotNull(workItems, "Work Items can not be null");
+      List<A> arts = new ArrayList<>();
+      for (IAtsWorkItem workItem : workItems) {
+         ArtifactReadable artifact = get(workItem, clazz);
+         if (artifact != null) {
+            arts.add((A) artifact);
+         }
+      }
+      return arts;
+   }
+
+   @Override
+   public IArtifactType getArtifactType(IAtsWorkItem workItem) {
+      Assert.isNotNull(workItem, "Work Item can not be null");
+      return ((ArtifactReadable) workItem.getStoreObject()).getArtifactType();
+   }
+
+   @Override
+   public boolean isOfType(ArtifactId artifact, IArtifactType artifactType) {
+      Assert.isNotNull(artifact, "Artifact can not be null");
+      Assert.isNotNull(artifactType, "Artifact Type can not be null");
+      return ((ArtifactReadable) services.getArtifact(artifact)).isOfType(artifactType);
+   }
+
+   @Override
+   public int getTransactionNumber(ArtifactId artifact) {
+      Assert.isNotNull(artifact, "Artifact can not be null");
+      return ((ArtifactReadable) services.getArtifact(artifact)).getTransaction();
+   }
+
+   @Override
+   public boolean isOfType(IAtsObject atsObject, IArtifactType artifactType) {
+      Assert.isNotNull(atsObject, "ATS Object can not be null");
+      Assert.isNotNull(artifactType, "Artifact Type can not be null");
+      return isOfType(atsObject.getStoreObject(), artifactType);
+   }
+
+   @Override
+   public boolean inheritsFrom(IArtifactType artType, IArtifactType parentArtType) {
+      return orcsApi.getOrcsTypes().getArtifactTypes().inheritsFrom(artType, parentArtType);
+   }
+
+}