insomnia/packages/insomnia-app/app/ui/redux/modules/entities.js

import clone from 'clone';
import * as db from '../../../common/database';
import * as models from '../../../models';

const ENTITY_CHANGES = 'entities/changes';
const ENTITY_INITIALIZE = 'entities/initialize';

// ~~~~~~~~ //
// Reducers //
// ~~~~~~~~ //

function getReducerName(type) {
  let trailer = 's';
  let chop = 0;

  // Things already ending with 's' stay that way
  if (type.match(/s$/)) {
    trailer = '';
    chop = 0;
  }

  // Things ending in 'y' convert to ies
  if (type.match(/y$/)) {
    trailer = 'ies';
    chop = 1;
  }

  // Lowercase first letter (camel case)
  const lowerFirstLetter = `${type.slice(0, 1).toLowerCase()}${type.slice(1)}`;

  // Add the trailer for pluralization
  return `${lowerFirstLetter.slice(0, lowerFirstLetter.length - chop)}${trailer}`;
}

const initialState = {};

for (const type of models.types()) {
  initialState[getReducerName(type)] = {};
}

export function reducer(state = initialState, action) {
  switch (action.type) {
    case ENTITY_INITIALIZE:
      const freshState = clone(initialState);
      const { docs } = action;
      for (const doc of docs) {
        const referenceName = getReducerName(doc.type);
        freshState[referenceName][doc._id] = doc;
      }
      return freshState;
    case ENTITY_CHANGES:
      const newState = clone(state);
      const { changes } = action;

      for (const [event, doc] of changes) {
        const referenceName = getReducerName(doc.type);

        switch (event) {
          case db.CHANGE_INSERT:
          case db.CHANGE_UPDATE:
            newState[referenceName][doc._id] = doc;
            break;

          case db.CHANGE_REMOVE:
            delete newState[referenceName][doc._id];
            break;

          default:
            break;
        }
      }

      return newState;
    default:
      return state;
  }
}

// ~~~~~~~ //
// Actions //
// ~~~~~~~ //

export function addChanges(changes) {
  return dispatch => {
    setTimeout(() => {
      dispatch(addChangesSync(changes));
    });
  };
}

export function addChangesSync(changes) {
  return { type: ENTITY_CHANGES, changes };
}

export function initialize() {
  return async dispatch => {
    const docs = await allDocs();
    dispatch(initializeWith(docs));
  };
}

export function initializeWith(docs) {
  return { type: ENTITY_INITIALIZE, docs };
}

export async function allDocs() {
  // NOTE: This list should be from most to least specific (ie. parents above children)
  return [
    ...(await models.settings.all()),
    ...(await models.workspace.all()),
    ...(await models.workspaceMeta.all()),
    ...(await models.environment.all()),
    ...(await models.cookieJar.all()),
    ...(await models.requestGroup.all()),
    ...(await models.requestGroupMeta.all()),
    ...(await models.request.all()),
    ...(await models.requestMeta.all()),
    ...(await models.requestVersion.all()),
    ...(await models.response.all()),
    ...(await models.oAuth2Token.all()),
    ...(await models.clientCertificate.all()),
  ];
}