feat: add isFlowNode and isArtifact in ShapeUtil

Author: tbouffard

src/model/bpmn/internal/shape/utils.ts

@@ -107,9 +107,23 @@ export class ShapeUtil {
     return Object.values(ShapeBpmnElementKind).filter(kind => !ShapeUtil.isPoolOrLane(kind));
   }
 
+  /**
+   * @since 0.48.0
+   */
+  static isFlowNode(kind: ShapeBpmnElementKind | string): boolean {
+    return ShapeUtil.flowNodeKinds().includes(kind as ShapeBpmnElementKind);
+  }
+
   static isPoolOrLane(kind: ShapeBpmnElementKind | string): boolean {
     return kind == ShapeBpmnElementKind.POOL || kind == ShapeBpmnElementKind.LANE;
   }
+
+  /**
+   * @since 0.48.0
+   */
+  static isArtifact(kind: ShapeBpmnElementKind | string): boolean {
+    return kind === ShapeBpmnElementKind.GROUP || kind === ShapeBpmnElementKind.TEXT_ANNOTATION;
+  }
 }
 
 function filterKind(suffix: string, options?: FilterParameter): ShapeBpmnElementKind[] {

test/unit/model/bpmn/internal/shape/utils.test.ts

@@ -14,7 +14,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 */
 
-import { ShapeBpmnElementKind, ShapeUtil } from '@lib/model/bpmn/internal';
+import { FlowKind, ShapeBpmnElementKind, ShapeUtil } from '@lib/model/bpmn/internal';
 
 describe('ShapeUtil', () => {
   it('top level bpmn event kinds', () => {
@@ -29,6 +29,14 @@ describe('ShapeUtil', () => {
     expect(tasks).toContain(ShapeBpmnElementKind.TASK_USER);
   });
 
+  it('flow node kinds', () => {
+    const flowNodeKinds = ShapeUtil.flowNodeKinds();
+    expect(flowNodeKinds).toContain(ShapeBpmnElementKind.TASK);
+    expect(flowNodeKinds).toContain(ShapeBpmnElementKind.EVENT_INTERMEDIATE_CATCH);
+    expect(flowNodeKinds).toContain(ShapeBpmnElementKind.GROUP); // artifact should not be included
+    expect(flowNodeKinds).not.toContain(ShapeBpmnElementKind.POOL);
+  });
+
   describe('Is pool or lane?', () => {
     it.each([
       [ShapeBpmnElementKind.CALL_ACTIVITY],
@@ -46,6 +54,46 @@ describe('ShapeUtil', () => {
     });
   });
 
+  describe('isArtifact', () => {
+    test.each`
+      kind                                     | expected
+      ${ShapeBpmnElementKind.EVENT_END}        | ${false}
+      ${ShapeBpmnElementKind.GATEWAY_PARALLEL} | ${false}
+      ${ShapeBpmnElementKind.TASK}             | ${false}
+      ${ShapeBpmnElementKind.SUB_PROCESS}      | ${false}
+      ${ShapeBpmnElementKind.CALL_ACTIVITY}    | ${false}
+      ${ShapeBpmnElementKind.POOL}             | ${false}
+      ${ShapeBpmnElementKind.LANE}             | ${false}
+      ${ShapeBpmnElementKind.GROUP}            | ${true}
+      ${ShapeBpmnElementKind.TEXT_ANNOTATION}  | ${true}
+      ${FlowKind.MESSAGE_FLOW}                 | ${false}
+      ${'unknown'}                             | ${false}
+      ${'receiveTask'}                         | ${false}
+    `('$kind isArtifact? $expected', ({ kind, expected }: Record<string, unknown>) => {
+      expect(ShapeUtil.isArtifact(kind as string)).toBe(expected);
+    });
+  });
+
+  describe('isFlowNode', () => {
+    test.each`
+      kind                                     | expected
+      ${ShapeBpmnElementKind.EVENT_END}        | ${true}
+      ${ShapeBpmnElementKind.GATEWAY_PARALLEL} | ${true}
+      ${ShapeBpmnElementKind.TASK}             | ${true}
+      ${ShapeBpmnElementKind.SUB_PROCESS}      | ${true}
+      ${ShapeBpmnElementKind.CALL_ACTIVITY}    | ${true}
+      ${ShapeBpmnElementKind.POOL}             | ${false}
+      ${ShapeBpmnElementKind.LANE}             | ${false}
+      ${ShapeBpmnElementKind.GROUP}            | ${true}
+      ${ShapeBpmnElementKind.TEXT_ANNOTATION}  | ${true}
+      ${FlowKind.MESSAGE_FLOW}                 | ${false}
+      ${'unknown'}                             | ${false}
+      ${'receiveTask'}                         | ${true}
+    `('$kind isFlowNode? $expected', ({ kind, expected }: Record<string, unknown>) => {
+      expect(ShapeUtil.isFlowNode(kind as string)).toBe(expected);
+    });
+  });
+
   describe('Reference kinds cannot be modified', () => {
     it.each`
       kind            | kindsFunction