2023-01-23 11:03:31 +01:00

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 );
}
}