path: root/org.eclipse.osee.framework.ui.skynet/src/org/eclipse/osee/framework/ui/skynet/Import/RootAndAttributeBasedArtifactResolver.java

/*******************************************************************************
 * Copyright (c) 2004, 2007 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.framework.ui.skynet.Import;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;
import java.util.logging.Level;
import org.eclipse.osee.framework.db.connection.exception.OseeCoreException;
import org.eclipse.osee.framework.jdk.core.type.HashCollection;
import org.eclipse.osee.framework.logging.OseeLog;
import org.eclipse.osee.framework.skynet.core.artifact.Artifact;
import org.eclipse.osee.framework.skynet.core.attribute.AttributeType;
import org.eclipse.osee.framework.ui.skynet.SkynetGuiPlugin;
import org.eclipse.osee.framework.ui.skynet.Import.RoughArtifact.NameAndVal;

/**
 * @author Robert A. Fisher
 */
public class RootAndAttributeBasedArtifactResolver extends NewArtifactImportResolver {
   private final LinkedList<AttributeType> identifyingAttributeDescriptors;
   private final Collection<String> EMPTY = new ArrayList<String>(0);
   private final boolean createNewIfNotExist;

   /**
    * @param identifyingAttributeDescriptors
    */
   public RootAndAttributeBasedArtifactResolver(Collection<AttributeType> identifyingAttributeDescriptors, boolean createNewIfNotExist) {
      if (identifyingAttributeDescriptors == null) throw new IllegalArgumentException(
            "identifyingAttributeDescriptors can not be null");
      if (identifyingAttributeDescriptors.isEmpty()) throw new IllegalArgumentException(
            "identifyingAttributeDescriptors can not be empty");

      this.identifyingAttributeDescriptors = new LinkedList<AttributeType>(identifyingAttributeDescriptors);
      this.createNewIfNotExist = createNewIfNotExist;
   }

   private boolean attributeValuesMatch(RoughArtifact roughArtifact, Artifact artifact) throws OseeCoreException {

      Collection<NameAndVal> roughAttributeCollection = roughArtifact.getAttributes();
      HashCollection<String, String> roughAttributeMap = new HashCollection<String, String>();
      for (NameAndVal roughAttribute : roughAttributeCollection) {
         roughAttributeMap.put(roughAttribute.getName(), roughAttribute.getValue());
      }

      for (AttributeType attributeType : identifyingAttributeDescriptors) {
         Collection<String> attributeValues = artifact.getAttributesToStringList(attributeType.getName());
         Collection<String> roughAttributes = roughAttributeMap.getValues(attributeType.getName());

         if (roughAttributes == null) {
            roughAttributes = EMPTY;
         }

         if (attributeValues.size() == roughAttributes.size()) {
            for (String attributeValue : attributeValues) {
               boolean attributeEqual = false;
               Iterator<String> iter = roughAttributes.iterator();

               String normalizedAttributeValue = normalizeAttributeValue(attributeValue);
               while (iter.hasNext()) {
                  String otherAttribute = iter.next();

                  if (normalizedAttributeValue.equals(normalizeAttributeValue(otherAttribute))) {
                     // Make sure we don't count this attribute more than once for equality
                     iter.remove();
                     attributeEqual = true;
                     break;
                  }
               }

               if (!attributeEqual) {
                  return false;
               }
            }
         }
      }

      return true;
   }

   private String normalizeAttributeValue(String value) {
      return value.trim().replaceAll("\\.$", "").toLowerCase();
   }

   public Artifact resolve(RoughArtifact roughArtifact) throws OseeCoreException {
      try {
         Artifact realArtifact = null;
         RoughArtifact roughParent = roughArtifact.getRoughParent();

         if (roughParent != null) {
            List<Artifact> siblings = roughParent.getAssociatedArtifact().getChildren();
            Collection<Artifact> candidates = new LinkedList<Artifact>();

            for (Artifact artifact : siblings) {
               if (attributeValuesMatch(roughArtifact, artifact)) {
                  candidates.add(artifact);
               }
            }

            if (candidates.size() == 1) {
               realArtifact = candidates.iterator().next();
               roughArtifact.updateValues(realArtifact);
            } else {
               OseeLog.log(SkynetGuiPlugin.class, Level.INFO,
                     "Found " + candidates.size() + " candidates during reuse import for " + roughArtifact.getName());
               if (createNewIfNotExist) {
                  realArtifact = super.resolve(roughArtifact);
               }
            }
         }

         return realArtifact;
      } catch (Exception ex) {
         throw new OseeCoreException(ex);
      }
   }
}