// @flow import type { BaseModel } from './index'; import * as db from '../common/database'; import * as models from './index'; export const name = 'Workspace'; export const type = 'Workspace'; export const prefix = 'wrk'; export const canDuplicate = true; type BaseWorkspace = { name: string, description: string, }; export type Workspace = BaseModel & BaseWorkspace; export function init() { return { name: 'New Workspace', description: '', }; } export async function migrate(doc: Workspace): Promise { return _migrateExtractClientCertificates(doc); } export function getById(id: string): Promise { return db.get(type, id); } export async function create(patch: Object = {}): Promise { return db.docCreate(type, patch); } export async function all(): Promise> { const workspaces = await db.all(type); if (workspaces.length === 0) { await create({ name: 'Insomnia' }); return all(); } else { return workspaces; } } export function count() { return db.count(type); } export function update(workspace: Workspace, patch: Object): Promise { return db.docUpdate(workspace, patch); } export function remove(workspace: Workspace): Promise { return db.remove(workspace); } async function _migrateExtractClientCertificates(workspace: Workspace): Promise { const certificates = (workspace: Object).certificates || null; if (!Array.isArray(certificates)) { // Already migrated return workspace; } for (const cert of certificates) { await models.clientCertificate.create({ parentId: workspace._id, host: cert.host || '', passphrase: cert.passphrase || null, cert: cert.cert || null, key: cert.key || null, pfx: cert.pfx || null, isPrivate: false, }); } delete (workspace: Object).certificates; // This will remove the now-missing `certificates` property // NOTE: Using db.update so we don't change things like modified time await db.update(workspace); return workspace; }