mirror of
https://github.com/denoland/deno.git
synced 2024-12-12 18:42:18 -05:00
80 lines
2.1 KiB
TypeScript
80 lines
2.1 KiB
TypeScript
import { listenAndServe } from "../../http/server.ts";
|
|
import {
|
|
acceptWebSocket,
|
|
acceptable,
|
|
WebSocket,
|
|
isWebSocketCloseEvent,
|
|
} from "../../ws/mod.ts";
|
|
import { fromFileUrl } from "../../path/mod.ts";
|
|
|
|
const clients = new Map<number, WebSocket>();
|
|
let clientId = 0;
|
|
function dispatch(msg: string): void {
|
|
for (const client of clients.values()) {
|
|
client.send(msg);
|
|
}
|
|
}
|
|
async function wsHandler(ws: WebSocket): Promise<void> {
|
|
const id = ++clientId;
|
|
clients.set(id, ws);
|
|
dispatch(`Connected: [${id}]`);
|
|
for await (const msg of ws) {
|
|
console.log(`msg:${id}`, msg);
|
|
if (typeof msg === "string") {
|
|
dispatch(`[${id}]: ${msg}`);
|
|
} else if (isWebSocketCloseEvent(msg)) {
|
|
clients.delete(id);
|
|
dispatch(`Closed: [${id}]`);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
listenAndServe({ port: 8080 }, async (req) => {
|
|
if (req.method === "GET" && req.url === "/") {
|
|
//Serve with hack
|
|
const u = new URL("./index.html", import.meta.url);
|
|
if (u.protocol.startsWith("http")) {
|
|
// server launched by deno run http(s)://.../server.ts,
|
|
fetch(u.href).then(async (resp) => {
|
|
const body = new Uint8Array(await resp.arrayBuffer());
|
|
return req.respond({
|
|
status: resp.status,
|
|
headers: new Headers({
|
|
"content-type": "text/html",
|
|
}),
|
|
body,
|
|
});
|
|
});
|
|
} else {
|
|
// server launched by deno run ./server.ts
|
|
const file = await Deno.open(fromFileUrl(u));
|
|
req.respond({
|
|
status: 200,
|
|
headers: new Headers({
|
|
"content-type": "text/html",
|
|
}),
|
|
body: file,
|
|
});
|
|
}
|
|
}
|
|
if (req.method === "GET" && req.url === "/favicon.ico") {
|
|
req.respond({
|
|
status: 302,
|
|
headers: new Headers({
|
|
location: "https://deno.land/favicon.ico",
|
|
}),
|
|
});
|
|
}
|
|
if (req.method === "GET" && req.url === "/ws") {
|
|
if (acceptable(req)) {
|
|
acceptWebSocket({
|
|
conn: req.conn,
|
|
bufReader: req.r,
|
|
bufWriter: req.w,
|
|
headers: req.headers,
|
|
}).then(wsHandler);
|
|
}
|
|
}
|
|
});
|
|
console.log("chat server starting on :8080....");
|