org.netbeans.html.wstyrus.TyrusContext Maven / Gradle / Ivy
/**
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
*
* Copyright 2013-2014 Oracle and/or its affiliates. All rights reserved.
*
* Oracle and Java are registered trademarks of Oracle and/or its affiliates.
* Other names may be trademarks of their respective owners.
*
* The contents of this file are subject to the terms of either the GNU
* General Public License Version 2 only ("GPL") or the Common
* Development and Distribution License("CDDL") (collectively, the
* "License"). You may not use this file except in compliance with the
* License. You can obtain a copy of the License at
* http://www.netbeans.org/cddl-gplv2.html
* or nbbuild/licenses/CDDL-GPL-2-CP. See the License for the
* specific language governing permissions and limitations under the
* License. When distributing the software, include this License Header
* Notice in each file and include the License file at
* nbbuild/licenses/CDDL-GPL-2-CP. Oracle designates this
* particular file as subject to the "Classpath" exception as provided
* by Oracle in the GPL Version 2 section of the License file that
* accompanied this code. If applicable, add the following below the
* License Header, with the fields enclosed by brackets [] replaced by
* your own identifying information:
* "Portions Copyrighted [year] [name of copyright owner]"
*
* Contributor(s):
*
* The Original Software is NetBeans. The Initial Developer of the Original
* Software is Oracle. Portions Copyright 2013-2014 Oracle. All Rights Reserved.
*
* If you wish your version of this file to be governed by only the CDDL
* or only the GPL Version 2, indicate your decision by adding
* "[Contributor] elects to include this software in this distribution
* under the [CDDL or GPL Version 2] license." If you do not indicate a
* single choice of license, a recipient has the option to distribute
* your version of this file under either the CDDL, the GPL Version 2 or
* to extend the choice of license to its licensees as provided above.
* However, if you add GPL Version 2 code and therefore, elected the GPL
* Version 2 license, then the option applies only if the new code is
* made subject to such option by the copyright holder.
*/
package org.netbeans.html.wstyrus;
import java.io.IOException;
import java.io.InputStream;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.Iterator;
import javax.websocket.ClientEndpoint;
import javax.websocket.ContainerProvider;
import javax.websocket.OnClose;
import javax.websocket.OnError;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.WebSocketContainer;
import net.java.html.json.OnReceive;
import org.netbeans.html.context.spi.Contexts;
import org.netbeans.html.json.spi.JSONCall;
import org.netbeans.html.json.spi.Transfer;
import org.netbeans.html.json.spi.WSTransfer;
import org.netbeans.html.wstyrus.TyrusContext.Comm;
import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;
import org.json.JSONTokener;
import org.openide.util.lookup.ServiceProvider;
/** This is an implementation module that provides support for
* WebSocket protocol for {@link OnReceive} communication end point for
* JDK7.
*
* Don't deal with this module directly, rather use the
* {@link OnReceive @OnReceive(url="ws://...", ...)} API to establish your
* WebSocket connection.
*
* There is no need to include this module in your application if you are
* running on JDK8. JDK8 WebView provides its own implementation of the
* WebSocket API based on WebSocket object inside a browser. This is included
* in the org.netbeans.html:ko4j:1.0 module.
*
* @author Jaroslav Tulach
*/
@Contexts.Id("tyrus")
@ServiceProvider(service = Contexts.Provider.class)
public final class TyrusContext
implements Contexts.Provider, WSTransfer, Transfer {
@Override
public void fillContext(Contexts.Builder context, Class requestor) {
// default WebSocket transfer implementation is registered
// in ko-fx module with 100, provide this one as a fallback only
context.register(WSTransfer.class, this, 1000);
context.register(Transfer.class, this, 1000);
}
@Override
public Comm open(String url, JSONCall callback) {
try {
return new Comm(new URI(url), callback);
} catch (URISyntaxException ex) {
throw new IllegalStateException(ex);
}
}
@Override
public void send(Comm socket, JSONCall data) {
socket.session.getAsyncRemote().sendText(data.getMessage());
}
@Override
public void close(Comm socket) {
try {
final Session s = socket.session;
if (s != null) {
s.close();
}
} catch (IOException ex) {
socket.callback.notifyError(ex);
}
}
@Override
public void extract(Object obj, String[] props, Object[] values) {
LoadJSON.extractJSON(obj, props, values);
}
@Override
public Object toJSON(InputStream is) throws IOException {
return LoadJSON.parse(is);
}
@Override
public void loadJSON(JSONCall call) {
LoadJSON.loadJSON(call);
}
/** Implementation class in an implementation. Represents a {@link ClientEndpoint} of the
* WebSocket channel. You are unlikely to get on hold of it.
*/
@ClientEndpoint
public static final class Comm {
private final JSONCall callback;
private Session session;
Comm(final URI url, JSONCall callback) {
this.callback = callback;
try {
final WebSocketContainer c = ContainerProvider.getWebSocketContainer();
c.connectToServer(Comm.this, url);
} catch (Exception ex) {
wasAnError(ex);
}
}
@OnOpen
public synchronized void open(Session s) {
this.session = s;
callback.notifySuccess(null);
}
@OnClose
public void close() {
this.session = null;
callback.notifyError(null);
}
@OnMessage
public void message(final String orig, Session s) {
Object json;
String data = orig.trim();
try {
JSONTokener tok = new JSONTokener(data);
Object obj = data.startsWith("[") ? new JSONArray(tok) : new JSONObject(tok);
json = convertToArray(obj);
} catch (JSONException ex) {
json = data;
}
callback.notifySuccess(json);
}
@OnError
public void wasAnError(Throwable t) {
callback.notifyError(t);
}
static Object convertToArray(Object o) throws JSONException {
if (o instanceof JSONArray) {
JSONArray ja = (JSONArray) o;
Object[] arr = new Object[ja.length()];
for (int i = 0; i < arr.length; i++) {
arr[i] = convertToArray(ja.get(i));
}
return arr;
} else if (o instanceof JSONObject) {
JSONObject obj = (JSONObject) o;
Iterator it = obj.keys();
while (it.hasNext()) {
String key = (String) it.next();
obj.put(key, convertToArray(obj.get(key)));
}
return obj;
} else {
return o;
}
}
} // end of Comm
}