X-Git-Url: http://git.argeo.org/?a=blobdiff_plain;f=server%2Fruntime%2Forg.argeo.server.core%2Fsrc%2Fmain%2Fjava%2Forg%2Fargeo%2Fserver%2Fmvc%2FSerializingView.java;h=b42a9a0d279300e6bad4e61bfcaa1ebaf662decc;hb=2e8d5eea45604d0f4686d54986be1ce6307aefdb;hp=02654e9ebf224d3928e9ff4c867649636c173d11;hpb=cdfd09387221a87b3a36f54fd3d5e8a6f4279baf;p=lgpl%2Fargeo-commons.git

diff --git a/server/runtime/org.argeo.server.core/src/main/java/org/argeo/server/mvc/SerializingView.java b/server/runtime/org.argeo.server.core/src/main/java/org/argeo/server/mvc/SerializingView.java
index 02654e9eb..b42a9a0d2 100644
--- a/server/runtime/org.argeo.server.core/src/main/java/org/argeo/server/mvc/SerializingView.java
+++ b/server/runtime/org.argeo.server.core/src/main/java/org/argeo/server/mvc/SerializingView.java
@@ -1,3 +1,18 @@
+/*
+ * Copyright (C) 2007-2012 Argeo GmbH
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *         http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
 package org.argeo.server.mvc;
 
 import java.util.Locale;
@@ -7,14 +22,26 @@ import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 
 import org.argeo.ArgeoException;
+import org.argeo.server.ServerAnswer;
 import org.argeo.server.ServerSerializer;
+import org.springframework.validation.BindingResult;
+import org.springframework.web.servlet.View;
 import org.springframework.web.servlet.view.AbstractView;
 
+/**
+ * Can be used as a standalone {@link View} or using
+ * {@link SerializingViewResolver}
+ */
 public class SerializingView extends AbstractView implements MvcConstants {
 	private final String viewName;
 	private final Locale locale;
 
-	private final ServerSerializer serializer;
+	private ServerSerializer serializer;
+
+	public SerializingView() {
+		this.viewName = null;
+		this.locale = Locale.getDefault();
+	}
 
 	public SerializingView(String viewName, Locale locale,
 			ServerSerializer serializer) {
@@ -23,24 +50,58 @@ public class SerializingView extends AbstractView implements MvcConstants {
 		this.serializer = serializer;
 	}
 
-	@SuppressWarnings("unchecked")
+	@SuppressWarnings({ "rawtypes" })
 	@Override
 	protected void renderMergedOutputModel(Map model,
 			HttpServletRequest request, HttpServletResponse response)
 			throws Exception {
-		final Object answer;
-		if (model.size() == 1) {
-			answer = model.values().iterator().next();
-		} else if (model.containsKey(ANSWER_MODEL_KEY)) {
-			answer = model.get(ANSWER_MODEL_KEY);
-		} else if (model.containsKey(viewName)) {
-			answer = model.get(viewName);
+		Boolean serverAnswersAsHtml = model
+				.containsKey(ANSWER_MODEL_KEY_AS_HTML);
+
+		final Object answer = findAnswerInModel(model);
+
+		if ((answer instanceof ServerAnswer) && serverAnswersAsHtml) {
+			response.setContentType("text/html");
+			ServerAnswer serverAnswer = (ServerAnswer) answer;
+			response.getWriter().append("<pre>");
+			response.getWriter().append(serverAnswer.getMessage());
+			response.getWriter().append("</pre>");
 		} else {
-			throw new ArgeoException(
-					"Model has a size different from 1. Specify a modelKey.");
+			serializer.serialize(answer, request, response);
 		}
+	}
+
+	@SuppressWarnings("rawtypes")
+	protected Object findAnswerInModel(Map model) {
+		if (model.size() == 1) {
+			return model.values().iterator().next();
+		} else if (model.size() == 2) {
+			boolean otherIsBindingResult = false;
+			Object answerValue = null;
+			for (Object value : model.values()) {
+				if (value instanceof BindingResult)
+					otherIsBindingResult = true;
+				else
+					answerValue = value;
+			}
 
-		serializer.serialize(answer, request, response);
+			if (otherIsBindingResult)
+				return answerValue;
+		}
+
+		if (model.containsKey(ANSWER_MODEL_KEY)) {
+			return model.get(ANSWER_MODEL_KEY);
+		} else if (model.containsKey(ANSWER_MODEL_KEY_AS_HTML)) {
+			return model.get(ANSWER_MODEL_KEY_AS_HTML);
+		} else if (viewName != null && model.containsKey(viewName)) {
+			return model.get(viewName);
+		} else {
+			if (model.size() == 0)
+				throw new ArgeoException("Model is empty.");
+			else
+				throw new ArgeoException(
+						"Model has a size different from 1. Specify a modelKey.");
+		}
 	}
 
 	public String getViewName() {
@@ -51,4 +112,8 @@ public class SerializingView extends AbstractView implements MvcConstants {
 		return locale;
 	}
 
+	public void setSerializer(ServerSerializer serializer) {
+		this.serializer = serializer;
+	}
+
 }