@atlaskit/editor-plugin-track-changes 2.1.0 → 2.2.0

package/CHANGELOG.md

@@ -1,5 +1,25 @@
 # @atlaskit/editor-plugin-track-changes
 
+## 2.2.0
+
+### Minor Changes
+
+- [#193685](https://bitbucket.org/atlassian/atlassian-frontend-monorepo/pull-requests/193685)
+  [`ee3ba46cb3d0a`](https://bitbucket.org/atlassian/atlassian-frontend-monorepo/commits/ee3ba46cb3d0a) -
+  [EDITOR-1073] Add undo/ redo & show diff to comment toolbar
+
+### Patch Changes
+
+- Updated dependencies
+
+## 2.1.1
+
+### Patch Changes
+
+- [#190819](https://bitbucket.org/atlassian/atlassian-frontend-monorepo/pull-requests/190819)
+  [`20420774e83cc`](https://bitbucket.org/atlassian/atlassian-frontend-monorepo/commits/20420774e83cc) -
+  Include Track chages plugin
+
 ## 2.1.0
 
 ### Minor Changes

package/afm-cc/tsconfig.json

@@ -18,7 +18,16 @@
 	],
 	"references": [
 		{
-			"path": "../../../design-system/tokens/afm-cc/tsconfig.json"
+			"path": "../../../design-system/button/afm-cc/tsconfig.json"
+		},
+		{
+			"path": "../../editor-plugin-primary-toolbar/afm-cc/tsconfig.json"
+		},
+		{
+			"path": "../../editor-plugin-show-diff/afm-cc/tsconfig.json"
+		},
+		{
+			"path": "../../../design-system/icon-lab/afm-cc/tsconfig.json"
 		},
 		{
 			"path": "../../editor-common/afm-cc/tsconfig.json"

package/afm-jira/tsconfig.json

@@ -2,7 +2,7 @@
 	"extends": "../../../../tsconfig.entry-points.jira.json",
 	"compilerOptions": {
 		"target": "es5",
-		"outDir": "../../../../../tsDist/@atlaskit__editor-plugin-toolbar-lists-indentation/app",
+		"outDir": "../../../../../tsDist/@atlaskit__editor-plugin-track-changes/app",
 		"rootDir": "../",
 		"composite": true
 	},
@@ -18,46 +18,16 @@
 	],
 	"references": [
 		{
-			"path": "../../../design-system/css/afm-jira/tsconfig.json"
-		},
-		{
-			"path": "../../editor-plugin-analytics/afm-jira/tsconfig.json"
-		},
-		{
-			"path": "../../editor-plugin-block-type/afm-jira/tsconfig.json"
-		},
-		{
-			"path": "../../editor-plugin-feature-flags/afm-jira/tsconfig.json"
-		},
-		{
-			"path": "../../editor-plugin-indentation/afm-jira/tsconfig.json"
-		},
-		{
-			"path": "../../editor-plugin-list/afm-jira/tsconfig.json"
+			"path": "../../../design-system/button/afm-jira/tsconfig.json"
 		},
 		{
 			"path": "../../editor-plugin-primary-toolbar/afm-jira/tsconfig.json"
 		},
 		{
-			"path": "../../editor-plugin-selection-toolbar/afm-jira/tsconfig.json"
-		},
-		{
-			"path": "../../editor-plugin-tasks-and-decisions/afm-jira/tsconfig.json"
-		},
-		{
-			"path": "../../editor-shared-styles/afm-jira/tsconfig.json"
-		},
-		{
-			"path": "../../../design-system/icon/afm-jira/tsconfig.json"
-		},
-		{
-			"path": "../../../platform/feature-flags/afm-jira/tsconfig.json"
-		},
-		{
-			"path": "../../tmp-editor-statsig/afm-jira/tsconfig.json"
+			"path": "../../editor-plugin-show-diff/afm-jira/tsconfig.json"
 		},
 		{
-			"path": "../../../design-system/tokens/afm-jira/tsconfig.json"
+			"path": "../../../design-system/icon-lab/afm-jira/tsconfig.json"
 		},
 		{
 			"path": "../../editor-common/afm-jira/tsconfig.json"

package/afm-post-office/tsconfig.json

@@ -2,7 +2,7 @@
 	"extends": "../../../../tsconfig.entry-points.post-office.json",
 	"compilerOptions": {
 		"target": "es5",
-		"outDir": "../../../../../post-office/tsDist/@atlaskit__editor-plugin-toolbar-lists-indentation/app",
+		"outDir": "../../../../../post-office/tsDist/@atlaskit__editor-plugin-track-changes/app",
 		"rootDir": "../",
 		"composite": true
 	},
@@ -18,10 +18,19 @@
 	],
 	"references": [
 		{
-			"path": "../editor-common/tsconfig.app.json"
+			"path": "../../../design-system/button/afm-post-office/tsconfig.json"
 		},
 		{
-			"path": "../../design-system/tokens/tsconfig.app.json"
+			"path": "../../editor-plugin-primary-toolbar/afm-post-office/tsconfig.json"
+		},
+		{
+			"path": "../../editor-plugin-show-diff/afm-post-office/tsconfig.json"
+		},
+		{
+			"path": "../../../design-system/icon-lab/afm-post-office/tsconfig.json"
+		},
+		{
+			"path": "../../editor-common/afm-post-office/tsconfig.json"
 		}
 	]
 }

package/afm-rovo-extension/tsconfig.json

@@ -0,0 +1,36 @@
+{
+	"extends": "../../../../tsconfig.entry-points.rovo-extension.json",
+	"compilerOptions": {
+		"target": "es5",
+		"outDir": "../../../../../rovo-extension/tsDist/@atlaskit__editor-plugin-track-changes/app",
+		"rootDir": "../",
+		"composite": true
+	},
+	"include": [
+		"../src/**/*.ts",
+		"../src/**/*.tsx"
+	],
+	"exclude": [
+		"../src/**/__tests__/*",
+		"../src/**/*.test.*",
+		"../src/**/test.*",
+		"../src/**/examples.*"
+	],
+	"references": [
+		{
+			"path": "../../../design-system/button/afm-rovo-extension/tsconfig.json"
+		},
+		{
+			"path": "../../editor-plugin-primary-toolbar/afm-rovo-extension/tsconfig.json"
+		},
+		{
+			"path": "../../editor-plugin-show-diff/afm-rovo-extension/tsconfig.json"
+		},
+		{
+			"path": "../../../design-system/icon-lab/afm-rovo-extension/tsconfig.json"
+		},
+		{
+			"path": "../../editor-common/afm-rovo-extension/tsconfig.json"
+		}
+	]
+}

package/afm-townsquare/tsconfig.json

@@ -0,0 +1,36 @@
+{
+	"extends": "../../../../tsconfig.entry-points.townsquare.json",
+	"compilerOptions": {
+		"target": "es5",
+		"outDir": "../../../../../townsquare/tsDist/@atlaskit__editor-plugin-track-changes/app",
+		"rootDir": "../",
+		"composite": true
+	},
+	"include": [
+		"../src/**/*.ts",
+		"../src/**/*.tsx"
+	],
+	"exclude": [
+		"../src/**/__tests__/*",
+		"../src/**/*.test.*",
+		"../src/**/test.*",
+		"../src/**/examples.*"
+	],
+	"references": [
+		{
+			"path": "../../../design-system/button/afm-townsquare/tsconfig.json"
+		},
+		{
+			"path": "../../editor-plugin-primary-toolbar/afm-townsquare/tsconfig.json"
+		},
+		{
+			"path": "../../editor-plugin-show-diff/afm-townsquare/tsconfig.json"
+		},
+		{
+			"path": "../../../design-system/icon-lab/afm-townsquare/tsconfig.json"
+		},
+		{
+			"path": "../../editor-common/afm-townsquare/tsconfig.json"
+		}
+	]
+}

package/dist/cjs/pm-plugins/main.js

@@ -4,35 +4,58 @@ var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefau
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
-exports.trackChangesPluginKey = exports.createTrackChangesPlugin = void 0;
+exports.trackChangesPluginKey = exports.createTrackChangesPlugin = exports.InvertableStep = void 0;
+var _toConsumableArray2 = _interopRequireDefault(require("@babel/runtime/helpers/toConsumableArray"));
 var _defineProperty2 = _interopRequireDefault(require("@babel/runtime/helpers/defineProperty"));
-var _prosemirrorChangeset = require("prosemirror-changeset");
+var _createClass2 = _interopRequireDefault(require("@babel/runtime/helpers/createClass"));
+var _classCallCheck2 = _interopRequireDefault(require("@babel/runtime/helpers/classCallCheck"));
 var _safePlugin = require("@atlaskit/editor-common/safe-plugin");
 var _state = require("@atlaskit/editor-prosemirror/state");
 var _transform = require("@atlaskit/editor-prosemirror/transform");
-var _view = require("@atlaskit/editor-prosemirror/view");
-var _decorations = require("./decorations");
 var _types = require("./types");
 function ownKeys(e, r) { var t = Object.keys(e); if (Object.getOwnPropertySymbols) { var o = Object.getOwnPropertySymbols(e); r && (o = o.filter(function (r) { return Object.getOwnPropertyDescriptor(e, r).enumerable; })), t.push.apply(t, o); } return t; }
 function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t = null != arguments[r] ? arguments[r] : {}; r % 2 ? ownKeys(Object(t), !0).forEach(function (r) { (0, _defineProperty2.default)(e, r, t[r]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(t)) : ownKeys(Object(t)).forEach(function (r) { Object.defineProperty(e, r, Object.getOwnPropertyDescriptor(t, r)); }); } return e; }
+function _createForOfIteratorHelper(r, e) { var t = "undefined" != typeof Symbol && r[Symbol.iterator] || r["@@iterator"]; if (!t) { if (Array.isArray(r) || (t = _unsupportedIterableToArray(r)) || e && r && "number" == typeof r.length) { t && (r = t); var _n = 0, F = function F() {}; return { s: F, n: function n() { return _n >= r.length ? { done: !0 } : { done: !1, value: r[_n++] }; }, e: function e(r) { throw r; }, f: F }; } throw new TypeError("Invalid attempt to iterate non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); } var o, a = !0, u = !1; return { s: function s() { t = t.call(r); }, n: function n() { var r = t.next(); return a = r.done, r; }, e: function e(r) { u = !0, o = r; }, f: function f() { try { a || null == t.return || t.return(); } finally { if (u) throw o; } } }; }
+function _unsupportedIterableToArray(r, a) { if (r) { if ("string" == typeof r) return _arrayLikeToArray(r, a); var t = {}.toString.call(r).slice(8, -1); return "Object" === t && r.constructor && (t = r.constructor.name), "Map" === t || "Set" === t ? Array.from(r) : "Arguments" === t || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(t) ? _arrayLikeToArray(r, a) : void 0; } }
+function _arrayLikeToArray(r, a) { (null == a || a > r.length) && (a = r.length); for (var e = 0, n = Array(a); e < a; e++) n[e] = r[e]; return n; }
 var trackChangesPluginKey = exports.trackChangesPluginKey = new _state.PluginKey('trackChangesPlugin');
-var createTrackChangesPlugin = exports.createTrackChangesPlugin = function createTrackChangesPlugin() {
+var InvertableStep = exports.InvertableStep = /*#__PURE__*/(0, _createClass2.default)(function InvertableStep(step, inverted) {
+  (0, _classCallCheck2.default)(this, InvertableStep);
+  this.step = step;
+  this.inverted = inverted;
+});
+var getBaselineFromSteps = function getBaselineFromSteps(doc, steps) {
+  var _iterator = _createForOfIteratorHelper(steps.slice().reverse()),
+    _step;
+  try {
+    for (_iterator.s(); !(_step = _iterator.n()).done;) {
+      var _step2 = _step.value;
+      var result = _step2.inverted.apply(doc);
+      if (result.failed === null && result.doc) {
+        doc = result.doc;
+      }
+    }
+  } catch (err) {
+    _iterator.e(err);
+  } finally {
+    _iterator.f();
+  }
+  return doc;
+};
+var createTrackChangesPlugin = exports.createTrackChangesPlugin = function createTrackChangesPlugin(api) {
   // Mark the state to be reset on next time the document has a meaningful change
   var resetBaseline = false;
   return new _safePlugin.SafePlugin({
     key: trackChangesPluginKey,
     state: {
-      init: function init(_, _ref) {
-        var doc = _ref.doc;
+      init: function init() {
         return {
-          changes: _prosemirrorChangeset.ChangeSet.create(doc),
+          steps: [],
           shouldChangesBeDisplayed: false,
-          isShowDiffAvailable: false,
-          baselineDoc: doc,
-          numOfChanges: 0
+          isShowDiffAvailable: false
         };
       },
-      apply: function apply(tr, state, oldState, newState) {
+      apply: function apply(tr, state) {
         var metadata = tr.getMeta(trackChangesPluginKey);
         if (metadata && metadata.action === _types.TOGGLE_TRACK_CHANGES_ACTION.TOGGLE_TRACK_CHANGES) {
           resetBaseline = true;
@@ -47,53 +70,42 @@ var createTrackChangesPlugin = exports.createTrackChangesPlugin = function creat
           // If no document changes, return the old changeSet
           return state;
         }
-        var changes = resetBaseline ? _prosemirrorChangeset.ChangeSet.create(tr.docs[0]).addSteps(tr.doc,
-        // The new document
-        tr.mapping.maps,
-        // The set of changes
-        tr.docs[0].content // The old document
-        ) : state.changes.addSteps(tr.doc,
-        // The new document
-        tr.mapping.maps,
-        // The set of changes
-        tr.docs[0].content // The old document
-        );
-        var baselineDoc = resetBaseline ? tr.docs[0] : state.baselineDoc;
+        var newSteps = tr.steps.map(function (step, idx) {
+          return new InvertableStep(step, step.invert(tr.docs[idx]));
+        });
+        var steps = resetBaseline ? newSteps : [].concat((0, _toConsumableArray2.default)(state.steps), (0, _toConsumableArray2.default)(newSteps));
         resetBaseline = false;
 
         // Create a new ChangeSet based on document changes
         return _objectSpread(_objectSpread({}, state), {}, {
-          baselineDoc: baselineDoc,
+          steps: steps,
           shouldChangesBeDisplayed: false,
-          changes: changes,
           isShowDiffAvailable: true
         });
       }
     },
-    props: {
-      decorations: function decorations(state) {
-        var pluginState = trackChangesPluginKey.getState(state);
-        if (!(pluginState !== null && pluginState !== void 0 && pluginState.shouldChangesBeDisplayed) || !pluginState.changes) {
-          return undefined;
-        }
-        var decoration = _view.DecorationSet.empty;
-        var decorations = [];
-        var changes = (0, _prosemirrorChangeset.simplifyChanges)(pluginState.changes.changes, state.doc);
-        var tr = state.tr;
-        changes.forEach(function (change) {
-          if (change.inserted.length > 0) {
-            decorations.push((0, _decorations.createInlineChangedDecoration)(change));
-          }
-          if (change.deleted.length > 0) {
-            decorations.push((0, _decorations.createDeletedContentDecoration)({
-              change: change,
-              doc: pluginState === null || pluginState === void 0 ? void 0 : pluginState.baselineDoc,
-              tr: tr
+    view: function view() {
+      return {
+        update: function update(view, prevState) {
+          var _trackChangesPluginKe;
+          var prevShouldChangesBeDisplayed = (_trackChangesPluginKe = trackChangesPluginKey.getState(prevState)) === null || _trackChangesPluginKe === void 0 ? void 0 : _trackChangesPluginKe.shouldChangesBeDisplayed;
+          var currentPluginState = trackChangesPluginKey.getState(view.state);
+          var currentShouldChangesBeDisplayed = currentPluginState === null || currentPluginState === void 0 ? void 0 : currentPluginState.shouldChangesBeDisplayed;
+          if (prevShouldChangesBeDisplayed === false && currentShouldChangesBeDisplayed === true) {
+            var _currentPluginState$s, _api$core, _api$showDiff;
+            var steps = (_currentPluginState$s = currentPluginState === null || currentPluginState === void 0 ? void 0 : currentPluginState.steps) !== null && _currentPluginState$s !== void 0 ? _currentPluginState$s : [];
+            api === null || api === void 0 || (_api$core = api.core) === null || _api$core === void 0 || _api$core.actions.execute(api === null || api === void 0 || (_api$showDiff = api.showDiff) === null || _api$showDiff === void 0 || (_api$showDiff = _api$showDiff.commands) === null || _api$showDiff === void 0 ? void 0 : _api$showDiff.showDiff({
+              originalDoc: getBaselineFromSteps(view.state.doc, steps),
+              steps: steps.map(function (s) {
+                return s.step;
+              })
             }));
+          } else if (currentShouldChangesBeDisplayed === false && prevShouldChangesBeDisplayed === true) {
+            var _api$core2, _api$showDiff2;
+            api === null || api === void 0 || (_api$core2 = api.core) === null || _api$core2 === void 0 || _api$core2.actions.execute(api === null || api === void 0 || (_api$showDiff2 = api.showDiff) === null || _api$showDiff2 === void 0 || (_api$showDiff2 = _api$showDiff2.commands) === null || _api$showDiff2 === void 0 ? void 0 : _api$showDiff2.hideDiff);
           }
-        });
-        return decoration.add(tr.doc, decorations);
-      }
+        }
+      };
     }
   });
 };

package/dist/cjs/trackChangesPlugin.js

@@ -1,23 +1,39 @@
 "use strict";
 
+var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
 exports.trackChangesPlugin = void 0;
+var _react = _interopRequireDefault(require("react"));
 var _main = require("./pm-plugins/main");
 var _types = require("./pm-plugins/types");
-var trackChangesPlugin = exports.trackChangesPlugin = function trackChangesPlugin() {
+var _TrackChangesToolbarButton = require("./ui/TrackChangesToolbarButton");
+var trackChangesPlugin = exports.trackChangesPlugin = function trackChangesPlugin(_ref) {
+  var _api$primaryToolbar;
+  var api = _ref.api;
+  var primaryToolbarComponent = function primaryToolbarComponent() {
+    return /*#__PURE__*/_react.default.createElement(_TrackChangesToolbarButton.TrackChangesToolbarButton, {
+      api: api
+    });
+  };
+  api === null || api === void 0 || (_api$primaryToolbar = api.primaryToolbar) === null || _api$primaryToolbar === void 0 || (_api$primaryToolbar = _api$primaryToolbar.actions) === null || _api$primaryToolbar === void 0 || _api$primaryToolbar.registerComponent({
+    name: 'trackChanges',
+    component: primaryToolbarComponent
+  });
   return {
     name: 'trackChanges',
     pmPlugins: function pmPlugins() {
       return [{
         name: 'trackChangesPlugin',
-        plugin: _main.createTrackChangesPlugin
+        plugin: function plugin() {
+          return (0, _main.createTrackChangesPlugin)(api);
+        }
       }];
     },
     commands: {
-      toggleChanges: function toggleChanges(_ref) {
-        var tr = _ref.tr;
+      toggleChanges: function toggleChanges(_ref2) {
+        var tr = _ref2.tr;
         return tr.setMeta(_main.trackChangesPluginKey, {
           action: _types.TOGGLE_TRACK_CHANGES_ACTION.TOGGLE_TRACK_CHANGES
         });

package/dist/cjs/ui/TrackChangesToolbarButton.js

@@ -0,0 +1,35 @@
+"use strict";
+
+var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.TrackChangesToolbarButton = void 0;
+var _react = _interopRequireDefault(require("react"));
+var _new = require("@atlaskit/button/new");
+var _hooks = require("@atlaskit/editor-common/hooks");
+var _history = _interopRequireDefault(require("@atlaskit/icon-lab/core/history"));
+var TrackChangesToolbarButton = exports.TrackChangesToolbarButton = function TrackChangesToolbarButton(_ref) {
+  var _api$trackChanges;
+  var api = _ref.api;
+  var _useSharedPluginState = (0, _hooks.useSharedPluginStateWithSelector)(api, ['trackChanges'], function (states) {
+      var _states$trackChangesS, _states$trackChangesS2;
+      return {
+        isDisplayingChanges: (_states$trackChangesS = states.trackChangesState) === null || _states$trackChangesS === void 0 ? void 0 : _states$trackChangesS.isDisplayingChanges,
+        isShowDiffAvailable: (_states$trackChangesS2 = states.trackChangesState) === null || _states$trackChangesS2 === void 0 ? void 0 : _states$trackChangesS2.isShowDiffAvailable
+      };
+    }),
+    isDisplayingChanges = _useSharedPluginState.isDisplayingChanges,
+    isShowDiffAvailable = _useSharedPluginState.isShowDiffAvailable;
+  var handleClick = _react.default.useCallback(function () {
+    api === null || api === void 0 || api.core.actions.execute(api === null || api === void 0 ? void 0 : api.trackChanges.commands.toggleChanges);
+  }, [api === null || api === void 0 || (_api$trackChanges = api.trackChanges) === null || _api$trackChanges === void 0 ? void 0 : _api$trackChanges.commands, api === null || api === void 0 ? void 0 : api.core.actions]);
+  return /*#__PURE__*/_react.default.createElement(_new.IconButton, {
+    icon: _history.default,
+    label: "Track Changes",
+    appearance: "subtle",
+    isDisabled: !isShowDiffAvailable,
+    isSelected: isDisplayingChanges,
+    onClick: handleClick
+  });
+};

package/dist/es2019/pm-plugins/main.js

@@ -1,29 +1,37 @@
-import { ChangeSet, simplifyChanges } from 'prosemirror-changeset';
 import { SafePlugin } from '@atlaskit/editor-common/safe-plugin';
 import { PluginKey } from '@atlaskit/editor-prosemirror/state';
 import { ReplaceAroundStep, ReplaceStep } from '@atlaskit/editor-prosemirror/transform';
-import { DecorationSet } from '@atlaskit/editor-prosemirror/view';
-import { createInlineChangedDecoration, createDeletedContentDecoration } from './decorations';
 import { TOGGLE_TRACK_CHANGES_ACTION as ACTION } from './types';
 export const trackChangesPluginKey = new PluginKey('trackChangesPlugin');
-export const createTrackChangesPlugin = () => {
+export class InvertableStep {
+  constructor(step, inverted) {
+    this.step = step;
+    this.inverted = inverted;
+  }
+}
+const getBaselineFromSteps = (doc, steps) => {
+  for (const step of steps.slice().reverse()) {
+    const result = step.inverted.apply(doc);
+    if (result.failed === null && result.doc) {
+      doc = result.doc;
+    }
+  }
+  return doc;
+};
+export const createTrackChangesPlugin = api => {
   // Mark the state to be reset on next time the document has a meaningful change
   let resetBaseline = false;
   return new SafePlugin({
     key: trackChangesPluginKey,
     state: {
-      init(_, {
-        doc
-      }) {
+      init() {
         return {
-          changes: ChangeSet.create(doc),
+          steps: [],
           shouldChangesBeDisplayed: false,
-          isShowDiffAvailable: false,
-          baselineDoc: doc,
-          numOfChanges: 0
+          isShowDiffAvailable: false
         };
       },
-      apply(tr, state, oldState, newState) {
+      apply(tr, state) {
         const metadata = tr.getMeta(trackChangesPluginKey);
         if (metadata && metadata.action === ACTION.TOGGLE_TRACK_CHANGES) {
           resetBaseline = true;
@@ -37,54 +45,39 @@ export const createTrackChangesPlugin = () => {
           // If no document changes, return the old changeSet
           return state;
         }
-        const changes = resetBaseline ? ChangeSet.create(tr.docs[0]).addSteps(tr.doc,
-        // The new document
-        tr.mapping.maps,
-        // The set of changes
-        tr.docs[0].content // The old document
-        ) : state.changes.addSteps(tr.doc,
-        // The new document
-        tr.mapping.maps,
-        // The set of changes
-        tr.docs[0].content // The old document
-        );
-        const baselineDoc = resetBaseline ? tr.docs[0] : state.baselineDoc;
+        const newSteps = tr.steps.map((step, idx) => new InvertableStep(step, step.invert(tr.docs[idx])));
+        const steps = resetBaseline ? newSteps : [...state.steps, ...newSteps];
         resetBaseline = false;
 
         // Create a new ChangeSet based on document changes
         return {
           ...state,
-          baselineDoc,
+          steps,
           shouldChangesBeDisplayed: false,
-          changes,
           isShowDiffAvailable: true
         };
       }
     },
-    props: {
-      decorations: state => {
-        const pluginState = trackChangesPluginKey.getState(state);
-        if (!(pluginState !== null && pluginState !== void 0 && pluginState.shouldChangesBeDisplayed) || !pluginState.changes) {
-          return undefined;
-        }
-        const decoration = DecorationSet.empty;
-        const decorations = [];
-        const changes = simplifyChanges(pluginState.changes.changes, state.doc);
-        const tr = state.tr;
-        changes.forEach(change => {
-          if (change.inserted.length > 0) {
-            decorations.push(createInlineChangedDecoration(change));
-          }
-          if (change.deleted.length > 0) {
-            decorations.push(createDeletedContentDecoration({
-              change,
-              doc: pluginState === null || pluginState === void 0 ? void 0 : pluginState.baselineDoc,
-              tr
+    view() {
+      return {
+        update(view, prevState) {
+          var _trackChangesPluginKe;
+          const prevShouldChangesBeDisplayed = (_trackChangesPluginKe = trackChangesPluginKey.getState(prevState)) === null || _trackChangesPluginKe === void 0 ? void 0 : _trackChangesPluginKe.shouldChangesBeDisplayed;
+          const currentPluginState = trackChangesPluginKey.getState(view.state);
+          const currentShouldChangesBeDisplayed = currentPluginState === null || currentPluginState === void 0 ? void 0 : currentPluginState.shouldChangesBeDisplayed;
+          if (prevShouldChangesBeDisplayed === false && currentShouldChangesBeDisplayed === true) {
+            var _currentPluginState$s, _api$core, _api$showDiff, _api$showDiff$command;
+            const steps = (_currentPluginState$s = currentPluginState === null || currentPluginState === void 0 ? void 0 : currentPluginState.steps) !== null && _currentPluginState$s !== void 0 ? _currentPluginState$s : [];
+            api === null || api === void 0 ? void 0 : (_api$core = api.core) === null || _api$core === void 0 ? void 0 : _api$core.actions.execute(api === null || api === void 0 ? void 0 : (_api$showDiff = api.showDiff) === null || _api$showDiff === void 0 ? void 0 : (_api$showDiff$command = _api$showDiff.commands) === null || _api$showDiff$command === void 0 ? void 0 : _api$showDiff$command.showDiff({
+              originalDoc: getBaselineFromSteps(view.state.doc, steps),
+              steps: steps.map(s => s.step)
             }));
+          } else if (currentShouldChangesBeDisplayed === false && prevShouldChangesBeDisplayed === true) {
+            var _api$core2, _api$showDiff2, _api$showDiff2$comman;
+            api === null || api === void 0 ? void 0 : (_api$core2 = api.core) === null || _api$core2 === void 0 ? void 0 : _api$core2.actions.execute(api === null || api === void 0 ? void 0 : (_api$showDiff2 = api.showDiff) === null || _api$showDiff2 === void 0 ? void 0 : (_api$showDiff2$comman = _api$showDiff2.commands) === null || _api$showDiff2$comman === void 0 ? void 0 : _api$showDiff2$comman.hideDiff);
           }
-        });
-        return decoration.add(tr.doc, decorations);
-      }
+        }
+      };
     }
   });
 };

package/dist/es2019/trackChangesPlugin.js

@@ -1,33 +1,49 @@
+import React from 'react';
 import { createTrackChangesPlugin, trackChangesPluginKey } from './pm-plugins/main';
 import { TOGGLE_TRACK_CHANGES_ACTION as ACTION } from './pm-plugins/types';
-export const trackChangesPlugin = () => ({
-  name: 'trackChanges',
-  pmPlugins() {
-    return [{
-      name: 'trackChangesPlugin',
-      plugin: createTrackChangesPlugin
-    }];
-  },
-  commands: {
-    toggleChanges: ({
-      tr
-    }) => {
-      return tr.setMeta(trackChangesPluginKey, {
-        action: ACTION.TOGGLE_TRACK_CHANGES
-      });
-    }
-  },
-  getSharedState: editorState => {
-    var _trackChangesPluginKe, _trackChangesPluginKe2;
-    if (!editorState) {
+import { TrackChangesToolbarButton } from './ui/TrackChangesToolbarButton';
+export const trackChangesPlugin = ({
+  api
+}) => {
+  var _api$primaryToolbar, _api$primaryToolbar$a;
+  const primaryToolbarComponent = () => {
+    return /*#__PURE__*/React.createElement(TrackChangesToolbarButton, {
+      api: api
+    });
+  };
+  api === null || api === void 0 ? void 0 : (_api$primaryToolbar = api.primaryToolbar) === null || _api$primaryToolbar === void 0 ? void 0 : (_api$primaryToolbar$a = _api$primaryToolbar.actions) === null || _api$primaryToolbar$a === void 0 ? void 0 : _api$primaryToolbar$a.registerComponent({
+    name: 'trackChanges',
+    component: primaryToolbarComponent
+  });
+  return {
+    name: 'trackChanges',
+    pmPlugins() {
+      return [{
+        name: 'trackChangesPlugin',
+        plugin: () => createTrackChangesPlugin(api)
+      }];
+    },
+    commands: {
+      toggleChanges: ({
+        tr
+      }) => {
+        return tr.setMeta(trackChangesPluginKey, {
+          action: ACTION.TOGGLE_TRACK_CHANGES
+        });
+      }
+    },
+    getSharedState: editorState => {
+      var _trackChangesPluginKe, _trackChangesPluginKe2;
+      if (!editorState) {
+        return {
+          isDisplayingChanges: false,
+          isShowDiffAvailable: false
+        };
+      }
       return {
-        isDisplayingChanges: false,
-        isShowDiffAvailable: false
+        isDisplayingChanges: Boolean((_trackChangesPluginKe = trackChangesPluginKey.getState(editorState)) === null || _trackChangesPluginKe === void 0 ? void 0 : _trackChangesPluginKe.shouldChangesBeDisplayed),
+        isShowDiffAvailable: Boolean((_trackChangesPluginKe2 = trackChangesPluginKey.getState(editorState)) === null || _trackChangesPluginKe2 === void 0 ? void 0 : _trackChangesPluginKe2.isShowDiffAvailable)
       };
     }
-    return {
-      isDisplayingChanges: Boolean((_trackChangesPluginKe = trackChangesPluginKey.getState(editorState)) === null || _trackChangesPluginKe === void 0 ? void 0 : _trackChangesPluginKe.shouldChangesBeDisplayed),
-      isShowDiffAvailable: Boolean((_trackChangesPluginKe2 = trackChangesPluginKey.getState(editorState)) === null || _trackChangesPluginKe2 === void 0 ? void 0 : _trackChangesPluginKe2.isShowDiffAvailable)
-    };
-  }
-});
+  };
+};