class-wpml-translation-element-factory.php 1.38 KB
<?php

class WPML_Translation_Element_Factory {
	const ELEMENT_TYPE_POST = 'Post';
	const ELEMENT_TYPE_TERM = 'Term';
	const ELEMENT_TYPE_MENU = 'Menu';

	/** @var SitePress */
	private $sitepress;

	/** @var WPML_WP_Cache */
	private $wpml_cache;

	/**
	 * @param SitePress     $sitepress
	 * @param WPML_WP_Cache $wpml_cache
	 */
	public function __construct( SitePress $sitepress, WPML_WP_Cache $wpml_cache = null ) {
		$this->sitepress  = $sitepress;
		$this->wpml_cache = $wpml_cache;
	}

	/**
	 * @param int    $id
	 * @param string $type any of `WPML_Translation_Element_Factory::ELEMENT_TYPE_POST`, `WPML_Translation_Element_Factory::ELEMENT_TYPE_TERM`, `WPML_Translation_Element_Factory::ELEMENT_TYPE_MENU`.
	 *
	 * @return WPML_Translation_Element
	 * @throws InvalidArgumentException InvalidArgumentException.
	 */
	public function create( $id, $type ) {
		$fn = 'create_' . $type;
		if ( method_exists( $this, $fn ) ) {
			return $this->$fn( $id );
		}

		throw new InvalidArgumentException( 'Element type: ' . $type . ' does not exist.' );
	}


	public function create_post( $id ) {
		return new WPML_Post_Element( $id, $this->sitepress, $this->wpml_cache );
	}

	public function create_term( $id ) {
		return new WPML_Term_Element( $id, $this->sitepress, '', $this->wpml_cache );
	}

	public function create_menu( $id ) {
		return new WPML_Menu_Element( $id, $this->sitepress, $this->wpml_cache );
	}
}