qdr.app.woo.plugins/studiou-wc-product-cat-manage/includes/class-studiou-wc-product-cat-manage-db.php

<?php
/**
 * Database operations class
 *
 * Handles database operations related to product categories
 *
 * @since      1.0.0
 * @package    Studiou_WC_Product_Cat_Manage
 * @subpackage Studiou_WC_Product_Cat_Manage/includes
 */

// If this file is called directly, abort.
if (!defined('WPINC')) {
    die;
}

class Studiou_WC_Product_Cat_Manage_DB {
    
    /**
     * Constructor
     */
    public function __construct() {
    }
    
    /**
     * Get all product categories
     * 
     * @return array Array of product categories with all required properties
     */
    public function get_all_product_categories() {
        // Get all categories regardless of hierarchy, including empty ones
        $args = array(
            'taxonomy'   => 'product_cat',
            'orderby'    => 'term_id', // Order by ID to maintain creation order
            'hide_empty' => false,
            'hierarchical' => false, // Don't nest them, get a flat list
            'get' => 'all', // Get all fields
        );
        
        $categories = get_terms($args);
        $formatted_categories = array();
        
        if (!is_wp_error($categories) && !empty($categories)) {
            // Sort by hierarchy: parent categories first, then children
            $sorted_categories = $this->sort_categories_by_hierarchy($categories);
            
            foreach ($sorted_categories as $category) {
                $formatted_categories[] = $this->format_category($category);
            }
        }
        
        return $formatted_categories;
    }
    
    /**
     * Sort categories by hierarchy - parents first, then children
     * 
     * @param array $categories Array of WP_Term objects
     * @return array Sorted array of categories
     */
    private function sort_categories_by_hierarchy($categories) {
        $parents = array();
        $children = array();
        
        // Separate parents and children
        foreach ($categories as $category) {
            if ($category->parent == 0) {
                $parents[] = $category;
            } else {
                $children[] = $category;
            }
        }
        
        // Sort parents alphabetically
        usort($parents, function($a, $b) {
            return strcmp($a->name, $b->name);
        });
        
        // Recursively add children after their parents
        $sorted = array();
        foreach ($parents as $parent) {
            $sorted[] = $parent;
            $sorted = array_merge($sorted, $this->get_children_recursive($parent->term_id, $children));
        }
        
        // Add any orphaned children (categories with non-existent parents)
        foreach ($children as $child) {
            if (!$this->is_child_added($child, $sorted)) {
                $sorted[] = $child;
            }
        }
        
        return $sorted;
    }
    
    /**
     * Recursively get children of a category
     * 
     * @param int $parent_id Parent category ID
     * @param array $all_children Array of all child categories
     * @return array Array of child categories
     */
    private function get_children_recursive($parent_id, $all_children) {
        $result = array();
        $direct_children = array();
        
        // Find direct children of this parent
        foreach ($all_children as $child) {
            if ($child->parent == $parent_id) {
                $direct_children[] = $child;
            }
        }
        
        // Sort direct children alphabetically
        usort($direct_children, function($a, $b) {
            return strcmp($a->name, $b->name);
        });
        
        // Add each child and its descendants
        foreach ($direct_children as $child) {
            $result[] = $child;
            $result = array_merge($result, $this->get_children_recursive($child->term_id, $all_children));
        }
        
        return $result;
    }
    
    /**
     * Check if a child category has already been added to the sorted array
     * 
     * @param WP_Term $child Child category to check
     * @param array $sorted Array of already sorted categories
     * @return bool True if child is already in sorted array
     */
    private function is_child_added($child, $sorted) {
        foreach ($sorted as $category) {
            if ($category->term_id == $child->term_id) {
                return true;
            }
        }
        return false;
    }
    
    /**
     * Format category data
     * 
     * @param object $category WP_Term object
     * @return array Formatted category data
     */
    private function format_category($category) {
        // Get parent category name
        $parent_name = '';
        if ($category->parent > 0) {
            $parent = get_term($category->parent, 'product_cat');
            if (!is_wp_error($parent)) {
                $parent_name = $parent->name;
            }
        }
        
        // Get display type
        $display_type = get_term_meta($category->term_id, 'display_type', true);
        if (empty($display_type)) {
            $display_type = 'default';
        }
        
        // Get protected category data if available
        $visibility = 'public';
        $protected_passwords = '';
        
        if (class_exists('WC_Protected_Categories')) {
            // Use Util class to get category visibility info
            $term_id = $category->term_id;
            
            // Use the plugin's own method to get visibility data if available
            if (class_exists('\\Barn2\\Plugin\\WC_Protected_Categories\\Util')) {
                $visibility_obj = \Barn2\Plugin\WC_Protected_Categories\Util::get_category_visibility($term_id);
                if ($visibility_obj) {
                    $visibility = $visibility_obj->get_visibility();
                    
                    if ($visibility === 'protected' && $visibility_obj->has_password_protection()) {
                        // Get passwords using the plugin's own method
                        $passwords = \Barn2\Plugin\WC_Protected_Categories\Util::get_term_passwords($term_id);
                        if (!empty($passwords)) {
                            $protected_passwords = implode('|', $passwords);
                        }
                    }
                }
            } else {
                // Fallback to direct meta access if plugin class not available
                $visibility = get_term_meta($term_id, 'visibility', true);
                if (empty($visibility)) {
                    $visibility = 'public';
                }
                
                if ($visibility === 'protected') {
                    $passwords = get_term_meta($term_id, 'password', false);
                    if ($passwords && !empty($passwords[0]) && is_array($passwords[0])) {
                        $passwords = $passwords[0];
                    }
                    
                    if (!empty($passwords)) {
                        $protected_passwords = implode('|', $passwords);
                    }
                }
            }
        }
        
        return array(
            'name' => $category->name,
            'url-name' => $category->slug,
            'parent-name' => $parent_name,
            'description' => $category->description,
            'display-type' => $display_type,
            'visibility' => $visibility,
            'protected-passwords' => $protected_passwords,
            'operation' => 'U' // Default operation for existing categories
        );
    }
    
    /**
     * Find category by name
     * 
     * @param string $name Category name
     * @return WP_Term|null Category term object or null if not found
     */
    public function find_category_by_name($name) {
        $args = array(
            'taxonomy'   => 'product_cat',
            'name'       => $name,
            'hide_empty' => false,
        );
        
        $categories = get_terms($args);
        
        if (!is_wp_error($categories) && !empty($categories)) {
            return $categories[0];
        }
        
        return null;
    }
    
    /**
     * Create product category
     * 
     * @param array $category_data Category data
     * @return array Result with status and message
     */
    public function create_category($category_data) {
        // Check if category already exists
        $existing = $this->find_category_by_name($category_data['name']);
        if ($existing) {
            return array(
                'status' => 'skipped',
                'message' => sprintf(
                    __("Product category '%s' already exists. Skipped.", 'studiou-wc-product-cat-manage'),
                    $category_data['name']
                )
            );
        }
        
        // Find parent category ID if parent name is provided
        $parent_id = 0;
        if (!empty($category_data['parent-name'])) {
            $parent = $this->find_category_by_name($category_data['parent-name']);
            if ($parent) {
                $parent_id = $parent->term_id;
            } else {
                return array(
                    'status' => 'error',
                    'message' => sprintf(
                        __("Parent category '%s' not found.", 'studiou-wc-product-cat-manage'),
                        $category_data['parent-name']
                    )
                );
            }
        }
        
        // Create category
        $args = array(
            'description' => $category_data['description'],
            'parent' => $parent_id,
            'slug' => $category_data['url-name'],
        );
        
        $result = wp_insert_term($category_data['name'], 'product_cat', $args);
        
        if (is_wp_error($result)) {
            return array(
                'status' => 'error',
                'message' => $result->get_error_message()
            );
        }
        
        $term_id = $result['term_id'];
        
        // Set display type
        update_term_meta($term_id, 'display_type', $category_data['display-type']);
        
        // Set visibility and password protection if WC Protected Categories is active
        if (class_exists('WC_Protected_Categories')) {
            // Set visibility - ensure the "visibility" meta key is used (not wc_protected_categories_visibility)
            update_term_meta($term_id, 'visibility', $category_data['visibility']);
            
            // Set password protection if visibility is protected and passwords are provided
            if ($category_data['visibility'] === 'protected' && !empty($category_data['protected-passwords'])) {
                $passwords = explode('|', $category_data['protected-passwords']);
                // Filter out empty passwords
                $passwords = array_filter($passwords);
                
                if (empty($passwords)) {
                    // If no valid passwords after filtering, set a default one as the plugin does
                    $passwords = ['password123'];
                }
                
                // Store the passwords array as a single meta entry
                add_term_meta($term_id, 'password', $passwords);
            }
        }
        
        return array(
            'status' => 'created',
            'message' => sprintf(
                __("Product category '%s' created.", 'studiou-wc-product-cat-manage'),
                $category_data['name']
            )
        );
    }
    
    /**
     * Update product category
     * 
     * @param array $category_data Category data
     * @return array Result with status and message
     */
    public function update_category($category_data) {
        // Check if category exists
        $existing = $this->find_category_by_name($category_data['name']);
        if (!$existing) {
            return array(
                'status' => 'error',
                'message' => sprintf(
                    __("Product category '%s' not found.", 'studiou-wc-product-cat-manage'),
                    $category_data['name']
                )
            );
        }
        
        // Find parent category ID if parent name is provided
        $parent_id = 0;
        if (!empty($category_data['parent-name'])) {
            $parent = $this->find_category_by_name($category_data['parent-name']);
            if ($parent) {
                $parent_id = $parent->term_id;
            } else {
                return array(
                    'status' => 'error',
                    'message' => sprintf(
                        __("Parent category '%s' not found.", 'studiou-wc-product-cat-manage'),
                        $category_data['parent-name']
                    )
                );
            }
        }
        
        // Update category
        $args = array(
            'description' => $category_data['description'],
            'parent' => $parent_id,
            'slug' => $category_data['url-name'],
        );
        
        $result = wp_update_term($existing->term_id, 'product_cat', $args);
        
        if (is_wp_error($result)) {
            return array(
                'status' => 'error',
                'message' => $result->get_error_message()
            );
        }
        
        // Set display type
        update_term_meta($existing->term_id, 'display_type', $category_data['display-type']);
        
        // Set visibility and password protection if WC Protected Categories is active
        if (class_exists('WC_Protected_Categories')) {
            // First remove existing protection meta
            delete_term_meta($existing->term_id, 'password');
            delete_term_meta($existing->term_id, 'user_roles');
            delete_term_meta($existing->term_id, 'users');
            
            // Set visibility - ensure the "visibility" meta key is used
            update_term_meta($existing->term_id, 'visibility', $category_data['visibility']);
            
            // Set password protection if visibility is protected and passwords are provided
            if ($category_data['visibility'] === 'protected' && !empty($category_data['protected-passwords'])) {
                $passwords = explode('|', $category_data['protected-passwords']);
                // Filter out empty passwords
                $passwords = array_filter($passwords);
                
                if (empty($passwords)) {
                    // If no valid passwords after filtering, set a default one as the plugin does
                    $passwords = ['password123'];
                }
                
                // Store the passwords array as a single meta entry
                add_term_meta($existing->term_id, 'password', $passwords);
            }
        }
        
        return array(
            'status' => 'updated',
            'message' => sprintf(
                __("Product category '%s' updated.", 'studiou-wc-product-cat-manage'),
                $category_data['name']
            )
        );
    }
    
    /**
     * Delete product category
     * 
     * @param string $category_name Category name
     * @return array Result with status and message
     */
    public function delete_category($category_name) {
        // Check if category exists
        $existing = $this->find_category_by_name($category_name);
        if (!$existing) {
            return array(
                'status' => 'error',
                'message' => sprintf(
                    __("Product category '%s' not found.", 'studiou-wc-product-cat-manage'),
                    $category_name
                )
            );
        }
        
        // Delete category
        $result = wp_delete_term($existing->term_id, 'product_cat');
        
        if (is_wp_error($result)) {
            return array(
                'status' => 'error',
                'message' => $result->get_error_message()
            );
        }
        
        return array(
            'status' => 'deleted',
            'message' => sprintf(
                __("Product category '%s' deleted.", 'studiou-wc-product-cat-manage'),
                $category_name
            )
        );
    }
}