net_alchim31_livereload.LRWebSocketHandler Maven / Gradle / Ivy

Go to download
package net_alchim31_livereload;

import java.io.IOException;
import java.util.concurrent.ConcurrentLinkedQueue;

import javax.servlet.http.HttpServletRequest;

import org.eclipse.jetty.websocket.WebSocket;
import org.eclipse.jetty.websocket.WebSocketHandler;

class LRWebSocketHandler extends WebSocketHandler {
  final ConcurrentLinkedQueue _broadcast = new ConcurrentLinkedQueue();
  final LRProtocol                         _protocol  = new LRProtocol();

  @Override
  public WebSocket doWebSocketConnect(HttpServletRequest request, String protocol) {
    if ("/livereload".equals(request.getPathInfo())) {
      return new LRWebSocket();
    }
    return new WebSocket() {
      @Override
      public void onOpen(Connection connection) {
        connection.close();
      }

      @Override
      public void onClose(int code, String msg) {}
    };
  }

  public void notifyChange(String path) throws Exception {
    String msg = _protocol.reload(path);
    for (LRWebSocket ws : _broadcast) {
      try {
        ws._connection.sendMessage(msg);
      } catch (IOException e) {
        _broadcast.remove(ws);
        e.printStackTrace();
      }
    }
  }

  class LRWebSocket implements WebSocket.OnTextMessage {
    protected Connection _connection;

    @Override
    public void onOpen(Connection connection) {
      _connection = connection;
      _broadcast.add(this);
    }

    @Override
    public void onClose(int code, String message) {
      _broadcast.remove(this);
    }

    @Override
    public void onMessage(final String data) {
      try {
        if (_protocol.isHello(data)) {
          _connection.sendMessage(_protocol.hello());
        }
      } catch (Exception exc) {
        exc.printStackTrace();
      }
    }
  }
}