insomnia/app/models/response.js

// @flow
import type {BaseModel} from './index';
import * as models from './index';

import fs from 'fs';
import crypto from 'crypto';
import path from 'path';
import mkdirp from 'mkdirp';
import * as electron from 'electron';
import {MAX_RESPONSES} from '../common/constants';
import * as db from '../common/database';
import {compress, decompress} from '../common/misc';

export const name = 'Response';
export const type = 'Response';
export const prefix = 'res';
export const canDuplicate = false;

export type ResponseHeader = {
  name: string,
  value: string
}

export type ResponseTimelineEntry = {
  name: string,
  value: string
}

type BaseResponse = {
  statusCode: number,
  statusMessage: string,
  contentType: string,
  url: string,
  bytesRead: number,
  elapsedTime: number,
  headers: Array<ResponseHeader>,
  timeline: Array<ResponseTimelineEntry>,
  bodyPath: string, // Actual bodies are stored on the filesystem
  error: string,
  requestVersionId: string | null,

  // Things from the request
  settingStoreCookies: boolean | null,
  settingSendCookies: boolean | null
};

export type Response = BaseModel & BaseResponse;

export function init (): BaseResponse {
  return {
    statusCode: 0,
    statusMessage: '',
    contentType: '',
    url: '',
    bytesRead: 0,
    elapsedTime: 0,
    headers: [],
    cookies: [],
    timeline: [],
    bodyPath: '', // Actual bodies are stored on the filesystem
    error: '',
    requestVersionId: null,

    // Things from the request
    settingStoreCookies: null,
    settingSendCookies: null
  };
}

export function migrate (doc: Object) {
  doc = migrateBody(doc);
  return doc;
}

export function getById (id: string) {
  return db.get(type, id);
}

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

export async function removeForRequest (parentId: string) {
  await db.removeWhere(type, {parentId});
}

export function remove (response: Response) {
  return db.remove(response);
}

export function findRecentForRequest (requestId: string, limit: number) {
  return db.findMostRecentlyModified(type, {parentId: requestId}, limit);
}

export async function getLatestForRequest (requestId: string) {
  const responses = await findRecentForRequest(requestId, 1);
  return responses[0] || null;
}

export async function create (patch: Object = {}, bodyBuffer: Buffer | null = null) {
  if (!patch.parentId) {
    throw new Error('New Response missing `parentId`');
  }

  const {parentId} = patch;

  // Create request version snapshot
  const request = await models.request.getById(parentId);
  const requestVersion = request ? await models.requestVersion.create(request) : null;
  patch.requestVersionId = requestVersion ? requestVersion._id : null;

  // Delete all other responses before creating the new one
  const allResponses = await db.findMostRecentlyModified(type, {parentId}, MAX_RESPONSES);
  const recentIds = allResponses.map(r => r._id);
  await db.removeWhere(type, {parentId, _id: {$nin: recentIds}});

  // Actually create the new response
  const bodyPath = bodyBuffer ? storeBodyBuffer(bodyBuffer) : '';
  return db.docCreate(type, {bodyPath}, patch);
}

export function getLatestByParentId (parentId: string) {
  return db.getMostRecentlyModified(type, {parentId});
}

export function getBodyBuffer (response: Response, readFailureValue: any = null) {
  // No body, so return empty Buffer
  if (!response.bodyPath) {
    return new Buffer([]);
  }

  try {
    return decompress(fs.readFileSync(response.bodyPath));
  } catch (err) {
    console.warn('Failed to read response body', err.message);
    return readFailureValue;
  }
}

export function storeBodyBuffer (bodyBuffer: Buffer | null) {
  const root = electron.remote.app.getPath('userData');
  const dir = path.join(root, 'responses');

  mkdirp.sync(dir);

  const hash = crypto.createHash('md5').update(bodyBuffer || '').digest('hex');
  const fullPath = path.join(dir, `${hash}.zip`);

  try {
    fs.writeFileSync(fullPath, compress(bodyBuffer || Buffer.from('')));
  } catch (err) {
    console.warn('Failed to write response body to file', err.message);
  }

  return fullPath;
}

function migrateBody (doc: Object) {
  if (doc.hasOwnProperty('body') && doc._id && !doc.bodyPath) {
    const bodyBuffer = Buffer.from(doc.body, doc.encoding || 'utf8');
    const bodyPath = storeBodyBuffer(bodyBuffer);
    const newDoc = Object.assign(doc, {bodyPath});
    db.docUpdate(newDoc);
    return newDoc;
  } else {
    return doc;
  }
}