@akanjs/server 0.0.53 → 0.0.55

package/src/websocket.mjs

@@ -0,0 +1,124 @@
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __decorateClass = (decorators, target, key, kind) => {
+  var result = kind > 1 ? void 0 : kind ? __getOwnPropDesc(target, key) : target;
+  for (var i = decorators.length - 1, decorator; i >= 0; i--)
+    if (decorator = decorators[i])
+      result = (kind ? decorator(target, key, result) : decorator(result)) || result;
+  if (kind && result)
+    __defProp(target, key, result);
+  return result;
+};
+import { lowerlize } from "@akanjs/common";
+import { Access, Account, getBodyPipes, Me, Self, UserIp, Ws } from "@akanjs/nest";
+import { isServiceEnabled } from "@akanjs/service";
+import { getArgMetas, getGqlMetas, getSigMeta } from "@akanjs/signal";
+import { Inject, Injectable, UseInterceptors } from "@nestjs/common";
+import { MessageBody, SubscribeMessage, WebSocketGateway, WebSocketServer } from "@nestjs/websockets";
+import { map } from "rxjs/operators";
+const internalArgMap = { Account, UserIp, Access, Self, Me, Ws };
+let TransformInterceptor = class {
+  intercept(context, next) {
+    const [, gqlKey] = [context.getArgByIndex(1), context.getArgByIndex(3)];
+    return next.handle().pipe(map((data) => ({ event: gqlKey, data })));
+  }
+};
+TransformInterceptor = __decorateClass([
+  Injectable()
+], TransformInterceptor);
+const makeRoomId = (gqlKey, argValues) => `${gqlKey}-${argValues.join("-")}`;
+const getPubsubInterceptor = (argMetas) => {
+  let PubsubInterceptor = class {
+    async intercept(context, next) {
+      const [socket, { __subscribe__, ...body }, gqlKey] = [context.getArgByIndex(0), context.getArgByIndex(1), context.getArgByIndex(3)];
+      const roomId = makeRoomId(
+        gqlKey,
+        argMetas.map((argMeta) => body[argMeta.name])
+      );
+      if (__subscribe__)
+        await socket.join(roomId);
+      else
+        await socket.leave(roomId);
+      return next.handle().pipe(map(() => ({ event: gqlKey, data: { roomId, __subscribe__ } })));
+    }
+  };
+  PubsubInterceptor = __decorateClass([
+    Injectable()
+  ], PubsubInterceptor);
+  return PubsubInterceptor;
+};
+const websocketOf = (sigRef, allSrvs) => {
+  const sigMeta = getSigMeta(sigRef);
+  class WsGateway {
+    __sigRef__ = sigRef;
+  }
+  Object.keys(allSrvs).forEach((srv) => {
+    if (!isServiceEnabled(allSrvs[srv]))
+      return;
+    Inject(allSrvs[srv])(WsGateway.prototype, lowerlize(srv));
+  });
+  const messageGqlMetas = getGqlMetas(sigRef).filter((gqlMeta) => gqlMeta.type === "Message");
+  for (const gqlMeta of messageGqlMetas) {
+    const descriptor = { ...Object.getOwnPropertyDescriptor(sigRef.prototype, gqlMeta.key) ?? {} };
+    Object.defineProperty(WsGateway.prototype, gqlMeta.key, descriptor);
+    const [argMetas, internalArgMetas] = getArgMetas(sigRef, gqlMeta.key);
+    argMetas.forEach((argMeta) => {
+      if (argMeta.type !== "Msg")
+        throw new Error(`Argument of Message should be Msg ${sigMeta.refName}-${gqlMeta.key}-${argMeta.name}`);
+      MessageBody(argMeta.name, ...getBodyPipes(argMeta))(WsGateway.prototype, gqlMeta.key, argMeta.idx);
+    });
+    internalArgMetas.forEach((internalArgMeta) => {
+      const internalDecorator = internalArgMap[internalArgMeta.type];
+      internalDecorator(internalArgMeta.option ?? {})(WsGateway.prototype, gqlMeta.key, internalArgMeta.idx);
+    });
+    UseInterceptors(TransformInterceptor)(WsGateway.prototype, gqlMeta.key, gqlMeta.descriptor);
+    SubscribeMessage(gqlMeta.key)(WsGateway.prototype, gqlMeta.key, descriptor);
+  }
+  const pubsubGqlMetas = getGqlMetas(sigRef).filter((gqlMeta) => gqlMeta.type === "Pubsub");
+  for (const gqlMeta of pubsubGqlMetas) {
+    const descriptor = { ...Object.getOwnPropertyDescriptor(sigRef.prototype, gqlMeta.key) ?? {} };
+    Object.defineProperty(WsGateway.prototype, gqlMeta.key, descriptor);
+    const [argMetas, internalArgMetas] = getArgMetas(sigRef, gqlMeta.key);
+    argMetas.forEach((argMeta) => {
+      if (argMeta.type !== "Room")
+        throw new Error(`Argument of Message should be Room ${sigMeta.refName}-${gqlMeta.key}-${argMeta.name}`);
+      MessageBody(argMeta.name, ...getBodyPipes(argMeta))(WsGateway.prototype, gqlMeta.key, argMeta.idx);
+    });
+    internalArgMetas.forEach((internalArgMeta) => {
+      const internalDecorator = internalArgMap[internalArgMeta.type];
+      internalDecorator(internalArgMeta.option ?? {})(WsGateway.prototype, gqlMeta.key, internalArgMeta.idx);
+    });
+    UseInterceptors(getPubsubInterceptor(argMetas))(WsGateway.prototype, gqlMeta.key, gqlMeta.descriptor);
+    SubscribeMessage(gqlMeta.key)(WsGateway.prototype, gqlMeta.key, descriptor);
+  }
+  WebSocketGateway({ cors: { origin: "*" }, transports: ["websocket"] })(WsGateway);
+  return WsGateway;
+};
+const websocketServerOf = (sigRef) => {
+  const pubsubGqlMetas = getGqlMetas(sigRef).filter((gqlMeta) => gqlMeta.type === "Pubsub");
+  let Websocket = class {
+    server;
+  };
+  __decorateClass([
+    WebSocketServer()
+  ], Websocket.prototype, "server", 2);
+  Websocket = __decorateClass([
+    Injectable(),
+    WebSocketGateway({ cors: { origin: "*" }, transports: ["websocket"] })
+  ], Websocket);
+  for (const gqlMeta of pubsubGqlMetas) {
+    const [argMetas] = getArgMetas(sigRef, gqlMeta.key);
+    Websocket.prototype[gqlMeta.key] = function(...args) {
+      const roomId = makeRoomId(
+        gqlMeta.key,
+        argMetas.map((argMeta) => args[argMeta.idx])
+      );
+      this.server.to(roomId).emit(roomId, args.at(-1));
+    };
+  }
+  return Websocket;
+};
+export {
+  websocketOf,
+  websocketServerOf
+};