wisemapping-frontend/packages/mindplot/src/components/persistence/XMLSerializer_Pela.js

/*
 *    Copyright [2015] [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, createDocument } from '@wisemapping/core-js';
import web2d from '@wisemapping/web2d';
import Mindmap from '../model/Mindmap';
import INodeModel, { TopicShape } from '../model/INodeModel';
import TopicFeature from '../TopicFeature';
import ConnectionLine from '../ConnectionLine';

/**
 * @class
 */
// eslint-disable-next-line camelcase
const XMLSerializer_Pela = new Class(
  /** @lends XMLSerializer_Pela */ {
    /**
         * @param mindmap
         * @throws will throw an error if mindmap is null or undefined
         * @return the created XML document (using the cross-browser implementation in core)
         */
    toXML(mindmap) {
      $assert(mindmap, 'Can not save a null mindmap');

      const document = createDocument();

      // Store map attributes ...
      const mapElem = document.createElement('map');
      const name = mindmap.getId();
      if ($defined(name)) {
        mapElem.setAttribute('name', this.rmXmlInv(name));
      }
      const version = mindmap.getVersion();
      if ($defined(version)) {
        mapElem.setAttribute('version', version);
      }

      document.appendChild(mapElem);

      // Create branches ...
      const topics = mindmap.getBranches();
      for (let i = 0; i < topics.length; i++) {
        const topic = topics[i];
        const topicDom = this._topicToXML(document, topic);
        mapElem.appendChild(topicDom);
      }

      // Create Relationships
      const relationships = mindmap.getRelationships();
      if (relationships.length > 0) {
        for (let j = 0; j < relationships.length; j++) {
          const relationship = relationships[j];
          if (
            mindmap.findNodeById(relationship.getFromNode()) !== null
            && mindmap.findNodeById(relationship.getToNode()) !== null
          ) {
            // Isolated relationships are not persisted ....
            const relationDom = this._relationshipToXML(document, relationship);
            mapElem.appendChild(relationDom);
          }
        }
      }

      return document;
    },

    _topicToXML(document, topic) {
      const parentTopic = document.createElement('topic');

      // Set topic attributes...
      if (topic.getType() === INodeModel.CENTRAL_TOPIC_TYPE) {
        parentTopic.setAttribute('central', 'true');
      } else {
        const pos = topic.getPosition();
        parentTopic.setAttribute('position', `${pos.x},${pos.y}`);

        const order = topic.getOrder();
        if (typeof order === 'number' && Number.isFinite(order)) { parentTopic.setAttribute('order', order); }
      }

      const text = topic.getText();
      if ($defined(text)) {
        this._noteTextToXML(document, parentTopic, text);
      }

      const shape = topic.getShapeType();
      if ($defined(shape)) {
        parentTopic.setAttribute('shape', shape);

        if (shape === TopicShape.IMAGE) {
          parentTopic.setAttribute(
            'image',
            `${topic.getImageSize().width},${topic.getImageSize().height
            }:${topic.getImageUrl()}`,
          );
        }
      }

      if (topic.areChildrenShrunken() && topic.getType() !== INodeModel.CENTRAL_TOPIC_TYPE) {
        parentTopic.setAttribute('shrink', 'true');
      }

      // Font properties ...
      const id = topic.getId();
      parentTopic.setAttribute('id', id);

      let font = '';

      const fontFamily = topic.getFontFamily();
      font += `${fontFamily || ''};`;

      const fontSize = topic.getFontSize();
      font += `${fontSize || ''};`;

      const fontColor = topic.getFontColor();
      font += `${fontColor || ''};`;

      const fontWeight = topic.getFontWeight();
      font += `${fontWeight || ''};`;

      const fontStyle = topic.getFontStyle();
      font += `${fontStyle || ''};`;

      if (
        $defined(fontFamily)
        || $defined(fontSize)
        || $defined(fontColor)
        || $defined(fontWeight)
        || $defined(fontStyle)
      ) {
        parentTopic.setAttribute('fontStyle', font);
      }

      const bgColor = topic.getBackgroundColor();
      if ($defined(bgColor)) {
        parentTopic.setAttribute('bgColor', bgColor);
      }

      const brColor = topic.getBorderColor();
      if ($defined(brColor)) {
        parentTopic.setAttribute('brColor', brColor);
      }

      const metadata = topic.getMetadata();
      if ($defined(metadata)) {
        parentTopic.setAttribute('metadata', metadata);
      }

      // Serialize features ...
      const features = topic.getFeatures();
      for (let i = 0; i < features.length; i++) {
        const feature = features[i];

        const featureType = feature.getType();
        const featureDom = document.createElement(featureType);
        const attributes = feature.getAttributes();

        for (const key in attributes) {
          const value = attributes[key];
          if (key === 'text') {
            const cdata = document.createCDATASection(this.rmXmlInv(value));
            featureDom.appendChild(cdata);
          } else {
            featureDom.setAttribute(key, this.rmXmlInv(value));
          }
        }
        parentTopic.appendChild(featureDom);
      }

      // CHILDREN TOPICS
      const childTopics = topic.getChildren();
      for (let j = 0; j < childTopics.length; j++) {
        const childTopic = childTopics[j];
        const childDom = this._topicToXML(document, childTopic);
        parentTopic.appendChild(childDom);
      }
      return parentTopic;
    },

    _noteTextToXML(document, elem, text) {
      if (text.indexOf('\n') === -1) {
        elem.setAttribute('text', this.rmXmlInv(text));
      } else {
        const textDom = document.createElement('text');
        const cdata = document.createCDATASection(this.rmXmlInv(text));
        textDom.appendChild(cdata);
        elem.appendChild(textDom);
      }
    },

    _relationshipToXML(document, relationship) {
      const result = document.createElement('relationship');
      result.setAttribute('srcTopicId', relationship.getFromNode());
      result.setAttribute('destTopicId', relationship.getToNode());

      const lineType = relationship.getLineType();
      result.setAttribute('lineType', lineType);
      if (lineType === ConnectionLine.CURVED || lineType === ConnectionLine.SIMPLE_CURVED) {
        if ($defined(relationship.getSrcCtrlPoint())) {
          const srcPoint = relationship.getSrcCtrlPoint();
          result.setAttribute(
            'srcCtrlPoint',
            `${Math.round(srcPoint.x)},${Math.round(srcPoint.y)}`,
          );
        }
        if ($defined(relationship.getDestCtrlPoint())) {
          const destPoint = relationship.getDestCtrlPoint();
          result.setAttribute(
            'destCtrlPoint',
            `${Math.round(destPoint.x)},${Math.round(destPoint.y)}`,
          );
        }
      }
      result.setAttribute('endArrow', relationship.getEndArrow());
      result.setAttribute('startArrow', relationship.getStartArrow());
      return result;
    },

    /**
     * @param dom
     * @param mapId
     * @throws will throw an error if dom is null or undefined
     * @throws will throw an error if mapId is null or undefined
     * @throws will throw an error if the document element is not consistent with a wisemap's root
     * element
     */
    loadFromDom(dom, mapId) {
      $assert(dom, 'dom can not be null');
      $assert(mapId, 'mapId can not be null');

      const rootElem = dom.documentElement;

      // Is a wisemap?.
      $assert(
        rootElem.tagName === XMLSerializer_Pela.MAP_ROOT_NODE,
        'This seem not to be a map document.',
      );

      this._idsMap = {};
      // Start the loading process ...
      const version = rootElem.getAttribute('version');

      const mindmap = new Mindmap(mapId, version);
      const children = rootElem.childNodes;
      for (let i = 0; i < children.length; i++) {
        const child = children[i];
        if (child.nodeType === 1) {
          switch (child.tagName) {
            case 'topic': {
              const topic = this._deserializeNode(child, mindmap);
              mindmap.addBranch(topic);
              break;
            }
            case 'relationship': {
              const relationship = this._deserializeRelationship(child, mindmap);
              if (relationship != null) {
                mindmap.addRelationship(relationship);
              }
              break;
            }
            default:
              break;
          }
        }
      }
      this._idsMap = null;
      mindmap.setId(mapId);
      return mindmap;
    },

    _deserializeNode(domElem, mindmap) {
      const type = domElem.getAttribute('central') != null
        ? INodeModel.CENTRAL_TOPIC_TYPE
        : INodeModel.MAIN_TOPIC_TYPE;

      // Load attributes...
      let id = domElem.getAttribute('id');
      if ($defined(id)) {
        id = parseInt(id, 10);
      }

      if (this._idsMap[id]) {
        id = null;
      } else {
        this._idsMap[id] = domElem;
      }

      const topic = mindmap.createNode(type, id);

      // Set text property is it;s defined...
      const text = domElem.getAttribute('text');
      if ($defined(text) && text) {
        topic.setText(text);
      }

      const fontStyle = domElem.getAttribute('fontStyle');
      if ($defined(fontStyle) && fontStyle) {
        const font = fontStyle.split(';');

        if (font[0]) {
          topic.setFontFamily(font[0]);
        }

        if (font[1]) {
          topic.setFontSize(font[1]);
        }

        if (font[2]) {
          topic.setFontColor(font[2]);
        }

        if (font[3]) {
          topic.setFontWeight(font[3]);
        }

        if (font[4]) {
          topic.setFontStyle(font[4]);
        }
      }

      const shape = domElem.getAttribute('shape');
      if ($defined(shape)) {
        topic.setShapeType(shape);

        if (shape === TopicShape.IMAGE) {
          const image = domElem.getAttribute('image');
          const size = image.substring(0, image.indexOf(':'));
          const url = image.substring(image.indexOf(':') + 1, image.length);
          topic.setImageUrl(url);

          const split = size.split(',');
          topic.setImageSize(split[0], split[1]);
        }
      }

      const bgColor = domElem.getAttribute('bgColor');
      if ($defined(bgColor)) {
        topic.setBackgroundColor(bgColor);
      }

      const borderColor = domElem.getAttribute('brColor');
      if ($defined(borderColor)) {
        topic.setBorderColor(borderColor);
      }

      const order = domElem.getAttribute('order');
      if ($defined(order) && order !== 'NaN') {
        // Hack for broken maps ...
        topic.setOrder(parseInt(order, 10));
      }

      const isShrink = domElem.getAttribute('shrink');
      // Hack: Some production maps has been stored with the central topic collapsed. This is a bug.
      if ($defined(isShrink) && type !== INodeModel.CENTRAL_TOPIC_TYPE) {
        topic.setChildrenShrunken(isShrink);
      }

      const position = domElem.getAttribute('position');
      if ($defined(position)) {
        const pos = position.split(',');
        topic.setPosition(pos[0], pos[1]);
      }

      const metadata = domElem.getAttribute('metadata');
      if ($defined(metadata)) {
        topic.setMetadata(metadata);
      }

      // Creating icons and children nodes
      const children = domElem.childNodes;
      for (let i = 0; i < children.length; i++) {
        const child = children[i];
        if (child.nodeType === Node.ELEMENT_NODE) {
          if (child.tagName === 'topic') {
            const childTopic = this._deserializeNode(child, mindmap);
            childTopic.connectTo(topic);
          } else if (TopicFeature.isSupported(child.tagName)) {
            // Load attributes ...
            const namedNodeMap = child.attributes;
            const attributes = {};
            for (let j = 0; j < namedNodeMap.length; j++) {
              const attribute = namedNodeMap.item(j);
              attributes[attribute.name] = attribute.value;
            }

            // Has text node ?.
            const textAttr = this._deserializeTextAttr(child);
            if (textAttr) {
              attributes.text = textAttr;
            }

            // Create a new element ....
            const featureType = child.tagName;
            const feature = TopicFeature.createModel(featureType, attributes);
            topic.addFeature(feature);
          } else if (child.tagName === 'text') {
            const nodeText = this._deserializeNodeText(child);
            topic.setText(nodeText);
          }
        }
      }
      return topic;
    },

    _deserializeTextAttr(domElem) {
      let value = domElem.getAttribute('text');
      if (!$defined(value)) {
        const children = domElem.childNodes;
        for (let i = 0; i < children.length; i++) {
          const child = children[i];
          if (child.nodeType === Node.CDATA_SECTION_NODE) {
            value = child.nodeValue;
          }
        }
      } else {
        // Notes must be decoded ...
        value = unescape(value);

        // Hack for empty nodes ...
        if (value === '') {
          value = ' ';
        }
      }

      return value;
    },

    _deserializeNodeText(domElem) {
      const children = domElem.childNodes;
      let value = null;
      for (let i = 0; i < children.length; i++) {
        const child = children[i];
        if (child.nodeType === Node.CDATA_SECTION_NODE) {
          value = child.nodeValue;
        }
      }
      return value;
    },

    _deserializeRelationship(domElement, mindmap) {
      const srcId = domElement.getAttribute('srcTopicId');
      const destId = domElement.getAttribute('destTopicId');
      const lineType = domElement.getAttribute('lineType');
      const srcCtrlPoint = domElement.getAttribute('srcCtrlPoint');
      const destCtrlPoint = domElement.getAttribute('destCtrlPoint');

      // If for some reason a relationship lines has source and dest nodes the same, don't import it.
      if (srcId === destId) {
        return null;
      }
      // Is the connections points valid ?. If it's not, do not load the relationship ...
      if (mindmap.findNodeById(srcId) == null || mindmap.findNodeById(destId) == null) {
        return null;
      }

      const model = mindmap.createRelationship(srcId, destId);
      model.setLineType(lineType);
      if ($defined(srcCtrlPoint) && srcCtrlPoint !== '') {
        model.setSrcCtrlPoint(web2d.Point.fromString(srcCtrlPoint));
      }
      if ($defined(destCtrlPoint) && destCtrlPoint !== '') {
        model.setDestCtrlPoint(web2d.Point.fromString(destCtrlPoint));
      }
      model.setEndArrow('false');
      model.setStartArrow('true');
      return model;
    },

    /**
         * This method ensures that the output String has only
         * valid XML unicode characters as specified by the
         * XML 1.0 standard. For reference, please see
         * <a href="http://www.w3.org/TR/2000/REC-xml-20001006#NT-Char">the
         * standard</a>. This method will return an empty
         * String if the input is null or empty.
         *
         * @param in The String whose non-valid characters we want to remove.
         * @return The in String, stripped of non-valid characters.
         */
    rmXmlInv(str) {
      if (str == null || str === undefined) return null;

      let result = '';
      for (let i = 0; i < str.length; i++) {
        const c = str.charCodeAt(i);
        if (
          c === 0x9
          || c === 0xa
          || c === 0xd
          || (c >= 0x20 && c <= 0xd7ff)
          || (c >= 0xe000 && c <= 0xfffd)
          || (c >= 0x10000 && c <= 0x10ffff)
        ) {
          result += str.charAt(i);
        }
      }
      return result;
    },
  },
);

/**
 * a wisemap's root element tag name
 * @constant
 * @type {String}
 * @default
 */
XMLSerializer_Pela.MAP_ROOT_NODE = 'map';

export default XMLSerializer_Pela;