86 lines
1.8 KiB
JavaScript
86 lines
1.8 KiB
JavaScript
import WebSocket from "ws"
|
|
|
|
class client {
|
|
constructor() {
|
|
this.url = `ws://localhost:${10000}`
|
|
this.ws = null
|
|
this.seq = 0
|
|
this.handlers = new Map()
|
|
console.log(process.cwd())
|
|
}
|
|
|
|
connect() {
|
|
return new Promise((resolve, reject) => {
|
|
this.ws = new WebSocket(this.url)
|
|
|
|
this.ws.on("message", msg => this.onMessage(msg))
|
|
|
|
this.ws.on("open", () => {
|
|
console.log("[forms] connected")
|
|
resolve()
|
|
})
|
|
|
|
this.ws.on("error", err => {
|
|
reject(err)
|
|
})
|
|
|
|
this.ws.on("close", () => {
|
|
console.log("[forms] disconnected")
|
|
})
|
|
})
|
|
}
|
|
|
|
register(type, schema) {
|
|
this.ws.send(JSON.stringify({
|
|
op: "register",
|
|
type,
|
|
schema
|
|
}))
|
|
}
|
|
|
|
add(type, data) {
|
|
this.ws.send(JSON.stringify({
|
|
op: "add",
|
|
type,
|
|
data
|
|
}))
|
|
}
|
|
|
|
watch(form, handler) {
|
|
if (!this.handlers.has(form)) {
|
|
this.handlers.set(form, new Set())
|
|
}
|
|
|
|
this.handlers.get(form).add(handler)
|
|
|
|
this.ws.send(JSON.stringify({
|
|
op: "replay",
|
|
form,
|
|
fromSeq: this.seq
|
|
}))
|
|
}
|
|
|
|
onMessage(raw) {
|
|
const msg = JSON.parse(raw.toString())
|
|
|
|
if (msg.type === "hello") {
|
|
this.seq = msg.seq
|
|
return
|
|
}
|
|
|
|
if (msg.type === "add") {
|
|
this.seq = msg.seq
|
|
|
|
const set = this.handlers.get(msg.form)
|
|
if (set) {
|
|
for (const fn of set) {
|
|
fn(msg.data, msg)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
export default {
|
|
client: client
|
|
} |