← All changes | includes/Shortcodes/ToC.php +150 -103 3.5.0 → 4.2.0 View file →

		@@ -5,127 +5,174 @@
5	5	use WPDeveloper\BetterDocs\Utils\Node;
6	6	use WPDeveloper\BetterDocs\Core\Shortcode;
7	7
8	8	class ToC extends Shortcode {
9		- protected $html_attributes = [];
	9	+ protected $html_attributes = [];
10	10
11		- public function get_name() {
12		- return 'betterdocs_toc';
13		- }
	11	+ public function get_name() {
	12	+ return 'betterdocs_toc';
	13	+ }
14	14
15		- public function get_style_depends() {
16		- return ['betterdocs-toc'];
17		- }
	15	+ public function get_style_depends() {
	16	+ return [ 'betterdocs-toc' ];
	17	+ }
18	18
19		- /**
20		- * Summary of default_attributes
21		- * @return array
22		- */
23		- public function default_attributes() {
24		- return [
25		- 'post_type' => 'docs',
26		- 'post_id' => get_the_ID(),
27		- 'htags' => '1,2,3,4,5,6',
28		- 'hierarchy' => '',
29		- 'list_number' => '',
30		- 'collapsible_on_mobile' => $this->settings->get( 'collapsible_toc_mobile', false ),
31		- 'toc_title' => ''
32		- ];
33		- }
	19	+ /**
	20	+ * Summary of default_attributes
	21	+ * @return array
	22	+ */
	23	+ public function default_attributes() {
	24	+ return [
	25	+ 'post_type' => 'docs',
	26	+ 'post_id' => get_the_ID(),
	27	+ 'htags' => '1,2,3,4,5,6',
	28	+ 'hierarchy' => '',
	29	+ 'list_number' => '',
	30	+ 'collapsible_on_mobile' => $this->settings->get( 'collapsible_toc_mobile', false ),
	31	+ 'toc_title' => ''
	32	+ ];
	33	+ }
34	34
35		- public function render( $atts, $content = null ) {
36		- $this->views( 'shortcodes/toc' );
37		- }
	35	+ public function render( $atts, $content = null ) {
	36	+ // Check if post is password protected and user hasn't provided correct password
	37	+ if ( post_password_required( $this->attributes['post_id'] ) ) {
	38	+ // Don't show ToC for password-protected posts until password is provided
	39	+ return '';
	40	+ }
38	41
39		- public function view_params() {
40		- return [
41		- 'post' => get_post( $this->attributes['post_id'] )
42		- ];
43		- }
	42	+ $this->views( 'shortcodes/toc' );
	43	+ }
44	44
45		- /**
46		- * This method is responsible for re-arranging the TOC data based on hierarchy or non-hierarchy
47		- *
48		- * @param string $post_content
49		- * @param string $htag_support
50		- * @return Node|null
51		- */
52		- public function format_toc_data( $post_content, $htag_support, $toc_hierarchy ) {
53		- $matches = [];
	45	+ public function view_params() {
	46	+ return [
	47	+ 'post' => get_post( $this->attributes['post_id'] )
	48	+ ];
	49	+ }
54	50
55		- if ( $htag_support != '' ) {
56		- preg_match_all( '/(<h([' . $htag_support . ']{1})[^>]*>).*<\/h\2>/msuU', $post_content, $matches, PREG_SET_ORDER );
57		- }
	51	+ /**
	52	+ * Process content for TOC generation without triggering heavy content filters
	53	+ * This handles special characters while avoiding memory issues from plugins like WPML
	54	+ *
	55	+ * @param string $content Raw post content
	56	+ * @return string Processed content
	57	+ */
	58	+ public function process_content_for_toc( $content ) {
	59	+ // Check if we should use the full content filter (for backward compatibility)
	60	+ $use_full_filter = apply_filters( 'betterdocs_toc_use_full_content_filter', false );
58	61
59		- if ( ! empty( $matches ) ) {
60		- /*
61		- |--------------------------------------------------------------------------
62		- | Backtracking Algorithm Using Iteration | Main Login For Hierarchy TOC
63		- |--------------------------------------------------------------------------
64		- |
65		- | Initially an object with key null and empty item of arrays are inserted into the stack of arrays.
66		- | When inside the loop condition for the first time, the last stack value is assigned in a variable $last_data.
67		- | And a new node object $new_data which is instantiated and the tag number is inserted for comparison as key, and empty items
68		- | as a array are inserted into items property of the new node object. On the 'if' condition it checks, if the current tag number
69		- | is smaller or equal to the last stack number. If the condition is true, the it enters into the while loop condition, which also
70		- | checks if the current last stack tag number is greater or equal to the current tag number. It pops values from the stack until and unless the
71		- | condition becomes false. The while loop condition becomes false only when the last stack value tag number becomes null and the last stack number is not
72		- | greater or equal to the current node tag number.
73		- |
74		- | Backtracking occurs when the current tag_number $number[2] is less than or equal to the stacks last tag_number which is assigned as $last_data
75		- |
76		- | And then the last value is inserted as the new last_data, and the new_data node is inserted into the last_data node items.
77		- | Additionally the $new_data is inserted into the stack to keep track of the used node. Somehow if the if condition becomes false
78		- | then the stack last data is taken, and the new_data is inserted into the last_data->items and the new_data is inserted into the stack.
79		- |
80		- */
	62	+ if ( $use_full_filter ) {
	63	+ // Use the full content filter if explicitly enabled
	64	+ return apply_filters( 'the_content', $content );
	65	+ }
81	66
82		- $dynamic_toc_title_switch = $this->settings->get( 'toc_dynamic_title' );
83		- $stack = [];
84		- $root = new Node();
85		- $root->key = null;
86		- $root->items = [];
87		- $tag_counter = 0;
	67	+ // Apply only essential content processing filters that handle special characters
	68	+ // without triggering heavy processing from plugins like WPML, page builders, etc.
88	69
89		- array_push( $stack, $root );
	70	+ // Handle shortcodes first (but don't execute them, just remove them to avoid conflicts)
	71	+ $content = strip_shortcodes( $content );
90	72
91		- foreach ( $matches as $number ) {
92		- $last_data = $stack[count( $stack ) - 1];
93		- $current_tag_number = isset( $number[2] ) ? $number[2] : '';
94		- $current_title = isset( $number[0] ) ? $number[0] : '';
95		- $current_tag = isset( $number[1] ) ? $number[1] : '';
	73	+ // Decode HTML entities to handle special characters properly
	74	+ $content = html_entity_decode( $content, ENT_QUOTES, 'UTF-8' );
96	75
97		- $heading_name = preg_replace( '/<[^<]+?>/', '', $current_title );
98		- $heading_name = ! empty( $heading_name ) ? strtolower( str_replace( " ", '-', preg_replace('/<[^>]+>|[^a-zA-Z\s\d]+/', "", html_entity_decode( $heading_name ) ) ) ) : '';
99		- preg_match('/id="(.+?)"/', $current_title, $matches_id);
100		- $heading_id = isset( $matches_id[1] ) ? strtolower( $matches_id[1] ) : '';
101		- $tag_number = ! empty( $heading_id ) ? $heading_id : ( ! empty( $heading_name ) && $this->settings->get( 'toc_dynamic_title' ) ? $heading_name : $tag_counter . '-toc-title' );
	76	+ // Convert line breaks to proper HTML if needed
	77	+ $content = wpautop( $content );
102	78
103		- $new_data = new Node();
104		- $new_data->key = $current_tag_number;
105		- $new_data->tag = $current_tag;
106		- $new_data->title = $current_title;
107		- $new_data->tag_number = $tag_number;
108		- $new_data->items = [];
	79	+ // Apply specific filters that are safe and necessary for TOC generation
	80	+ // These are lightweight filters that handle character encoding and basic formatting
	81	+ $content = apply_filters( 'betterdocs_toc_content_processing', $content );
109	82
110		- if ( $last_data->key != null && $current_tag_number <= $last_data->key ) {
111		- while ( $stack[count( $stack ) - 1]->key != null && $stack[count( $stack ) - 1]->key >= $current_tag_number ) {
112		- array_pop( $stack );
113		- }
114		- $last_data = $stack[count( $stack ) - 1];
115		- }
	83	+ // Additional safety: limit content size to prevent memory issues
	84	+ $max_content_length = apply_filters( 'betterdocs_toc_max_content_length', 500000 ); // 500KB default
	85	+ if ( strlen( $content ) > $max_content_length ) {
	86	+ $content = substr( $content, 0, $max_content_length );
	87	+ }
116	88
117		- array_push( $last_data->items, $new_data );
	89	+ return $content;
	90	+ }
118	91
119		- if ( $toc_hierarchy != 'off' && $toc_hierarchy != '' ) {
120		- array_push( $stack, $new_data );
121		- }
	92	+ /**
	93	+ * This method is responsible for re-arranging the TOC data based on hierarchy or non-hierarchy
	94	+ *
	95	+ * @param string $post_content
	96	+ * @param string $htag_support
	97	+ * @return Node|null
	98	+ */
	99	+ public function format_toc_data( $post_content, $htag_support, $toc_hierarchy ) {
	100	+ $matches = [];
122	101
123		- $tag_counter++;
124		- }
	102	+ if ( $htag_support != '' ) {
	103	+ preg_match_all( '/(<h([' . $htag_support . ']{1})[^>]*>).*<\/h\2>/msuU', $post_content, $matches, PREG_SET_ORDER );
	104	+ }
125	105
126		- return $root;
127		- }
	106	+ if ( ! empty( $matches ) ) {
	107	+ /*
	108	+ |--------------------------------------------------------------------------
	109	+ | Backtracking Algorithm Using Iteration | Main Login For Hierarchy TOC
	110	+ |--------------------------------------------------------------------------
	111	+ |
	112	+ | Initially an object with key null and empty item of arrays are inserted into the stack of arrays.
	113	+ | When inside the loop condition for the first time, the last stack value is assigned in a variable $last_data.
	114	+ | And a new node object $new_data which is instantiated and the tag number is inserted for comparison as key, and empty items
	115	+ | as a array are inserted into items property of the new node object. On the 'if' condition it checks, if the current tag number
	116	+ | is smaller or equal to the last stack number. If the condition is true, the it enters into the while loop condition, which also
	117	+ | checks if the current last stack tag number is greater or equal to the current tag number. It pops values from the stack until and unless the
	118	+ | condition becomes false. The while loop condition becomes false only when the last stack value tag number becomes null and the last stack number is not
	119	+ | greater or equal to the current node tag number.
	120	+ |
	121	+ | Backtracking occurs when the current tag_number $number[2] is less than or equal to the stacks last tag_number which is assigned as $last_data
	122	+ |
	123	+ | And then the last value is inserted as the new last_data, and the new_data node is inserted into the last_data node items.
	124	+ | Additionally the $new_data is inserted into the stack to keep track of the used node. Somehow if the if condition becomes false
	125	+ | then the stack last data is taken, and the new_data is inserted into the last_data->items and the new_data is inserted into the stack.
	126	+ |
	127	+ */
128	128
129		- return null;
130		- }
	129	+ $dynamic_toc_title_switch = $this->settings->get( 'toc_dynamic_title' );
	130	+ $stack = [];
	131	+ $root = new Node();
	132	+ $root->key = null;
	133	+ $root->items = [];
	134	+ $tag_counter = 0;
	135	+
	136	+ array_push( $stack, $root );
	137	+
	138	+ foreach ( $matches as $number ) {
	139	+ $last_data = $stack[ count( $stack ) - 1 ];
	140	+ $current_tag_number = isset( $number[2] ) ? $number[2] : '';
	141	+ $current_title = isset( $number[0] ) ? $number[0] : '';
	142	+ $current_tag = isset( $number[1] ) ? $number[1] : '';
	143	+
	144	+ $heading_name = preg_replace( '/<[^<]+?>/', '', $current_title );
	145	+ $heading_name = ! empty( $heading_name ) ? strtolower( str_replace( ' ', '-', preg_replace( '/[^\p{L}\p{N}\s]/u', '', $heading_name ) ) ) : '';
	146	+ preg_match( '/id="(.+?)"/', $current_title, $matches_id );
	147	+ $heading_id = isset( $matches_id[1] ) ? strtolower( $matches_id[1] ) : '';
	148	+ $tag_number = ! empty( $heading_id ) ? $heading_id : ( ! empty( $heading_name ) && $dynamic_toc_title_switch ? $heading_name : $tag_counter . '-toc-title' );
	149	+
	150	+ $new_data = new Node();
	151	+ $new_data->key = $current_tag_number;
	152	+ $new_data->tag = $current_tag;
	153	+ $new_data->title = $current_title;
	154	+ $new_data->tag_number = $tag_number;
	155	+ $new_data->items = [];
	156	+
	157	+ if ( $last_data->key != null && $current_tag_number <= $last_data->key ) {
	158	+ while ( $stack[ count( $stack ) - 1 ]->key != null && $stack[ count( $stack ) - 1 ]->key >= $current_tag_number ) {
	159	+ array_pop( $stack );
	160	+ }
	161	+ $last_data = $stack[ count( $stack ) - 1 ];
	162	+ }
	163	+
	164	+ array_push( $last_data->items, $new_data );
	165	+
	166	+ if ( $toc_hierarchy != 'off' && $toc_hierarchy != '' ) {
	167	+ array_push( $stack, $new_data );
	168	+ }
	169	+
	170	+ ++$tag_counter;
	171	+ }
	172	+
	173	+ return $root;
	174	+ }
	175	+
	176	+ return null;
	177	+ }
131	178	}