【WordPress】特定の投稿にだけパスワード保護を自動で適用する方法

function auto_password_protect_limited_category($post_id, $post, $update, $post_before) {
  // 自動保存・リビジョン時はスキップ
  if (defined('DOING_AUTOSAVE') && DOING_AUTOSAVE) return;
  if (wp_is_post_revision($post_id)) return;

  // 投稿タイプチェック
  $post_type = get_post_type($post_id);
  if ($post_type !== 'post') return;

  // カテゴリが「限定公開」（例: slug が 'limited'）か確認
  if (has_category('limited', $post_id)) {
    // 既にパスワード保護されていない場合のみ設定
    if (get_post_field('post_password', $post_id) === '') {
      // 任意のパスワードを設定（例：secret123）
      $update_args = array(
        'ID' => $post_id,
        'post_password' => 'secret123',
      );
      remove_action('wp_after_insert_post', 'auto_password_protect_limited_category'); // 再帰防止
      wp_update_post($update_args);
      add_action('wp_after_insert_post', 'auto_password_protect_limited_category', 10, 4);
    }
  }
}
add_action('wp_after_insert_post', 'auto_password_protect_limited_category', 10, 4);

function auto_password_protect_custom_post_type($post_id, $post, $update, $post_before) {
  if (defined('DOING_AUTOSAVE') && DOING_AUTOSAVE) return;
  if (wp_is_post_revision($post_id)) return;

  $post_type = get_post_type($post_id);
  if ($post_type !== 'event') return;

  if (get_post_field('post_password', $post_id) === '') {
    $update_args = array(
      'ID' => $post_id,
      'post_password' => 'eventpass',
    );
    remove_action('wp_after_insert_post', 'auto_password_protect_custom_post_type');
    wp_update_post($update_args);
    add_action('wp_after_insert_post', 'auto_password_protect_custom_post_type', 10, 4);
  }
}
add_action('wp_after_insert_post', 'auto_password_protect_custom_post_type', 10, 4);

$custom_password = 'pw-' . get_post_field('post_name', $post_id);