| // Copyright 2017 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| SDK.CookieModel = class extends SDK.SDKModel { |
| /** |
| * @param {!SDK.Target} target |
| */ |
| constructor(target) { |
| super(target); |
| } |
| |
| /** |
| * @param {!Protocol.Network.Cookie} protocolCookie |
| * @return {!SDK.Cookie} |
| */ |
| static _parseProtocolCookie(protocolCookie) { |
| const cookie = new SDK.Cookie(protocolCookie.name, protocolCookie.value, null); |
| cookie.addAttribute('domain', protocolCookie['domain']); |
| cookie.addAttribute('path', protocolCookie['path']); |
| cookie.addAttribute('port', protocolCookie['port']); |
| if (protocolCookie['expires']) |
| cookie.addAttribute('expires', protocolCookie['expires'] * 1000); |
| if (protocolCookie['httpOnly']) |
| cookie.addAttribute('httpOnly'); |
| if (protocolCookie['secure']) |
| cookie.addAttribute('secure'); |
| if (protocolCookie['sameSite']) |
| cookie.addAttribute('sameSite', protocolCookie['sameSite']); |
| cookie.setSize(protocolCookie['size']); |
| return cookie; |
| } |
| |
| /** |
| * @param {!SDK.Cookie} cookie |
| * @param {string} resourceURL |
| * @return {boolean} |
| */ |
| static cookieMatchesResourceURL(cookie, resourceURL) { |
| const url = resourceURL.asParsedURL(); |
| if (!url || !SDK.CookieModel.cookieDomainMatchesResourceDomain(cookie.domain(), url.host)) |
| return false; |
| return ( |
| url.path.startsWith(cookie.path()) && (!cookie.port() || url.port === cookie.port()) && |
| (!cookie.secure() || url.scheme === 'https')); |
| } |
| |
| /** |
| * @param {string} cookieDomain |
| * @param {string} resourceDomain |
| * @return {boolean} |
| */ |
| static cookieDomainMatchesResourceDomain(cookieDomain, resourceDomain) { |
| if (cookieDomain.charAt(0) !== '.') |
| return resourceDomain === cookieDomain; |
| return !!resourceDomain.match( |
| new RegExp('^([^\\.]+\\.)*' + cookieDomain.substring(1).escapeForRegExp() + '$', 'i')); |
| } |
| |
| /** |
| * @param {!Array<string>} urls |
| * @return {!Promise<!Array<!SDK.Cookie>>} |
| */ |
| getCookies(urls) { |
| return this.target().networkAgent().getCookies(urls).then( |
| cookies => (cookies || []).map(cookie => SDK.CookieModel._parseProtocolCookie(cookie))); |
| } |
| |
| /** |
| * @param {!SDK.Cookie} cookie |
| * @param {function()=} callback |
| */ |
| deleteCookie(cookie, callback) { |
| this._deleteAll([cookie], callback); |
| } |
| |
| /** |
| * @param {string=} domain |
| * @param {function()=} callback |
| */ |
| clear(domain, callback) { |
| this.getCookiesForDomain(domain || null).then(cookies => this._deleteAll(cookies, callback)); |
| } |
| |
| /** |
| * @param {!SDK.Cookie} cookie |
| * @return {!Promise<boolean>} |
| */ |
| saveCookie(cookie) { |
| let domain = cookie.domain(); |
| if (!domain.startsWith('.')) |
| domain = ''; |
| let expires = undefined; |
| if (cookie.expires()) |
| expires = Math.floor(Date.parse(cookie.expires()) / 1000); |
| return this.target() |
| .networkAgent() |
| .setCookie( |
| cookie.name(), cookie.value(), cookie.url(), domain, cookie.path(), cookie.secure(), cookie.httpOnly(), |
| cookie.sameSite(), expires) |
| .then(success => !!success); |
| } |
| |
| /** |
| * @param {?string} domain |
| * @return {!Promise<!Array<!SDK.Cookie>>} |
| */ |
| getCookiesForDomain(domain) { |
| const resourceURLs = []; |
| /** |
| * @param {!SDK.Resource} resource |
| */ |
| function populateResourceURLs(resource) { |
| const url = resource.documentURL.asParsedURL(); |
| if (url && (!domain || url.securityOrigin() === domain)) |
| resourceURLs.push(resource.url); |
| } |
| const resourceTreeModel = this.target().model(SDK.ResourceTreeModel); |
| if (resourceTreeModel) |
| resourceTreeModel.forAllResources(populateResourceURLs); |
| return this.getCookies(resourceURLs); |
| } |
| |
| /** |
| * @param {!Array<!SDK.Cookie>} cookies |
| * @param {function()=} callback |
| */ |
| _deleteAll(cookies, callback) { |
| const networkAgent = this.target().networkAgent(); |
| Promise |
| .all( |
| cookies.map(cookie => networkAgent.deleteCookies(cookie.name(), undefined, cookie.domain(), cookie.path()))) |
| .then(callback || function() {}); |
| } |
| }; |
| |
| SDK.SDKModel.register(SDK.CookieModel, SDK.Target.Capability.Network, false); |