| /** |
| * @license |
| * Copyright Google Inc. All Rights Reserved. |
| * |
| * Use of this source code is governed by an MIT-style license that can be |
| * found in the LICENSE file at https://angular.io/license |
| */ |
| /** |
| * @fileoverview |
| * @suppress {missingRequire} |
| */ |
| |
| Zone.__load_patch('fetch', (global: any, Zone: ZoneType, api: _ZonePrivate) => { |
| interface FetchTaskData extends TaskData { |
| fetchArgs?: any[]; |
| } |
| let fetch = global['fetch']; |
| if (typeof fetch !== 'function') { |
| return; |
| } |
| const originalFetch = global[api.symbol('fetch')]; |
| if (originalFetch) { |
| // restore unpatched fetch first |
| fetch = originalFetch; |
| } |
| const ZoneAwarePromise = global.Promise; |
| const symbolThenPatched = api.symbol('thenPatched'); |
| const fetchTaskScheduling = api.symbol('fetchTaskScheduling'); |
| const fetchTaskAborting = api.symbol('fetchTaskAborting'); |
| const OriginalAbortController = global['AbortController']; |
| const supportAbort = typeof OriginalAbortController === 'function'; |
| let abortNative: Function|null = null; |
| if (supportAbort) { |
| global['AbortController'] = function() { |
| const abortController = new OriginalAbortController(); |
| const signal = abortController.signal; |
| signal.abortController = abortController; |
| return abortController; |
| }; |
| abortNative = api.patchMethod( |
| OriginalAbortController.prototype, 'abort', |
| (delegate: Function) => (self: any, args: any) => { |
| if (self.task) { |
| return self.task.zone.cancelTask(self.task); |
| } |
| return delegate.apply(self, args); |
| }); |
| } |
| const placeholder = function() {}; |
| global['fetch'] = function() { |
| const args = Array.prototype.slice.call(arguments); |
| const options = args.length > 1 ? args[1] : null; |
| const signal = options && options.signal; |
| return new Promise((res, rej) => { |
| const task = Zone.current.scheduleMacroTask( |
| 'fetch', placeholder, {fetchArgs: args} as FetchTaskData, |
| () => { |
| let fetchPromise; |
| let zone = Zone.current; |
| try { |
| (zone as any)[fetchTaskScheduling] = true; |
| fetchPromise = fetch.apply(this, args); |
| } catch (error) { |
| rej(error); |
| return; |
| } finally { |
| (zone as any)[fetchTaskScheduling] = false; |
| } |
| |
| if (!(fetchPromise instanceof ZoneAwarePromise)) { |
| let ctor = fetchPromise.constructor; |
| if (!ctor[symbolThenPatched]) { |
| api.patchThen(ctor); |
| } |
| } |
| fetchPromise.then( |
| (resource: any) => { |
| if (task.state !== 'notScheduled') { |
| task.invoke(); |
| } |
| res(resource); |
| }, |
| (error: any) => { |
| if (task.state !== 'notScheduled') { |
| task.invoke(); |
| } |
| rej(error); |
| }); |
| }, |
| () => { |
| if (!supportAbort) { |
| rej('No AbortController supported, can not cancel fetch'); |
| return; |
| } |
| if (signal && signal.abortController && !signal.aborted && |
| typeof signal.abortController.abort === 'function' && abortNative) { |
| try { |
| (Zone.current as any)[fetchTaskAborting] = true; |
| abortNative.call(signal.abortController); |
| } finally { |
| (Zone.current as any)[fetchTaskAborting] = false; |
| } |
| } else { |
| rej('cancel fetch need a AbortController.signal'); |
| } |
| }); |
| if (signal && signal.abortController) { |
| signal.abortController.task = task; |
| } |
| }); |
| }; |
| }); |