mirror of
https://github.com/Kong/insomnia
synced 2024-11-08 14:49:53 +00:00
86 lines
2.2 KiB
JavaScript
86 lines
2.2 KiB
JavaScript
import * as _stats from './stats';
|
|
import * as _settings from './settings';
|
|
import * as _workspace from './workspace';
|
|
import * as _environment from './environment';
|
|
import * as _cookieJar from './cookieJar';
|
|
import * as _requestGroup from './requestGroup';
|
|
import * as _request from './request';
|
|
import * as _response from './response';
|
|
|
|
// Reference to each model
|
|
export const stats = _stats;
|
|
export const settings = _settings;
|
|
export const workspace = _workspace;
|
|
export const environment = _environment;
|
|
export const cookieJar = _cookieJar;
|
|
export const requestGroup = _requestGroup;
|
|
export const request = _request;
|
|
export const response = _response;
|
|
|
|
const _models = {
|
|
[stats.type]: stats,
|
|
[settings.type]: settings,
|
|
[workspace.type]: workspace,
|
|
[environment.type]: environment,
|
|
[cookieJar.type]: cookieJar,
|
|
[requestGroup.type]: requestGroup,
|
|
[request.type]: request,
|
|
[response.type]: response,
|
|
};
|
|
|
|
export function all () {
|
|
return Object.keys(_models).map(type => _models[type]);
|
|
}
|
|
|
|
export function types () {
|
|
return all().map(model => model.type);
|
|
}
|
|
|
|
export function getModel (type) {
|
|
return _models[type] || null;
|
|
}
|
|
|
|
export function getModelName (type, count = 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, ...sources) {
|
|
const model = getModel(type);
|
|
|
|
// Define global default fields
|
|
const objectDefaults = Object.assign({
|
|
type: type,
|
|
_id: null,
|
|
_schema: 0,
|
|
parentId: null,
|
|
modified: Date.now(),
|
|
created: Date.now(),
|
|
}, model.init());
|
|
|
|
// Make a new object
|
|
const fullObject = Object.assign({}, objectDefaults, ...sources);
|
|
|
|
// Migrate the model
|
|
// NOTE: Do migration before pruning because we might need to look at those fields
|
|
const migratedObject = model.migrate(fullObject);
|
|
|
|
// Prune extra keys from doc
|
|
for (const key of Object.keys(migratedObject)) {
|
|
if (!objectDefaults.hasOwnProperty(key)) {
|
|
delete migratedObject[key];
|
|
}
|
|
}
|
|
|
|
return migratedObject;
|
|
}
|