first commit

wp-content/plugins/sitepress-multilingual-cms/classes/ATE/Review/NonPublicCPTPreview.php

@@ -0,0 +1,43 @@
+<?php
+
+namespace WPML\TM\ATE\Review;
+
+use WPML\FP\Fns;
+use WPML\FP\Logic;
+use WPML\FP\Lst;
+use WPML\FP\Obj;
+
+/**
+ * This will allow displaying private CPT reviews on the frontend.
+ */
+class NonPublicCPTPreview {
+
+	const POST_TYPE = 'wpmlReviewPostType';
+
+	/**
+	 * @param array $args
+	 *
+	 * @return array
+	 */
+	public static function addArgs( array $args ) {
+		return Obj::assoc( self::POST_TYPE, \get_post_type( $args['preview_id'] ), $args );
+	}
+
+	/**
+	 * @return callable
+	 */
+	public static function allowReviewPostTypeQueryVar() {
+		return Lst::append( self::POST_TYPE );
+	}
+
+	/**
+	 * @return callable
+	 */
+	public static function enforceReviewPostTypeIfSet() {
+		return Logic::ifElse(
+			Obj::prop( self::POST_TYPE ),
+			Obj::renameProp( self::POST_TYPE, 'post_type' ),
+			Fns::identity()
+		) ;
+	}
+}