prettier-plugin-solidity/src/slang-comments/handlers/handle-else-branch-comments.ts

A Prettier Plugin for automatically formatting your Solidity code.

import { NonterminalKind } from '@nomicfoundation/slang/cst';
import { util } from 'prettier';
import addCollectionFirstComment from './add-collection-first-comment.js';

import type { HandlerParams } from './types.d.ts';

const { addLeadingComment } = util;

export default function handleElseBranchComments({
  enclosingNode,
  followingNode,
  comment
}: HandlerParams): boolean {
  if (enclosingNode?.kind !== NonterminalKind.ElseBranch || !followingNode) {
    return false;
  }

  if (
    followingNode === enclosingNode.body &&
    followingNode.variant.kind === NonterminalKind.IfStatement
  ) {
    if (followingNode.variant.body.variant.kind === NonterminalKind.Block) {
      addCollectionFirstComment(
        followingNode.variant.body.variant.statements,
        comment
      );
    } else {
      addLeadingComment(followingNode.variant.body.variant, comment);
    }
    return true;
  }

  return false;
}