aboutsummaryrefslogtreecommitdiff
path: root/src/bun.js/ws.exports.js
blob: 90d3f89114a6973fb3c80e85b4b853b6c32b85d7 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
// this just wraps WebSocket to look like an EventEmitter
// without actually using an EventEmitter polyfill

class BunWebSocket extends globalThis.WebSocket {
  constructor(url, ...args) {
    super(url, ...args);
    this.#wrappedHandlers = new WeakMap();
  }
  #wrappedHandlers = new WeakMap();

  on(event, callback) {
    if (event === "message") {
      var handler = ({ data }) => {
        try {
          callback(data);
        } catch (e) {
          globalThis.reportError(e);
        }
      };

      this.#wrappedHandlers.set(callback, handler);
      this.addEventListener(event, handler);
    } else {
      this.addEventListener(event, callback);
    }
  }

  once(event, callback) {
    if (event === "message") {
      var handler = ({ data }) => {
        try {
          callback(data);
        } catch (e) {
          globalThis.reportError(e);
        }
      };

      this.#wrappedHandlers.set(callback, handler);
      this.addEventListener(event, handler, { once: true });
    } else {
      this.addEventListener(event, callback, { once: true });
    }
  }

  emit(event, data) {
    if (event === "message") {
      this.dispatchEvent(new MessageEvent("message", { data }));
    } else {
      this.dispatchEvent(new CustomEvent(event, { detail: data }));
    }
  }

  off(event, callback) {
    var wrapped = this.#wrappedHandlers.get(callback);
    if (wrapped) {
      this.removeEventListener(event, wrapped);
      this.#wrappedHandlers.delete(callback);
    } else {
      this.removeEventListener(event, callback);
    }
  }
}

BunWebSocket.WebSocket = BunWebSocket;
var WebSocketServer = (BunWebSocket.WebSocketServer = class WebSocketServer {
  constructor() {
    throw new Error("Not implemented yet!");
  }
});

var Sender = (BunWebSocket.Sender = class Sender {
  constructor() {
    throw new Error("Not supported in Bun");
  }
});

var Receiver = (BunWebSocket.Receiver = class Receiver {
  constructor() {
    throw new Error("Not supported in Bun");
  }
});

var createWebSocketStream = (BunWebSocket.createWebSocketStream = function (
  ws
) {
  throw new Error("Not supported in Bun");
});

export default BunWebSocket;

export {
  createWebSocketStream,
  Sender,
  Receiver,
  BunWebSocket as WebSocket,
  WebSocketServer,
};