如何在进入数学模式和退出数学模式时插入命令？

Question

我想\special在进入数学模式时和退出数学模式时插入。

我使用everyhook.sty编写了以下源代码：

\documentclass{amsart}
\usepackage{everyhook}
\newcommand{\boe}{\special{b}}
\newcommand{\eoe}{\special{e}}
\PushPostHook{math}{\boe\aftergroup\eoe}
\PushPostHook{display}{\boe\aftergroup\eoe}
\begin{document}
\(x\)
\[y\]
\begin{align}z\end{align}
\end{document}

但这不起作用。我遇到以下错误：

! Improper \halign inside $$'s.
<recently read> \halign 
                        
l.10 \begin{align}z\end{align}

是否有适用于以下三种数学模式的好的解决方案？：（ \( \)内联）、\[ \]（显示）和\begin{align} \end{align}（amsmath）。