insomnia/packages/insomnia-app/app/models/cookie-jar.js
2018-12-12 12:36:11 -05:00

87 lines
1.8 KiB
JavaScript

// @flow
import * as db from '../common/database';
import type { BaseModel } from './index';
export const name = 'Cookie Jar';
export const type = 'CookieJar';
export const prefix = 'jar';
export const canDuplicate = true;
export type Cookie = {
id: string,
key: string,
value: string,
expires: Date | string | number | null,
domain: string,
path: string,
secure: boolean,
httpOnly: boolean,
extensions?: Array<any>,
creation?: Date,
creationIndex?: number,
hostOnly?: boolean,
pathIsDefault?: boolean,
lastAccessed?: Date,
};
type BaseCookieJar = {
name: string,
cookies: Array<Cookie>,
};
export type CookieJar = BaseModel & BaseCookieJar;
export function init() {
return {
name: 'Default Jar',
cookies: [],
};
}
export function migrate(doc: CookieJar): CookieJar {
doc = migrateCookieId(doc);
return doc;
}
export function create(patch: Object = {}) {
if (!patch.parentId) {
throw new Error(`New CookieJar missing \`parentId\`: ${JSON.stringify(patch)}`);
}
return db.docCreate(type, patch);
}
export async function getOrCreateForParentId(parentId: string) {
const cookieJars = await db.find(type, { parentId });
if (cookieJars.length === 0) {
return create({ parentId });
} else {
return cookieJars[0];
}
}
export function all() {
return db.all(type);
}
export function getById(id: string) {
return db.get(type, id);
}
export function update(cookieJar: CookieJar, patch: Object = {}) {
return db.docUpdate(cookieJar, patch);
}
/** Ensure every cookie has an ID property */
function migrateCookieId(cookieJar: CookieJar) {
for (const cookie of cookieJar.cookies) {
if (!cookie.id) {
cookie.id = Math.random()
.toString()
.replace('0.', '');
}
}
return cookieJar;
}