Merge pull request #40 from MetaCell/feature/39

#39 extending and making serialisation generic

src/react-diagrams/MetaLinkModel.ts

@@ -1,5 +1,5 @@
 import { DefaultLinkModel } from '@projectstorm/react-diagrams';
-import { ReactDiagramMetaTypes } from '../constants';
+import { ReactDiagramMetaTypes, CallbackTypes } from '../constants';
 
 export class MetaLinkModel extends DefaultLinkModel {
   constructor(options = {}) {
@@ -8,4 +8,35 @@ export class MetaLinkModel extends DefaultLinkModel {
       type: ReactDiagramMetaTypes.META_LINK,
     });
   }
+
+  setOption(label: string, newValue: any, triggerUpdate?: boolean | undefined) {
+    // @ts-ignore
+    this.options[label] = newValue;
+    if (triggerUpdate) {
+      this.flagUpdate(CallbackTypes.OPTIONS_UPDATED);
+    }
+  }
+
+  getOption(label: string): any {
+    // @ts-ignore
+    return this.getOptions()[label];
+  }
+
+  flagUpdate(updateType: CallbackTypes, extraCondition?: CallbackTypes) {
+    this.fireEvent(
+      { node: this, function: updateType, extraCondition: extraCondition },
+      updateType
+    );
+  }
+
+  serialise(params: Array<string>) {
+    const additionalParams = Object.create({});
+    params.forEach(param => {
+      additionalParams[param] = this.getOption(param); 
+    });
+		return {
+			...super.serialize(),
+			...additionalParams,
+		};
+	}
 }

src/react-diagrams/MetaNodeModel.ts

@@ -125,4 +125,15 @@ export class MetaNodeModel extends NodeModel {
   getNodeBoundingBox(): BoundingBox {
     return this.boundingBox;
   }
+
+  serialise(params: Array<string>) {
+    const additionalParams = Object.create({});
+    params.forEach(param => {
+      additionalParams[param] = this.getOption(param); 
+    });
+		return {
+			...super.serialize(),
+			...additionalParams,
+		};
+	}
 }