first commit
This commit is contained in:
@@ -0,0 +1,55 @@
|
||||
<?php
|
||||
|
||||
class WPML_Media_Usage {
|
||||
|
||||
const FIELD_NAME = '_wpml_media_usage';
|
||||
|
||||
/**
|
||||
* @var int
|
||||
*/
|
||||
private $attachment_id;
|
||||
/**
|
||||
* @var array
|
||||
*/
|
||||
private $usage;
|
||||
|
||||
/**
|
||||
* @param int $attachment_id
|
||||
*/
|
||||
public function __construct( $attachment_id ) {
|
||||
$this->attachment_id = $attachment_id;
|
||||
|
||||
$usage = get_post_meta( $this->attachment_id, self::FIELD_NAME, true );
|
||||
$this->usage = empty( $usage ) ? array() : $usage;
|
||||
}
|
||||
|
||||
/**
|
||||
* @return array
|
||||
*/
|
||||
public function get_posts() {
|
||||
return empty( $this->usage['posts'] ) ? array() : $this->usage['posts'];
|
||||
}
|
||||
|
||||
/**
|
||||
* @param int $post_id
|
||||
*/
|
||||
public function add_post( $post_id ) {
|
||||
$posts = $this->get_posts();
|
||||
$posts[] = $post_id;
|
||||
$this->usage['posts'] = array_unique( $posts );
|
||||
$this->update_usage();
|
||||
}
|
||||
|
||||
/**
|
||||
* @param int $post_id
|
||||
*/
|
||||
public function remove_post( $post_id ) {
|
||||
$this->usage['posts'] = array_values( array_diff( (array) $this->usage['posts'], array( $post_id ) ) );
|
||||
$this->update_usage();
|
||||
}
|
||||
|
||||
private function update_usage() {
|
||||
update_post_meta( $this->attachment_id, self::FIELD_NAME, $this->usage );
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user