wisemapping-frontend/packages/mindplot/src/components/DesignerModel.ts
2023-02-26 20:37:06 -03:00

127 lines
3.4 KiB
TypeScript

/*
* Copyright [2021] [wisemapping]
*
* Licensed under WiseMapping Public License, Version 1.0 (the "License").
* It is basically the Apache License, Version 2.0 (the "License") plus the
* "powered by wisemapping" text requirement on every single page;
* you may not use this file except in compliance with the License.
* You may obtain a copy of the license at
*
* http://www.wisemapping.org/license
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
import { $assert } from '@wisemapping/core-js';
import CentralTopic from './CentralTopic';
import { DesignerOptions } from './DesignerOptionsBuilder';
import Relationship from './Relationship';
import Topic from './Topic';
import { $notify } from './model/ToolbarNotifier';
class DesignerModel {
private _zoom: number;
private _topics: Topic[];
private _relationships: Relationship[];
constructor(options: DesignerOptions) {
this._zoom = options.zoom;
this._topics = [];
this._relationships = [];
}
getZoom(): number {
return this._zoom;
}
setZoom(zoom: number): void {
this._zoom = zoom;
}
getTopics(): Topic[] {
return this._topics;
}
getRelationships(): Relationship[] {
return this._relationships;
}
getCentralTopic(): CentralTopic {
const topics = this.getTopics();
return topics[0] as unknown as CentralTopic;
}
filterSelectedTopics(): Topic[] {
return this._topics.filter((t) => t.isOnFocus());
}
filterSelectedRelationships(): Relationship[] {
return this._relationships.filter((r) => r.isOnFocus());
}
getEntities(): (Relationship | Topic)[] {
let result: (Relationship | Topic)[] = [];
result = result.concat(this._topics);
result = result.concat(this._relationships);
return result;
}
removeTopic(topic: Topic): void {
$assert(topic, 'topic can not be null');
this._topics = this._topics.filter((t) => t !== topic);
}
removeRelationship(rel: Relationship): void {
$assert(rel, 'rel can not be null');
this._relationships = this._relationships.filter((r) => r !== rel);
}
addTopic(topic: Topic): void {
$assert(topic, 'topic can not be null');
$assert(typeof topic.getId() === 'number', `id is not a number:${topic.getId()}`);
this._topics.push(topic);
}
addRelationship(rel: Relationship): void {
$assert(rel, 'rel can not be null');
this._relationships.push(rel);
}
filterTopicsIds(validate?: (topic: Topic) => boolean, errorMsg?: string): number[] {
const result: number[] = [];
const topics = this.filterSelectedTopics();
let isValid = true;
topics.forEach((topic) => {
if (validate) {
isValid = validate(topic);
}
// Add node only if it's valid.
if (isValid) {
result.push(topic.getId());
} else if (errorMsg) {
$notify(errorMsg);
}
});
return result;
}
selectedTopic(): Topic | undefined {
const topics = this.filterSelectedTopics();
return topics.length > 0 ? topics[0] : undefined;
}
findTopicById(id: number): Topic | undefined {
return this._topics.find((t) => t.getId() === id);
}
}
export default DesignerModel;