Add license headers

[lgpl/argeo-commons.git] / server / runtime / org.argeo.server.core / src / main / java / org / argeo / server / mvc / SerializingView.java

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 02654e9ebf224d3928e9ff4c867649636c173d11..2e502b58bbb84e2aa5575e1e5eb0aba4decc15f9 100644 (file)
--- 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,19 @@
+/*
+ * Copyright (C) 2010 Mathieu Baudier <mbaudier@argeo.org>
+ *
+ * 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 +23,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 +51,58 @@ public class SerializingView extends AbstractView implements MvcConstants {
                this.serializer = serializer;
        }
 
-       @SuppressWarnings("unchecked")
+       @SuppressWarnings( { "unchecked", "restriction" })
        @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("unchecked")
+       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;
+                       }
+
+                       if (otherIsBindingResult)
+                               return answerValue;
                }
 
-               serializer.serialize(answer, request, response);
+               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 +113,8 @@ public class SerializingView extends AbstractView implements MvcConstants {
                return locale;
        }
 
+       public void setSerializer(ServerSerializer serializer) {
+               this.serializer = serializer;
+       }
+
 }