229 lines
4.6 KiB
PHP
229 lines
4.6 KiB
PHP
<?php
|
|
/**
|
|
* Class AMP_Content
|
|
*
|
|
* @package AMP
|
|
*/
|
|
|
|
/**
|
|
* Class AMP_Content
|
|
*/
|
|
class AMP_Content {
|
|
|
|
/**
|
|
* Content.
|
|
*
|
|
* @var string
|
|
*/
|
|
private $content;
|
|
|
|
/**
|
|
* AMP content.
|
|
*
|
|
* @var string
|
|
*/
|
|
private $amp_content = '';
|
|
|
|
/**
|
|
* AMP scripts.
|
|
*
|
|
* @var array
|
|
*/
|
|
private $amp_scripts = array();
|
|
|
|
/**
|
|
* AMP styles.
|
|
*
|
|
* @deprecated
|
|
* @var array
|
|
*/
|
|
private $amp_styles = array();
|
|
|
|
/**
|
|
* AMP stylesheets.
|
|
*
|
|
* @since 1.0
|
|
* @var array
|
|
*/
|
|
private $amp_stylesheets = array();
|
|
|
|
/**
|
|
* Args.
|
|
*
|
|
* @var array
|
|
*/
|
|
private $args = array();
|
|
|
|
/**
|
|
* Embed handlers.
|
|
*
|
|
* @var AMP_Base_Embed_Handler[] AMP_Base_Embed_Handler[]
|
|
*/
|
|
private $embed_handlers = array();
|
|
|
|
/**
|
|
* Sanitizer class names.
|
|
*
|
|
* @var string[]
|
|
*/
|
|
private $sanitizer_classes = array();
|
|
|
|
/**
|
|
* AMP_Content constructor.
|
|
*
|
|
* @param string $content Content.
|
|
* @param string[] $embed_handler_classes Embed handler class names.
|
|
* @param string[] $sanitizer_classes Sanitizer class names.
|
|
* @param array $args Args.
|
|
*/
|
|
public function __construct( $content, $embed_handler_classes, $sanitizer_classes, $args = array() ) {
|
|
$this->content = $content;
|
|
$this->args = $args;
|
|
$this->embed_handlers = $this->register_embed_handlers( $embed_handler_classes );
|
|
$this->sanitizer_classes = $sanitizer_classes;
|
|
|
|
$this->sanitizer_classes['AMP_Embed_Sanitizer']['embed_handlers'] = $this->embed_handlers;
|
|
|
|
$this->transform();
|
|
}
|
|
|
|
/**
|
|
* Get AMP content.
|
|
*
|
|
* @return string
|
|
*/
|
|
public function get_amp_content() {
|
|
return $this->amp_content;
|
|
}
|
|
|
|
/**
|
|
* Get AMP scripts.
|
|
*
|
|
* @return array
|
|
*/
|
|
public function get_amp_scripts() {
|
|
return $this->amp_scripts;
|
|
}
|
|
|
|
/**
|
|
* Get AMP styles.
|
|
*
|
|
* @deprecated Since 1.0 in favor of the get_amp_stylesheets method.
|
|
* @return array Empty list.
|
|
*/
|
|
public function get_amp_styles() {
|
|
_deprecated_function( __METHOD__, '1.0', __CLASS__ . '::get_amp_stylesheets' );
|
|
return array();
|
|
}
|
|
|
|
/**
|
|
* Get AMP styles.
|
|
*
|
|
* @since 1.0
|
|
* @return array
|
|
*/
|
|
public function get_amp_stylesheets() {
|
|
return $this->amp_stylesheets;
|
|
}
|
|
|
|
/**
|
|
* Transform.
|
|
*/
|
|
private function transform() {
|
|
$content = $this->content;
|
|
|
|
// First, embeds + the_content filter.
|
|
$content = apply_filters( 'the_content', $content );
|
|
$this->unregister_embed_handlers( $this->embed_handlers );
|
|
|
|
// Then, sanitize to strip and/or convert non-amp content.
|
|
$content = $this->sanitize( $content );
|
|
|
|
$this->amp_content = $content;
|
|
}
|
|
|
|
/**
|
|
* Add scripts.
|
|
*
|
|
* @param array $scripts Scripts.
|
|
*/
|
|
private function add_scripts( $scripts ) {
|
|
$this->amp_scripts = array_merge( $this->amp_scripts, $scripts );
|
|
}
|
|
|
|
/**
|
|
* Add stylesheets.
|
|
*
|
|
* @since 1.0
|
|
* @param array $stylesheets Styles.
|
|
*/
|
|
private function add_stylesheets( $stylesheets ) {
|
|
$this->amp_stylesheets = array_merge( $this->amp_stylesheets, $stylesheets );
|
|
}
|
|
|
|
/**
|
|
* Register embed handlers.
|
|
*
|
|
* @param string[] $embed_handler_classes Embed handler class names.
|
|
* @return array
|
|
*/
|
|
private function register_embed_handlers( $embed_handler_classes ) {
|
|
$embed_handlers = array();
|
|
|
|
foreach ( $embed_handler_classes as $embed_handler_class => $args ) {
|
|
$embed_handler = new $embed_handler_class( array_merge( $this->args, $args ) );
|
|
|
|
if ( ! is_subclass_of( $embed_handler, 'AMP_Base_Embed_Handler' ) ) {
|
|
_doing_it_wrong(
|
|
__METHOD__,
|
|
esc_html(
|
|
sprintf(
|
|
/* translators: 1: embed handler. 2: AMP_Embed_Handler */
|
|
__( 'Embed Handler (%1$s) must extend `%2$s`', 'amp' ),
|
|
esc_html( $embed_handler_class ),
|
|
'AMP_Embed_Handler'
|
|
)
|
|
),
|
|
'0.1'
|
|
);
|
|
continue;
|
|
}
|
|
|
|
$embed_handler->register_embed();
|
|
$embed_handlers[] = $embed_handler;
|
|
}
|
|
|
|
return $embed_handlers;
|
|
}
|
|
|
|
/**
|
|
* Unregister embed handlers.
|
|
*
|
|
* @param array $embed_handlers Embed handlers.
|
|
*/
|
|
private function unregister_embed_handlers( $embed_handlers ) {
|
|
foreach ( $embed_handlers as $embed_handler ) {
|
|
$this->add_scripts( $embed_handler->get_scripts() );
|
|
$embed_handler->unregister_embed();
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Sanitize.
|
|
*
|
|
* @see AMP_Content_Sanitizer::sanitize()
|
|
* @param string $content Content.
|
|
* @return string Sanitized content.
|
|
*/
|
|
private function sanitize( $content ) {
|
|
$dom = AMP_DOM_Utils::get_dom_from_content( $content );
|
|
|
|
$results = AMP_Content_Sanitizer::sanitize_document( $dom, $this->sanitizer_classes, $this->args );
|
|
|
|
$this->add_scripts( $results['scripts'] );
|
|
$this->add_stylesheets( $results['stylesheets'] );
|
|
|
|
return AMP_DOM_Utils::get_content_from_dom( $dom );
|
|
}
|
|
}
|