使用短代码添加条件按钮

Question

我正在尝试使用[button]短代码添加几个按钮。我想要实现的是第一个按钮BUTTON-01向所有用户显示，而另一个按钮BUTTON-02仅显示给＆＃34;作者＆＃34;用户角色。

这是我的代码：

function shortcode_button() {

    global $current_user;
    get_currentuserinfo();

    // Show this button to ALL
    return '<a class="button" href="http://example.com/01">FOR ALL</a>';

    // Show button to AUTHORS ONLY
    if ( user_can( $current_user, "author" ) ) {
            return '<a class="button" href="http://example.com/02">FOR AUTHORS ONLY</a>';
        }

}
add_shortcode( 'button', 'shortcode_button' );

以AUTHOR身份登录后，我可以看到第一个按钮，但第二个按钮不会显示。

我错过了什么吗？是否应该有endif;声明？