v0.1.7

classes/Base/Group.php

@@ -42,6 +42,13 @@ class Group {
 	 */
 	protected $post_types;
 
+	/**
+	 * Page Types on which to display group
+	 *
+	 * @var $page_types    array
+	 */
+	protected $page_types;
+
 	/**
 	 * Templates on which to display group
 	 *
@@ -191,6 +198,16 @@ public function setPostTypes( $post_types = [] ) {
 		$this->post_types = $post_types;
 	}
 
+	/**
+	 * Set the post types on which to show the group
+	 *
+	 * @param array $post_types
+	 * @since 0.1.0
+	 */
+	public function setPageTypes( $page_types = [] ) {
+		$this->page_types = $page_types;
+	}
+
 	/**
 	 * Set the post types on which to ignore the group
 	 *
@@ -317,6 +334,18 @@ public function getLocations() {
 			}
 		}
 
+		if( ! empty( $this->page_types ) ) {
+			foreach( $this->page_types as $page_type) {
+				$toAdd =
+					[
+						'param'    => 'page_type',
+						'operator' => '==',
+						'value'    => $page_type,
+					];
+				array_push($results, $toAdd);
+			}
+		}
+
 		return [$results];
 	}