request-iframe 0.0.1

package/library/core/message-handler.js

@@ -0,0 +1,320 @@
+"use strict";
+
+require("core-js/modules/es.array.filter.js");
+require("core-js/modules/es.object.get-own-property-descriptors.js");
+var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.IframeMessageHandler = void 0;
+require("core-js/modules/es.array.iterator.js");
+require("core-js/modules/es.map.js");
+require("core-js/modules/es.set.js");
+require("core-js/modules/es.string.starts-with.js");
+require("core-js/modules/web.dom-collections.for-each.js");
+require("core-js/modules/web.dom-collections.iterator.js");
+var _defineProperty2 = _interopRequireDefault(require("@babel/runtime/helpers/defineProperty"));
+var _utils = require("../utils");
+var _constants = require("../constants");
+function ownKeys(e, r) { var t = Object.keys(e); if (Object.getOwnPropertySymbols) { var o = Object.getOwnPropertySymbols(e); r && (o = o.filter(function (r) { return Object.getOwnPropertyDescriptor(e, r).enumerable; })), t.push.apply(t, o); } return t; }
+function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t = null != arguments[r] ? arguments[r] : {}; r % 2 ? ownKeys(Object(t), !0).forEach(function (r) { (0, _defineProperty2.default)(e, r, t[r]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(t)) : ownKeys(Object(t)).forEach(function (r) { Object.defineProperty(e, r, Object.getOwnPropertyDescriptor(t, r)); }); } return e; }
+/**
+ * 消息处理器配置
+ */
+
+/**
+ * 消息处理回调
+ */
+
+/**
+ * 事件监听器集合
+ */
+
+/**
+ * 等待确认的响应
+ */
+
+/**
+ * 等待响应的请求
+ */
+
+/**
+ * IframeMessageHandler - 底层消息处理器
+ * 负责 postMessage 监听、协议验证、uniqueKey 隔离、事件分发
+ */
+class IframeMessageHandler {
+  /** 消息监听函数（绑定 this） */
+
+  constructor(options) {
+    var _options$ackTimeout;
+    (0, _defineProperty2.default)(this, "listeners", {
+      request: new Set(),
+      ack: new Set(),
+      async: new Set(),
+      response: new Set(),
+      error: new Set(),
+      received: new Set(),
+      ping: new Set(),
+      pong: new Set()
+    });
+    /** 等待客户端确认的响应 */
+    (0, _defineProperty2.default)(this, "pendingAcks", new Map());
+    /** 等待响应的请求 */
+    (0, _defineProperty2.default)(this, "pendingRequests", new Map());
+    /** 消息处理回调 */
+    (0, _defineProperty2.default)(this, "callbacks", {});
+    this.uniqueKey = options === null || options === void 0 ? void 0 : options.uniqueKey;
+    this.ackTimeout = (_options$ackTimeout = options === null || options === void 0 ? void 0 : options.ackTimeout) !== null && _options$ackTimeout !== void 0 ? _options$ackTimeout : _constants.DefaultTimeout.SERVER_ACK;
+    this.boundOnMessage = this.onMessage.bind(this);
+    window.addEventListener('message', this.boundOnMessage);
+  }
+
+  /**
+   * 设置消息处理回调
+   */
+  setCallbacks(callbacks) {
+    this.callbacks = _objectSpread(_objectSpread({}, this.callbacks), callbacks);
+  }
+
+  /**
+   * 消息处理入口
+   */
+  onMessage(event) {
+    var data = event.data;
+
+    // 检查是否是 request-iframe 框架的消息
+    if (!(0, _utils.isValidPostMessage)(data)) {
+      return;
+    }
+
+    // 检查协议版本兼容性
+    var version = (0, _utils.getProtocolVersion)(data);
+    if (version !== undefined && !(0, _utils.isCompatibleVersion)(version)) {
+      this.sendProtocolError(data, event, version);
+      return;
+    }
+
+    // uniqueKey 隔离
+    if (this.uniqueKey) {
+      if (data.uniqueKey !== this.uniqueKey) return;
+    } else {
+      if (data.uniqueKey) return;
+    }
+
+    // 触发事件监听器
+    this.emit(data.type, data, event);
+
+    // 分发到具体处理器
+    this.dispatchMessage(data, event);
+  }
+
+  /**
+   * 分发消息到具体处理器
+   */
+  dispatchMessage(data, event) {
+    var type = data.type;
+
+    // 请求
+    if (type === _constants.MessageType.REQUEST) {
+      var _this$callbacks$onReq, _this$callbacks;
+      (_this$callbacks$onReq = (_this$callbacks = this.callbacks).onRequest) === null || _this$callbacks$onReq === void 0 || _this$callbacks$onReq.call(_this$callbacks, data, event);
+      return;
+    }
+
+    // 客户端响应
+    if (type === _constants.MessageType.RESPONSE || type === _constants.MessageType.ERROR || type === _constants.MessageType.ACK || type === _constants.MessageType.ASYNC) {
+      var _this$callbacks$onCli, _this$callbacks2;
+      this.handleClientResponse(data, event);
+      (_this$callbacks$onCli = (_this$callbacks2 = this.callbacks).onClientResponse) === null || _this$callbacks$onCli === void 0 || _this$callbacks$onCli.call(_this$callbacks2, data, event);
+      return;
+    }
+
+    // received 确认
+    if (type === _constants.MessageType.RECEIVED) {
+      var _this$callbacks$onRec, _this$callbacks3;
+      this.handleReceived(data);
+      (_this$callbacks$onRec = (_this$callbacks3 = this.callbacks).onReceived) === null || _this$callbacks$onRec === void 0 || _this$callbacks$onRec.call(_this$callbacks3, data, event);
+      return;
+    }
+
+    // ping
+    if (type === _constants.MessageType.PING) {
+      var _this$callbacks$onPin, _this$callbacks4;
+      this.handlePing(data, event);
+      (_this$callbacks$onPin = (_this$callbacks4 = this.callbacks).onPing) === null || _this$callbacks$onPin === void 0 || _this$callbacks$onPin.call(_this$callbacks4, data, event);
+      return;
+    }
+
+    // pong
+    if (type === _constants.MessageType.PONG) {
+      var _this$callbacks$onPon, _this$callbacks5;
+      this.handlePong(data, event);
+      (_this$callbacks$onPon = (_this$callbacks5 = this.callbacks).onPong) === null || _this$callbacks$onPon === void 0 || _this$callbacks$onPon.call(_this$callbacks5, data, event);
+      return;
+    }
+
+    // 流消息（stream_start, stream_data, stream_end, stream_error, stream_cancel）
+    if (type.startsWith('stream_')) {
+      var _this$callbacks$onStr, _this$callbacks6;
+      (_this$callbacks$onStr = (_this$callbacks6 = this.callbacks).onStream) === null || _this$callbacks$onStr === void 0 || _this$callbacks$onStr.call(_this$callbacks6, data, event);
+      return;
+    }
+  }
+
+  /**
+   * 处理客户端响应
+   */
+  handleClientResponse(data, event) {
+    var pending = this.pendingRequests.get(data.requestId);
+    if (pending) {
+      // 验证 origin
+      if (pending.origin && pending.origin !== '*' && event.origin !== pending.origin) {
+        return;
+      }
+      // ack 和 async 不删除 pending
+      if (data.type === _constants.MessageType.ACK || data.type === _constants.MessageType.ASYNC) {
+        pending.resolve(data);
+        return;
+      }
+      // response 和 error 删除 pending
+      this.pendingRequests.delete(data.requestId);
+      pending.resolve(data);
+    }
+  }
+
+  /**
+   * 处理 received 确认
+   */
+  handleReceived(data) {
+    var pending = this.pendingAcks.get(data.requestId);
+    if (pending) {
+      clearTimeout(pending.timeoutId);
+      this.pendingAcks.delete(data.requestId);
+      pending.resolve(true);
+    }
+  }
+
+  /**
+   * 处理 ping
+   */
+  handlePing(data, event) {
+    if (!event.source) return;
+    event.source.postMessage((0, _utils.createPostMessage)(_constants.MessageType.PONG, data.requestId, {
+      uniqueKey: data.uniqueKey
+    }), event.origin);
+  }
+
+  /**
+   * 处理 pong
+   */
+  handlePong(data, event) {
+    var pending = this.pendingRequests.get(data.requestId);
+    if (pending) {
+      if (pending.origin && pending.origin !== '*' && event.origin !== pending.origin) {
+        return;
+      }
+      this.pendingRequests.delete(data.requestId);
+      pending.resolve(data);
+    }
+  }
+
+  /**
+   * 注册等待确认的响应
+   */
+  registerPendingAck(requestId, resolve, reject) {
+    var timeoutId = setTimeout(() => {
+      this.pendingAcks.delete(requestId);
+      resolve(false);
+    }, this.ackTimeout);
+    this.pendingAcks.set(requestId, {
+      resolve,
+      reject,
+      timeoutId
+    });
+  }
+
+  /**
+   * 注册等待响应的请求
+   */
+  registerPendingRequest(requestId, resolve, reject, origin) {
+    this.pendingRequests.set(requestId, {
+      resolve,
+      reject,
+      origin
+    });
+  }
+
+  /**
+   * 取消等待响应
+   */
+  unregisterPendingRequest(requestId) {
+    this.pendingRequests.delete(requestId);
+  }
+
+  /**
+   * 监听事件
+   */
+  on(event, fn) {
+    this.listeners[event].add(fn);
+  }
+
+  /**
+   * 取消监听
+   */
+  off(event, fn) {
+    if (!fn) {
+      this.listeners[event].clear();
+      return;
+    }
+    this.listeners[event].delete(fn);
+  }
+
+  /**
+   * 触发事件
+   */
+  emit(event, payload, messageEvent) {
+    this.listeners[event].forEach(fn => {
+      try {
+        fn(payload, messageEvent);
+      } catch (e) {
+        // 忽略监听器异常
+      }
+    });
+  }
+
+  /**
+   * 添加 path 前缀
+   */
+  prefixPath(path) {
+    return this.uniqueKey ? `${this.uniqueKey}:${path}` : path;
+  }
+
+  /**
+   * 发送协议版本错误
+   */
+  sendProtocolError(data, event, remoteVersion) {
+    if (!event.source) return;
+    event.source.postMessage((0, _utils.createPostMessage)(_constants.MessageType.ERROR, data.requestId, {
+      path: data.path,
+      uniqueKey: data.uniqueKey,
+      status: _constants.HttpStatus.BAD_REQUEST,
+      statusText: _constants.Messages.PROTOCOL_VERSION_UNSUPPORTED,
+      error: {
+        message: (0, _constants.formatMessage)(_constants.Messages.PROTOCOL_VERSION_TOO_LOW, remoteVersion, _constants.ProtocolVersion.MIN_SUPPORTED),
+        code: _constants.ErrorCode.PROTOCOL_UNSUPPORTED
+      }
+    }), event.origin);
+  }
+
+  /**
+   * 销毁
+   */
+  destroy() {
+    window.removeEventListener('message', this.boundOnMessage);
+    this.pendingRequests.clear();
+    this.pendingAcks.forEach(pending => clearTimeout(pending.timeoutId));
+    this.pendingAcks.clear();
+    Object.keys(this.listeners).forEach(k => this.listeners[k].clear());
+  }
+}
+exports.IframeMessageHandler = IframeMessageHandler;

package/library/core/request-response.d.ts

@@ -0,0 +1,59 @@
+import { ServerRequest, ServerResponse, PostMessageData, CookieOptions, SendOptions, SendFileOptions } from '../types';
+import { IframeWritableStream } from '../stream';
+/**
+ * ServerRequest 实现
+ */
+export declare class ServerRequestImpl implements ServerRequest {
+    body: any;
+    headers: Record<string, string>;
+    cookies: Record<string, string>;
+    path: string;
+    requestId: string;
+    origin: string;
+    source: Window;
+    res: ServerResponse;
+    constructor(data: PostMessageData, event: MessageEvent, response: ServerResponse);
+}
+/**
+ * 等待客户端确认收到的回调
+ */
+type AckCallback = (received: boolean) => void;
+/**
+ * ServerResponse 实现
+ */
+export declare class ServerResponseImpl implements ServerResponse {
+    statusCode: number;
+    headers: Record<string, string | string[]>;
+    private readonly requestId;
+    private readonly path;
+    private readonly uniqueKey?;
+    private readonly targetWindow;
+    private readonly targetOrigin;
+    private onAckCallback?;
+    _sent: boolean;
+    constructor(requestId: string, path: string, uniqueKey: string | undefined, targetWindow: Window, targetOrigin: string);
+    /**
+     * 设置等待客户端确认的回调
+     */
+    _setOnAckCallback(callback: AckCallback): void;
+    /**
+     * 触发客户端确认回调
+     */
+    _triggerAck(received: boolean): void;
+    send(data: any, options?: SendOptions): Promise<boolean>;
+    json(data: any, options?: SendOptions): Promise<boolean>;
+    sendFile(content: string | Blob | File, options?: SendFileOptions): Promise<boolean>;
+    /**
+     * 发送流响应
+     * 将流绑定到当前请求上下文并启动流传输
+     */
+    sendStream(stream: IframeWritableStream): Promise<void>;
+    status(code: number): ServerResponse;
+    setHeader(name: string, value: string | number | string[]): void;
+    set(name: string, value: string | number | string[]): ServerResponse;
+    cookie(name: string, value: string, options?: CookieOptions): ServerResponse;
+    clearCookie(name: string, options?: CookieOptions): ServerResponse;
+    private blobToBase64;
+}
+export {};
+//# sourceMappingURL=request-response.d.ts.map

package/library/core/request-response.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"request-response.d.ts","sourceRoot":"","sources":["../../src/core/request-response.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,cAAc,EAAE,eAAe,EAAE,aAAa,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAGvH,OAAO,EAAE,oBAAoB,EAAE,MAAM,WAAW,CAAC;AAEjD;;GAEG;AACH,qBAAa,iBAAkB,YAAW,aAAa;IAC9C,IAAI,EAAE,GAAG,CAAC;IACV,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;IACf,GAAG,EAAE,cAAc,CAAC;gBAGzB,IAAI,EAAE,eAAe,EACrB,KAAK,EAAE,YAAY,EACnB,QAAQ,EAAE,cAAc;CAiB3B;AAED;;GAEG;AACH,KAAK,WAAW,GAAG,CAAC,QAAQ,EAAE,OAAO,KAAK,IAAI,CAAC;AAE/C;;GAEG;AACH,qBAAa,kBAAmB,YAAW,cAAc;IAChD,UAAU,EAAE,MAAM,CAAiB;IACnC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC,CAAM;IACvD,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAS;IAC9B,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAS;IACpC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAS;IACtC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAS;IACtC,OAAO,CAAC,aAAa,CAAC,CAAc;IAC7B,KAAK,UAAS;gBAGnB,SAAS,EAAE,MAAM,EACjB,IAAI,EAAE,MAAM,EACZ,SAAS,EAAE,MAAM,GAAG,SAAS,EAC7B,YAAY,EAAE,MAAM,EACpB,YAAY,EAAE,MAAM;IAStB;;OAEG;IACI,iBAAiB,CAAC,QAAQ,EAAE,WAAW,GAAG,IAAI;IAIrD;;OAEG;IACI,WAAW,CAAC,QAAQ,EAAE,OAAO,GAAG,IAAI;IAOpC,IAAI,CAAC,IAAI,EAAE,GAAG,EAAE,OAAO,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC;IA0CxD,IAAI,CAAC,IAAI,EAAE,GAAG,EAAE,OAAO,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC;IAKlD,QAAQ,CACnB,OAAO,EAAE,MAAM,GAAG,IAAI,GAAG,IAAI,EAC7B,OAAO,CAAC,EAAE,eAAe,GACxB,OAAO,CAAC,OAAO,CAAC;IAyDnB;;;OAGG;IACU,UAAU,CAAC,MAAM,EAAE,oBAAoB,GAAG,OAAO,CAAC,IAAI,CAAC;IAgB7D,MAAM,CAAC,IAAI,EAAE,MAAM,GAAG,cAAc;IAKpC,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM,EAAE,GAAG,IAAI;IAwBhE,GAAG,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM,EAAE,GAAG,cAAc;IAMpE,MAAM,CACX,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,MAAM,EACb,OAAO,CAAC,EAAE,aAAa,GACtB,cAAc;IA4BV,WAAW,CAChB,IAAI,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE,aAAa,GACtB,cAAc;YAkBH,YAAY;CAa3B"}