possible option is to return indent length instead of actual indent (speculative since indent works locally, but not in pipeline)

Author: lee-treehouse

packages/mermaid/src/diagrams/treeView/parser.ts

@@ -6,7 +6,9 @@ import { parse, type TreeView } from '@mermaid-js/parser';
 
 const populate = (ast: TreeView) => {
   populateCommonDb(ast, db);
-  ast.nodes.map((node) => db.addNode(node.indent?.length ?? 0, node.name));
+  ast.nodes.map((node) =>
+    db.addNode(node.indentLength ? parseInt(node.indentLength) : 0, node.name)
+  );
 };
 
 export const parser: ParserDefinition = {

packages/parser/src/language/treeView/module.ts

@@ -12,7 +12,7 @@ import {
   inject,
 } from 'langium';
 
-import { CommonValueConverter } from '../common/valueConverter.js';
+import { TreeViewValueConverter } from './valueConverter.js';
 import {
   MermaidGeneratedSharedModule,
   TreeViewGrammarGeneratedModule as TreeViewGeneratedModule,
@@ -25,7 +25,7 @@ import { TreeViewTokenBuilder } from './tokenBuilder.js';
 interface TreeViewAddedServices {
   parser: {
     TokenBuilder: TreeViewTokenBuilder;
-    ValueConverter: CommonValueConverter;
+    ValueConverter: TreeViewValueConverter;
   };
 }
 
@@ -44,7 +44,7 @@ export const TreeViewModule: Module<
 > = {
   parser: {
     TokenBuilder: () => new TreeViewTokenBuilder(),
-    ValueConverter: () => new CommonValueConverter(),
+    ValueConverter: () => new TreeViewValueConverter(),
   },
 };
 

packages/parser/src/language/treeView/treeView.langium

@@ -34,7 +34,8 @@ hidden terminal ML_COMMENT: /\%\%[^\n]*/;
 hidden terminal NL: /\r?\n/;
 
 TreeNode: 
-    indent=INDENTATION?
+    // actually contains the indent, but is converted to a length value by valueConverter
+    indentLength=INDENTATION?
     name=TEXT
 ;
 

packages/parser/src/language/treeView/valueConverter.ts

@@ -0,0 +1,15 @@
+import type { CstNode, GrammarAST, ValueType } from 'langium';
+import { AbstractMermaidValueConverter } from '../common/index.js';
+
+export class TreeViewValueConverter extends AbstractMermaidValueConverter {
+  protected runCustomConverter(
+    rule: GrammarAST.AbstractRule,
+    input: string,
+    _cstNode: CstNode
+  ): ValueType | undefined {
+    if (rule.name === 'INDENTATION') {
+      return input?.length || 0;
+    }
+    return undefined;
+  }
+}

packages/parser/tests/treeView.test.ts

@@ -26,7 +26,7 @@ describe('TreeView Parser', () => {
       expect(result.value.$type).toBe('TreeView');
       expect(result.value.nodes).toHaveLength(1);
       expect(result.value.nodes[0].name).toBe('Root');
-      expect(result.value.nodes[0].indent).toBe(undefined);
+      expect(result.value.nodes[0].indentLength).toBe(undefined);
     });
 
     it('should parse a treeView with child nodes', () => {
@@ -36,16 +36,16 @@ describe('TreeView Parser', () => {
       expect(result.value.nodes).toHaveLength(4);
 
       expect(result.value.nodes[0].name).toBe('Root');
-      expect(result.value.nodes[0].indent).toBe(undefined);
+      expect(result.value.nodes[0].indentLength).toBe(undefined);
 
       expect(result.value.nodes[1].name).toBe('Child1');
-      // expect(result.value.nodes[1].indent?.length).toBe(4);
+      expect(result.value.nodes[1].indentLength).toBe(4);
 
       expect(result.value.nodes[2].name).toBe('Child2');
-      // expect(result.value.nodes[2].indent?.length).toBe(4);
+      expect(result.value.nodes[2].indentLength).toBe(4);
 
       expect(result.value.nodes[3].name).toBe('Child3');
-      // expect(result.value.nodes[3].indent?.length).toBe(8);
+      expect(result.value.nodes[3].indentLength).toBe(8);
     });
   });