1 files changed, 136 insertions, 0 deletions
diff --git a/winery/org.eclipse.winery.repository/src/main/java/org/eclipse/winery/repository/resources/servicetemplates/plans/PlansResourceData.java b/winery/org.eclipse.winery.repository/src/main/java/org/eclipse/winery/repository/resources/servicetemplates/plans/PlansResourceData.java
new file mode 100644
index 0000000..8527eb4
--- /dev/null
+++ b/winery/org.eclipse.winery.repository/src/main/java/org/eclipse/winery/repository/resources/servicetemplates/plans/PlansResourceData.java
@@ -0,0 +1,136 @@
+/*******************************************************************************
+ * Copyright (c) 2012-2013 University of Stuttgart.
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the Eclipse Public License v1.0
+ * and the Apache License 2.0 which both accompany this distribution,
+ * and are available at http://www.eclipse.org/legal/epl-v10.html
+ * and http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Contributors:
+ *     Oliver Kopp - initial API and implementation
+ *******************************************************************************/
+package org.eclipse.winery.repository.resources.servicetemplates.plans;
+
+import java.io.IOException;
+import java.io.StringWriter;
+import java.util.Collection;
+import java.util.List;
+
+import org.eclipse.winery.model.tosca.TPlan;
+import org.eclipse.winery.model.tosca.TPlan.PlanModelReference;
+import org.eclipse.winery.repository.datatypes.TypeWithShortName;
+import org.eclipse.winery.repository.resources.admin.types.PlanLanguagesManager;
+import org.eclipse.winery.repository.resources.admin.types.PlanTypesManager;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import com.fasterxml.jackson.core.JsonFactory;
+import com.fasterxml.jackson.core.JsonGenerationException;
+import com.fasterxml.jackson.core.JsonGenerator;
+
+public class PlansResourceData {
+	
+	private static final Logger logger = LoggerFactory.getLogger(PlansResourceData.class);
+	
+	// data: [ [id, pre, name, type, lang]* ]
+	private String embeddedPlansTableData;
+	
+	// data: [ [id, pre, name, type, lang, reference]* ]
+	private String linkedPlansTableData;
+	
+	
+	/**
+	 * Data object for the JSP
+	 * 
+	 * @param serviceTemplateResource the service template the plans belong to
+	 */
+	public PlansResourceData(List<TPlan> plans) {
+		if (plans.isEmpty()) {
+			this.embeddedPlansTableData = "[]";
+			this.linkedPlansTableData = "[]";
+			return;
+		}
+		JsonFactory jsonFactory = new JsonFactory();
+		StringWriter embeddedPlansTableDataSW = new StringWriter();
+		StringWriter linkedPlansTableDataSW = new StringWriter();
+		try {
+			JsonGenerator jGeneratorEmbedded = jsonFactory.createGenerator(embeddedPlansTableDataSW);
+			JsonGenerator jGeneratorLinked = jsonFactory.createGenerator(linkedPlansTableDataSW);
+			
+			jGeneratorEmbedded.writeStartArray();
+			jGeneratorLinked.writeStartArray();
+			
+			for (TPlan plan : plans) {
+				String name = plan.getName();
+				if (name == null) {
+					// name defaults to id
+					name = plan.getId();
+				}
+				String type = PlanTypesManager.INSTANCE.getShortName(plan.getPlanType());
+				String language = PlanLanguagesManager.INSTANCE.getShortName(plan.getPlanLanguage());
+				PlanModelReference planModelReference = plan.getPlanModelReference();
+				String reference = planModelReference != null ? planModelReference.getReference() : null;
+				JsonGenerator gen;
+				boolean writeReference;
+				if (reference == null) {
+					gen = jGeneratorEmbedded;
+					writeReference = false;
+				} else if (reference.startsWith("../")) {
+					gen = jGeneratorEmbedded;
+					writeReference = false;
+				} else {
+					gen = jGeneratorLinked;
+					writeReference = true;
+				}
+				
+				gen.writeStartArray();
+				gen.writeString(plan.getId());
+				gen.writeString(""); // precondition
+				gen.writeString(name);
+				gen.writeString(type);
+				gen.writeString(language);
+				if (writeReference) {
+					gen.writeString(reference);
+				}
+				gen.writeEndArray();
+			}
+			
+			jGeneratorEmbedded.writeEndArray();
+			jGeneratorLinked.writeEndArray();
+			
+			jGeneratorEmbedded.close();
+			embeddedPlansTableDataSW.close();
+			jGeneratorLinked.close();
+			linkedPlansTableDataSW.close();
+		} catch (JsonGenerationException e) {
+			PlansResourceData.logger.error(e.getMessage(), e);
+			this.embeddedPlansTableData = "[]";
+			this.linkedPlansTableData = "[]";
+			return;
+		} catch (IOException e) {
+			PlansResourceData.logger.error("", e);
+			this.embeddedPlansTableData = "[]";
+			this.linkedPlansTableData = "[]";
+			return;
+		}
+		this.embeddedPlansTableData = embeddedPlansTableDataSW.toString();
+		this.linkedPlansTableData = linkedPlansTableDataSW.toString();
+	}
+	
+	public String getEmbeddedPlansTableData() {
+		return this.embeddedPlansTableData;
+	}
+	
+	public String getLinkedPlansTableData() {
+		return this.linkedPlansTableData;
+	}
+	
+	public Collection<TypeWithShortName> getPlanTypes() {
+		return PlanTypesManager.INSTANCE.getTypes();
+	}
+	
+	public Collection<TypeWithShortName> getPlanLanguages() {
+		return PlanLanguagesManager.INSTANCE.getTypes();
+	}
+	
+}