// @ts-nocheck
|
import CommandStack from 'diagram-js/lib/command/CommandStack';
|
import { bpmnGroup } from '../config/variableName';
|
|
class CustomCommandStack extends CommandStack {
|
constructor(eventBus, modeling) {
|
super(eventBus, modeling);
|
}
|
}
|
|
let sameId = '';
|
let isVerticalSameId = false;
|
let isHorizontalSameId = false;
|
|
CommandStack.prototype._internalExecute = function (action, redo) {
|
const command = action.command;
|
const context = action.context;
|
const handler = this._getHandler(command);
|
if (!handler) throw new Error('no command handler registered for <' + command + '>');
|
this._pushAction(action);
|
if (!redo) {
|
this._fire(command, 'preExecute', action);
|
if (handler.preExecute) handler.preExecute(context);
|
this._fire(command, 'preExecuted', action);
|
}
|
// guard against illegal nested command stack invocations
|
this._atomicDo(() => {
|
this._fire(command, 'execute', action);
|
if (handler.execute) this._markDirty(handler.execute(context));
|
if (action.command == 'element.updateProperties' || sameId == action.id) {
|
sameId = action.id;
|
} else {
|
if (action.id != 'updateProperties') this._executedAction(action, redo);
|
}
|
this._fire(command, 'executed', action);
|
});
|
if (!redo) {
|
this._fire(command, 'postExecute', action);
|
if (handler.postExecute) handler.postExecute(context);
|
this._fire(command, 'postExecuted', action);
|
}
|
this._popAction(action);
|
};
|
CommandStack.prototype._pushAction = function (action, isSameId = false) {
|
const execution = this._currentExecution;
|
const actions = execution.actions;
|
const baseAction = actions[0];
|
if (execution.atomic) throw new Error('illegal invocation in <execute> or <revert> phase (action: ' + action.command + ')');
|
if (action.command == 'element.updateProperties') action.id = 'updateProperties';
|
if (!action.id) action.id = (baseAction && baseAction.id) || this._createId(isSameId);
|
if (isVerticalSameId) action.type = 'beautification.vertical';
|
if (isHorizontalSameId) action.type = 'beautification.horizontal';
|
actions.push(action);
|
};
|
CommandStack.prototype._createId = function (isSameId = false) {
|
if (isSameId) {
|
this._uid = this._uid - 1;
|
}
|
return isHorizontalSameId || isVerticalSameId ? this._uid : this._uid++;
|
};
|
CommandStack.prototype.execute = function (command, context) {
|
let isSameId = false;
|
if (!command) throw new Error('command required');
|
// 设置一个开始和一个结束状态 如果遇到遇到开始 则全局设置该值为true 后续的入栈id设置成相同id
|
if (command === 'vertical.action.same.id.start') return (isVerticalSameId = true);
|
if (command === 'vertical.action.same.id.end') {
|
this._uid++;
|
return (isVerticalSameId = false);
|
}
|
if (command === 'horizontal.action.same.id.start') return (isHorizontalSameId = true);
|
if (command === 'horizontal.action.same.id.end') {
|
this._uid++;
|
return (isHorizontalSameId = false);
|
}
|
|
if (command === 'shape.create' && context.shape?.type === bpmnGroup) {
|
isSameId = true;
|
}
|
if (command === 'shape.resize') {
|
if (!(isVerticalSameId || isHorizontalSameId)) isSameId = true;
|
}
|
|
this._currentExecution.trigger = 'execute';
|
const action = { command: command, context: context };
|
this._pushAction(action, isSameId);
|
this._internalExecute(action);
|
this._popAction(action);
|
};
|
CommandStack.prototype.undo = function () {
|
let action = this._getUndoAction(),
|
next;
|
if (action) {
|
this._currentExecution.trigger = 'undo';
|
this._pushAction(action);
|
while (action) {
|
this._internalUndo(action);
|
next = this._getUndoAction();
|
if (!next || next.id !== action.id) {
|
break;
|
}
|
action = next;
|
}
|
if (action.command === 'shape.resize' || (action.command === 'shape.create' && action.context.shape?.type === bpmnGroup)) {
|
let id = action.id - 1;
|
while (action) {
|
this._internalUndo(action);
|
next = this._getUndoAction();
|
if (!next || next.id !== id) break;
|
action = next;
|
}
|
}
|
this._popAction();
|
}
|
};
|
CommandStack.prototype.redo = function () {
|
let action = this._getRedoAction(),
|
resizeNext,
|
next;
|
if (action) {
|
this._currentExecution.trigger = 'redo';
|
this._pushAction(action);
|
while (action) {
|
this._internalExecute(action, true);
|
next = this._getRedoAction();
|
if (!next || next.id !== action.id) {
|
if (next?.command === 'shape.resize') {
|
resizeNext = next;
|
}
|
break;
|
}
|
action = next;
|
}
|
if (resizeNext) {
|
let id = resizeNext.id + 1;
|
while (resizeNext) {
|
this._internalExecute(resizeNext, true);
|
next = this._getRedoAction();
|
if (!next || next.id !== id) {
|
break;
|
}
|
action = next;
|
}
|
}
|
this._popAction();
|
}
|
};
|
|
export default CustomCommandStack;
|
