| @@ -20,21 +20,23 @@ | ||
| 20 | 20 | protected $block_name = 'ecm-upvote-count'; |
| 21 | 21 | |
| 22 | 22 | public function build_css( $attributes ) { |
| 23 | 23 | $css_generator = new CssGenerator( $attributes, $this->block_name ); |
| 24 | - $wrapper_class = ! empty( $attributes['wrapper_class'] ) ? $attributes['wrapper_class'] : 'upvote-count-wrapper'; | |
| 24 | + $wrapper_class = sanitize_html_class( $attributes['wrapper_class'] ?? '', 'upvote-count-wrapper' ); | |
| 25 | 25 | |
| 26 | 26 | $css_generator->add_class_styles( |
| 27 | 27 | '{{WRAPPER}} .' . $wrapper_class, |
| 28 | 28 | $this->get_coutineu_button_wrapper_css( $attributes ), |
| 29 | 29 | $this->get_coutineu_button_wrapper_css( $attributes, 'Tablet' ), |
| 30 | - $this->get_coutineu_button_wrapper_css( $attributes, 'Mobile' ) | |
| 30 | + $this->get_coutineu_button_wrapper_css( $attributes, 'Mobile' ), | |
| 31 | + $css_generator->custom_device_map( function ( $device ) use ( $attributes ) { return $this->get_coutineu_button_wrapper_css( $attributes, $device ); } ) | |
| 31 | 32 | ); |
| 32 | 33 | $css_generator->add_class_styles( |
| 33 | 34 | '{{WRAPPER}} .' . $wrapper_class, |
| 34 | 35 | $this->get_coutineu_button_css( $attributes ), |
| 35 | 36 | $this->get_coutineu_button_css( $attributes, 'Tablet' ), |
| 36 | - $this->get_coutineu_button_css( $attributes, 'Mobile' ) | |
| 37 | + $this->get_coutineu_button_css( $attributes, 'Mobile' ), | |
| 38 | + $css_generator->custom_device_map( function ( $device ) use ( $attributes ) { return $this->get_coutineu_button_css( $attributes, $device ); } ) | |
| 37 | 39 | ); |
| 38 | 40 | |
| 39 | 41 | return $css_generator->generate_css(); |
| 40 | 42 | } |