post-fixtures/classes/FixtureBuilder.inc

<?php

class FixtureBuilder {
	public $deferred_builds;
	private $current_object;

	function __construct() {
		$this->deferred_builds = array();
		$this->current_object = null;
	}

	function ensure_type($type) {
		if (!isset($this->deferred_builds[$type])) {
			$this->deferred_builds[$type] = array();
		}
	}

	function check_correct_type($name) {
		$ok = false;
		if (!empty($this->current_object)) {
			if ($name != $this->current_object['type']) {
				throw new LogicException("You can't call ${name} at this point.");
			}
			$ok = true;
		}
		return $ok;
	}

	function category($name) {
		if ($this->check_correct_type('category')) {
			$this->defer();
		}

		$this->current_object = array(
			'type' => 'category', 'name' => $name
		);

		return $this;
	}

	function post($title) {
		if ($this->check_correct_type('post')) {
			$this->defer();
		}

		$this->current_object = array(
			'type' => 'post', 'post' => array(
				'post_title' => $title,
				'post_type' => 'post'
			)
		);
		return $this;
	}

	function date($date) {
		$this->check_correct_type('post');

		$this->current_object['post']['post_date'] = $date;
		return $this;
	}

	function content($content) {
		$this->check_correct_type('post');

		$this->current_object['post']['post_content'] = $content;
		return $this;
	}

	function tags($tags) {
		$this->check_correct_type('post');

		$this->current_object['post']['tags'] = explode(',', $tags);
		return $this;
	}

	function defer() {
		if (!empty($this->current_object)) {
			$this->ensure_type($this->current_object['type']);
			$this->{"defer_{$this->current_object['type']}"}($this->current_object);
		}
		unset($this->current_object);
	}

	function build() {
		if (!empty($this->current_object)) {
			$this->ensure_type($this->current_object['type']);
			$result = $this->{"build_{$this->current_object['type']}"}($this->current_object);
			unset($this->current_object);
			return $result;
		}
	}

	function categories($categories) {
		$this->check_correct_type('post');

		$this->current_object['post']['categories'] = explode(',', $categories);
		return $this;
	}

	function metadata($key, $value) {
		$this->check_correct_type('post');

		$this->current_object['post']['metadata'][$key] = $value;
		return $this;
	}

	function option($name, $value) {
		update_option($name, $value);
		return $this;
	}

	function defer_category($obj) {
		$this->deferred_builds['category'] = array_merge($this->deferred_builds['category'], explode(',', $obj['name']));
	}

	function defer_post($obj) {
		$this->deferred_builds['post'][] = $obj['post'];
	}

	function build_category($obj) {
		return PostFixtures::create_category($obj['name']);
	}

	function build_post($obj) {
		$post_obj = $obj['post'];
		foreach (array('categories', 'metadata', 'tags') as $field) {
			unset($post_obj[$field]);
		}
		$post_id = PostFixtures::create_post($post_obj);
		if ($post_id != 0) {
			$category_ids  = array();
			if (isset($obj['post']['categories'])) {
				$category_ids_by_slug = array();
				foreach ($obj['post']['categories'] as $category) {
					$category_ids_by_slug = array_merge($category_ids_by_slug, PostFixtures::create_category($category));
				}
				$category_ids = array_values($category_ids_by_slug);
				PostFixtures::set_post_categories($post_id, $category_ids);
			}

			$metadata = array();
			if (isset($obj['post']['metadata'])) {
				foreach ($obj['post']['metadata'] as $key => $value) {
					PostFixtures::create_post_metadata($post_id, $key, $value);
				}
			}

			if (isset($obj['post']['tags'])) {
				PostFixtures::create_post_tags($post_id, $obj['post']['tags']);
			}

			return get_post($post_id);
		}
		// @codeCoverageIgnoreStart
		return false;
		// @codeCoverageIgnoreEnd
	}

	function finalize() {
		$this->defer();

		foreach ($this->deferred_builds as $type => $entries) {
			foreach ($entries as $entry) {
				$object = null;
				switch ($type) {
					case 'category':
						$object = array('name' => $entry);
						break;
					case 'post':
						$object = array('post' => $entry);
						break;
				}
				$this->{"build_${type}"}($object);
			}
		}
	}
}