org.eclipse.osee.define/src/org/eclipse/osee/define/DefineArtifactFactory.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.define;

import java.util.Arrays;
import java.util.List;
import org.eclipse.osee.define.artifact.Spreadsheet;
import org.eclipse.osee.framework.db.connection.exception.OseeArgumentException;
import org.eclipse.osee.framework.db.connection.exception.OseeCoreException;
import org.eclipse.osee.framework.jdk.core.util.Collections;
import org.eclipse.osee.framework.skynet.core.artifact.Artifact;
import org.eclipse.osee.framework.skynet.core.artifact.ArtifactFactory;
import org.eclipse.osee.framework.skynet.core.artifact.ArtifactType;
import org.eclipse.osee.framework.skynet.core.artifact.Branch;
import org.eclipse.osee.framework.skynet.core.artifact.WordArtifact;
import org.eclipse.osee.framework.skynet.core.utility.Requirements;

/**
 * @author Ryan D. Brooks
 */
public class DefineArtifactFactory extends ArtifactFactory {
   private static List<String> WORD_ARTIFACTS =
         Arrays.asList(Requirements.SYSTEM_REQUIREMENT, Requirements.SUBSYSTEM_REQUIREMENT,
               Requirements.TEST_PROCEDURE, Requirements.TEST_PROCEDURE, "Test Procedure WML",
               Requirements.HARDWARE_REQUIREMENT, Requirements.SOFTWARE_DESIGN, Requirements.SOFTWARE_REQUIREMENT,
               Requirements.INDIRECT_SOFTWARE_REQUIREMENT, Requirements.SYSTEM_DESIGN, Requirements.SUBSYSTEM_DESIGN,
               Requirements.INTERFACE_REQUIREMENT, Requirements.SUBSYSTEM_FUNCTION, Requirements.SYSTEM_FUNCTION,
               Requirements.TEST_INFORMATION_SHEET);

   public DefineArtifactFactory() {
      super(Collections.setUnion(WORD_ARTIFACTS, Arrays.asList("Spreadsheet")));
   }

   @Override
   public Artifact getArtifactInstance(String guid, String humandReadableId, Branch branch, ArtifactType artifactType) throws OseeCoreException {
      if (artifactType.getName().equals("Spreadsheet")) {
         return new Spreadsheet(this, guid, humandReadableId, branch, artifactType);
      }
      if (WORD_ARTIFACTS.contains(artifactType.getName())) {
         return new WordArtifact(this, guid, humandReadableId, branch, artifactType);
      }
      throw new OseeArgumentException("did not recognize the artifact type: " + artifactType.getName());
   }
}