insomnia/packages/insomnia-app/app/models/grpc-request-meta.js

// @flow
import * as db from '../common/database';
import type { BaseModel } from './index';
import { isGrpcRequestId } from './helpers/is-model';

export const name = 'gRPC Request Meta';
export const type = 'GrpcRequestMeta';
export const prefix = 'greqm';
export const canDuplicate = false;
export const canSync = false;

type BaseGrpcRequestMeta = {
  pinned: boolean,
  lastActive: number,
};

export type GrpcRequestMeta = BaseModel & BaseGrpcRequestMeta;

export function init() {
  return {
    pinned: false,
    lastActive: 0,
  };
}

export function migrate(doc: GrpcRequestMeta): GrpcRequestMeta {
  return doc;
}

export function create(patch: $Shape<GrpcRequestMeta> = {}): Promise<GrpcRequestMeta> {
  if (!patch.parentId) {
    throw new Error('New GrpcRequestMeta missing `parentId`');
  }

  expectParentToBeGrpcRequest(patch.parentId);

  return db.docCreate(type, patch);
}

export function update(
  requestMeta: GrpcRequestMeta,
  patch: $Shape<GrpcRequestMeta>,
): Promise<GrpcRequestMeta> {
  expectParentToBeGrpcRequest(patch.parentId || requestMeta.parentId);
  return db.docUpdate(requestMeta, patch);
}

export function getByParentId(parentId: string): Promise<GrpcRequestMeta> {
  expectParentToBeGrpcRequest(parentId);
  return db.getWhere(type, { parentId });
}

export async function getOrCreateByParentId(parentId: string): Promise<GrpcRequestMeta> {
  const requestMeta = await getByParentId(parentId);

  if (requestMeta) {
    return requestMeta;
  }

  return create({ parentId });
}

export async function updateOrCreateByParentId(parentId: string, patch: $Shape<GrpcRequestMeta>) {
  const requestMeta = await getByParentId(parentId);

  if (requestMeta) {
    return update(requestMeta, patch);
  } else {
    const newPatch = Object.assign({ parentId }, patch);
    return create(newPatch);
  }
}

export function all(): Promise<Array<GrpcRequestMeta>> {
  return db.all(type);
}

function expectParentToBeGrpcRequest(parentId: string) {
  if (!isGrpcRequestId(parentId)) {
    throw new Error('Expected the parent of GrpcRequestMeta to be a GrpcRequest');
  }
}