!7

cmodule ScribbleDemo > DynPrintLog {
  !include #1029545 // API for Eleu

  transient Set<virtual WebSocket> webSockets = syncWeakHashSet();

  O html(virtual Request request) {
    S uri = cast get(request, 'uri);
    SS params = cast get(request, 'params);
    ret hhtml(hhead(htitle("Scribble")
      + loadJQuery()
      + hreconnectingWebSockets()
      + hcss([[
        body { margin: 0 !important; padding: 0 !important; }
        #peopleCounter {
          position: absolute;
          top: 10px; right: 10px;
          width: 100px;
          text-align: right;
          z-index: 1000;
          color: white;
        }
      ]]))
      + hbody(
        hfulltag canvas("", id := "scribble")
        + div("?", id := "peopleCounter")
        + hjssnippet(#1029696)
        + hjs([[
          function resized() {
            canvas.width = document.body.clientWidth;
            canvas.height = document.body.clientHeight;
          }
          resized();
          $(window).bind("resize", resized);

          // JavaScript WebSocket handling
          var wsReady = false;
          var ws = new ReconnectingWebSocket(((window.location.protocol === "https:") ? "wss://" : "ws://") + window.location.host + "/");
          ws.onopen = function(event) { wsReady = true; };
          ws.onmessage = function(event) {
            var data = JSON.parse(event.data);
            if (data.counter)
              $("#peopleCounter").html(data.counter);
            else
              drawStroke(data.x1, data.y1, data.x2, data.y2, data.mouseDown);
          };
          sendStroke = function(data) {
            //console.log(data);
            if (wsReady)
              ws.send(JSON.stringify(data));
          };
        ]])
      ));
  }
  
  void handleWebSocket(virtual WebSocket ws) {
    set(ws, onClose := r { webSockets.remove(ws); wsCounterUpdate(); });
    
    setFieldToIVF1Proxy(ws, onMessage := msg -> { temp enter(); pcall {
      S data = rcall_string getTextPayload(msg);
      distributeEvent(data, ws);
    }});

    set(ws, onOpen := rEnter {  
      print("WebSocket opened!");
      webSockets.add(ws); wsCounterUpdate();
    });
  }

  void wsCounterUpdate {
    distributeEvent(jsonEncode(litmap(counter := nUsers(webSockets) + " online")));
  }

  void distributeEvent(S data, virtual WebSocket sendingWebSocket default null) {
    L<virtual WebSocket> sockets = cloneList(webSockets);
    print("Distributing event to " + n2(sockets, "web socket") + ": " + data);
    for (virtual WebSocket ws : sockets) pcall {
      if (ws == sendingWebSocket) continue;
      try {
        call(ws, "send", data);
      } catch e {
        print("Removing faulty WebSocket");
        webSockets.remove(ws); wsCounterUpdate();
        call(ws, "close");
      }
    }
  }
}