Blame view

lib/jsdom/browser/resources/per-document-resource-loader.js 2.62 KB
858f2bdf5   Boyan Georgiev   fixes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
  "use strict";
  const idlUtils = require("../../living/generated/utils");
  const { fireAnEvent } = require("../../living/helpers/events");
  
  module.exports = class PerDocumentResourceLoader {
    constructor(document) {
      this._document = document;
      this._defaultEncoding = document._encoding;
      this._resourceLoader = document._defaultView ? document._defaultView._resourceLoader : null;
      this._requestManager = document._requestManager;
      this._queue = document._queue;
      this._deferQueue = document._deferQueue;
      this._asyncQueue = document._asyncQueue;
    }
  
    fetch(url, { element, onLoad, onError }) {
      const request = this._resourceLoader.fetch(url, {
        cookieJar: this._document._cookieJar,
        element: idlUtils.wrapperForImpl(element),
        referrer: this._document.URL
      });
  
      if (request === null) {
        return null;
      }
  
      this._requestManager.add(request);
  
      const onErrorWrapped = error => {
        this._requestManager.remove(request);
  
        if (onError) {
          onError(error);
        }
  
        fireAnEvent("error", element);
  
        const err = new Error(`Could not load ${element.localName}: "${url}"`);
        err.type = "resource loading";
        err.detail = error;
  
        this._document._defaultView._virtualConsole.emit("jsdomError", err);
  
        return Promise.resolve();
      };
  
      const onLoadWrapped = data => {
        this._requestManager.remove(request);
  
        this._addCookies(url, request.response ? request.response.headers : {});
  
        try {
          const result = onLoad ? onLoad(data) : undefined;
  
          return Promise.resolve(result)
            .then(() => {
              fireAnEvent("load", element);
  
              return Promise.resolve();
            })
            .catch(err => {
              return onErrorWrapped(err);
            });
        } catch (err) {
          return onErrorWrapped(err);
        }
      };
  
      if (element.localName === "script" && element.hasAttributeNS(null, "async")) {
        this._asyncQueue.push(request, onLoadWrapped, onErrorWrapped, this._queue.getLastScript());
      } else if (element.localName === "script" && element.hasAttributeNS(null, "defer")) {
        this._deferQueue.push(request, onLoadWrapped, onErrorWrapped, false, element);
      } else {
        this._queue.push(request, onLoadWrapped, onErrorWrapped, false, element);
      }
  
      return request;
    }
  
    _addCookies(url, headers) {
      let cookies = headers["set-cookie"];
  
      if (!cookies) {
        return;
      }
  
      if (!Array.isArray(cookies)) {
        cookies = [cookies];
      }
  
      cookies.forEach(cookie => {
        this._document._cookieJar.setCookieSync(cookie, url, { http: true, ignoreError: true });
      });
    }
  };