wisemapping-frontend/packages/mindplot/src/components/persistence/Beta2PelaMigrator.ts

/*
 *    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, $defined } from '@wisemapping/core-js';
import { Mindmap } from '../..';
import NodeModel from '../model/NodeModel';
import ModelCodeName from './ModelCodeName';
import XMLMindmapSerializer from './XMLMindmapSerializer';
import XMLSerializerPela from './XMLSerializerPela';

class Beta2PelaMigrator implements XMLMindmapSerializer {
  private _betaSerializer: XMLMindmapSerializer;
  private _pelaSerializer: XMLSerializerPela;
  constructor(betaSerializer: XMLSerializerPela) {
    this._betaSerializer = betaSerializer;
    this._pelaSerializer = new XMLSerializerPela();
  }

  toXML(mindmap: Mindmap) {
    return this._pelaSerializer.toXML(mindmap);
  }

  loadFromDom(dom: Document, mapId: string): Mindmap {
    $assert($defined(mapId), 'mapId can not be null');
    const mindmap = this._betaSerializer.loadFromDom(dom, mapId);
    mindmap.setVersion(ModelCodeName.PELA);

    // Beta does not set position on second level nodes ...
    const branches = mindmap.getBranches();
    const me = this;
    branches.forEach((model) => {
      me._fixPosition(model);
    });

    return mindmap;
  }

  private _fixPosition(parentModel: NodeModel) {
    const parentPos = parentModel.getPosition();
    const isRight = parentPos.x > 0;
    const me = this;
    parentModel.getChildren().forEach((child) => {
      if (!child.getPosition()) {
        child.setPosition(parentPos.x + (isRight ? 1 : -1), parentPos.y);
      }
      me._fixPosition(child);
    });
  }
}

export default Beta2PelaMigrator;