/**

  * The MIT License (MIT)

  *

  * Copyright (C) 2013 Jaroslav Tulach <jaroslav.tulach@apidesign.org>

  *

  * Permission is hereby granted, free of charge, to any person obtaining a copy

  * of this software and associated documentation files (the "Software"), to deal

  * in the Software without restriction, including without limitation the rights

  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell

  * copies of the Software, and to permit persons to whom the Software is

  * furnished to do so, subject to the following conditions:

  *

  * The above copyright notice and this permission notice shall be included in

  * all copies or substantial portions of the Software.

  *

  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR

  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,

  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE

  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER

  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,

  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN

  * THE SOFTWARE.

  */

 package org.apidesign.demo.chat.server;

 import java.io.ByteArrayInputStream;

 import java.io.IOException;

 import java.util.ArrayList;

 import java.util.HashSet;

 import java.util.List;

 import java.util.Set;

 import java.util.logging.Logger;

 import net.java.html.BrwsrCtx;

 import net.java.html.json.Models;

 import org.apidesign.demo.chat.shared.Message;

 import org.apidesign.demo.chat.shared.Query;

 import org.apidesign.demo.chat.shared.Reply;

 import org.glassfish.grizzly.websockets.DataFrame;

 import org.glassfish.grizzly.websockets.WebSocket;

 import org.glassfish.grizzly.websockets.WebSocketApplication;

 /** Server side of the chat application.*/

 public final class ChatServerResource extends WebSocketApplication {

     private static final Logger LOG = Logger.getLogger(ChatServerResource.class.getName());

     private static final long started = System.currentTimeMillis() - 10;

     private List<Message> msgs = new ArrayList<>();

     {

         Message welcome = new Message();

         welcome.setUser("system");

         welcome.setComment("Welcome and enjoy!");

         welcome.setSince(10);

         msgs.add(welcome);

     }

     private final Set<WebSocket> connected = new HashSet<>();

 /*

     @Produces(MediaType.APPLICATION_JSON)

     @GET public synchronized void getResources(

         @QueryParam("since") @DefaultValue("0") long since,

         @Suspended AsyncResponse ar

     ) {

         Query q = new Query();

         for (Message m : msgs) {

             if (m.getSince()>= since) {

                 q.getMessages().add(m);

             }

         }

         if (!q.getMessages().isEmpty()) {

             ar.resume(q);

         } else {

             awaiting.put(ar, since);

         }

     }

     */

     private void handleAwaiting(Message msg) {

         assert Thread.holdsLock(this);

         Reply reply = new Reply();

         reply.getMessages().add(msg);

         String txt = reply.toString();

         for (WebSocket webSocket : connected) {

             webSocket.send(txt);

         }

     }

     private synchronized Message publish(Message msg) {

         msg.setSince(System.currentTimeMillis() - started);

         msgs.add(msg);

         handleAwaiting(msg);

         return msg;

     }

     @Override

     public void onConnect(WebSocket socket) {

         connected.add(socket);

     }

     @Override

     public void onClose(WebSocket socket, DataFrame frame) {

         connected.remove(socket);

     }

     @Override

     public void onMessage(WebSocket socket, String text) {

         try {

             ByteArrayInputStream is = new ByteArrayInputStream(text.getBytes("UTF-8"));

             Query q = Models.parse(BrwsrCtx.findDefault(ChatServerResource.class), Query.class, is);

             if (q.getPost() != null) {

                 publish(q.getPost());

             }

             if (q.isAll()) {

                 socket.send(new Reply(true, msgs.toArray(new Message[0])).toString());

             }

         } catch (IOException ex) {

             throw new IllegalStateException(ex);

         }

     }

 }