insomnia/packages/insomnia-app/app/models/o-auth-2-token.js

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

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

type BaseOAuth2Token = {
  refreshToken: string,
  accessToken: string,
  expiresAt: number | null, // Should be Date.now() if valid

  // Error handling
  error: string,
  errorDescription: string,
  errorUri: string
};

export type OAuth2Token = BaseModel & BaseOAuth2Token;

export const name = 'OAuth 2.0 Token';
export const type = 'OAuth2Token';
export const prefix = 'oa2';
export const canDuplicate = false;

export function init (): BaseOAuth2Token {
  return {
    refreshToken: '',
    accessToken: '',
    expiresAt: null, // Should be Date.now() if valid

    // Error handling
    error: '',
    errorDescription: '',
    errorUri: ''
  };
}

export function migrate<T> (doc: T): T {
  return doc;
}

export function create (patch: Object = {}): Promise<OAuth2Token> {
  if (!patch.parentId) {
    throw new Error(`New OAuth2Token missing \`parentId\` ${JSON.stringify(patch)}`);
  }

  return db.docCreate(type, patch);
}

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

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

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

export async function getOrCreateByParentId (parentId: string): Promise<OAuth2Token> {
  let token = await db.getWhere(type, {parentId});

  if (!token) {
    token = await create({parentId});
  }

  return token;
}

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