138 lines
3.5 KiB
JavaScript
138 lines
3.5 KiB
JavaScript
import crypto from 'crypto'
|
|
|
|
const async_build_js_script = async function (path_js_entry_script, build_options) {
|
|
const result = await Bun.build({
|
|
entrypoints: [ path_js_entry_script ],
|
|
env: 'disable',
|
|
minify: false,
|
|
...build_options,
|
|
})
|
|
const str_out = await result.outputs[0].text()
|
|
return str_out
|
|
}
|
|
|
|
const get_siteroot_html = function ({ page_title }) {return `
|
|
<!DOCTYPE html>
|
|
<html lang=en>
|
|
<head>
|
|
<meta charset=utf-8>
|
|
<link rel="icon" href="data:image/svg+xml,<svg xmlns='http://www.w3.org/2000/svg'/>">
|
|
<meta name="viewport" content="width=device-width, initial-scale=1">
|
|
<title>${page_title || 'Page Title'}</title>
|
|
</head>
|
|
<body>
|
|
<script type="text/javascript" src="/index.js">
|
|
</script>
|
|
</body>
|
|
</html>
|
|
`.trim()}
|
|
|
|
const start_server = function ({ context_meta, config, jsbuild_app_frontend }) {
|
|
const server = Bun.serve({
|
|
hostname: config.host || '127.0.0.1',
|
|
port : config.port || 8800,
|
|
async fetch (req, server) {
|
|
const url = new URL(req.url)
|
|
let resp
|
|
if (url.pathname === '/') {
|
|
resp = new Response(get_siteroot_html({ page_title: config.page_title }), { status: 200, headers: { 'Content-Type': 'text/html' } })
|
|
}
|
|
else if (url.pathname === '/index.js') {
|
|
const str_js = await jsbuild_app_frontend()
|
|
resp = new Response(
|
|
await async_build_js_script(import.meta.dir + '/../frontend/index.js', {
|
|
define: {
|
|
JS_APP_FRONTEND: JSON.stringify(str_js),
|
|
}
|
|
}),
|
|
{
|
|
status: 200,
|
|
headers: { 'Content-Type': 'text/javascript' }
|
|
}
|
|
)
|
|
}
|
|
else if (url.pathname === '/ws') {
|
|
const uid = crypto.randomUUID()
|
|
if (server.upgrade(req, {
|
|
data: {
|
|
uid,
|
|
},
|
|
})) {
|
|
// do not return a Response if upgrade succeeds
|
|
resp = undefined
|
|
}
|
|
resp = new Response('Upgrade failed', { status: 500 })
|
|
}
|
|
else {
|
|
resp = new Response(null, { status: 404 })
|
|
}
|
|
return resp
|
|
},
|
|
websocket: {
|
|
open: function (ws) {
|
|
const map_obj = {
|
|
ws,
|
|
closed_by_server: false,
|
|
}
|
|
context_meta.ws_map.set(ws.data.uid, map_obj)
|
|
return
|
|
},
|
|
close: function (ws, code, message) {
|
|
const map_obj = context_meta.ws_map.get(ws.data.uid)
|
|
if (map_obj === undefined) {
|
|
console.warn(['Unexpected: ws_map missing uid', ws.data.uid])
|
|
}
|
|
else {
|
|
if (map_obj.closed_by_server === true) {
|
|
console.info(['ws closed by server', ws.data.uid])
|
|
// cleanup will happen in outer context
|
|
}
|
|
else {
|
|
console.info(['ws closed by client', ws.data.uid, { code, message }])
|
|
// cleanup from this context
|
|
context_meta.ws_map.delete(ws.data.uid)
|
|
}
|
|
}
|
|
return
|
|
},
|
|
message: function (ws, message) {},
|
|
},
|
|
})
|
|
return server
|
|
}
|
|
|
|
const close_active_ws_sessions = function ({ context_meta }) {
|
|
const keys_to_delete = []
|
|
for (const [ uid, map_obj ] of context_meta.ws_map) {
|
|
const code = 1001
|
|
const reason = 'Going Away'
|
|
map_obj.closed_by_server = true
|
|
map_obj.ws.close(code, reason)
|
|
keys_to_delete.push(uid)
|
|
}
|
|
for (let key of keys_to_delete) {
|
|
context_meta.ws_map.delete(key)
|
|
}
|
|
return
|
|
}
|
|
|
|
const async_run = async function ({
|
|
config,
|
|
jsbuild_app_frontend,
|
|
}) {
|
|
const context_meta = {
|
|
ws_map: new Map(),
|
|
}
|
|
const server = start_server({ context_meta, config, jsbuild_app_frontend })
|
|
console.info(server)
|
|
process.on('SIGINT', async () => {
|
|
console.log('SIGINT intercepted')
|
|
close_active_ws_sessions({ context_meta })
|
|
// NOTE: wait for a second for teardown to fully complete
|
|
await Bun.sleep(2000)
|
|
process.exit()
|
|
})
|
|
return
|
|
}
|
|
|
|
export { async_run }
|