| package org.apache.turbine.modules.screens; |
| |
| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you 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. |
| */ |
| |
| import java.io.BufferedReader; |
| import java.io.CharArrayWriter; |
| import java.io.InputStreamReader; |
| import java.io.OutputStreamWriter; |
| import java.io.PrintWriter; |
| import java.nio.charset.Charset; |
| import java.nio.charset.StandardCharsets; |
| |
| import javax.servlet.http.HttpServletRequest; |
| |
| import org.apache.turbine.annotation.TurbineService; |
| import org.apache.turbine.pipeline.PipelineData; |
| import org.apache.turbine.services.jsonrpc.JsonRpcService; |
| import org.apache.turbine.util.RunData; |
| import org.jabsorb.JSONRPCBridge; |
| |
| |
| /** |
| * A Screen class for dealing with JSON-RPC requests. Typically you would |
| * extend this class and override the doOutput() method to use TurbineJsonRpc |
| * to register the POJOs that will provide the functions you are making |
| * available via JSON-RPC. Use JSONSecureScreen if you need the user to be |
| * logged in prior to executing the functions you provide. |
| * |
| * <p>Here is an example from a superclass: |
| * <code> |
| * public void doOutput(PipelineData data) throws Exception |
| * { |
| * User user = data.getUser(); |
| * |
| * MyJsonFunctions myFunctions = new MyJsonFunctions(user.getName()); |
| * |
| * // Session specific |
| * TurbineJsonRpc.registerObject(data.getSession(), "myFunctions", myFunctions); |
| * |
| * // Global |
| * //TurbineJsonRpc.registerObjectGlobal("testGlobal", testObject); |
| * |
| * super.doOutput(data); |
| * } |
| * </code> |
| * |
| * <p>The class MyFunctions would be something like: |
| * <code> |
| * public class MyJsonFunctions |
| * { |
| * private String getHello(String clientParameter) |
| * { |
| * return "Hello " + clientParameter; |
| * } |
| * } |
| * </code> |
| * |
| * <p>This code is derived from the com.metaparadigm.jsonrpc.JSONRPCServlet and (after upgrade) checked against |
| * org.jabsorb.JSONRPCServlet. |
| * |
| * @author brad@folkens.com |
| * @author <a href="mailto:seade@backstagetech.com.au">Scott Eade</a> |
| * @version $Id$ |
| */ |
| public class JSONScreen extends RawScreen |
| { |
| protected static final String JSONRPC_CONTENT_TYPE = "application/json;charset=utf-8"; |
| |
| protected final static int BUFFER_SIZE = 4096; |
| |
| /** Injected service instance */ |
| @TurbineService |
| protected JsonRpcService jsonRpcService; |
| |
| /** |
| * @see org.apache.turbine.modules.screens.RawScreen#getContentType(org.apache.turbine.pipeline.PipelineData) |
| */ |
| @Override |
| protected String getContentType(PipelineData pipelineData) |
| { |
| return JSONRPC_CONTENT_TYPE; |
| } |
| |
| /** |
| * Output the dynamic content. |
| * |
| * Encoding is UTF-8. @{@link #JSONRPC_CONTENT_TYPE}: {@value #JSONRPC_CONTENT_TYPE}. |
| * |
| * @param pipelineData The PipelineData object. |
| */ |
| @Override |
| protected void doOutput(PipelineData pipelineData) throws Exception |
| { |
| RunData data = pipelineData.getRunData(); |
| data.declareDirectResponse(); |
| HttpServletRequest request = data.getRequest(); |
| |
| // we require utf-8, cft |
| Charset charset = StandardCharsets.UTF_8; //request.getCharacterEncoding(); |
| BufferedReader in = new BufferedReader(new InputStreamReader(request.getInputStream(), charset)); |
| |
| // Read the request |
| CharArrayWriter cdata = new CharArrayWriter(); |
| char buf[] = new char[BUFFER_SIZE]; |
| int ret; |
| while ((ret = in.read(buf, 0, BUFFER_SIZE)) != -1) |
| { |
| cdata.write(buf, 0, ret); |
| } |
| |
| // Find the JSONRPCBridge for this session or create one |
| // if it doesn't exist |
| JSONRPCBridge json_bridge = jsonRpcService.getBridge(data.getSession()); |
| |
| // Process the request |
| Object json_res = jsonRpcService.processCall(cdata, json_bridge, request); |
| |
| try (PrintWriter out = new PrintWriter( |
| new OutputStreamWriter( |
| data.getResponse().getOutputStream(), charset))) |
| { |
| out.print(json_res.toString()); |
| out.flush(); |
| } |
| } |
| } |