class.wpcom-json-api-render-endpoint.php — Jetpack – WP Security, Backup, Speed, & Growth 10.7.2

jetpack / json-endpoints / class.wpcom-json-api-render-endpoint.php

jetpack / json-endpoints Last commit date

class.wpcom-json-api-render-endpoint.php

150 lines

1	<?php
2
3	// these are helpers for the shortcode and embed render endpoints
4	abstract class WPCOM_JSON_API_Render_Endpoint extends WPCOM_JSON_API_Endpoint {
5
6	/*
7	* Figure out what scripts and styles to load.
8	* props to o2's o2_Read_API::poll() function for inspiration.
9	*
10	* In short we figure out what scripts load for a "normal" page load by executing wp_head and wp_footer
11	* then we render the embed/shortcode (to both get our result, and to have the shortcode files enqueue their resources)
12	* then we load wp_head and wp_footer again to see what new resources were added
13	* finally we find out the url to the source file and any extra info (like media or init js)
14	*/
15	function process_render( $callback, $callback_arg ) {
16	global $wp_scripts, $wp_styles;
17
18	if ( false === defined( 'STYLESHEETPATH' ) ) {
19	wp_templating_constants();
20	}
21
22	// initial scripts & styles (to subtract)
23	ob_start();
24	wp_head();
25	wp_footer();
26	ob_end_clean();
27	$initial_scripts = $wp_scripts->done;
28	$initial_styles = $wp_styles->done;
29
30	// actually render the shortcode, get the result, and do the resource loading again so we can subtract..
31	ob_start();
32	wp_head();
33	ob_end_clean();
34	$result = call_user_func( $callback, $callback_arg );
35	ob_start();
36	wp_footer();
37	ob_end_clean();
38
39	// find the difference (the new resource files)
40	$loaded_scripts = array_diff( $wp_scripts->done, $initial_scripts );
41	$loaded_styles = array_diff( $wp_styles->done, $initial_styles );
42	return array(
43	'result' => $result,
44	'loaded_scripts' => $loaded_scripts,
45	'loaded_styles' => $loaded_styles,
46	);
47	}
48
49	/**
50	* Takes the list of styles and scripts and adds them to the JSON response
51	*/
52	function add_assets( $return, $loaded_scripts, $loaded_styles ) {
53	global $wp_scripts, $wp_styles;
54	// scripts first, just cuz
55	if ( count( $loaded_scripts ) > 0 ) {
56	$scripts = array();
57	foreach ( $loaded_scripts as $handle ) {
58	if ( !isset( $wp_scripts->registered[ $handle ] ) )
59	continue;
60
61	$src = $wp_scripts->registered[ $handle ]->src;
62
63	// attach version and an extra query parameters
64	$ver = $this->get_version( $wp_scripts->registered[ $handle ]->ver, $wp_scripts->default_version );
65	if ( isset( $wp_scripts->args[ $handle ] ) ) {
66	$ver = $ver ? $ver . '&' . $wp_scripts->args[$handle] : $wp_scripts->args[$handle];
67	}
68	$src = add_query_arg( 'ver', $ver, $src );
69
70	// add to an aray so we can return all this info
71	$scripts[ $handle ] = array(
72	'src' => $src,
73	);
74	$extra = $wp_scripts->print_extra_script( $handle, false );
75	if ( !empty( $extra ) ) {
76	$scripts[$handle]['extra'] = $extra;
77	}
78	}
79	$return['scripts'] = $scripts;
80	}
81	// now styles
82	if ( count( $loaded_styles ) > 0 ) {
83	$styles = array();
84	foreach ( $loaded_styles as $handle ) {
85	if ( !isset( $wp_styles->registered[ $handle ] ) )
86	continue;
87
88	$src = $wp_styles->registered[ $handle ]->src;
89
90	// attach version and an extra query parameters
91	$ver = $this->get_version( $wp_styles->registered[ $handle ]->ver, $wp_styles->default_version );
92	if ( isset( $wp_styles->args[ $handle ] ) ) {
93	$ver = $ver ? $ver . '&' . $wp_styles->args[$handle] : $wp_styles->args[$handle];
94	}
95	$src = add_query_arg( 'ver', $ver, $src );
96
97	// is there a special media (print, screen, etc) for this? if not, default to 'all'
98	$media = 'all';
99	if ( isset( $wp_styles->registered[ $handle ]->args ) ) {
100	$media = esc_attr( $wp_styles->registered[ $handle ]->args );
101	}
102
103	// add to an array so we can return all this info
104	$styles[ $handle ] = array (
105	'src' => $src,
106	'media' => $media,
107	);
108	}
109
110	$return['styles'] = $styles;
111	}
112
113	return $return;
114	}
115
116	/**
117	* Returns the 'version' string set by the shortcode so different versions of scripts/styles can be loaded
118	*/
119	function get_version( $this_scripts_version, $default_version ) {
120	if ( null === $this_scripts_version ) {
121	$ver = '';
122	} else {
123	$ver = $this_scripts_version ? $this_scripts_version : $default_version;
124	}
125	return $ver;
126	}
127
128	/**
129	* given a shortcode, process and return the result
130	*/
131	function do_shortcode( $shortcode ) {
132	return do_shortcode( $shortcode );
133	}
134
135	/**
136	* given a one-line embed URL, process and return the result
137	*/
138	function do_embed( $embed_url ) {
139	// in order for oEmbed to fire in the `$wp_embed->shortcode` method, we need to set a post as the current post
140	$_posts = get_posts( array( 'posts_per_page' => 1, 'suppress_filters' => false ) );
141	if ( ! empty( $_posts ) ) {
142	global $post;
143	$post = array_shift( $_posts );
144	}
145
146	global $wp_embed;
147	return $wp_embed->shortcode( array(), $embed_url );
148	}
149	}
150