prettier-plugin-solidity
Version:
A Prettier Plugin for automatically formatting your Solidity code.
73 lines (62 loc) • 2.3 kB
text/typescript
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');
}
}