Make it possible to intercept RPC calls without copy/paste

Change-Id: I50c2e4a95b492a42c4291e8dcbede3de87f1be6f

5 files changed, 320 insertions, 0 deletions

diff --git a/uitest/src/com/vaadin/tests/rpclogger/LoggingServerRpcHandler.java b/uitest/src/com/vaadin/tests/rpclogger/LoggingServerRpcHandler.java
new file mode 100644
index 0000000000..1cb66972ef
--- /dev/null
+++ b/uitest/src/com/vaadin/tests/rpclogger/LoggingServerRpcHandler.java
@@ -0,0 +1,40 @@
+/*
+ * Copyright 2000-2014 Vaadin Ltd.
+ * 
+ * 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 com.vaadin.tests.rpclogger;
+
+import com.vaadin.server.ClientConnector;
+import com.vaadin.server.ServerRpcMethodInvocation;
+import com.vaadin.server.communication.ServerRpcHandler;
+import com.vaadin.shared.communication.LegacyChangeVariablesInvocation;
+import com.vaadin.ui.UI;
+
+public class LoggingServerRpcHandler extends ServerRpcHandler {
+
+    @Override
+    protected void handleInvocation(UI ui, ClientConnector connector,
+            LegacyChangeVariablesInvocation legacyInvocation) {
+        ((RPCLoggerUI) ui).recordInvocation(connector, legacyInvocation);
+        super.handleInvocation(ui, connector, legacyInvocation);
+    }
+
+    @Override
+    protected void handleInvocation(UI ui, ClientConnector connector,
+            ServerRpcMethodInvocation invocation) {
+        ((RPCLoggerUI) ui).recordInvocation(connector, invocation);
+        super.handleInvocation(ui, connector, invocation);
+    }
+
+}
diff --git a/uitest/src/com/vaadin/tests/rpclogger/LoggingUidlRequestHandler.java b/uitest/src/com/vaadin/tests/rpclogger/LoggingUidlRequestHandler.java
new file mode 100644
index 0000000000..5eff9b56d9
--- /dev/null
+++ b/uitest/src/com/vaadin/tests/rpclogger/LoggingUidlRequestHandler.java
@@ -0,0 +1,28 @@
+/*
+ * Copyright 2000-2014 Vaadin Ltd.
+ * 
+ * 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 com.vaadin.tests.rpclogger;
+
+import com.vaadin.server.communication.ServerRpcHandler;
+import com.vaadin.server.communication.UidlRequestHandler;
+
+public class LoggingUidlRequestHandler extends UidlRequestHandler {
+
+    @Override
+    protected ServerRpcHandler createRpcHandler() {
+        return new LoggingServerRpcHandler();
+    }
+
+}
diff --git a/uitest/src/com/vaadin/tests/rpclogger/RPCLoggerService.java b/uitest/src/com/vaadin/tests/rpclogger/RPCLoggerService.java
new file mode 100644
index 0000000000..a70a51b603
--- /dev/null
+++ b/uitest/src/com/vaadin/tests/rpclogger/RPCLoggerService.java
@@ -0,0 +1,41 @@
+/*
+ * Copyright 2000-2014 Vaadin Ltd.
+ * 
+ * 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 com.vaadin.tests.rpclogger;
+
+import java.util.List;
+
+import com.vaadin.server.DeploymentConfiguration;
+import com.vaadin.server.RequestHandler;
+import com.vaadin.server.ServiceException;
+import com.vaadin.server.VaadinServlet;
+import com.vaadin.server.VaadinServletService;
+
+public class RPCLoggerService extends VaadinServletService {
+
+    public RPCLoggerService(VaadinServlet servlet,
+            DeploymentConfiguration deploymentConfiguration)
+            throws ServiceException {
+        super(servlet, deploymentConfiguration);
+    }
+
+    @Override
+    protected List<RequestHandler> createRequestHandlers()
+            throws ServiceException {
+        List<RequestHandler> handlers = super.createRequestHandlers();
+        handlers.add(new LoggingUidlRequestHandler());
+        return handlers;
+    }
+}
diff --git a/uitest/src/com/vaadin/tests/rpclogger/RPCLoggerServlet.java b/uitest/src/com/vaadin/tests/rpclogger/RPCLoggerServlet.java
new file mode 100644
index 0000000000..adb4302ba3
--- /dev/null
+++ b/uitest/src/com/vaadin/tests/rpclogger/RPCLoggerServlet.java
@@ -0,0 +1,34 @@
+/*
+ * Copyright 2000-2014 Vaadin Ltd.
+ * 
+ * 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 com.vaadin.tests.rpclogger;
+
+import com.vaadin.server.DeploymentConfiguration;
+import com.vaadin.server.ServiceException;
+import com.vaadin.server.VaadinServlet;
+import com.vaadin.server.VaadinServletService;
+
+public class RPCLoggerServlet extends VaadinServlet {
+
+    @Override
+    protected VaadinServletService createServletService(
+            DeploymentConfiguration deploymentConfiguration)
+            throws ServiceException {
+        RPCLoggerService service = new RPCLoggerService(this,
+                deploymentConfiguration);
+        service.init();
+        return service;
+    }
+}
diff --git a/uitest/src/com/vaadin/tests/rpclogger/RPCLoggerUI.java b/uitest/src/com/vaadin/tests/rpclogger/RPCLoggerUI.java
new file mode 100644
index 0000000000..bac82f232d
--- /dev/null
+++ b/uitest/src/com/vaadin/tests/rpclogger/RPCLoggerUI.java
@@ -0,0 +1,177 @@
+/*
+ * Copyright 2000-2014 Vaadin Ltd.
+ * 
+ * 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 com.vaadin.tests.rpclogger;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+
+import com.vaadin.server.ClientConnector;
+import com.vaadin.server.ErrorHandler;
+import com.vaadin.server.ServerRpcMethodInvocation;
+import com.vaadin.server.VaadinRequest;
+import com.vaadin.shared.communication.LegacyChangeVariablesInvocation;
+import com.vaadin.shared.communication.MethodInvocation;
+import com.vaadin.tests.components.AbstractTestUIWithLog;
+import com.vaadin.ui.AbstractComponent;
+import com.vaadin.ui.Button;
+import com.vaadin.ui.Button.ClickEvent;
+import com.vaadin.ui.Button.ClickListener;
+import com.vaadin.ui.ListSelect;
+import com.vaadin.ui.Notification;
+import com.vaadin.ui.Notification.Type;
+
+public class RPCLoggerUI extends AbstractTestUIWithLog implements ErrorHandler {
+
+    private List<Action> lastActions = new ArrayList<Action>();
+
+    public static class Action {
+        public Action(ClientConnector connector, MethodInvocation invocation) {
+            target = connector;
+            this.invocation = invocation;
+        }
+
+        private MethodInvocation invocation;
+        private ClientConnector target;
+    }
+
+    @Override
+    protected int getLogSize() {
+        return 10;
+    }
+
+    @Override
+    protected void setup(VaadinRequest request) {
+        setErrorHandler(this);
+        addComponent(new Button("Do something"));
+        ListSelect s = new ListSelect();
+        s.setMultiSelect(true);
+        s.addItem("foo");
+        s.addItem("bar");
+        addComponent(s);
+
+        addComponent(new Button("Action, which will fail", new ClickListener() {
+
+            @Override
+            public void buttonClick(ClickEvent event) {
+                throw new RuntimeException("Something went wrong");
+            }
+        }));
+    }
+
+    public void recordInvocation(ClientConnector connector,
+            LegacyChangeVariablesInvocation legacyInvocation) {
+        addAction(new Action(connector, legacyInvocation));
+    }
+
+    public void recordInvocation(ClientConnector connector,
+            ServerRpcMethodInvocation invocation) {
+        addAction(new Action(connector, invocation));
+    }
+
+    private void addAction(Action action) {
+        while (lastActions.size() >= 5) {
+            lastActions.remove(0);
+        }
+        lastActions.add(action);
+    }
+
+    public String formatAction(ClientConnector connector,
+            LegacyChangeVariablesInvocation legacyInvocation) {
+        String connectorIdentifier = getConnectorIdentifier(connector);
+        Map<String, Object> changes = legacyInvocation.getVariableChanges();
+        String rpcInfo = "";
+        for (String key : changes.keySet()) {
+            Object value = changes.get(key);
+            rpcInfo += key + ": " + formatValue(value);
+        }
+        return "Legacy RPC " + rpcInfo + " for " + connectorIdentifier;
+
+    }
+
+    public String formatAction(ClientConnector connector,
+            ServerRpcMethodInvocation invocation) {
+        String connectorIdentifier = getConnectorIdentifier(connector);
+        String rpcInfo = invocation.getInterfaceName() + "."
+                + invocation.getMethodName() + " (";
+        for (Object o : invocation.getParameters()) {
+            rpcInfo += formatValue(o);
+            rpcInfo += ",";
+        }
+        rpcInfo = rpcInfo.substring(0, rpcInfo.length() - 2) + ")";
+        return "RPC " + rpcInfo + " for " + connectorIdentifier;
+
+    }
+
+    private String formatValue(Object value) {
+        if (value == null) {
+            return "null";
+        }
+        if (value instanceof String) {
+            return (String) value;
+        } else if (value instanceof Object[]) {
+            String formatted = "";
+            for (Object o : ((Object[]) value)) {
+                formatted += formatValue(o) + ",";
+            }
+            return formatted;
+        } else {
+            return value.toString();
+        }
+    }
+
+    private String getConnectorIdentifier(ClientConnector connector) {
+        String connectorIdentifier = connector.getClass().getSimpleName();
+        if (connector instanceof AbstractComponent) {
+            String caption = ((AbstractComponent) connector).getCaption();
+            if (caption != null) {
+                connectorIdentifier += " - " + caption;
+            }
+        }
+        return "'" + connectorIdentifier + "'";
+    }
+
+    @Override
+    public void error(com.vaadin.server.ErrorEvent event) {
+        String msg = "";
+
+        for (int i = 0; i < lastActions.size(); i++) {
+            Action action = lastActions.get(i);
+            if (action.invocation instanceof ServerRpcMethodInvocation) {
+                msg += "\n"
+                        + (i + 1)
+                        + " "
+                        + formatAction(action.target,
+                                (ServerRpcMethodInvocation) action.invocation);
+            } else {
+                msg += "\n"
+                        + (i + 1)
+                        + " "
+                        + formatAction(
+                                action.target,
+                                (LegacyChangeVariablesInvocation) action.invocation);
+            }
+        }
+
+        msg += "\n";
+        msg += "\n";
+        msg += "This error should not really be shown but logged for later analysis.";
+        Notification.show(
+                "Something went wrong. Actions leading up to this error were:",
+                msg, Type.ERROR_MESSAGE);
+        // log(msg);
+    }
+}