WordPress友链申请功能开发踩坑记录

<?php
function friendlink_apply_handle_post() {
  if ($_SERVER['REQUEST_METHOD'] !== 'POST') return;
  if (empty($_POST['friendlink_apply_submit'])) return;
  if (defined('FRIENDLINK_APPLY_HANDLED')) return;
  define('FRIENDLINK_APPLY_HANDLED', true);

  if (!isset($_POST['_wpnonce']) || !wp_verify_nonce($_POST['_wpnonce'], 'friendlink_apply')) {
    friendlink_apply_redirect_with_status('nonce_fail');
  }

  if (!function_exists('wp_insert_link')) {
    require_once ABSPATH . 'wp-admin/includes/bookmark.php';
  }

  $name  = sanitize_text_field($_POST['fl_name'] ?? '');
  $url   = esc_url_raw($_POST['fl_url'] ?? '');
  $cover = esc_url_raw($_POST['fl_cover'] ?? '');
  $desc  = sanitize_textarea_field($_POST['fl_desc'] ?? '');

  if (!$name || !$url) {
    friendlink_apply_redirect_with_status('missing');
  }

  $linkdata = [
    'link_name'        => $name,
    'link_url'         => $url,
    'link_image'       => $cover,
    'link_description' => $desc,
    'link_visible'     => 'N',
    'link_notes'       => 'pending|' . time(),
  ];

  $new_id = wp_insert_link($linkdata, true);
  if (is_wp_error($new_id)) {
    friendlink_apply_redirect_with_status('insert_fail');
  }

  friendlink_apply_redirect_with_status('success');
}
add_action('template_redirect', 'friendlink_apply_handle_post');

function friendlink_apply_redirect_with_status($status) {
  $page_id = function_exists('get_queried_object_id') ? get_queried_object_id() : 0;
  $base = $page_id ? get_permalink($page_id) : '';
  if (!$base) $base = function_exists('wp_get_raw_referer') ? wp_get_raw_referer() : '';
  if (!$base) $base = home_url('/');

  $redirect = add_query_arg(['fl_apply' => $status], $base);
  $redirect .= '#friendlink-apply';
  wp_safe_redirect($redirect);
  exit;
}

function friendlink_apply_shortcode() {
  $msg = '';
  $status = isset($_GET['fl_apply']) ? sanitize_text_field($_GET['fl_apply']) : '';

  if ($status === 'success') {
    $msg = '<div class="alert alert-success">感谢申请！等待审核通过后展示。</div>';
  } elseif ($status === 'missing') {
    $msg = '<div class="alert alert-warning">请填写站点名称和网址。</div>';
  } elseif ($status === 'nonce_fail') {
    $msg = '<div class="alert alert-danger">安全校验失败，刷新重试。</div>';
  } elseif ($status === 'insert_fail') {
    $msg = '<div class="alert alert-danger">提交失败，稍后重试。</div>';
  }

  ob_start();
  ?>
  <div id="friendlink-apply" class="friendlink-apply card shadow-sm">
    <div class="card-body">
      <h3 class="card-title">申请友链</h3>
      <?php echo $msg; ?>
      <form method="post">
        <?php wp_nonce_field('friendlink_apply'); ?>
        <div class="mb-3">
          <label class="form-label">站点名称</label>
          <input class="form-control" name="fl_name" required placeholder="YHalo Blog">
        </div>
        <div class="mb-3">
          <label class="form-label">网站地址</label>
          <input class="form-control" name="fl_url" required placeholder="https://example.com">
        </div>
        <div class="mb-3">
          <label class="form-label">封面图</label>
          <input class="form-control" name="fl_cover" placeholder="https://example.com/cover.jpg">
        </div>
        <div class="mb-3">
          <label class="form-label">站点描述</label>
          <textarea class="form-control" name="fl_desc" rows="3" placeholder="一句话介绍"></textarea>
        </div>
        <button class="btn btn-primary" type="submit" name="friendlink_apply_submit" value="1">提交</button>
      </form>
    </div>
  </div>

  <?php if ($status === 'success') : ?>
  <script>
  (function () {
    var checkSvg = "data:image/svg+xml,%3Csvg xmlns='http://www.w3.org/2000/svg' viewBox='0 0 24 24'%3E%3Cpath fill='none' stroke='%23fff' stroke-width='3.2' stroke-linecap='round' stroke-linejoin='round' d='M20 6L9 17l-5-5'/%3E%3C/svg%3E";
    if (!window.iziToast) return;
    iziToast.show({
      class: 'fl-apply-toast',
      title: '提交成功',
      message: '等待审核～',
      position: 'topRight',
      transitionIn: 'fadeInUp',
      transitionOut: 'fadeOutUp',
      timeout: 4000,
      progressBar: true,
      progressBarColor: 'rgba(255,255,255,.9)',
      backgroundColor: 'rgb(45, 206, 137)',
      iconUrl: checkSvg,
      close: true,
      zindex: 9999
    });
  })();
  </script>
  <?php endif; ?>

  <style>
  .friendlink-apply{border-radius:16px;border:1px solid rgba(0,0,0,.08);overflow:hidden}
  .friendlink-apply .card-body{padding:20px}
  .friendlink-apply .card-title{margin:0 0 12px;font-size:20px}
  .friendlink-apply .form-label{display:block;margin:0 0 6px;opacity:.85}
  .friendlink-apply .form-control{width:100%;border-radius:12px;border:1px solid rgba(0,0,0,.12);padding:10px 12px}
  .friendlink-apply .btn{border-radius:12px;padding:10px 14px}
  .friendlink-apply .alert{margin:12px 0;padding:10px 12px;border-radius:12px}
  .friendlink-apply textarea{resize:none}
  .iziToast.fl-apply-toast{border-radius:6px!important}
  .iziToast.fl-apply-toast .iziToast-title,
  .iziToast.fl-apply-toast .iziToast-message{color:#fff!important}
  .iziToast.fl-apply-toast .iziToast-icon{width:22px!important;height:22px!important}
  </style>
  <?php
  return ob_get_clean();
}
add_shortcode('friendlink_apply', 'friendlink_apply_shortcode');