org.eclipse.osee.framework.ui.data.model.editor/src/org/eclipse/osee/framework/ui/data/model/editor/input/OseeDataTypeDatastore.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.data.model.editor.input;

import java.util.ArrayList;
import java.util.List;
import java.util.logging.Level;
import org.eclipse.osee.framework.db.connection.ConnectionHandlerStatement;
import org.eclipse.osee.framework.db.connection.exception.OseeCoreException;
import org.eclipse.osee.framework.db.connection.exception.OseeDataStoreException;
import org.eclipse.osee.framework.db.connection.exception.OseeTypeDoesNotExist;
import org.eclipse.osee.framework.jdk.core.type.CompositeKey;
import org.eclipse.osee.framework.jdk.core.type.CompositeKeyHashMap;
import org.eclipse.osee.framework.jdk.core.type.HashCollection;
import org.eclipse.osee.framework.jdk.core.type.ObjectPair;
import org.eclipse.osee.framework.logging.OseeLog;
import org.eclipse.osee.framework.skynet.core.artifact.ArtifactType;
import org.eclipse.osee.framework.skynet.core.artifact.ArtifactTypeManager;
import org.eclipse.osee.framework.skynet.core.attribute.AttributeType;
import org.eclipse.osee.framework.skynet.core.attribute.AttributeTypeManager;
import org.eclipse.osee.framework.skynet.core.relation.RelationType;
import org.eclipse.osee.framework.skynet.core.relation.RelationTypeManager;
import org.eclipse.osee.framework.ui.data.model.editor.model.ArtifactDataType;
import org.eclipse.osee.framework.ui.data.model.editor.model.AttributeDataType;
import org.eclipse.osee.framework.ui.data.model.editor.model.RelationDataType;
import org.eclipse.osee.framework.ui.skynet.ImageManager;
import org.eclipse.osee.framework.ui.skynet.SkynetGuiPlugin;

/**
 * @author Roberto E. Escobar
 */
public class OseeDataTypeDatastore {
   private static final String SELECT_ATTRIBUTE_VALIDITY = "SELECT * FROM osee_valid_attributes";
   private static final String SELECT_RELATION_VALIDITY = "SELECT * FROM osee_valid_relations";

   private OseeDataTypeDatastore() {
   }

   public static List<AttributeDataType> getAttributeTypes() throws OseeDataStoreException {
      List<AttributeDataType> attributeDataTypes = new ArrayList<AttributeDataType>();
      for (AttributeType attributeType : AttributeTypeManager.getAllTypes()) {
         AttributeDataType attributeDataType =
               new AttributeDataType(String.valueOf(attributeType.getAttrTypeId()), attributeType.getNamespace(),
                     attributeType.getName(), attributeType.getBaseAttributeClass().getCanonicalName(),
                     attributeType.getDefaultValue(), attributeType.getFileTypeExtension(),
                     attributeType.getMaxOccurrences(), attributeType.getMinOccurrences(),
                     attributeType.getProviderAttributeClass().getCanonicalName(), attributeType.getTaggerId(),
                     attributeType.getTipText(), attributeType.getOseeEnumTypeId());
         attributeDataTypes.add(attributeDataType);
      }
      return attributeDataTypes;
   }

   public static List<RelationDataType> getRelationDataTypes() throws OseeDataStoreException, OseeTypeDoesNotExist {
      List<RelationDataType> relationDataTypes = new ArrayList<RelationDataType>();
      for (RelationType relationType : RelationTypeManager.getAllTypes()) {
         RelationDataType relationDataType =
               new RelationDataType(String.valueOf(relationType.getRelationTypeId()), relationType.getNamespace(),
                     relationType.getTypeName(), relationType.getAToBPhrasing(), relationType.getBToAPhrasing(),
                     relationType.isOrdered(), relationType.getShortName(), relationType.getSideAName(),
                     relationType.getSideBName());
         relationDataTypes.add(relationDataType);
      }
      return relationDataTypes;
   }

   public static List<ArtifactDataType> getArtifactDataTypes() throws OseeDataStoreException {
      List<ArtifactDataType> artifactDataTypes = new ArrayList<ArtifactDataType>();
      for (ArtifactType artifactType : ArtifactTypeManager.getAllTypes()) {
         ArtifactDataType artifactDataType =
               new ArtifactDataType(String.valueOf(artifactType.getArtTypeId()), artifactType.getNamespace(),
                     artifactType.getName(), ImageManager.getImage(artifactType));
         artifactDataTypes.add(artifactDataType);
      }
      return artifactDataTypes;
   }

   public static HashCollection<String, String> getArtifactToAttributeEntries() throws OseeCoreException {
      HashCollection<String, String> toReturn = new HashCollection<String, String>();
      ConnectionHandlerStatement chStmt = new ConnectionHandlerStatement();
      try {
         chStmt.runPreparedQuery(2000, SELECT_ATTRIBUTE_VALIDITY);
         while (chStmt.next()) {
            try {
               toReturn.put(chStmt.getString("art_type_id"), chStmt.getString("attr_type_id"));
            } catch (OseeCoreException ex) {
               OseeLog.log(SkynetGuiPlugin.class, Level.SEVERE, ex);
            }
         }
      } finally {
         chStmt.close();
      }
      return toReturn;
   }

   public static CompositeKeyHashMap<String, String, ObjectPair<Integer, Integer>> getArtifactToRelationEntries() throws OseeCoreException {
      CompositeKeyHashMap<String, String, ObjectPair<Integer, Integer>> toReturn =
            new CompositeKeyHashMap<String, String, ObjectPair<Integer, Integer>>();
      ConnectionHandlerStatement chStmt = new ConnectionHandlerStatement();
      try {
         chStmt.runPreparedQuery(2000, SELECT_RELATION_VALIDITY);
         while (chStmt.next()) {
            try {
               CompositeKey<String, String> key =
                     new CompositeKey<String, String>(chStmt.getString("art_type_id"),
                           chStmt.getString("rel_link_type_id"));

               ObjectPair<Integer, Integer> multiplicity =
                     new ObjectPair<Integer, Integer>(chStmt.getInt("side_a_max"), chStmt.getInt("side_b_max"));

               toReturn.put(key, multiplicity);
            } catch (OseeCoreException ex) {
               OseeLog.log(SkynetGuiPlugin.class, Level.SEVERE, ex);
            }
         }
      } finally {
         chStmt.close();
      }
      return toReturn;
   }

   public static HashCollection<String, String> getArtifactInheritance() throws OseeCoreException {
      HashCollection<String, String> toReturn = new HashCollection<String, String>();
      ConnectionHandlerStatement chStmt = new ConnectionHandlerStatement();
      try {
         chStmt.runPreparedQuery(2000, "select * from osee_artifact_type_inheritance");
         while (chStmt.next()) {
            try {
               toReturn.put(chStmt.getString("art_type_ancestor_id"), chStmt.getString("art_type_descendant_id"));
            } catch (OseeCoreException ex) {
               OseeLog.log(SkynetGuiPlugin.class, Level.SEVERE, ex);
            }
         }
      } finally {
         chStmt.close();
      }
      return toReturn;
   }
}