add attrs to figure node

wax-prosemirror-services/src/ImageService/ImageService.js

@@ -22,9 +22,12 @@ class ImageService extends Service {
     this.container.bind('Image').to(Image);
     const createNode = this.container.get('CreateNode');
     const createOverlay = this.container.get('CreateOverlay');
-    createNode({
-      figure: figureNode,
-    });
+    createNode(
+      {
+        figure: figureNode,
+      },
+      { toWaxSchema: true },
+    );
 
     createNode(
       {

wax-prosemirror-services/src/ImageService/schema/figureNode.js

@@ -2,10 +2,21 @@ const figureNode = {
   content: 'image* figcaption{0,1}',
   group: 'block',
   marks: '',
-  parseDOM: [{ tag: 'figure' }],
+  parseDOM: [
+    {
+      tag: 'figure',
+      getAttrs(hook, next) {
+        next();
+      },
+    },
+  ],
   toDOM() {
     return ['figure', 0];
   },
+  toDOM(hook, next) {
+    hook.value = ['figure', 0];
+    next();
+  },
 };
 
 export default figureNode;

wax-prosemirror-services/src/TrackChangeService/schema/trackChangesNodes/figureTrackNode.js

+import { SchemaHelpers } from 'wax-prosemirror-core';
+
+const figureTrackNode = {
+  attrs: {
+    id: { default: '' },
+    track: { default: [] },
+    group: { default: '' },
+    viewid: { default: '' },
+  },
+  parseDOM: [
+    {
+      tag: 'p.author',
+      getAttrs(hook, next) {
+        Object.assign(hook, {
+          id: hook.dom.dataset.id,
+          track: SchemaHelpers.parseTracks(hook.dom.dataset.track),
+          group: hook.dom.dataset.group,
+          viewid: hook.dom.dataset.viewid,
+        });
+        next();
+      },
+    },
+  ],
+  toDOM(hook, next) {
+    if (hook.node.attrs.track && hook.node.attrs.track.length) {
+      Object.assign(hook.value[1], {
+        'data-id': hook.node.attrs.id,
+        'data-track': JSON.stringify(hook.node.attrs.track),
+        'data-group': hook.node.attrs.group,
+        'data-viewid': hook.node.attrs.viewid,
+      });
+    }
+    next();
+  },
+};
+
+export default figureTrackNode;

wax-prosemirror-services/src/TrackChangeService/schema/trackChangesNodes/index.js

@@ -15,6 +15,7 @@ import sourceNoteTrackNode from './sourceNoteTrackNode';
 // import bulletListTrackNode from './bulletListTrackNode';
 import orderedListTrackNode from './orderedListTrackNode';
 // import listItemTrackNode from './listItemTrackNode';
+import figureTrackNode from './figureTrackNode';
 // import imageTrackNode from './imageTrackNode';
 import customBlockTrackNode from './customBlockTrackNode';
 
@@ -37,5 +38,6 @@ export default {
   customTagBlock: customBlockTrackNode,
   // bulletlist: bulletListTrackNode,
   // list_item: listItemTrackNode,
+  figure: figureTrackNode,
   // image: imageTrackNode
 };