From 1d98ead5c2e4dfed1dfddbb4df3ef997d6aff3d8 Mon Sep 17 00:00:00 2001
From: Subhash Halder <halder.subhash@gmail.com>
Date: Sun, 2 Jul 2023 13:18:28 +0530
Subject: [PATCH] Added support for Diagram title

---
 packages/mermaid/src/diagrams/xychart/parser/xychart.jison | 1 +
 .../src/diagrams/xychart/parser/xychart.jison.spec.ts      | 7 +++++++
 2 files changed, 8 insertions(+)

diff --git a/packages/mermaid/src/diagrams/xychart/parser/xychart.jison b/packages/mermaid/src/diagrams/xychart/parser/xychart.jison
index 047eeebc5..0f211fe87 100644
--- a/packages/mermaid/src/diagrams/xychart/parser/xychart.jison
+++ b/packages/mermaid/src/diagrams/xychart/parser/xychart.jison
@@ -154,6 +154,7 @@ statement
   :
   | SPACE statement
 	| directive
+	| title title_value  { $$=$2.trim();yy.setDiagramTitle($$); }
   | X_AXIS parseXAxis
   | Y_AXIS parseYAxis
   | parseLine
diff --git a/packages/mermaid/src/diagrams/xychart/parser/xychart.jison.spec.ts b/packages/mermaid/src/diagrams/xychart/parser/xychart.jison.spec.ts
index 1ce73abf3..a999d9b8d 100644
--- a/packages/mermaid/src/diagrams/xychart/parser/xychart.jison.spec.ts
+++ b/packages/mermaid/src/diagrams/xychart/parser/xychart.jison.spec.ts
@@ -11,6 +11,7 @@ const parserFnConstructor = (str: string) => {
 const mockDB: Record<string, Mock<any, any>> = {
   parseDirective: vi.fn(),
   setOrientation: vi.fn(),
+  setDiagramTitle: vi.fn(),
   setXAxisTitle: vi.fn(),
   setXAxisRangeData: vi.fn(),
   addXAxisBand: vi.fn(),
@@ -43,6 +44,12 @@ describe('Testing xychart jison file', () => {
     expect(parserFnConstructor(str)).not.toThrow();
   });
 
+  it('parse title of the chart', () => {
+    const str = 'xychart-beta \n title This is a title';
+    expect(parserFnConstructor(str)).not.toThrow();
+    expect(mockDB.setDiagramTitle).toHaveBeenCalledWith('This is a title');
+  });
+
   it('should be able to parse directive', () => {
     const str =
       '%%{init: {"xychart": {"chartWidth": 600, "chartHeight": 600} } }%%  \n xychart-beta';