mirror of
https://github.com/Kong/insomnia
synced 2024-11-08 14:49:53 +00:00
121 lines
3.3 KiB
JavaScript
121 lines
3.3 KiB
JavaScript
// @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;
|
|
}
|