123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270 |
- /* global attachEvent */
- import XMLHttpRequest from "./xmlhttprequest.js";
- import debugModule from "debug"; // debug()
- import globalThis from "../globalThis.js";
- import { installTimerFunctions, pick } from "../util.js";
- import { Emitter } from "@socket.io/component-emitter";
- import { Polling } from "./polling.js";
- const debug = debugModule("engine.io-client:polling-xhr"); // debug()
- /**
- * Empty function
- */
- function empty() { }
- const hasXHR2 = (function () {
- const xhr = new XMLHttpRequest({
- xdomain: false
- });
- return null != xhr.responseType;
- })();
- export class XHR extends Polling {
- /**
- * XHR Polling constructor.
- *
- * @param {Object} opts
- * @api public
- */
- constructor(opts) {
- super(opts);
- if (typeof location !== "undefined") {
- const isSSL = "https:" === location.protocol;
- let port = location.port;
- // some user agents have empty `location.port`
- if (!port) {
- port = isSSL ? "443" : "80";
- }
- this.xd =
- (typeof location !== "undefined" &&
- opts.hostname !== location.hostname) ||
- port !== opts.port;
- this.xs = opts.secure !== isSSL;
- }
- /**
- * XHR supports binary
- */
- const forceBase64 = opts && opts.forceBase64;
- this.supportsBinary = hasXHR2 && !forceBase64;
- }
- /**
- * Creates a request.
- *
- * @param {String} method
- * @api private
- */
- request(opts = {}) {
- Object.assign(opts, { xd: this.xd, xs: this.xs }, this.opts);
- return new Request(this.uri(), opts);
- }
- /**
- * Sends data.
- *
- * @param {String} data to send.
- * @param {Function} called upon flush.
- * @api private
- */
- doWrite(data, fn) {
- const req = this.request({
- method: "POST",
- data: data
- });
- req.on("success", fn);
- req.on("error", err => {
- this.onError("xhr post error", err);
- });
- }
- /**
- * Starts a poll cycle.
- *
- * @api private
- */
- doPoll() {
- debug("xhr poll");
- const req = this.request();
- req.on("data", this.onData.bind(this));
- req.on("error", err => {
- this.onError("xhr poll error", err);
- });
- this.pollXhr = req;
- }
- }
- export class Request extends Emitter {
- /**
- * Request constructor
- *
- * @param {Object} options
- * @api public
- */
- constructor(uri, opts) {
- super();
- installTimerFunctions(this, opts);
- this.opts = opts;
- this.method = opts.method || "GET";
- this.uri = uri;
- this.async = false !== opts.async;
- this.data = undefined !== opts.data ? opts.data : null;
- this.create();
- }
- /**
- * Creates the XHR object and sends the request.
- *
- * @api private
- */
- create() {
- const opts = pick(this.opts, "agent", "pfx", "key", "passphrase", "cert", "ca", "ciphers", "rejectUnauthorized", "autoUnref");
- opts.xdomain = !!this.opts.xd;
- opts.xscheme = !!this.opts.xs;
- const xhr = (this.xhr = new XMLHttpRequest(opts));
- try {
- debug("xhr open %s: %s", this.method, this.uri);
- xhr.open(this.method, this.uri, this.async);
- try {
- if (this.opts.extraHeaders) {
- xhr.setDisableHeaderCheck && xhr.setDisableHeaderCheck(true);
- for (let i in this.opts.extraHeaders) {
- if (this.opts.extraHeaders.hasOwnProperty(i)) {
- xhr.setRequestHeader(i, this.opts.extraHeaders[i]);
- }
- }
- }
- }
- catch (e) { }
- if ("POST" === this.method) {
- try {
- xhr.setRequestHeader("Content-type", "text/plain;charset=UTF-8");
- }
- catch (e) { }
- }
- try {
- xhr.setRequestHeader("Accept", "*/*");
- }
- catch (e) { }
- // ie6 check
- if ("withCredentials" in xhr) {
- xhr.withCredentials = this.opts.withCredentials;
- }
- if (this.opts.requestTimeout) {
- xhr.timeout = this.opts.requestTimeout;
- }
- xhr.onreadystatechange = () => {
- if (4 !== xhr.readyState)
- return;
- if (200 === xhr.status || 1223 === xhr.status) {
- this.onLoad();
- }
- else {
- // make sure the `error` event handler that's user-set
- // does not throw in the same tick and gets caught here
- this.setTimeoutFn(() => {
- this.onError(typeof xhr.status === "number" ? xhr.status : 0);
- }, 0);
- }
- };
- debug("xhr data %s", this.data);
- xhr.send(this.data);
- }
- catch (e) {
- // Need to defer since .create() is called directly from the constructor
- // and thus the 'error' event can only be only bound *after* this exception
- // occurs. Therefore, also, we cannot throw here at all.
- this.setTimeoutFn(() => {
- this.onError(e);
- }, 0);
- return;
- }
- if (typeof document !== "undefined") {
- this.index = Request.requestsCount++;
- Request.requests[this.index] = this;
- }
- }
- /**
- * Called upon successful response.
- *
- * @api private
- */
- onSuccess() {
- this.emit("success");
- this.cleanup();
- }
- /**
- * Called if we have data.
- *
- * @api private
- */
- onData(data) {
- this.emit("data", data);
- this.onSuccess();
- }
- /**
- * Called upon error.
- *
- * @api private
- */
- onError(err) {
- this.emit("error", err);
- this.cleanup(true);
- }
- /**
- * Cleans up house.
- *
- * @api private
- */
- cleanup(fromError) {
- if ("undefined" === typeof this.xhr || null === this.xhr) {
- return;
- }
- this.xhr.onreadystatechange = empty;
- if (fromError) {
- try {
- this.xhr.abort();
- }
- catch (e) { }
- }
- if (typeof document !== "undefined") {
- delete Request.requests[this.index];
- }
- this.xhr = null;
- }
- /**
- * Called upon load.
- *
- * @api private
- */
- onLoad() {
- const data = this.xhr.responseText;
- if (data !== null) {
- this.onData(data);
- }
- }
- /**
- * Aborts the request.
- *
- * @api public
- */
- abort() {
- this.cleanup();
- }
- }
- Request.requestsCount = 0;
- Request.requests = {};
- /**
- * Aborts pending requests when unloading the window. This is needed to prevent
- * memory leaks (e.g. when using IE) and to ensure that no spurious error is
- * emitted.
- */
- if (typeof document !== "undefined") {
- // @ts-ignore
- if (typeof attachEvent === "function") {
- // @ts-ignore
- attachEvent("onunload", unloadHandler);
- }
- else if (typeof addEventListener === "function") {
- const terminationEvent = "onpagehide" in globalThis ? "pagehide" : "unload";
- addEventListener(terminationEvent, unloadHandler, false);
- }
- }
- function unloadHandler() {
- for (let i in Request.requests) {
- if (Request.requests.hasOwnProperty(i)) {
- Request.requests[i].abort();
- }
- }
- }
|