insomnia/app/models/workspace.js

// @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,
  certificates: Array<{
    host: string,
    passphrase: string,
    cert: string,
    key: string,
    pfx: string
  }>
};

export type Workspace = BaseModel & BaseWorkspace;

export function init () {
  return {
    name: 'New Workspace',
    description: '',
    certificates: [
      // {host, port, cert, key, pfx, passphrase}
    ]
  };
}

export async function migrate (doc: Workspace): Promise<Workspace> {
  // There was a bug on import that would set this to the current workspace ID.
  // Let's remove it here so that nothing bad happens.
  if (doc.parentId !== null) {
    // Save it to the DB for this one
    process.nextTick(() => update(doc, {parentId: null}));
  }

  await _ensureDependencies(doc);

  return doc;
}

export function getById (id: string): Promise<Workspace | null> {
  return db.get(type, id);
}

export async function create (patch: Object = {}): Promise<Workspace> {
  const doc = await db.docCreate(type, patch);
  await _ensureDependencies(doc);
  return doc;
}

export async function all (): Promise<Array<Workspace>> {
  const workspaces = await db.all(type);

  if (workspaces.length === 0) {
    await create({name: 'Insomnia'});
    return await all();
  } else {
    return workspaces;
  }
}

export function count () {
  return db.count(type);
}

export function update (workspace: Workspace, patch: Object): Promise<Workspace> {
  return db.docUpdate(workspace, patch);
}

export function remove (workspace: Workspace): Promise<void> {
  return db.remove(workspace);
}

async function _ensureDependencies (workspace: Workspace) {
  await models.cookieJar.getOrCreateForParentId(workspace._id);
  await models.environment.getOrCreateForWorkspaceId(workspace._id);
}