/**

  * 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.

  * <p>

  * Don't deal with this module directly, rather use the 

  * {@link OnReceive @OnReceive(url="ws://...", ...)} API to establish your

  * WebSocket connection.

  * <p>

  * 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 <code>org.netbeans.html:ko4j:1.0</code> module.

  *

  * @author Jaroslav Tulach

  */

 @Contexts.Id("tyrus")

 @ServiceProvider(service = Contexts.Provider.class)

 public final class TyrusContext 

 implements Contexts.Provider, WSTransfer<Comm>, 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

 }