insomnia/app/common/cookies.js
2017-11-03 19:32:22 +01:00

77 lines
1.7 KiB
JavaScript

import {CookieJar, Cookie} from 'tough-cookie';
/**
* Get a list of cookie objects from a request.jar()
*
* @param jar
*/
export function cookiesFromJar (jar) {
return new Promise(resolve => {
jar.store.getAllCookies((err, cookies) => {
if (err) {
console.warn('Failed to get cookies form jar', err);
resolve([]);
} else {
// NOTE: Perform toJSON so we have a plain JS object instead of Cookie instance
resolve(cookies.map(c => c.toJSON()));
}
});
});
}
/**
* Get cookies header
* @param jar
* @param uri
* @returns {Promise}
*/
export function cookieHeaderValueForUri (jar, uri) {
return new Promise(resolve => {
jar.getCookies(uri, (err, cookies) => {
cookies = err ? [] : cookies;
resolve(cookies.map(c => c.cookieString()).join('; '));
});
});
}
/**
* Get a request.jar() from a list of cookie objects
*
* @param cookies
*/
export function jarFromCookies (cookies) {
let jar;
try {
// For some reason, fromJSON modifies `cookies`. Create a copy first
// just to be sure
const copy = JSON.stringify({cookies});
jar = CookieJar.fromJSON(copy);
} catch (e) {
console.log('Failed to initialize cookie jar', e);
jar = new CookieJar();
}
jar.rejectPublicSuffixes = false;
jar.looseMode = true;
return jar;
}
export function cookieToString (cookie) {
// Cookie can either be a plain JS object or Cookie instance
if (!(cookie instanceof Cookie)) {
cookie = Cookie.fromJSON(cookie);
}
let str = cookie.toString();
// tough-cookie toString() doesn't put domain on all the time.
// This hack adds when tough-cookie won't
if (cookie.domain && cookie.hostOnly) {
str += '; Domain=' + cookie.domain;
}
return str;
}