Edit File by line

<?php

[0] Fix | Delete

[1] Fix | Delete

/**

[2] Fix | Delete

* Provide utilities to interact and replace the current post akin to setup_postdata()

[3] Fix | Delete

* but in a stacking manner.

[4] Fix | Delete

[5] Fix | Delete

* @since 4.0

[6] Fix | Delete

[7] Fix | Delete

class ET_Post_Stack {

[8] Fix | Delete

/**

[9] Fix | Delete

* Post override stack.

[10] Fix | Delete

[11] Fix | Delete

* @var (WP_Post|null)[]

[12] Fix | Delete

[13] Fix | Delete

protected static $stack = array();

[14] Fix | Delete

[15] Fix | Delete

/**

[16] Fix | Delete

* Get the top post from the stack or the current one if the stack is empty.

[17] Fix | Delete

[18] Fix | Delete

* @since 4.0

[19] Fix | Delete

[20] Fix | Delete

* @param integer $offset Offset from the end of the array, 0 being the last post. Use negative integers.

[21] Fix | Delete

[22] Fix | Delete

* @return WP_Post|null

[23] Fix | Delete

[24] Fix | Delete

public static function get( $offset = 0 ) {

[25] Fix | Delete

global $post;

[26] Fix | Delete

[27] Fix | Delete

$index = count( self::$stack ) - 1 + $offset;

[28] Fix | Delete

[29] Fix | Delete

if ( empty( self::$stack ) && 0 === $index ) {

[30] Fix | Delete

return $post;

[31] Fix | Delete

}

[32] Fix | Delete

[33] Fix | Delete

return isset( self::$stack[ $index ] ) ? self::$stack[ $index ] : null;

[34] Fix | Delete

}

[35] Fix | Delete

[36] Fix | Delete

/**

[37] Fix | Delete

* Pop the top post off of the stack.

[38] Fix | Delete

[39] Fix | Delete

* @since 4.0

[40] Fix | Delete

[41] Fix | Delete

* @return void

[42] Fix | Delete

[43] Fix | Delete

public static function pop() {

[44] Fix | Delete

array_pop( self::$stack );

[45] Fix | Delete

}

[46] Fix | Delete

[47] Fix | Delete

/**

[48] Fix | Delete

* Setup a post as the global $post.

[49] Fix | Delete

[50] Fix | Delete

* @since 4.0

[51] Fix | Delete

[52] Fix | Delete

* @param WP_Post|null $with

[53] Fix | Delete

* @param boolean $force

[54] Fix | Delete

[55] Fix | Delete

* @return void

[56] Fix | Delete

[57] Fix | Delete

protected static function setup( $with, $force = false ) {

[58] Fix | Delete

global $post;

[59] Fix | Delete

[60] Fix | Delete

if ( $force || ! ( $with && $post && $with->ID === $post->ID ) ) {

[61] Fix | Delete

$post = $with;

[62] Fix | Delete

setup_postdata( $post );

[63] Fix | Delete

}

[64] Fix | Delete

}

[65] Fix | Delete

[66] Fix | Delete

/**

[67] Fix | Delete

* Equivalent to setup_postdata() but keeps a stack of posts so it can be nested.

[68] Fix | Delete

* Pushes the specified post on the stack.

[69] Fix | Delete

[70] Fix | Delete

* @since 4.0

[71] Fix | Delete

[72] Fix | Delete

* @param WP_Post $with

[73] Fix | Delete

[74] Fix | Delete

* @return void

[75] Fix | Delete

[76] Fix | Delete

public static function replace( $with ) {

[77] Fix | Delete

global $post;

[78] Fix | Delete

[79] Fix | Delete

$force = empty( self::$stack );

[80] Fix | Delete

[81] Fix | Delete

if ( empty( self::$stack ) ) {

[82] Fix | Delete

// Add the current post as the first in the stack even if it does not exist.

[83] Fix | Delete

self::$stack[] = $post;

[84] Fix | Delete

}

[85] Fix | Delete

[86] Fix | Delete

self::$stack[] = $with;

[87] Fix | Delete

[88] Fix | Delete

self::setup( $with, $force );

[89] Fix | Delete

}

[90] Fix | Delete

[91] Fix | Delete

/**

[92] Fix | Delete

* Restores the last post from the stack.

[93] Fix | Delete

* The final restore will setup the post that was setup when the stack was last empty.

[94] Fix | Delete

[95] Fix | Delete

* @since 4.0

[96] Fix | Delete

[97] Fix | Delete

* @return void

[98] Fix | Delete

[99] Fix | Delete

public static function restore() {

[100] Fix | Delete

self::pop();

[101] Fix | Delete

self::reset();

[102] Fix | Delete

}

[103] Fix | Delete

[104] Fix | Delete

/**

[105] Fix | Delete

* Resets the post to the one at the top of the stack.

[106] Fix | Delete

[107] Fix | Delete

* @since 4.0

[108] Fix | Delete

[109] Fix | Delete

* @return void

[110] Fix | Delete

[111] Fix | Delete

public static function reset() {

[112] Fix | Delete

if ( ! empty( self::$stack ) ) {

[113] Fix | Delete

self::setup( self::get() );

[114] Fix | Delete

}

[115] Fix | Delete

}

[116] Fix | Delete

[117] Fix | Delete

/**

[118] Fix | Delete

* Returns the $post for the current true WordPress query post ignoring any posts that may be

[119] Fix | Delete

* forced using setup_postdata().

[120] Fix | Delete

[121] Fix | Delete

* @since 4.0

[122] Fix | Delete

[123] Fix | Delete

* @return WP_Post|null

[124] Fix | Delete

[125] Fix | Delete

public static function get_main_post() {

[126] Fix | Delete

global $wp_query;

[127] Fix | Delete

[128] Fix | Delete

if ( ! $wp_query || 0 === $wp_query->post_count ) {

[129] Fix | Delete

// Handle special case where there is no current post but once the_content()

[130] Fix | Delete

// gets called the global $post is polluted and will no longer reset to null.

[131] Fix | Delete

return null;

[132] Fix | Delete

}

[133] Fix | Delete

[134] Fix | Delete

if ( empty( $wp_query->post ) ) {

[135] Fix | Delete

return null;

[136] Fix | Delete

}

[137] Fix | Delete

[138] Fix | Delete

return $wp_query->post;

[139] Fix | Delete

}

[140] Fix | Delete

[141] Fix | Delete

/**

[142] Fix | Delete

* Returns the post ID for the current true WordPress query post.

[143] Fix | Delete

* See ::get_main_post() for more information.

[144] Fix | Delete

[145] Fix | Delete

* @since 4.0

[146] Fix | Delete

[147] Fix | Delete

* @return integer

[148] Fix | Delete

[149] Fix | Delete

public static function get_main_post_id() {

[150] Fix | Delete

$post = self::get_main_post();

[151] Fix | Delete

[152] Fix | Delete

return $post ? (int) $post->ID : 0;

[153] Fix | Delete

}

[154] Fix | Delete

}

[155] Fix | Delete

[156] Fix | Delete