UNPKG

prettier-plugin-solidity

Version:

A Prettier Plugin for automatically formatting your Solidity code.

40 lines (28 loc) 1.18 kB
import { NonterminalKind } from '@nomicfoundation/slang/cst'; import { getNodeMetadata, updateMetadata } from '../slang-utils/metadata.js'; import { VersionExpressionSet } from './VersionExpressionSet.js'; import type * as ast from '@nomicfoundation/slang/ast'; import type { AstPath, Doc } from 'prettier'; import type { PrintFunction, SlangNode } from '../types.d.ts'; export class VersionExpressionSets implements SlangNode { readonly kind = NonterminalKind.VersionExpressionSets; comments; loc; items: VersionExpressionSet[]; separators: string[]; constructor(ast: ast.VersionExpressionSets) { let metadata = getNodeMetadata(ast, true); this.items = ast.items.map((item) => new VersionExpressionSet(item)); this.separators = ast.separators.map((separator) => separator.unparse()); metadata = updateMetadata(metadata, [this.items]); this.comments = metadata.comments; this.loc = metadata.loc; } print(path: AstPath<VersionExpressionSets>, print: PrintFunction): Doc { return path .map(print, 'items') .map((item, index) => index === 0 ? item : [` ${this.separators[index - 1]} `, item] ); } }