insomnia/app/models/index.js

// @flow
import * as _stats from './stats';
import * as _settings from './settings';
import * as _workspace from './workspace';
import * as _workspaceMeta from './workspace-meta';
import * as _environment from './environment';
import * as _cookieJar from './cookie-jar';
import * as _requestGroup from './request-group';
import * as _requestGroupMeta from './request-group-meta';
import * as _request from './request';
import * as _requestVersion from './request-version';
import * as _requestMeta from './request-meta';
import * as _response from './response';
import * as _oAuth2Token from './o-auth-2-token';
import {generateId} from '../common/misc';

export type BaseModel = {
  _id: string,
  modified: number,
  created: number
}

// Reference to each model
export const stats = _stats;
export const settings = _settings;
export const workspace = _workspace;
export const workspaceMeta = _workspaceMeta;
export const environment = _environment;
export const cookieJar = _cookieJar;
export const requestGroup = _requestGroup;
export const requestGroupMeta = _requestGroupMeta;
export const request = _request;
export const requestVersion = _requestVersion;
export const requestMeta = _requestMeta;
export const response = _response;
export const oAuth2Token = _oAuth2Token;

const _models = {
  [stats.type]: stats,
  [settings.type]: settings,
  [workspace.type]: workspace,
  [workspaceMeta.type]: workspaceMeta,
  [environment.type]: environment,
  [cookieJar.type]: cookieJar,
  [requestGroup.type]: requestGroup,
  [requestGroupMeta.type]: requestGroupMeta,
  [request.type]: request,
  [requestVersion.type]: requestVersion,
  [requestMeta.type]: requestMeta,
  [response.type]: response,
  [oAuth2Token.type]: oAuth2Token
};

export function all () {
  return Object.keys(_models).map(type => _models[type]);
}

export function types () {
  return all().map(model => model.type);
}

export function getModel (type: string) {
  return _models[type] || null;
}

export function canDuplicate (type: string) {
  const model = getModel(type);
  return model ? model.canDuplicate : false;
}

export function getModelName (type: string, count: number = 1) {
  const model = getModel(type);
  if (!model) {
    return 'Unknown';
  } else if (count === 1) {
    return model.name;
  } else if (!model.name.match(/s$/)) {
    // Add an 's' if it doesn't already end in one
    return `${model.name}s`;
  } else {
    return model.name;
  }
}

export function initModel (type: string, ...sources: Array<Object>) {
  const model = getModel(type);

  if (!model) {
    throw new Error(`Tried to init invalid model type ${type}`);
  }

  // Define global default fields
  const objectDefaults = Object.assign({
    _id: null,
    type: type,
    parentId: null,
    modified: Date.now(),
    created: Date.now()
  }, model.init());

  const fullObject = Object.assign({}, objectDefaults, ...sources);

  // Generate an _id if there isn't one yet
  if (!fullObject._id) {
    fullObject._id = generateId(model.prefix);
  }

  // Migrate the model
  // NOTE: Do migration before pruning because we might need to look at those fields
  model.migrate(fullObject);

  // Prune extra keys from doc
  for (const key of Object.keys(fullObject)) {
    if (!objectDefaults.hasOwnProperty(key)) {
      delete fullObject[key];
    }
  }

  return fullObject;
}