【WordPress】管理画面でカテゴリーの並び順を自由に変更する方法

class Custom_Order_Category_Checklist_Walker extends Walker_Category_Checklist {
    protected $custom_order;

    public function __construct($custom_order = array()) {
        $this->custom_order = $custom_order;
    }

    public function walk($elements, $max_depth, ...$args) {
        $custom_order = $this->custom_order;

        usort($elements, function ($a, $b) use ($custom_order) {
            $pos_a = array_search($a->term_id, $custom_order);
            $pos_b = array_search($b->term_id, $custom_order);
            // 指定リストにないカテゴリーは末尾に回す
            $pos_a = ($pos_a === false) ? PHP_INT_MAX : $pos_a;
            $pos_b = ($pos_b === false) ? PHP_INT_MAX : $pos_b;
            return $pos_a <=> $pos_b;
        });

        return parent::walk($elements, $max_depth, ...$args);
    }
}

function custom_category_order_in_admin($args, $post_id) {
    // カスタム順序にしたいカテゴリーIDの配列（任意の順番で指定）
    $custom_order = array(5, 12, 8, 3, 1);

    // タクソノミーがカテゴリーであることを確認
    if ($args['taxonomy'] === 'category') {
        $args['walker'] = new Custom_Order_Category_Checklist_Walker($custom_order);
    }

    return $args;
}
add_filter('wp_terms_checklist_args', 'custom_category_order_in_admin', 10, 2);