Merge pull request #5 from humanmade/get-2-colors

restrict-and-validate

inc/class-plugin.php

@@ -49,7 +49,7 @@ public function plugins_loaded() {
 
 	/**
 	 * Adds the style attribute to the image HTML.
-	 * 
+	 *
 	 * @param $html
 	 * @param $id
 	 * @param $caption
@@ -161,8 +161,8 @@ public function extract_colors( $image_path ) {
 
 		$image = $client->loadJpeg( $image_path );
 
-		// Get three most used color hex code
-		return $image->extract( 4 );
+		// Get 2 most used color hex code
+		return $image->extract( 2 );
 
 	}
 
@@ -200,8 +200,18 @@ public function get_colors_for_attachment( $id ) {
 	 * @param $id
 	 * @param array $colors
 	 */
-	public function save_colors_for_attachment( $id, $colors = array() ) {
+	public function save_colors_for_attachment( $id, Array $colors = array() ) {
+
+		// Validate hex codes.
+		$colors = array_filter( $colors, function( $color ) {
+			return preg_match( '/^#([A-Fa-f0-9]{6}|[A-Fa-f0-9]{3})$/', $color );
+		} );
+
+		// Restrict to 2 items.
+		$colors = array_slice( $colors, 0, 2 );
+
 		update_post_meta( $id, 'hmgp_image_colors', $colors );
+
 	}
 
 	/**