Commit realizado el 12:13:52 08-04-2024
This commit is contained in:
@@ -0,0 +1,403 @@
|
||||
<?php
|
||||
/**
|
||||
* Class Media_Source_Taxonomy
|
||||
*
|
||||
* @link https://github.com/googleforcreators/web-stories-wp
|
||||
*
|
||||
* @copyright 2021 Google LLC
|
||||
* @license https://www.apache.org/licenses/LICENSE-2.0 Apache License 2.0
|
||||
*/
|
||||
|
||||
/**
|
||||
* Copyright 2021 Google LLC
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
declare(strict_types = 1);
|
||||
|
||||
namespace Google\Web_Stories\Media;
|
||||
|
||||
use Google\Web_Stories\Context;
|
||||
use Google\Web_Stories\REST_API\Stories_Terms_Controller;
|
||||
use Google\Web_Stories\Taxonomy\Taxonomy_Base;
|
||||
use ReflectionClass;
|
||||
use WP_Post;
|
||||
use WP_Query;
|
||||
use WP_Site;
|
||||
|
||||
/**
|
||||
* Class Media_Source_Taxonomy
|
||||
*
|
||||
* @phpstan-import-type TaxonomyArgs from \Google\Web_Stories\Taxonomy\Taxonomy_Base
|
||||
*/
|
||||
class Media_Source_Taxonomy extends Taxonomy_Base {
|
||||
public const TERM_EDITOR = 'editor';
|
||||
public const TERM_POSTER_GENERATION = 'poster-generation';
|
||||
public const TERM_SOURCE_VIDEO = 'source-video';
|
||||
public const TERM_SOURCE_IMAGE = 'source-image';
|
||||
public const TERM_VIDEO_OPTIMIZATION = 'video-optimization';
|
||||
public const TERM_PAGE_TEMPLATE = 'page-template';
|
||||
public const TERM_GIF_CONVERSION = 'gif-conversion';
|
||||
public const TERM_RECORDING = 'recording';
|
||||
|
||||
/**
|
||||
* Media Source key.
|
||||
*/
|
||||
public const MEDIA_SOURCE_KEY = 'web_stories_media_source';
|
||||
|
||||
/**
|
||||
* Context instance.
|
||||
*/
|
||||
private Context $context;
|
||||
|
||||
/**
|
||||
* Single constructor.
|
||||
*
|
||||
* @param Context $context Context instance.
|
||||
*/
|
||||
public function __construct( Context $context ) {
|
||||
$this->context = $context;
|
||||
$this->taxonomy_slug = 'web_story_media_source';
|
||||
$this->taxonomy_post_type = 'attachment';
|
||||
}
|
||||
|
||||
/**
|
||||
* Init.
|
||||
*
|
||||
* @since 1.10.0
|
||||
*/
|
||||
public function register(): void {
|
||||
$this->register_taxonomy();
|
||||
|
||||
add_action( 'rest_api_init', [ $this, 'rest_api_init' ] );
|
||||
add_filter( 'wp_prepare_attachment_for_js', [ $this, 'wp_prepare_attachment_for_js' ] );
|
||||
|
||||
// Hide video posters from Media grid view.
|
||||
add_filter( 'ajax_query_attachments_args', [ $this, 'filter_ajax_query_attachments_args' ], PHP_INT_MAX );
|
||||
// Hide video posters from Media list view.
|
||||
add_action( 'pre_get_posts', [ $this, 'filter_generated_media_attachments' ], PHP_INT_MAX );
|
||||
// Hide video posters from web-stories/v1/media REST API requests.
|
||||
add_filter( 'web_stories_rest_attachment_query', [ $this, 'filter_rest_generated_media_attachments' ], PHP_INT_MAX );
|
||||
}
|
||||
|
||||
/**
|
||||
* Act on site initialization.
|
||||
*
|
||||
* @since 1.29.0
|
||||
*
|
||||
* @param WP_Site $site The site being initialized.
|
||||
*/
|
||||
public function on_site_initialization( WP_Site $site ): void {
|
||||
parent::on_site_initialization( $site );
|
||||
|
||||
$this->add_missing_terms();
|
||||
}
|
||||
|
||||
/**
|
||||
* Act on plugin activation.
|
||||
*
|
||||
* @since 1.29.0
|
||||
*
|
||||
* @param bool $network_wide Whether the activation was done network-wide.
|
||||
*/
|
||||
public function on_plugin_activation( bool $network_wide ): void {
|
||||
parent::on_plugin_activation( $network_wide );
|
||||
|
||||
$this->add_missing_terms();
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns all defined media source term names.
|
||||
*
|
||||
* @since 1.29.0
|
||||
*
|
||||
* @return string[] Media sources
|
||||
*/
|
||||
public function get_all_terms(): array {
|
||||
$consts = ( new ReflectionClass( $this ) )->getConstants();
|
||||
|
||||
/**
|
||||
* List of terms.
|
||||
*
|
||||
* @var string[] $terms
|
||||
*/
|
||||
$terms = array_values(
|
||||
array_filter(
|
||||
$consts,
|
||||
static fn( $key ) => str_starts_with( $key, 'TERM_' ),
|
||||
ARRAY_FILTER_USE_KEY
|
||||
)
|
||||
);
|
||||
|
||||
return $terms;
|
||||
}
|
||||
|
||||
/**
|
||||
* Registers additional REST API fields upon API initialization.
|
||||
*
|
||||
* @since 1.10.0
|
||||
*/
|
||||
public function rest_api_init(): void {
|
||||
// Custom field, as built in term update require term id and not slug.
|
||||
register_rest_field(
|
||||
$this->taxonomy_post_type,
|
||||
self::MEDIA_SOURCE_KEY,
|
||||
[
|
||||
|
||||
'get_callback' => [ $this, 'get_callback_media_source' ],
|
||||
'schema' => [
|
||||
'description' => __( 'Media source.', 'web-stories' ),
|
||||
'type' => 'string',
|
||||
'enum' => $this->get_all_terms(),
|
||||
'context' => [ 'view', 'edit', 'embed' ],
|
||||
],
|
||||
'update_callback' => [ $this, 'update_callback_media_source' ],
|
||||
]
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* Filters the attachment data prepared for JavaScript.
|
||||
*
|
||||
* @since 1.0.0
|
||||
*
|
||||
* @param array|mixed $response Array of prepared attachment data.
|
||||
* @return array|mixed $response Filtered attachment data.
|
||||
*
|
||||
* @template T
|
||||
*
|
||||
* @phpstan-return ($response is array<T> ? array<T> : mixed)
|
||||
*/
|
||||
public function wp_prepare_attachment_for_js( $response ) {
|
||||
if ( ! \is_array( $response ) ) {
|
||||
return $response;
|
||||
}
|
||||
|
||||
$response[ self::MEDIA_SOURCE_KEY ] = $this->get_callback_media_source( $response );
|
||||
|
||||
return $response;
|
||||
}
|
||||
|
||||
/**
|
||||
* Force media attachment as string instead of the default array.
|
||||
*
|
||||
* @since 1.0.0
|
||||
*
|
||||
* @param array<string, mixed> $prepared Prepared data before response.
|
||||
*/
|
||||
public function get_callback_media_source( array $prepared ): string {
|
||||
/**
|
||||
* Taxonomy ID.
|
||||
*
|
||||
* @var int $id
|
||||
*/
|
||||
$id = $prepared['id'];
|
||||
|
||||
$terms = get_the_terms( $id, $this->taxonomy_slug );
|
||||
if ( \is_array( $terms ) && ! empty( $terms ) ) {
|
||||
return array_shift( $terms )->slug;
|
||||
}
|
||||
|
||||
return '';
|
||||
}
|
||||
|
||||
/**
|
||||
* Update rest field callback.
|
||||
*
|
||||
* @since 1.0.0
|
||||
*
|
||||
* @param string $value Value to update.
|
||||
* @param WP_Post $post Object to update on.
|
||||
* @return true|\WP_Error
|
||||
*/
|
||||
public function update_callback_media_source( string $value, WP_Post $post ) {
|
||||
$check = wp_set_object_terms( $post->ID, $value, $this->taxonomy_slug );
|
||||
if ( is_wp_error( $check ) ) {
|
||||
return $check;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* Filters the attachment query args to hide generated video poster images.
|
||||
*
|
||||
* Reduces unnecessary noise in the Media grid view.
|
||||
*
|
||||
* @since 1.10.0
|
||||
*
|
||||
* @param array<string, mixed>|mixed $args Query args.
|
||||
* @return array<string, mixed>|mixed Filtered query args.
|
||||
*
|
||||
* @template T
|
||||
*
|
||||
* @phpstan-return ($args is array<T> ? array<T> : mixed)
|
||||
*/
|
||||
public function filter_ajax_query_attachments_args( $args ) {
|
||||
if ( ! \is_array( $args ) ) {
|
||||
return $args;
|
||||
}
|
||||
|
||||
$args['tax_query'] = $this->get_exclude_tax_query( $args ); // phpcs:ignore WordPress.DB.SlowDBQuery.slow_db_query_tax_query
|
||||
|
||||
return $args;
|
||||
}
|
||||
|
||||
/**
|
||||
* Filters the current query to hide generated video poster images and source video.
|
||||
*
|
||||
* Reduces unnecessary noise in the Media list view.
|
||||
*
|
||||
* @since 1.10.0
|
||||
*
|
||||
* @param WP_Query $query WP_Query instance, passed by reference.
|
||||
*/
|
||||
public function filter_generated_media_attachments( WP_Query $query ): void {
|
||||
if ( is_admin() && $query->is_main_query() && $this->context->is_upload_screen() ) {
|
||||
$tax_query = $query->get( 'tax_query' );
|
||||
|
||||
$query->set( 'tax_query', $this->get_exclude_tax_query( [ 'tax_query' => $tax_query ] ) ); // phpcs:ignore WordPress.DB.SlowDBQuery.slow_db_query_tax_query
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Filters the current query to hide generated video poster images.
|
||||
*
|
||||
* Reduces unnecessary noise in media REST API requests.
|
||||
*
|
||||
* @since 1.10.0
|
||||
*
|
||||
* @param array<string, mixed>|mixed $args Query args.
|
||||
* @return array<string, mixed>|mixed Filtered query args.
|
||||
*
|
||||
* @template T
|
||||
*
|
||||
* @phpstan-return ($args is array<T> ? array<T> : mixed)
|
||||
*/
|
||||
public function filter_rest_generated_media_attachments( $args ) {
|
||||
if ( ! \is_array( $args ) ) {
|
||||
return $args;
|
||||
}
|
||||
|
||||
$args['tax_query'] = $this->get_exclude_tax_query( $args ); // phpcs:ignore WordPress.DB.SlowDBQuery.slow_db_query_tax_query
|
||||
|
||||
return $args;
|
||||
}
|
||||
|
||||
/**
|
||||
* Adds missing terms to the taxonomy.
|
||||
*
|
||||
* @since 1.29.0
|
||||
*/
|
||||
private function add_missing_terms(): void {
|
||||
$existing_terms = get_terms(
|
||||
[
|
||||
'taxonomy' => $this->get_taxonomy_slug(),
|
||||
'hide_empty' => false,
|
||||
'fields' => 'slugs',
|
||||
]
|
||||
);
|
||||
|
||||
if ( is_wp_error( $existing_terms ) ) {
|
||||
return;
|
||||
}
|
||||
|
||||
$missing_terms = array_diff( $this->get_all_terms(), $existing_terms );
|
||||
|
||||
foreach ( $missing_terms as $term ) {
|
||||
wp_insert_term( $term, $this->get_taxonomy_slug() );
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Taxonomy args.
|
||||
*
|
||||
* @since 1.12.0
|
||||
*
|
||||
* @return array<string,mixed> Taxonomy args.
|
||||
*
|
||||
* @phpstan-return TaxonomyArgs
|
||||
*/
|
||||
protected function taxonomy_args(): array {
|
||||
return [
|
||||
'label' => __( 'Source', 'web-stories' ),
|
||||
'public' => false,
|
||||
'rewrite' => false,
|
||||
'hierarchical' => false,
|
||||
'show_in_rest' => true,
|
||||
'rest_namespace' => self::REST_NAMESPACE,
|
||||
'rest_controller_class' => Stories_Terms_Controller::class,
|
||||
];
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns the tax query needed to exclude generated video poster images and source videos.
|
||||
*
|
||||
* @param array<string, mixed> $args Existing WP_Query args.
|
||||
* @return array<int|string, mixed> Tax query arg.
|
||||
*/
|
||||
private function get_exclude_tax_query( array $args ): array {
|
||||
/**
|
||||
* Tax query.
|
||||
*
|
||||
* @var array<int|string, mixed> $tax_query
|
||||
*/
|
||||
$tax_query = ! empty( $args['tax_query'] ) ? $args['tax_query'] : [];
|
||||
|
||||
/**
|
||||
* Filter whether generated attachments should be hidden in the media library.
|
||||
*
|
||||
* @since 1.16.0
|
||||
*
|
||||
* @param bool $enabled Whether the taxonomy check should be applied.
|
||||
* @param array $args Existing WP_Query args.
|
||||
*/
|
||||
$enabled = apply_filters( 'web_stories_hide_auto_generated_attachments', true, $args );
|
||||
if ( true !== $enabled ) {
|
||||
return $tax_query;
|
||||
}
|
||||
|
||||
/**
|
||||
* Merge with existing tax query if needed,
|
||||
* in a nested way so WordPress will run them
|
||||
* with an 'AND' relation. Example:
|
||||
*
|
||||
* [
|
||||
* 'relation' => 'AND', // implicit.
|
||||
* [ this query ],
|
||||
* [ [ any ], [ existing ], [ tax queries] ]
|
||||
* ]
|
||||
*/
|
||||
$new_tax_query = [
|
||||
'relation' => 'AND',
|
||||
[
|
||||
'taxonomy' => $this->taxonomy_slug,
|
||||
'field' => 'slug',
|
||||
'terms' => [
|
||||
self::TERM_POSTER_GENERATION,
|
||||
self::TERM_SOURCE_VIDEO,
|
||||
self::TERM_SOURCE_IMAGE,
|
||||
self::TERM_PAGE_TEMPLATE,
|
||||
],
|
||||
'operator' => 'NOT IN',
|
||||
],
|
||||
];
|
||||
|
||||
if ( ! empty( $tax_query ) ) {
|
||||
$new_tax_query[] = [ $tax_query ];
|
||||
}
|
||||
|
||||
return $new_tax_query;
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user