feat: support to control websockets in a easy way

brisa-build · Sep 26, 2023 · cf0e26e · cf0e26e
1 parent 2647487
commit cf0e26e
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 4 deletions.
diff --git a/src/brisa/request-context/index.ts b/src/brisa/request-context/index.ts
@@ -1,14 +1,16 @@
-import { MatchedRoute } from "bun";
+import { MatchedRoute, ServerWebSocket } from "bun";
 import { I18nFromRequest } from "../../types";
 
 export default class RequestContext extends Request {
  constructor(request: Request, route?: MatchedRoute) {
  super(request);
  this.route = route;
  this.context = new Map<string, any>();
+ this.ws = globalThis.ws;
  }
 
  route?: MatchedRoute;
  context: Map<string, any>;
  i18n?: I18nFromRequest;
+ ws?: ServerWebSocket<unknown>;
 }
diff --git a/src/cli/serve.tsx b/src/cli/serve.tsx
@@ -10,6 +10,7 @@ import importFileIfExists from "../utils/import-file-if-exists";
 import getConstants from "../constants";
 import handleI18n from "../utils/handle-i18n";
 import redirectTrailingSlash from "../utils/redirect-trailing-slash";
+import getImportableFilepath from "../utils/get-importable-filepath";
 
 const {
  IS_PRODUCTION,
@@ -22,6 +23,9 @@ const {
  ASSETS_DIR,
 } = getConstants();
 
+const WEBSOCKET_PATH = getImportableFilepath("websocket", ROOT_DIR);
+const wsModule = WEBSOCKET_PATH ? await import(WEBSOCKET_PATH) : null
+
 declare global {
  var ws: ServerWebSocket<unknown> | undefined;
 }
@@ -92,13 +96,18 @@ Bun.serve({
  websocket: {
  open: (ws: ServerWebSocket<unknown>) => {
  globalThis.ws = ws;
+ wsModule?.open?.(ws);
  },
- close: () => {
+ close: (...args) => {
  globalThis.ws = undefined;
+ wsModule?.close?.(...args);
  },
- message: () => {
- /* void */
+ message: (ws, message) => {
+ wsModule?.message?.(ws, message);
  },
+ drain: (ws) => {
+ wsModule?.drain?.(ws);
+ }
  },
 });
 

diff --git a/src/types/index.d.ts b/src/types/index.d.ts
@@ -2,6 +2,7 @@ export interface RequestContext extends Request {
  context: Map<string, any>;
  route?: MatchedRoute;
  i18n?: I18nFromRequest;
+ ws?: ServerWebSocket<unknown>;
 }
 
 type Props = Record<string, unknown> & {

diff --git a/src/utils/load-layout/index.tsx b/src/utils/load-layout/index.tsx
@@ -14,6 +14,8 @@ export default async function LoadLayout({
  return (
  <html>
  <head>
+ <meta charset="UTF-8" />
+ <meta name="viewport" content="width=device-width, initial-scale=1.0" />
  <title>Brisa</title>
  </head>
  <body>{children}</body>