tachybase_todo/packages/core/data-source-manager/src/collection.ts

57 lines
1.4 KiB
TypeScript

import merge from 'deepmerge';
import { default as lodash } from 'lodash';
import { CollectionField } from './collection-field';
import { CollectionOptions, ICollection, ICollectionManager, IField, IRepository } from './types';
export class Collection implements ICollection {
repository: IRepository;
fields: Map<string, IField> = new Map<string, IField>();
constructor(
protected options: CollectionOptions,
protected collectionManager: ICollectionManager,
) {
this.setRepository(options.repository);
if (options.fields) {
for (const field of options.fields) {
this.setField(field.name, field);
}
}
}
updateOptions(options: CollectionOptions, mergeOptions?: any) {
let newOptions = lodash.cloneDeep(options);
newOptions = merge(this.options, newOptions, mergeOptions);
this.options = newOptions;
if (options.repository) {
this.setRepository(options.repository);
}
return this;
}
setField(name: string, options: any) {
const field = new CollectionField(options);
this.fields.set(name, field);
return field;
}
removeField(name: string) {
this.fields.delete(name);
}
getField(name: string) {
return this.fields.get(name);
}
getFields() {
return [...this.fields.values()];
}
protected setRepository(repository: any) {
this.repository = this.collectionManager.getRegisteredRepository(repository || 'Repository');
}
}