【WordPress】管理画面の一覧テーブルにCSVダウンロードボタンを追加する方法

function export_custom_post_csv() {
  if (!is_admin() || !current_user_can('manage_options')) return;
  if (!isset($_GET['export_csv']) || $_GET['export_csv'] !== 'true') return;

  $post_type = isset($_GET['post_type']) ? sanitize_key($_GET['post_type']) : 'post';

  header('Content-Type: text/csv; charset=UTF-8');
  header('Content-Disposition: attachment; filename="' . $post_type . '-list.csv"');
  $output = fopen('php://output', 'w');

  // ExcelでUTF-8のCSVを開くと日本語が文字化けするため、先頭にBOMを書き込む
  fwrite($output, "\xEF\xBB\xBF");

  $args = array(
    'post_type'      => $post_type,
    'post_status'    => 'publish',
    'posts_per_page' => -1,
  );
  $query = new WP_Query($args);

  fputcsv($output, array('ID', 'タイトル', '公開日'));

  if ($query->have_posts()) {
    while ($query->have_posts()) {
      $query->the_post();
      fputcsv($output, array(get_the_ID(), get_the_title(), get_the_date('Y-m-d')));
    }
  }

  fclose($output);
  exit;
}
add_action('admin_init', 'export_custom_post_csv');

function add_csv_export_button($which) {
  // 上部（top）と下部（bottom）の両方で発火するため、二重表示を防ぐため上部のみ出力する
  if ($which !== 'top') return;

  $screen = get_current_screen();
  if ($screen->base === 'edit' && post_type_exists($screen->post_type)) {
    $url = add_query_arg(array(
      'export_csv' => 'true',
      'post_type'  => $screen->post_type
    ), admin_url('edit.php'));

    echo '<div class="wrap"><a href="' . esc_url($url) . '" class="button button-primary">CSVダウンロード</a></div>';
  }
}
add_action('manage_posts_extra_tablenav', 'add_csv_export_button');

$author = get_the_author();
$custom_value = get_post_meta(get_the_ID(), 'your_custom_field', true);
fputcsv($output, array(get_the_ID(), get_the_title(), $author, $custom_value));