Files
BIHomeWP/wp-content/plugins/wordpress-seo/src/repositories/indexable-repository.php
KhaiNguyen 59401cb805 khaihihi
2020-02-13 10:39:37 +07:00

183 lines
5.2 KiB
PHP

<?php
/**
* Yoast extension of the Model class.
*
* @package Yoast\YoastSEO\ORM\Repositories
*/
namespace Yoast\WP\SEO\Repositories;
use Yoast\WP\SEO\Builders\Indexable_Author_Builder;
use Yoast\WP\SEO\Builders\Indexable_Post_Builder;
use Yoast\WP\SEO\Builders\Indexable_Term_Builder;
use Yoast\WP\SEO\Loggers\Logger;
use Yoast\WP\SEO\ORM\ORMWrapper;
use Yoast\WP\SEO\ORM\Yoast_Model;
/**
* Class Indexable_Repository
*
* @package Yoast\WP\SEO\ORM\Repositories
*/
class Indexable_Repository extends ORMWrapper {
/**
* @var \Yoast\WP\SEO\Builders\Indexable_Author_Builder
*/
protected $author_builder;
/**
* @var \Yoast\WP\SEO\Builders\Indexable_Post_Builder
*/
protected $post_builder;
/**
* @var \Yoast\WP\SEO\Builders\Indexable_Term_Builder
*/
protected $term_builder;
/**
* @var \Psr\Log\LoggerInterface
*/
protected $logger;
/**
* Returns the instance of this class constructed through the ORM Wrapper.
*
* @param \Yoast\WP\SEO\Builders\Indexable_Author_Builder $author_builder The author builder for creating missing indexables.
* @param \Yoast\WP\SEO\Builders\Indexable_Post_Builder $post_builder The post builder for creating missing indexables.
* @param \Yoast\WP\SEO\Builders\Indexable_Term_Builder $term_builder The term builder for creating missing indexables.
* @param \Yoast\WP\SEO\Loggers\Logger $logger The logger.
*
* @return \Yoast\WP\SEO\Repositories\Indexable_Repository
*/
public static function get_instance(
Indexable_Author_Builder $author_builder,
Indexable_Post_Builder $post_builder,
Indexable_Term_Builder $term_builder,
Logger $logger
) {
ORMWrapper::$repositories[ Yoast_Model::get_table_name( 'Indexable' ) ] = self::class;
/**
* @var $instance self
*/
$instance = Yoast_Model::of_type( 'Indexable' );
$instance->author_builder = $author_builder;
$instance->post_builder = $post_builder;
$instance->term_builder = $term_builder;
$instance->logger = $logger;
return $instance;
}
/**
* Retrieves an indexable by it's URL.
*
* @param string $url The indexable url.
*/
public function find_by_url( $url ) {
$url = \trailingslashit( $url );
$url_hash = \strlen( $url ) . ':' . \md5( $url );
// Find by both url_hash and url, url_hash is indexed so will be used first by the DB to optimize the query.
return $this->where( 'url_hash', $url_hash )
->where( 'url', $url )
->find_one();
}
/**
* Retrieves an indexable by its ID and type.
*
* @param int $object_id The indexable object ID.
* @param string $object_type The indexable object type.
* @param bool $auto_create Optional. Create the indexable if it does not exist.
*
* @return bool|\Yoast\WP\SEO\Models\Indexable Instance of indexable.
*/
public function find_by_id_and_type( $object_id, $object_type, $auto_create = true ) {
$indexable = $this->where( 'object_id', $object_id )
->where( 'object_type', $object_type )
->find_one();
if ( $auto_create && ! $indexable ) {
$indexable = $this->create_for_id_and_type( $object_id, $object_type );
}
return $indexable;
}
/**
* Retrieves multiple indexables at once by their IDs and type.
*
* @param int[] $object_ids The array of indexable object IDs.
* @param string $object_type The indexable object type.
* @param bool $auto_create Optional. Create the indexable if it does not exist.
*
* @return \Yoast\WP\SEO\Models\Indexable[] An array of indexables.
*/
public function find_by_multiple_ids_and_type( $object_ids, $object_type, $auto_create = true ) {
$indexables = $this
->where_in( 'object_id', $object_ids )
->where( 'object_type', $object_type )
->find_many();
if ( $auto_create ) {
$indexables_available = \array_column( $indexables, 'object_id' );
$indexables_to_create = \array_diff( $object_ids, $indexables_available );
foreach ( $indexables_to_create as $indexable_to_create ) {
$indexable = $this->create_for_id_and_type( $indexable_to_create, $object_type );
$indexable->save();
$indexables[] = $indexable;
}
}
return $indexables;
}
/**
* Creates an indexable by its ID and type.
*
* @param int $object_id The indexable object ID.
* @param string $object_type The indexable object type.
*
* @return bool|\Yoast\WP\SEO\Models\Indexable Instance of indexable.
*/
public function create_for_id_and_type( $object_id, $object_type ) {
/**
* Indexable instance.
*
* @var \Yoast\WP\SEO\Models\Indexable $indexable
*/
$indexable = $this->create();
$indexable->object_id = $object_id;
$indexable->object_type = $object_type;
switch ( $object_type ) {
case 'post':
$indexable = $this->post_builder->build( $object_id, $indexable );
break;
case 'user':
$indexable = $this->author_builder->build( $object_id, $indexable );
break;
case 'term':
$indexable = $this->term_builder->build( $object_id, $indexable );
break;
}
$this->logger->debug(
\sprintf(
/* translators: 1: object ID; 2: object type. */
\__( 'Indexable created for object %1$s with type %2$s', 'wordpress-seo' ),
$object_id,
$object_type
),
\get_object_vars( $indexable )
);
return $indexable;
}
}