Как добавить поле «Отдел» и ограничить видимость категорий в WordPress Overcod.net

Как добавить поле «Отдел» и ограничить видимость категорий в WordPress

Иногда нужно, чтобы не все пользователи сайта видели одинаковые категории. Например, у вас есть два отдела: Горний цех и Гараж. Каждый из них должен видеть только свои категории.

В WordPress это можно сделать без плагинов — достаточно добавить немного кода.

в админке на странице «Рубрики» (Категории)

1. Добавляем новые поля в категории

Код ниже добавляет в редактирование категории два новых блока:

Кому видна категория — список пользователей с чекбоксами.
Отдел — выпадающий список с вариантами: “Горний цех” и “Гараж”.

Вставьте этот код в файл functions.php вашей темы (или дочерней темы).

// ===============
// 1. Поля в категории (Кому видна и Отдел)
// ===============
add_action('category_edit_form_fields', function($term) {
    $users = get_users(['fields' => ['ID', 'display_name']]);
    $allowed = get_term_meta($term->term_id, 'allowed_users', true) ?: [];
    $department = get_term_meta($term->term_id, 'department', true);
    ?>
    <tr class="form-field">
        <th scope="row"><label>Кому видна категория</label></th>
        <td>
            <?php foreach ($users as $user): ?>
                <label>
                    <input type="checkbox" name="allowed_users[]" value="<?php echo $user->ID; ?>"
                        <?php checked(in_array($user->ID, $allowed)); ?>>
                    <?php echo esc_html($user->display_name); ?>
                </label><br>
            <?php endforeach; ?>
        </td>
    </tr>

    <tr class="form-field">
        <th scope="row"><label for="department">Отдел</label></th>
        <td>
            <select name="department" id="department">
                <option value="">— Не выбран —</option>
                <option value="горний цех" <?php selected($department, 'горний цех'); ?>>Горний цех</option>
                <option value="гараж" <?php selected($department, 'гараж'); ?>>Гараж</option>
            </select>
            <p class="description">Выберите отдел, к которому относится категория.</p>
        </td>
    </tr>
    <?php
});

После этого при редактировании категории появятся два новых поля.

2. Сохраняем введённые данные

Этот код сохраняет выбранных пользователей и отдел, когда вы обновляете категорию (файл functions.php).

// ===============
// 2. Сохраняем данные
// ===============
add_action('edited_category', function($term_id) {
    // пользователи
    if (isset($_POST['allowed_users'])) {
        update_term_meta($term_id, 'allowed_users', array_map('intval', $_POST['allowed_users']));
    } else {
        delete_term_meta($term_id, 'allowed_users');
    }

    // отдел
    if (isset($_POST['department'])) {
        update_term_meta($term_id, 'department', sanitize_text_field($_POST['department']));
    }
});

3. Скрываем категории от других пользователей

Этот блок фильтрует список постов:
если пользователь не имеет доступа к категории, то записи из неё не показываются (файл functions.php).

// ===============
// 3. Фильтр видимости категорий
// ===============
add_action('pre_get_posts', function($query) {
    if (is_admin() || !$query->is_main_query()) return;

    $user_id = get_current_user_id();
    if (!$user_id) return;

    $categories = get_categories(['hide_empty' => false]);
    $allowed = [];

    foreach ($categories as $cat) {
        $allowed_users = get_term_meta($cat->term_id, 'allowed_users', true);
        if (empty($allowed_users) || in_array($user_id, $allowed_users)) {
            $allowed[] = $cat->term_id;
        }
    }

    if ($allowed) {
        $query->set('category__in', $allowed);
    } else {
        $query->set('category__in', [0]);
    }
});

4. Как вывести категории по отделу

Если нужно отобразить только категории определённого отдела, добавьте функцию (например в sidebar.php, home.php, index.php …):

<?php
function my_thecat($department_filter = '') { 
    $arg_cat = array(
        'orderby'      => 'name',
        'order'        => 'ASC',
        'parent'       => 0,
        'hide_empty'   => 1,
    );

    $categories = get_categories($arg_cat);

    if ($categories) {
        foreach ($categories as $cat) {
            $department = get_term_meta($cat->term_id, 'department', true);
            if ($department_filter && $department != $department_filter) continue;

            echo '<a href="' . get_category_link($cat->term_id) . '" class="list-group-item">' . esc_html($cat->name) . '</a>';
        }
    }
}
?>

Пример использования:

<?php
my_thecat('горний цех'); // категории отдела Горний цех
my_thecat('гараж');      // категории отдела Гараж
?>

Результат

В админке у каждой категории есть поле “Кому видна” и выпадающий “Отдел”.
На сайте показываются только те категории, которые доступны текущему пользователю.
Можно легко вывести категории конкретного отдела.