如何编写此命令行？

Question

我正在使用oh-my-zsh。

我想制作ZSH命令mx google.com，

并运行以下内容：

dig +short mx $1

然后，如果我

a p

在“ p”代表“以前”的地方运行（我在此处向其他更常见的约定开放）

我希望它运行（假设是否以前的行输出3 IP）：

dig + short a ip1
dig + short a ip2
dig + short a ip3

对于每个结果，返回了上一个命令中。

我要支持上述约定的命令是a，txt，mx，cname

这是当前全部我有：

alias a='f() { dig +short a $1 };f'

但是我不确定如何结合两者。