You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

55 lines
1.7 KiB

4 years ago
var denodeify = require('es6-denodeify')(Promise)
var tough = require('tough-cookie')
module.exports = function fetchCookieDecorator (fetch, jar) {
fetch = fetch || window.fetch
jar = jar || new tough.CookieJar()
var getCookieString = denodeify(jar.getCookieString.bind(jar))
var setCookie = denodeify(jar.setCookie.bind(jar))
async function fetchCookie (url, opts) {
opts = opts || {}
// Prepare request
const cookie = await getCookieString(url)
if (url.headers && typeof url.headers.append === 'function') {
url.headers.append('cookie', cookie)
} else if (opts.headers && typeof opts.headers.append === 'function') {
opts.headers.append('cookie', cookie)
} else {
opts.headers = Object.assign(
opts.headers || {},
cookie ? { cookie: cookie } : {}
)
}
// Actual request
const res = await fetch(url, opts)
// Get cookie header
var cookies = []
if (res.headers.getAll) {
// node-fetch v1
cookies = res.headers.getAll('set-cookie')
// console.warn("You are using a fetch version that supports 'Headers.getAll' which is deprecated!")
// console.warn("In the future 'fetch-cookie-v2' may discontinue supporting that fetch implementation.")
// console.warn('Details: https://developer.mozilla.org/en-US/docs/Web/API/Headers/getAll')
} else {
// node-fetch v2
const headers = res.headers.raw()
if (headers['set-cookie'] !== undefined) {
cookies = headers['set-cookie']
}
}
// Store all present cookies
await Promise.all(cookies.map((cookie) => setCookie(cookie, res.url)))
return res
}
return fetchCookie
}