feat: [sequence] add participant boxes

hikerpig · May 9, 2022 · f22f85c · f22f85c
1 parent 5ecb6db
commit f22f85c
Show file tree

Hide file tree

Showing 11 changed files with 645 additions and 309 deletions.
diff --git a/.changeset/metal-parents-work.md b/.changeset/metal-parents-work.md
@@ -0,0 +1,5 @@
+---
+'@pintora/diagrams': patch
+---
+
+feat: [sequence] add participant boxes
diff --git a/packages/pintora-diagrams/src/sequence/__tests__/sequence-parser.spec.js b/packages/pintora-diagrams/src/sequence/__tests__/sequence-parser.spec.js
@@ -10,7 +10,7 @@ describe('sequence parser', () => {
   it('can parse unicode chars', () => {
     const backquoteExample = stripStartEmptyLines(`
 sequenceDiagram
-  autonumber 
+  autonumber
   用户 ->> Pintora: 帮我画张时序图
   activate Pintora
   alt DSL 正确
@@ -106,10 +106,10 @@ sequenceDiagram
 
   it('can parse participant', () => {
     const example = stripStartEmptyLines(`
-sequenceDiagram 
+sequenceDiagram
   participant A as "Alice"
   participant B as Bobby Bob
-  participant C  
+  participant C
   A-->B: hello
   A-->C: yoho
   `)
@@ -175,6 +175,58 @@ sequenceDiagram
     })
   })
 
+  it('can parse participant inside box', () => {
+    const example = stripStartEmptyLines(`
+sequenceDiagram
+  box #d6d3fa "group participants"
+  participant A as "Alice"
+  participant B
+  endbox
+  `)
+    parse(example)
+    const ir = db.getDiagramIR()
+    // console.log(JSON.stringify(ir, null, 2))
+    expect(ir.participantBoxes).toMatchObject({
+      1: {
+        id: '1',
+        text: 'group participants',
+        actors: ['A', 'B'],
+        background: '#d6d3fa',
+      },
+    })
+    expect(ir.actors).toMatchObject({
+      A: {
+        name: 'A',
+        description: 'Alice',
+        boxId: '1',
+      },
+      B: {
+        name: 'B',
+        boxId: '1',
+      },
+    })
+  })
+
+  it('can parse participant box without title', () => {
+    const example = stripStartEmptyLines(`
+sequenceDiagram
+  box
+  participant A as "Alice"
+  participant B
+  endbox
+  `)
+    parse(example)
+    const ir = db.getDiagramIR()
+    // console.log(JSON.stringify(ir, null, 2))
+    expect(ir.participantBoxes).toMatchObject({
+      1: {
+        id: '1',
+        text: null,
+        actors: ['A', 'B'],
+      },
+    })
+  })
+
   it('can parse multiline message', () => {
     const example = stripStartEmptyLines(`
 sequenceDiagram