UNPKG

prettier-plugin-solidity

Version:

A Prettier Plugin for automatically formatting your Solidity code.

73 lines (62 loc) 2.3 kB
import { NonterminalKind, TerminalNode } from '@nomicfoundation/slang/cst'; import { getNodeMetadata, updateMetadata } from '../slang-utils/metadata.js'; import { ExpressionStatement } from './ExpressionStatement.js'; import { VariableDeclarationStatement } from './VariableDeclarationStatement.js'; import { TupleDeconstructionStatement } from './TupleDeconstructionStatement.js'; import type * as ast from '@nomicfoundation/slang/ast'; import type { AstPath, Doc, ParserOptions } from 'prettier'; import type { AstNode } from './types.d.ts'; import type { PrintFunction, SlangNode } from '../types.d.ts'; export class ForStatementInitialization implements SlangNode { readonly kind = NonterminalKind.ForStatementInitialization; comments; loc; variant: | ExpressionStatement | VariableDeclarationStatement | TupleDeconstructionStatement | string; constructor( ast: ast.ForStatementInitialization, options: ParserOptions<AstNode> ) { let metadata = getNodeMetadata(ast); if (ast.variant instanceof TerminalNode) { this.variant = ast.variant.unparse(); } else { switch (ast.variant.cst.kind) { case NonterminalKind.ExpressionStatement: this.variant = new ExpressionStatement( ast.variant as ast.ExpressionStatement, options ); break; case NonterminalKind.VariableDeclarationStatement: this.variant = new VariableDeclarationStatement( ast.variant as ast.VariableDeclarationStatement, options ); break; case NonterminalKind.TupleDeconstructionStatement: this.variant = new TupleDeconstructionStatement( ast.variant as ast.TupleDeconstructionStatement, options ); break; default: throw new Error(`Unexpected variant: ${ast.variant.cst.kind}`); } } metadata = updateMetadata( metadata, typeof this.variant === 'string' ? [] : [this.variant] ); this.comments = metadata.comments; this.loc = metadata.loc; } print(path: AstPath<ForStatementInitialization>, print: PrintFunction): Doc { return typeof this.variant === 'string' ? this.variant : path.call(print, 'variant'); } }