Skip to content

Commit 16760eb

Browse files
committed
New validation in Block@add_classes(). Param can be array or string
1 parent 1886ea7 commit 16760eb

File tree

1 file changed

+12
-5
lines changed

1 file changed

+12
-5
lines changed

includes/Block.php

Lines changed: 12 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -492,13 +492,20 @@ public function set_classes()
492492
/**
493493
* Add classes to the main HTML element.
494494
*
495-
* @param array $classes
495+
* @param array | string $classes
496496
*/
497-
public function add_classes(array $classes)
497+
public function add_classes( $classes = '' )
498498
{
499-
foreach ($classes as $class) {
500-
$this->classes[] = $class;
501-
}
499+
if ( empty($classes) ) return;
500+
501+
if ( is_array( $classes ) ) {
502+
foreach ($classes as $class) {
503+
$this->classes[] = $class;
504+
}
505+
}else if ( is_string( $classes ) ){
506+
$this->classes[] = $classes;
507+
}
508+
502509
}
503510

504511
/**

0 commit comments

Comments
 (0)