| 6dd74de | | | 1 | import type { ValidationAcceptor, ValidationChecks } from 'langium'; |
| 6dd74de | | | 2 | import type { MermaidAstType, Treemap } from '../generated/ast.js'; |
| 6dd74de | | | 3 | import type { TreemapServices } from './module.js'; |
| 6dd74de | | | 4 | |
| 6dd74de | | | 5 | /** |
| 6dd74de | | | 6 | * Register custom validation checks. |
| 6dd74de | | | 7 | */ |
| 6dd74de | | | 8 | export function registerValidationChecks(services: TreemapServices) { |
| 6dd74de | | | 9 | const validator = services.validation.TreemapValidator; |
| 6dd74de | | | 10 | const registry = services.validation.ValidationRegistry; |
| 6dd74de | | | 11 | if (registry) { |
| 6dd74de | | | 12 | // Use any to bypass type checking since we know Treemap is part of the AST |
| 6dd74de | | | 13 | // but the type system is having trouble with it |
| 6dd74de | | | 14 | const checks: ValidationChecks<MermaidAstType> = { |
| 6dd74de | | | 15 | Treemap: validator.checkSingleRoot.bind(validator), |
| 6dd74de | | | 16 | // Remove unused validation for TreemapRow |
| 6dd74de | | | 17 | }; |
| 6dd74de | | | 18 | registry.register(checks, validator); |
| 6dd74de | | | 19 | } |
| 6dd74de | | | 20 | } |
| 6dd74de | | | 21 | |
| 6dd74de | | | 22 | /** |
| 6dd74de | | | 23 | * Implementation of custom validations. |
| 6dd74de | | | 24 | */ |
| 6dd74de | | | 25 | export class TreemapValidator { |
| 6dd74de | | | 26 | /** |
| 6dd74de | | | 27 | * Validates that a treemap has only one root node. |
| 6dd74de | | | 28 | * A root node is defined as a node that has no indentation. |
| 6dd74de | | | 29 | */ |
| 6dd74de | | | 30 | checkSingleRoot(doc: Treemap, accept: ValidationAcceptor): void { |
| 6dd74de | | | 31 | let rootNodeIndentation; |
| 6dd74de | | | 32 | |
| 6dd74de | | | 33 | for (const row of doc.TreemapRows) { |
| 6dd74de | | | 34 | // Skip non-node items or items without a type |
| 6dd74de | | | 35 | if (!row.item) { |
| 6dd74de | | | 36 | continue; |
| 6dd74de | | | 37 | } |
| 6dd74de | | | 38 | |
| 6dd74de | | | 39 | if ( |
| 6dd74de | | | 40 | rootNodeIndentation === undefined && // Check if this is a root node (no indentation) |
| 6dd74de | | | 41 | row.indent === undefined |
| 6dd74de | | | 42 | ) { |
| 6dd74de | | | 43 | rootNodeIndentation = 0; |
| 6dd74de | | | 44 | } else if (row.indent === undefined) { |
| 6dd74de | | | 45 | // If we've already found a root node, report an error |
| 6dd74de | | | 46 | accept('error', 'Multiple root nodes are not allowed in a treemap.', { |
| 6dd74de | | | 47 | node: row, |
| 6dd74de | | | 48 | property: 'item', |
| 6dd74de | | | 49 | }); |
| 6dd74de | | | 50 | } else if ( |
| 6dd74de | | | 51 | rootNodeIndentation !== undefined && |
| 6dd74de | | | 52 | rootNodeIndentation >= parseInt(row.indent, 10) |
| 6dd74de | | | 53 | ) { |
| 6dd74de | | | 54 | accept('error', 'Multiple root nodes are not allowed in a treemap.', { |
| 6dd74de | | | 55 | node: row, |
| 6dd74de | | | 56 | property: 'item', |
| 6dd74de | | | 57 | }); |
| 6dd74de | | | 58 | } |
| 6dd74de | | | 59 | } |
| 6dd74de | | | 60 | } |
| 6dd74de | | | 61 | } |