<?php 
/** 
 * Colors block support flag. 
 * 
 * @package WordPress 
 * @since 5.6.0 
 */ 
 
/** 
 * Registers the style and colors block attributes for block types that support it. 
 * 
 * @since 5.6.0 
 * @since 6.1.0 Improved $color_support assignment optimization. 
 * @access private 
 * 
 * @param WP_Block_Type $block_type Block Type. 
 */ 
function wp_register_colors_support( $block_type ) { 
    $color_support = false; 
    if ( $block_type instanceof WP_Block_Type ) { 
        $color_support = isset( $block_type->supports['color'] ) ? $block_type->supports['color'] : false; 
    } 
    $has_text_colors_support       = true === $color_support || 
        ( isset( $color_support['text'] ) && $color_support['text'] ) || 
        ( is_array( $color_support ) && ! isset( $color_support['text'] ) ); 
    $has_background_colors_support = true === $color_support || 
        ( isset( $color_support['background'] ) && $color_support['background'] ) || 
        ( is_array( $color_support ) && ! isset( $color_support['background'] ) ); 
    $has_gradients_support         = isset( $color_support['gradients'] ) ? $color_support['gradients'] : false; 
    $has_link_colors_support       = isset( $color_support['link'] ) ? $color_support['link'] : false; 
    $has_button_colors_support     = isset( $color_support['button'] ) ? $color_support['button'] : false; 
    $has_heading_colors_support    = isset( $color_support['heading'] ) ? $color_support['heading'] : false; 
    $has_color_support             = $has_text_colors_support || 
        $has_background_colors_support || 
        $has_gradients_support || 
        $has_link_colors_support || 
        $has_button_colors_support || 
        $has_heading_colors_support; 
 
    if ( ! $block_type->attributes ) { 
        $block_type->attributes = array(); 
    } 
 
    if ( $has_color_support && ! array_key_exists( 'style', $block_type->attributes ) ) { 
        $block_type->attributes['style'] = array( 
            'type' => 'object', 
        ); 
    } 
 
    if ( $has_background_colors_support && ! array_key_exists( 'backgroundColor', $block_type->attributes ) ) { 
        $block_type->attributes['backgroundColor'] = array( 
            'type' => 'string', 
        ); 
    } 
 
    if ( $has_text_colors_support && ! array_key_exists( 'textColor', $block_type->attributes ) ) { 
        $block_type->attributes['textColor'] = array( 
            'type' => 'string', 
        ); 
    } 
 
    if ( $has_gradients_support && ! array_key_exists( 'gradient', $block_type->attributes ) ) { 
        $block_type->attributes['gradient'] = array( 
            'type' => 'string', 
        ); 
    } 
} 
 
 
/** 
 * Adds CSS classes and inline styles for colors to the incoming attributes array. 
 * This will be applied to the block markup in the front-end. 
 * 
 * @since 5.6.0 
 * @since 6.1.0 Implemented the style engine to generate CSS and classnames. 
 * @access private 
 * 
 * @param  WP_Block_Type $block_type       Block type. 
 * @param  array         $block_attributes Block attributes. 
 * 
 * @return array Colors CSS classes and inline styles. 
 */ 
function wp_apply_colors_support( $block_type, $block_attributes ) { 
    $color_support = isset( $block_type->supports['color'] ) ? $block_type->supports['color'] : false; 
 
    if ( 
        is_array( $color_support ) && 
        wp_should_skip_block_supports_serialization( $block_type, 'color' ) 
    ) { 
        return array(); 
    } 
 
    $has_text_colors_support       = true === $color_support || 
        ( isset( $color_support['text'] ) && $color_support['text'] ) || 
        ( is_array( $color_support ) && ! isset( $color_support['text'] ) ); 
    $has_background_colors_support = true === $color_support || 
        ( isset( $color_support['background'] ) && $color_support['background'] ) || 
        ( is_array( $color_support ) && ! isset( $color_support['background'] ) ); 
    $has_gradients_support         = isset( $color_support['gradients'] ) ? $color_support['gradients'] : false; 
    $color_block_styles            = array(); 
 
    // Text colors. 
    if ( $has_text_colors_support && ! wp_should_skip_block_supports_serialization( $block_type, 'color', 'text' ) ) { 
        $preset_text_color          = array_key_exists( 'textColor', $block_attributes ) ? "var:preset|color|{$block_attributes['textColor']}" : null; 
        $custom_text_color          = isset( $block_attributes['style']['color']['text'] ) ? $block_attributes['style']['color']['text'] : null; 
        $color_block_styles['text'] = $preset_text_color ? $preset_text_color : $custom_text_color; 
    } 
 
    // Background colors. 
    if ( $has_background_colors_support && ! wp_should_skip_block_supports_serialization( $block_type, 'color', 'background' ) ) { 
        $preset_background_color          = array_key_exists( 'backgroundColor', $block_attributes ) ? "var:preset|color|{$block_attributes['backgroundColor']}" : null; 
        $custom_background_color          = isset( $block_attributes['style']['color']['background'] ) ? $block_attributes['style']['color']['background'] : null; 
        $color_block_styles['background'] = $preset_background_color ? $preset_background_color : $custom_background_color; 
    } 
 
    // Gradients. 
    if ( $has_gradients_support && ! wp_should_skip_block_supports_serialization( $block_type, 'color', 'gradients' ) ) { 
        $preset_gradient_color          = array_key_exists( 'gradient', $block_attributes ) ? "var:preset|gradient|{$block_attributes['gradient']}" : null; 
        $custom_gradient_color          = isset( $block_attributes['style']['color']['gradient'] ) ? $block_attributes['style']['color']['gradient'] : null; 
        $color_block_styles['gradient'] = $preset_gradient_color ? $preset_gradient_color : $custom_gradient_color; 
    } 
 
    $attributes = array(); 
    $styles     = wp_style_engine_get_styles( array( 'color' => $color_block_styles ), array( 'convert_vars_to_classnames' => true ) ); 
 
    if ( ! empty( $styles['classnames'] ) ) { 
        $attributes['class'] = $styles['classnames']; 
    } 
 
    if ( ! empty( $styles['css'] ) ) { 
        $attributes['style'] = $styles['css']; 
    } 
 
    return $attributes; 
} 
 
// Register the block support. 
WP_Block_Supports::get_instance()->register( 
    'colors', 
    array( 
        'register_attribute' => 'wp_register_colors_support', 
        'apply'              => 'wp_apply_colors_support', 
    ) 
); 
 
 |