UNPKG

prettier-plugin-solidity

Version:

A Prettier Plugin for automatically formatting your Solidity code.

49 lines 2.15 kB
import { coerce, satisfies } from 'semver'; import { NonterminalKind } from '@nomicfoundation/slang/cst'; import { printFunction } from '../slang-printers/print-function.js'; import { getNodeMetadata, updateMetadata } from '../slang-utils/metadata.js'; import { FunctionName } from './FunctionName.js'; import { ParametersDeclaration } from './ParametersDeclaration.js'; import { FunctionAttributes } from './FunctionAttributes.js'; import { ReturnsDeclaration } from './ReturnsDeclaration.js'; import { FunctionBody } from './FunctionBody.js'; export class FunctionDefinition { constructor(ast, options) { this.kind = NonterminalKind.FunctionDefinition; let metadata = getNodeMetadata(ast); this.name = new FunctionName(ast.name); this.parameters = new ParametersDeclaration(ast.parameters, options); this.attributes = new FunctionAttributes(ast.attributes, options); if (ast.returns) { this.returns = new ReturnsDeclaration(ast.returns, options); } this.body = new FunctionBody(ast.body, options); metadata = updateMetadata(metadata, [ this.name, this.parameters, this.attributes, this.returns, this.body ]); this.comments = metadata.comments; this.loc = metadata.loc; // Older versions of Solidity defined a constructor as a function having // the same name as the contract. const compiler = coerce(options.compiler); if (compiler && satisfies(compiler, '>=0.5.0')) { this.cleanModifierInvocationArguments(); } } cleanModifierInvocationArguments() { for (const attribute of this.attributes.items) { if (typeof attribute.variant !== 'string' && attribute.variant.kind === NonterminalKind.ModifierInvocation) { attribute.variant.cleanModifierInvocationArguments(); } } } print(path, print) { return printFunction(['function ', path.call(print, 'name')], this, path, print); } } //# sourceMappingURL=FunctionDefinition.js.map