mirror of
https://github.com/DanielnetoDotCom/YouPHPTube
synced 2025-10-03 17:59:55 +02:00
54 lines
1.7 KiB
JavaScript
54 lines
1.7 KiB
JavaScript
const { promisify } = require('util')
|
|
const tough = require('tough-cookie')
|
|
|
|
module.exports = function fetchCookieDecorator (fetch, jar, ignoreError = true) {
|
|
fetch = fetch || window.fetch
|
|
jar = jar || new tough.CookieJar()
|
|
|
|
const getCookieString = promisify(jar.getCookieString.bind(jar))
|
|
const setCookie = promisify(jar.setCookie.bind(jar))
|
|
|
|
async function fetchCookie (url, opts) {
|
|
opts = opts || {}
|
|
|
|
// Prepare request
|
|
const cookie = await getCookieString(typeof url === 'string' ? url : url.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, { ignoreError })))
|
|
|
|
return res
|
|
}
|
|
|
|
return fetchCookie
|
|
}
|