intraxia / jaxion

src/Axolotl/Relationship/BelongsToOne.php

Indentation +106 added lines, -106 removed lines

@@ -13,110 +13,110 @@
  * @subpackage Axolotl\Relationship
  */
 class BelongsToOne extends Root {
-	/**
-	 * Relationship's primary key name
-	 * on related model.
-	 *
-	 * @var string
-	 */
-	protected $local_key = '';
-
-	/**
-	 * HasMany constructor.
-	 *
-	 * @param Model  $model
-	 * @param string $class
-	 * @param string $type
-	 * @param string $local_key
-	 */
-	public function __construct( Model $model, $class, $type, $local_key ) {
-		$this->local_key = $local_key;
-
-		parent::__construct( $model, $class, $type );
-	}
-
-	/**
-	 * {@inheritDoc}
-	 *
-	 * @param EntityManager $database
-	 *
-	 * @throws LogicException
-	 */
-	public function attach_relation( EntityManager $database ) {
-		if ( $this->get_model()->is_filling() ) {
-			return;
-		}
-
-		$this->get_model()->set_filling( true );
-
-		$id = $this->make_target_id();
-
-		if ( $id ) {
-			$target = $database->find(
-				$this->get_class(),
-				$id
-			);
-
-			$this->set_target( $target );
-		}
-
-		$this->get_model()->set_filling( false );
-	}
-
-	/**
-	 * {@inheritDoc}
-	 *
-	 * @return string
-	 */
-	public function get_sha() {
-		return sha1(
-			__CLASS__ .
-			get_class( $this->model ) .
-			$this->class .
-			$this->type .
-			$this->local_key
-		);
-	}
-
-	/**
-	 * Get the relationship's local key.
-	 *
-	 * @return string
-	 */
-	public function get_local_key() {
-		return $this->local_key;
-	}
-
-	/**
-	 * Gets the ID value for the target model to search by.
-	 *
-	 * @return int|false
-	 *
-	 * @throws LogicException
-	 */
-	protected function make_target_id() {
-		$class = $this->get_class();
-
-		switch ( $this->get_relationship_type() ) {
-			case 'post_post':
-				return $this->get_model()
-					->get_underlying_wp_object()
-					->{$this->local_key};
-			case 'post_term':
-				$terms = wp_get_post_terms(
-					$this->get_model()
-						->get_underlying_wp_object()
-						->ID,
-					$class::get_taxonomy()
-				);
-
-				if ( ! $terms || is_wp_error( $terms ) ) {
-					return false;
-				}
-
-				return $terms[0]->term_id;
-			default:
-				throw new LogicException;
-		}
-	}
+    /**
+     * Relationship's primary key name
+     * on related model.
+     *
+     * @var string
+     */
+    protected $local_key = '';
+
+    /**
+     * HasMany constructor.
+     *
+     * @param Model  $model
+     * @param string $class
+     * @param string $type
+     * @param string $local_key
+     */
+    public function __construct( Model $model, $class, $type, $local_key ) {
+        $this->local_key = $local_key;
+
+        parent::__construct( $model, $class, $type );
+    }
+
+    /**
+     * {@inheritDoc}
+     *
+     * @param EntityManager $database
+     *
+     * @throws LogicException
+     */
+    public function attach_relation( EntityManager $database ) {
+        if ( $this->get_model()->is_filling() ) {
+            return;
+        }
+
+        $this->get_model()->set_filling( true );
+
+        $id = $this->make_target_id();
+
+        if ( $id ) {
+            $target = $database->find(
+                $this->get_class(),
+                $id
+            );
+
+            $this->set_target( $target );
+        }
+
+        $this->get_model()->set_filling( false );
+    }
+
+    /**
+     * {@inheritDoc}
+     *
+     * @return string
+     */
+    public function get_sha() {
+        return sha1(
+            __CLASS__ .
+            get_class( $this->model ) .
+            $this->class .
+            $this->type .
+            $this->local_key
+        );
+    }
+
+    /**
+     * Get the relationship's local key.
+     *
+     * @return string
+     */
+    public function get_local_key() {
+        return $this->local_key;
+    }
+
+    /**
+     * Gets the ID value for the target model to search by.
+     *
+     * @return int|false
+     *
+     * @throws LogicException
+     */
+    protected function make_target_id() {
+        $class = $this->get_class();
+
+        switch ( $this->get_relationship_type() ) {
+            case 'post_post':
+                return $this->get_model()
+                    ->get_underlying_wp_object()
+                    ->{$this->local_key};
+            case 'post_term':
+                $terms = wp_get_post_terms(
+                    $this->get_model()
+                        ->get_underlying_wp_object()
+                        ->ID,
+                    $class::get_taxonomy()
+                );
+
+                if ( ! $terms || is_wp_error( $terms ) ) {
+                    return false;
+                }
+
+                return $terms[0]->term_id;
+            default:
+                throw new LogicException;
+        }
+    }
 }

src/Axolotl/Relationship/HasMany.php

Indentation +99 added lines, -99 removed lines

@@ -12,113 +12,113 @@
  * @subpackage Axolotl\Relationship
  */
 class HasMany extends Root {
-	/**
-	 * Relationship's primary key name
-	 * on related model.
-	 *
-	 * @var string
-	 */
-	protected $foreign_key = '';
+    /**
+     * Relationship's primary key name
+     * on related model.
+     *
+     * @var string
+     */
+    protected $foreign_key = '';
 
-	/**
-	 * HasMany constructor.
-	 *
-	 * @param Model  $model
-	 * @param string $class
-	 * @param string $type
-	 * @param string $foreign_key
-	 */
-	public function __construct( Model $model, $class, $type, $foreign_key ) {
-		$this->foreign_key = $foreign_key;
+    /**
+     * HasMany constructor.
+     *
+     * @param Model  $model
+     * @param string $class
+     * @param string $type
+     * @param string $foreign_key
+     */
+    public function __construct( Model $model, $class, $type, $foreign_key ) {
+        $this->foreign_key = $foreign_key;
 
-		parent::__construct( $model, $class, $type );
-	}
+        parent::__construct( $model, $class, $type );
+    }
 
-	/**
-	 * {@inheritDoc}
-	 *
-	 * @param EntityManager $database
-	 *
-	 * @throws LogicException
-	 */
-	public function attach_relation( EntityManager $database ) {
-		if ( $this->get_model()->is_filling() ) {
-			return;
-		}
+    /**
+     * {@inheritDoc}
+     *
+     * @param EntityManager $database
+     *
+     * @throws LogicException
+     */
+    public function attach_relation( EntityManager $database ) {
+        if ( $this->get_model()->is_filling() ) {
+            return;
+        }
 
-		$this->get_model()->set_filling( true );
+        $this->get_model()->set_filling( true );
 
-		switch ( $this->get_type() ) {
-			case 'object':
-				$target = $database->find_by(
-					$this->get_class(),
-					$this->make_target_params()
-				);
-				break;
-			case 'table': // @todo implement
-			default:
-				throw new LogicException;
-		}
+        switch ( $this->get_type() ) {
+            case 'object':
+                $target = $database->find_by(
+                    $this->get_class(),
+                    $this->make_target_params()
+                );
+                break;
+            case 'table': // @todo implement
+            default:
+                throw new LogicException;
+        }
 
-		$this->set_target( $target );
+        $this->set_target( $target );
 
-		$this->get_model()->set_filling( false );
-	}
+        $this->get_model()->set_filling( false );
+    }
 
-	/**
-	 * {@inheritDoc}
-	 *
-	 * @return string
-	 */
-	public function get_sha() {
-		return sha1(
-			__CLASS__ .
-			get_class( $this->model ) .
-			$this->class .
-			$this->type .
-			$this->foreign_key
-		);
-	}
+    /**
+     * {@inheritDoc}
+     *
+     * @return string
+     */
+    public function get_sha() {
+        return sha1(
+            __CLASS__ .
+            get_class( $this->model ) .
+            $this->class .
+            $this->type .
+            $this->foreign_key
+        );
+    }
 
-	/**
-	 * Gets the relationship's foreign key.
-	 *
-	 * @return string
-	 */
-	public function get_foreign_key() {
-		return $this->foreign_key;
-	}
+    /**
+     * Gets the relationship's foreign key.
+     *
+     * @return string
+     */
+    public function get_foreign_key() {
+        return $this->foreign_key;
+    }
 
-	/**
-	 * Gets the params required for the EntityManager to find the target models.
-	 *
-	 * @return array
-	 *
-	 * @throws LogicException
-	 */
-	protected function make_target_params() {
-		switch ( $this->get_relationship_type() ) {
-			case 'post_post':
-				return array(
-					$this->foreign_key => $this->get_model()->get_primary_id(),
-					'nopaging'         => true,
-				);
-			case 'post_term':
-				return array();
-			case 'term_post':
-				$model = $this->get_model();
-				return array(
-					'nopaging'  => true,
-					'tax_query' => array(
-						array(
-							'taxonomy' => $model::get_taxonomy(),
-							'field'    => 'term_id',
-							'terms'    => $model->get_primary_id(),
-						),
-					),
-				);
-			default:
-				throw new LogicException;
-		}
-	}
+    /**
+     * Gets the params required for the EntityManager to find the target models.
+     *
+     * @return array
+     *
+     * @throws LogicException
+     */
+    protected function make_target_params() {
+        switch ( $this->get_relationship_type() ) {
+            case 'post_post':
+                return array(
+                    $this->foreign_key => $this->get_model()->get_primary_id(),
+                    'nopaging'         => true,
+                );
+            case 'post_term':
+                return array();
+            case 'term_post':
+                $model = $this->get_model();
+                return array(
+                    'nopaging'  => true,
+                    'tax_query' => array(
+                        array(
+                            'taxonomy' => $model::get_taxonomy(),
+                            'field'    => 'term_id',
+                            'terms'    => $model->get_primary_id(),
+                        ),
+                    ),
+                );
+            default:
+                throw new LogicException;
+        }
+    }
 }

src/Axolotl/Relationship/Root.php

Indentation +122 added lines, -122 removed lines

@@ -18,126 +18,126 @@
  * @subpackage Axolotl\Relationship
  */
 abstract class Root {
-	/**
-	 * Model the relationship is on.
-	 *
-	 * @var Model
-	 */
-	protected $model;
-
-	/**
-	 * Class name the Model is related to.
-	 *
-	 * @var string
-	 */
-	protected $class;
-
-	/**
-	 * Whether the Relationship is by object or model.
-	 *
-	 * @var string
-	 */
-	protected $type;
-
-	/**
-	 * Root relationship constructor.
-	 *
-	 * @param Model  $model
-	 * @param string $class
-	 * @param string $type
-	 */
-	public function __construct( Model $model, $class, $type ) {
-		$this->model = $model;
-		$this->class = $class;
-		$this->type  = $type;
-	}
-
-	/**
-	 * Returns the relationship's Model.
-	 *
-	 * @return Model
-	 */
-	public function get_model() {
-		return $this->model;
-	}
-
-	/**
-	 * Returns the relationship's related class.
-	 *
-	 * @return string
-	 */
-	public function get_class() {
-		return $this->class;
-	}
-
-	/**
-	 * Returns whether the relation is on the object or table.
-	 *
-	 * @return string
-	 */
-	public function get_type() {
-		return $this->type;
-	}
-
-	/**
-	 * Sets the relationship target on the model.
-	 *
-	 * @param Model|Collection $target
-	 */
-	public function set_target( $target ) {
-		$this->get_model()->set_related( $this->get_sha(), $target );
-	}
-
-	/**
-	 * Returns the relationship type between source and destination Model.
-	 *
-	 * @return string {source}_{dest}
-	 */
-	public function get_relationship_type() {
-		$relationship = '';
-
-		if ( $this->get_model() instanceof UsesWordPressPost ) {
-			$relationship .= 'post';
-		} elseif ( $this->get_model() instanceof UsesWordPressTerm ) {
-			$relationship .= 'term';
-		} elseif ( $this->get_model() instanceof UsesCustomTable ) {
-			$relationship .= 'table';
-		}
-
-		$relationship .= '_';
-
-		if ( is_subclass_of(
-			$this->get_class(),
-			'Intraxia\Jaxion\Contract\Axolotl\UsesWordPressPost'
-		) ) {
-			$relationship .= 'post';
-		} elseif ( is_subclass_of(
-			$this->get_class(),
-			'Intraxia\Jaxion\Contract\Axolotl\UsesWordPressTerm'
-		) ) {
-			$relationship .= 'term';
-		} elseif ( is_subclass_of(
-			$this->get_class(),
-			'Intraxia\Jaxion\Contract\Axolotl\UsesCustomTable'
-		) ) {
-			$relationship .= 'table';
-		}
-
-		return $relationship;
-	}
-
-	/**
-	 * Returns the query args to
-	 * fetch the provided relation.
-	 *
-	 * @param EntityManager $database
-	 */
-	abstract public function attach_relation( EntityManager $database );
-
-	/**
-	 * Generate the sha for the relation.
-	 *
-	 * @return string
-	 */
-	abstract public function get_sha();
+    /**
+     * Model the relationship is on.
+     *
+     * @var Model
+     */
+    protected $model;
+
+    /**
+     * Class name the Model is related to.
+     *
+     * @var string
+     */
+    protected $class;
+
+    /**
+     * Whether the Relationship is by object or model.
+     *
+     * @var string
+     */
+    protected $type;
+
+    /**
+     * Root relationship constructor.
+     *
+     * @param Model  $model
+     * @param string $class
+     * @param string $type
+     */
+    public function __construct( Model $model, $class, $type ) {
+        $this->model = $model;
+        $this->class = $class;
+        $this->type  = $type;
+    }
+
+    /**
+     * Returns the relationship's Model.
+     *
+     * @return Model
+     */
+    public function get_model() {
+        return $this->model;
+    }
+
+    /**
+     * Returns the relationship's related class.
+     *
+     * @return string
+     */
+    public function get_class() {
+        return $this->class;
+    }
+
+    /**
+     * Returns whether the relation is on the object or table.
+     *
+     * @return string
+     */
+    public function get_type() {
+        return $this->type;
+    }
+
+    /**
+     * Sets the relationship target on the model.
+     *
+     * @param Model|Collection $target
+     */
+    public function set_target( $target ) {
+        $this->get_model()->set_related( $this->get_sha(), $target );
+    }
+
+    /**
+     * Returns the relationship type between source and destination Model.
+     *
+     * @return string {source}_{dest}
+     */
+    public function get_relationship_type() {
+        $relationship = '';
+
+        if ( $this->get_model() instanceof UsesWordPressPost ) {
+            $relationship .= 'post';
+        } elseif ( $this->get_model() instanceof UsesWordPressTerm ) {
+            $relationship .= 'term';
+        } elseif ( $this->get_model() instanceof UsesCustomTable ) {
+            $relationship .= 'table';
+        }
+
+        $relationship .= '_';
+
+        if ( is_subclass_of(
+            $this->get_class(),
+            'Intraxia\Jaxion\Contract\Axolotl\UsesWordPressPost'
+        ) ) {
+            $relationship .= 'post';
+        } elseif ( is_subclass_of(
+            $this->get_class(),
+            'Intraxia\Jaxion\Contract\Axolotl\UsesWordPressTerm'
+        ) ) {
+            $relationship .= 'term';
+        } elseif ( is_subclass_of(
+            $this->get_class(),
+            'Intraxia\Jaxion\Contract\Axolotl\UsesCustomTable'
+        ) ) {
+            $relationship .= 'table';
+        }
+
+        return $relationship;
+    }
+
+    /**
+     * Returns the query args to
+     * fetch the provided relation.
+     *
+     * @param EntityManager $database
+     */
+    abstract public function attach_relation( EntityManager $database );
+
+    /**
+     * Generate the sha for the relation.
+     *
+     * @return string
+     */
+    abstract public function get_sha();
 }

src/Axolotl/EntityManager.php

Indentation +241 added lines, -241 removed lines

@@ -22,245 +22,245 @@
  * @subpackage Axolotl
  */
 class EntityManager implements EntityManagerContract {
-	/**
-	 * Post meta prefix.
-	 *
-	 * @var string
-	 */
-	protected $prefix;
-
-	/**
-	 * WP_Query instance.
-	 *
-	 * @var WP_Query
-	 */
-	protected $main;
-
-	/**
-	 * Global WPDB instance.
-	 *
-	 * @var wpdb
-	 */
-	protected $wpdb;
-
-	/**
-	 * EntityManager constructor.
-	 *
-	 * @param WP_Query $main
-	 * @param string   $prefix Post meta prefix.
-	 */
-	public function __construct( WP_Query $main, $prefix ) {
-		global $wpdb;
-
-		$this->wpdb   = $wpdb;
-		$this->main   = $main;
-		$this->prefix = $prefix;
-	}
-
-	/**
-	 * {@inheritDoc}
-	 *
-	 * @param string $class
-	 * @param int    $id
-	 *
-	 * @return Model|WP_Error
-	 *
-	 * @throws LogicException
-	 */
-	public function find( $class, $id ) {
-		$repository = $this->get_repository( $class );
-		$model      = $repository->find( $id );
-
-		if ( is_wp_error( $model ) ) {
-			return $model;
-		}
-
-		$this->handle_model( $repository, $model );
-
-		return $model;
-	}
-
-	/**
-	 * {@inheritDoc}
-	 *
-	 * @param string $class
-	 * @param array  $params
-	 *
-	 * @return Collection
-	 *
-	 * @throws LogicException
-	 */
-	public function find_by( $class, $params = array() ) {
-		$repository = $this->get_repository( $class );
-		$collection = $repository->find_by( $params );
-
-		foreach ( $collection as $model ) {
-			$this->handle_model( $repository, $model );
-		}
-
-		return $collection;
-	}
-
-	/**
-	 * {@inheritDoc}
-	 *
-	 * @param string $class
-	 * @param array  $data
-	 *
-	 * @return Model|WP_Error
-	 */
-	public function create( $class, $data = array() ) {
-		$repository = $this->get_repository( $class );
-		$model      = $repository->create( $data );
-
-		$this->handle_model( $repository, $model );
-
-		return $model;
-	}
-
-	/**
-	 * {@inheritDoc}
-	 *
-	 * @param Model $model
-	 *
-	 * @return Model|WP_Error
-	 */
-	public function persist( Model $model ) {
-		return $this->get_repository( get_class( $model ) )->persist( $model );
-	}
-
-	/**
-	 * {@inheritDoc}
-	 *
-	 * @param Model $model
-	 * @param bool  $force
-	 *
-	 * @return Model|WP_Error
-	 */
-	public function delete( Model $model, $force = false ) {
-		return $this->get_repository( get_class( $model ) )->delete( $model, $force );
-	}
-
-	/**
-	 * {@inheritDoc}
-	 */
-	public function free() {
-		AbstractRepository::free();
-	}
-
-	/**
-	 * Get the EntityManager prefix.
-	 *
-	 * @return string
-	 */
-	public function get_prefix() {
-		return $this->prefix;
-	}
-
-	/**
-	 * Get the main WP_Query instance.
-	 *
-	 * @return WP_Query
-	 */
-	public function get_main_query() {
-		return $this->main;
-	}
-
-	/**
-	 * Get the wpdb connection instance.
-	 *
-	 * @return wpdb
-	 */
-	public function get_wpdb() {
-		return $this->wpdb;
-	}
-
-	/**
-	 * Retrieves the repository for the given class.
-	 *
-	 * @param string $class
-	 *
-	 * @return Repository\AbstractRepository
-	 *
-	 * @throws LogicException
-	 */
-	protected function get_repository( $class ) {
-		// We can only use Axolotl models.
-		if ( ! is_subclass_of( $class, 'Intraxia\Jaxion\Axolotl\Model' ) ) {
-			throw new LogicException;
-		}
-
-		if ( is_subclass_of(
-			$class,
-			'Intraxia\Jaxion\Contract\Axolotl\UsesWordPressPost'
-		) ) {
-			return new WordPressPostRepository( $this, $class );
-		}
-
-		if ( is_subclass_of(
-			$class,
-			'Intraxia\Jaxion\Contract\Axolotl\UsesWordPressTerm'
-		) ) {
-			return new WordPressTermRepository( $this, $class );
-		}
-
-		if ( is_subclass_of(
-			$class,
-			'Intraxia\Jaxion\Contract\Axolotl\UsesCustomTable'
-		) ) {
-			throw new LogicException;
-		}
-
-		// If a model doesn't have a backing data source,
-		// the developer needs to fix this immediately.
-		throw new LogicException;
-	}
-
-	/**
-	 * Ensures the model is registered with the model and fills its relations.
-	 *
-	 * @param AbstractRepository $repository
-	 * @param Model              $model
-	 */
-	protected function handle_model( AbstractRepository $repository, Model $model ) {
-		$repository->register_model( $model );
-
-		if ( $model instanceof HasEagerRelationships ) {
-			$this->fill_related( $model, $model::get_eager_relationships() );
-		}
-	}
-
-	/**
-	 * Fills the Model with the provided relations.
-	 *
-	 * If no relations are provided, all relations are filled.
-	 *
-	 * @param Model $model
-	 * @param array $relations
-	 *
-	 * @throws LogicException
-	 */
-	protected function fill_related( Model $model, array $relations = array() ) {
-		if ( ! $relations ) {
-			$relations = $model->get_related_keys();
-		}
-
-		foreach ( $relations as $relation ) {
-			if ( ! in_array( $relation, $model->get_related_keys() ) ) {
-				throw new LogicException;
-			}
-
-			if ( $model->relation_is_filled( $relation ) ) {
-				continue;
-			}
-
-			/**
-			 * Model relationship.
-			 *
-			 * @var Relationship $relation
-			 */
-			$relation = $model->{"related_{$relation}"}();
-			$relation->attach_relation( $this );
-		}
-	}
+    /**
+     * Post meta prefix.
+     *
+     * @var string
+     */
+    protected $prefix;
+
+    /**
+     * WP_Query instance.
+     *
+     * @var WP_Query
+     */
+    protected $main;
+
+    /**
+     * Global WPDB instance.
+     *
+     * @var wpdb
+     */
+    protected $wpdb;
+
+    /**
+     * EntityManager constructor.
+     *
+     * @param WP_Query $main
+     * @param string   $prefix Post meta prefix.
+     */
+    public function __construct( WP_Query $main, $prefix ) {
+        global $wpdb;
+
+        $this->wpdb   = $wpdb;
+        $this->main   = $main;
+        $this->prefix = $prefix;
+    }
+
+    /**
+     * {@inheritDoc}
+     *
+     * @param string $class
+     * @param int    $id
+     *
+     * @return Model|WP_Error
+     *
+     * @throws LogicException
+     */
+    public function find( $class, $id ) {
+        $repository = $this->get_repository( $class );
+        $model      = $repository->find( $id );
+
+        if ( is_wp_error( $model ) ) {
+            return $model;
+        }
+
+        $this->handle_model( $repository, $model );
+
+        return $model;
+    }
+
+    /**
+     * {@inheritDoc}
+     *
+     * @param string $class
+     * @param array  $params
+     *
+     * @return Collection
+     *
+     * @throws LogicException
+     */
+    public function find_by( $class, $params = array() ) {
+        $repository = $this->get_repository( $class );
+        $collection = $repository->find_by( $params );
+
+        foreach ( $collection as $model ) {
+            $this->handle_model( $repository, $model );
+        }
+
+        return $collection;
+    }
+
+    /**
+     * {@inheritDoc}
+     *
+     * @param string $class
+     * @param array  $data
+     *
+     * @return Model|WP_Error
+     */
+    public function create( $class, $data = array() ) {
+        $repository = $this->get_repository( $class );
+        $model      = $repository->create( $data );
+
+        $this->handle_model( $repository, $model );
+
+        return $model;
+    }
+
+    /**
+     * {@inheritDoc}
+     *
+     * @param Model $model
+     *
+     * @return Model|WP_Error
+     */
+    public function persist( Model $model ) {
+        return $this->get_repository( get_class( $model ) )->persist( $model );
+    }
+
+    /**
+     * {@inheritDoc}
+     *
+     * @param Model $model
+     * @param bool  $force
+     *
+     * @return Model|WP_Error
+     */
+    public function delete( Model $model, $force = false ) {
+        return $this->get_repository( get_class( $model ) )->delete( $model, $force );
+    }
+
+    /**
+     * {@inheritDoc}
+     */
+    public function free() {
+        AbstractRepository::free();
+    }
+
+    /**
+     * Get the EntityManager prefix.
+     *
+     * @return string
+     */
+    public function get_prefix() {
+        return $this->prefix;
+    }
+
+    /**
+     * Get the main WP_Query instance.
+     *
+     * @return WP_Query
+     */
+    public function get_main_query() {
+        return $this->main;
+    }
+
+    /**
+     * Get the wpdb connection instance.
+     *
+     * @return wpdb
+     */
+    public function get_wpdb() {
+        return $this->wpdb;
+    }
+
+    /**
+     * Retrieves the repository for the given class.
+     *
+     * @param string $class
+     *
+     * @return Repository\AbstractRepository
+     *
+     * @throws LogicException
+     */
+    protected function get_repository( $class ) {
+        // We can only use Axolotl models.
+        if ( ! is_subclass_of( $class, 'Intraxia\Jaxion\Axolotl\Model' ) ) {
+            throw new LogicException;
+        }
+
+        if ( is_subclass_of(
+            $class,
+            'Intraxia\Jaxion\Contract\Axolotl\UsesWordPressPost'
+        ) ) {
+            return new WordPressPostRepository( $this, $class );
+        }
+
+        if ( is_subclass_of(
+            $class,
+            'Intraxia\Jaxion\Contract\Axolotl\UsesWordPressTerm'
+        ) ) {
+            return new WordPressTermRepository( $this, $class );
+        }
+
+        if ( is_subclass_of(
+            $class,
+            'Intraxia\Jaxion\Contract\Axolotl\UsesCustomTable'
+        ) ) {
+            throw new LogicException;
+        }
+
+        // If a model doesn't have a backing data source,
+        // the developer needs to fix this immediately.
+        throw new LogicException;
+    }
+
+    /**
+     * Ensures the model is registered with the model and fills its relations.
+     *
+     * @param AbstractRepository $repository
+     * @param Model              $model
+     */
+    protected function handle_model( AbstractRepository $repository, Model $model ) {
+        $repository->register_model( $model );
+
+        if ( $model instanceof HasEagerRelationships ) {
+            $this->fill_related( $model, $model::get_eager_relationships() );
+        }
+    }
+
+    /**
+     * Fills the Model with the provided relations.
+     *
+     * If no relations are provided, all relations are filled.
+     *
+     * @param Model $model
+     * @param array $relations
+     *
+     * @throws LogicException
+     */
+    protected function fill_related( Model $model, array $relations = array() ) {
+        if ( ! $relations ) {
+            $relations = $model->get_related_keys();
+        }
+
+        foreach ( $relations as $relation ) {
+            if ( ! in_array( $relation, $model->get_related_keys() ) ) {
+                throw new LogicException;
+            }
+
+            if ( $model->relation_is_filled( $relation ) ) {
+                continue;
+            }
+
+            /**
+             * Model relationship.
+             *
+             * @var Relationship $relation
+             */
+            $relation = $model->{"related_{$relation}"}();
+            $relation->attach_relation( $this );
+        }
+    }
 }

src/Axolotl/Model.php

Indentation +826 added lines, -826 removed lines

@@ -23,830 +23,830 @@
  * @since      0.1.0
  */
 abstract class Model implements Serializes {
-	/**
-	 * Memoized values for class methods.
-	 *
-	 * @var array
-	 */
-	private static $memo = array();
-
-	/**
-	 * Model attributes.
-	 *
-	 * @var array
-	 */
-	private $attributes = array(
-		'table'  => array(),
-		'object' => null,
-	);
-
-	/**
-	 * Model's original attributes.
-	 *
-	 * @var array
-	 */
-	private $original = array(
-		'table'  => array(),
-		'object' => null,
-	);
-
-	/**
-	 * Properties which are allowed to be set on the model.
-	 *
-	 * If this array is empty, any attributes can be set on the model.
-	 *
-	 * @var string[]
-	 */
-	protected $fillable = array();
-
-	/**
-	 * Properties which cannot be automatically filled on the model.
-	 *
-	 * If the model is unguarded, these properties can be filled.
-	 *
-	 * @var array
-	 */
-	protected $guarded = array();
-
-	/**
-	 * Properties which should not be serialized.
-	 *
-	 * @var array
-	 */
-	protected $hidden = array();
-
-	/**
-	 * Properties which should be serialized.
-	 *
-	 * @var array
-	 */
-	protected $visible = array();
-
-	/**
-	 * Relations saved on the Model.
-	 *
-	 * @var array
-	 */
-	protected $related = array();
-
-	/**
-	 * Whether the model's properties are guarded.
-	 *
-	 * When false, allows guarded properties to be filled.
-	 *
-	 * @var bool
-	 */
-	protected $is_guarded = true;
-
-	/**
-	 * Whether the Model is having its relations filled.
-	 *
-	 * @var bool
-	 */
-	protected $filling = false;
-
-	/**
-	 * Constructs a new model with provided attributes.
-	 *
-	 * If 'object' is passed as one of the attributes, the underlying post
-	 * will be overwritten.
-	 *
-	 * @param array <string, mixed> $attributes
-	 */
-	public function __construct( array $attributes = array() ) {
-		$this->maybe_boot();
-		$this->sync_original();
-
-		if ( $this->uses_wp_object() ) {
-			$this->create_wp_object();
-		}
-
-		$this->refresh( $attributes );
-	}
-
-	/**
-	 * Refreshes the model's current attributes with the provided array.
-	 *
-	 * The model's attributes will match what was provided in the array,
-	 * and any attributes not passed
-	 *
-	 * @param array $attributes
-	 */
-	public function refresh( array $attributes ) {
-		$this->clear();
-
-		foreach ( $attributes as $name => $value ) {
-			$this->set_attribute( $name, $value );
-		}
-	}
-
-	/**
-	 * Get the model's table attributes.
-	 *
-	 * Returns the array of for the model that will either need to be
-	 * saved in postmeta or a separate table.
-	 *
-	 * @return array
-	 */
-	public function get_table_attributes() {
-		return $this->attributes['table'];
-	}
-
-	/**
-	 * Get the model's original attributes.
-	 *
-	 * @return array
-	 */
-	public function get_original_table_attributes() {
-		return $this->original['table'];
-	}
-
-	/**
-	 * Retrieve an array of the attributes on the model
-	 * that have changed compared to the model's
-	 * original data.
-	 *
-	 * @return array
-	 */
-	public function get_changed_table_attributes() {
-		$changed = array();
-
-		foreach ( $this->get_table_attributes() as $attribute ) {
-			if ( $this->get_attribute( $attribute ) !==
-			     $this->get_original_attribute( $attribute )
-			) {
-				$changed[ $attribute ] = $this->get_attribute( $attribute );
-			}
-		}
-
-		return $changed;
-	}
-
-	/**
-	 * Get the model's underlying post.
-	 *
-	 * Returns the underlying WP_Post object for the model, representing
-	 * the data that will be save in the wp_posts table.
-	 *
-	 * @return false|WP_Post|WP_Term
-	 */
-	public function get_underlying_wp_object() {
-		if ( isset( $this->attributes['object'] ) ) {
-			return $this->attributes['object'];
-		}
-
-		return false;
-	}
-
-	/**
-	 * Get the model's original underlying post.
-	 *
-	 * @return WP_Post
-	 */
-	public function get_original_underlying_wp_object() {
-		return $this->original['object'];
-	}
-
-	/**
-	 * Get the model attributes on the WordPress object
-	 * that have changed compared to the model's
-	 * original attributes.
-	 *
-	 * @return array
-	 */
-	public function get_changed_wp_object_attributes() {
-		$changed = array();
-
-		foreach ( $this->get_wp_object_keys() as $key ) {
-			if ( $this->get_attribute( $key ) !==
-			     $this->get_original_attribute( $key )
-			) {
-				$changed[ $key ] = $this->get_attribute( $key );
-			}
-		}
-
-		return $changed;
-	}
-
-	/**
-	 * Magic __set method.
-	 *
-	 * Passes the name and value to set_attribute, which is where the magic happens.
-	 *
-	 * @param string $name
-	 * @param mixed  $value
-	 */
-	public function __set( $name, $value ) {
-		$this->set_attribute( $name, $value );
-	}
-
-	/**
-	 * Sets the model attributes.
-	 *
-	 * Checks whether the model attribute can be set, check if it
-	 * maps to the WP_Post property, otherwise, assigns it to the
-	 * table attribute array.
-	 *
-	 * @param string $name
-	 * @param mixed  $value
-	 *
-	 * @return $this
-	 *
-	 * @throws GuardedPropertyException
-	 */
-	public function set_attribute( $name, $value ) {
-		if ( 'object' === $name ) {
-			return $this->override_wp_object( $value );
-		}
-
-		if ( ! $this->is_fillable( $name ) ) {
-			throw new GuardedPropertyException;
-		}
-
-		if ( $method = $this->has_map_method( $name ) ) {
-			$this->attributes['object']->{$this->{$method}()} = $value;
-		} else {
-			$this->attributes['table'][ $name ] = $value;
-		}
-
-		return $this;
-	}
-
-	/**
-	 * Retrieves all the attribute keys for the model.
-	 *
-	 * @return array
-	 */
-	public function get_attribute_keys() {
-		if ( isset( self::$memo[ get_called_class() ][ __METHOD__ ] ) ) {
-			return self::$memo[ get_called_class() ][ __METHOD__ ];
-		}
-
-		return self::$memo[ get_called_class() ][ __METHOD__ ]
-			= array_merge(
-				$this->fillable,
-				$this->guarded,
-				$this->get_compute_methods(),
-				$this->get_related_methods()
-			);
-	}
-
-	/**
-	 * Retrieves the attribute keys that aren't mapped to a post.
-	 *
-	 * @return array
-	 */
-	public function get_table_keys() {
-		if ( isset( self::$memo[ get_called_class() ][ __METHOD__ ] ) ) {
-			return self::$memo[ get_called_class() ][ __METHOD__ ];
-		}
-
-		$keys = array();
-
-		foreach ( $this->get_attribute_keys() as $key ) {
-			if ( ! $this->has_map_method( $key ) &&
-			     ! $this->has_compute_method( $key ) &&
-			     ! $this->has_related_method( $key )
-			) {
-				$keys[] = $key;
-			}
-		}
-
-		return self::$memo[ get_called_class() ][ __METHOD__ ] = $keys;
-	}
-
-	/**
-	 * Retrieves the attribute keys that are mapped to a post.
-	 *
-	 * @return array
-	 */
-	public function get_wp_object_keys() {
-		if ( isset( self::$memo[ get_called_class() ][ __METHOD__ ] ) ) {
-			return self::$memo[ get_called_class() ][ __METHOD__ ];
-		}
-
-		$keys = array();
-
-		foreach ( $this->get_attribute_keys() as $key ) {
-			if ( $this->has_map_method( $key ) ) {
-				$keys[] = $key;
-			}
-		}
-
-		return self::$memo[ get_called_class() ][ __METHOD__ ] = $keys;
-	}
-
-	/**
-	 * Returns the model's keys that are computed at call time.
-	 *
-	 * @return array
-	 */
-	public function get_computed_keys() {
-		if ( isset( self::$memo[ get_called_class() ][ __METHOD__ ] ) ) {
-			return self::$memo[ get_called_class() ][ __METHOD__ ];
-		}
-
-		$keys = array();
-
-		foreach ( $this->get_attribute_keys() as $key ) {
-			if ( $this->has_compute_method( $key ) ) {
-				$keys[] = $key;
-			}
-		}
-
-		return self::$memo[ get_called_class() ][ __METHOD__ ] = $keys;
-	}
-
-	/**
-	 * Returns the model's keys that are related to other Models.
-	 *
-	 * @return array
-	 */
-	public function get_related_keys() {
-		if ( isset( self::$memo[ get_called_class() ][ __METHOD__ ] ) ) {
-			return self::$memo[ get_called_class() ][ __METHOD__ ];
-		}
-
-		$keys = array();
-
-		foreach ( $this->get_attribute_keys() as $key ) {
-			if ( $this->has_related_method( $key ) ) {
-				$keys[] = $key;
-			}
-		}
-
-		return self::$memo[ get_called_class() ][ __METHOD__ ] = $keys;
-	}
-
-	/**
-	 * Serializes the model's public data into an array.
-	 *
-	 * @return array
-	 */
-	public function serialize() {
-		$attributes = array();
-
-		if ( $this->visible ) {
-			// If visible attributes are set, we'll only reveal those.
-			foreach ( $this->visible as $key ) {
-				$attributes[ $key ] = $this->get_attribute( $key );
-			}
-		} elseif ( $this->hidden ) {
-			// If hidden attributes are set, we'll grab everything and hide those.
-			foreach ( $this->get_attribute_keys() as $key ) {
-				if ( ! in_array( $key, $this->hidden ) ) {
-					$attributes[ $key ] = $this->get_attribute( $key );
-				}
-			}
-		} else {
-			// If nothing is hidden/visible, we'll grab and reveal everything.
-			foreach ( $this->get_attribute_keys() as $key ) {
-				$attributes[ $key ] = $this->get_attribute( $key );
-			}
-		}
-
-		return array_map( function ( $attribute ) {
-			if ( $attribute instanceof Serializes ) {
-				return $attribute->serialize();
-			}
-
-			return $attribute;
-		}, $attributes );
-	}
-
-	/**
-	 * Syncs the current attributes to the model's original.
-	 *
-	 * @return $this
-	 */
-	public function sync_original() {
-		$this->original = $this->attributes;
-
-		if ( $this->attributes['object'] ) {
-			$this->original['object'] = clone $this->attributes['object'];
-		}
-
-		return $this;
-	}
-
-	/**
-	 * Checks if a given attribute is mass-fillable.
-	 *
-	 * Returns true if the attribute can be filled, false if it can't.
-	 *
-	 * @param string $name
-	 *
-	 * @return bool
-	 */
-	private function is_fillable( $name ) {
-		// If this model isn't guarded, everything is fillable.
-		if ( ! $this->is_guarded ) {
-			return true;
-		}
-
-		// If it's in the fillable array, then it's fillable.
-		if ( in_array( $name, $this->fillable ) ) {
-			return true;
-		}
-
-		// If it's explicitly guarded, then it's not fillable.
-		if ( in_array( $name, $this->guarded ) ) {
-			return false;
-		}
-
-		// If fillable hasn't been defined, then everything else fillable.
-		return ! $this->fillable;
-	}
-
-	/**
-	 * Overrides the current WP_Post with a provided one.
-	 *
-	 * Resets the post's default values and stores it in the attributes.
-	 *
-	 * @param WP_Post $value
-	 *
-	 * @return $this
-	 */
-	private function override_wp_object( $value ) {
-		$this->attributes['object'] = $this->set_wp_object_constants( $value );
-
-		return $this;
-	}
-
-	/**
-	 * Create and set with a new blank post.
-	 *
-	 * Creates a new WP_Post object, assigns it the default attributes,
-	 * and stores it in the attributes.
-	 *
-	 * @throws LogicException
-	 */
-	private function create_wp_object() {
-		switch ( true ) {
-			case $this instanceof UsesWordPressPost:
-				$object = new WP_Post( (object) array() );
-				break;
-			case $this instanceof UsesWordPressTerm:
-				$object = new WP_Term( (object) array() );
-				break;
-			default:
-				throw new LogicException;
-				break;
-		}
-
-		$this->attributes['object'] = $this->set_wp_object_constants( $object );
-	}
-
-	/**
-	 * Enforces values on the post that can't change.
-	 *
-	 * Primarily, this is used to make sure the post_type always maps
-	 * to the model's "$type" property, but this can all be overridden
-	 * by the developer to enforce other values in the model.
-	 *
-	 * @param object $object
-	 *
-	 * @return object
-	 */
-	protected function set_wp_object_constants( $object ) {
-		if ( $this instanceof UsesWordPressPost ) {
-			$object->post_type = $this::get_post_type();
-		}
-
-		if ( $this instanceof UsesWordPressTerm ) {
-			$object->taxonomy = $this::get_taxonomy();
-		}
-
-		return $object;
-	}
-
-	/**
-	 * Magic __get method.
-	 *
-	 * Passes the name and value to get_attribute, which is where the magic happens.
-	 *
-	 * @param string $name
-	 *
-	 * @return mixed
-	 */
-	public function __get( $name ) {
-		return $this->get_attribute( $name );
-	}
-
-	/**
-	 * Retrieves the model attribute.
-	 *
-	 * @param string $name
-	 *
-	 * @return mixed
-	 *
-	 * @throws PropertyDoesNotExistException If property isn't found.
-	 */
-	public function get_attribute( $name ) {
-		if ( $method = $this->has_map_method( $name ) ) {
-			$value = $this->attributes['object']->{$this->{$method}()};
-		} elseif ( $method = $this->has_compute_method( $name ) ) {
-			$value = $this->{$method}();
-		} else if ( $method = $this->has_related_method( $name ) ) {
-			$value = $this->get_related( $this->{$method}()->get_sha() );
-		} else {
-			if ( ! isset( $this->attributes['table'][ $name ] ) ) {
-				throw new PropertyDoesNotExistException;
-			}
-
-			$value = $this->attributes['table'][ $name ];
-		}
-
-		return $value;
-	}
-
-	/**
-	 * Retrieve the model's original attribute value.
-	 *
-	 * @param string $name
-	 *
-	 * @return mixed
-	 *
-	 * @throws PropertyDoesNotExistException If property isn't found.
-	 */
-	public function get_original_attribute( $name ) {
-		$original = new static( $this->original );
-
-		return $original->get_attribute( $name );
-	}
-
-	/**
-	 * Fetches the Model's primary ID, depending on the model
-	 * implementation.
-	 *
-	 * @return int
-	 *
-	 * @throws LogicException
-	 */
-	public function get_primary_id() {
-		if ( $this instanceof UsesWordPressPost ) {
-			return $this->get_underlying_wp_object()->ID;
-		}
-
-		if ( $this instanceof UsesWordPressTerm ) {
-			return $this->get_underlying_wp_object()->term_id;
-		}
-
-		// Model w/o wp_object not yet supported.
-		throw new LogicException;
-	}
-
-	/**
-	 * Generates the table foreign key, depending on the model
-	 * implementation.
-	 *
-	 * @return string
-	 *
-	 * @throws LogicException
-	 */
-	public function get_foreign_key() {
-		if ( $this instanceof UsesWordPressPost ) {
-			return 'post_id';
-		}
-
-		// Model w/o wp_object not yet supported.
-		throw new LogicException;
-	}
-
-	/**
-	 * Gets the related Model or Collection for the given sha.
-	 *
-	 * @param string $sha
-	 *
-	 * @return Model|Collection
-	 */
-	public function get_related( $sha ) {
-		return $this->related[ $sha ];
-	}
-
-	/**
-	 * Sets the related Model or Collection for the given sha.
-	 *
-	 * @param string           $sha
-	 * @param Model|Collection $target
-	 *
-	 * @throws RuntimeException
-	 */
-	public function set_related( $sha, $target ) {
-		if ( ! ( $target instanceof Model ) && ! ( $target instanceof Collection ) ) {
-			throw new RuntimeException;
-		}
-
-		$this->related[ $sha ] = $target;
-	}
-
-	/**
-	 * Checks whether the attribute has a map method.
-	 *
-	 * This is used to determine whether the attribute maps to a
-	 * property on the underlying WP_Post object. Returns the
-	 * method if one exists, returns false if it doesn't.
-	 *
-	 * @param string $name
-	 *
-	 * @return false|string
-	 */
-	protected function has_map_method( $name ) {
-		if ( method_exists( $this, $method = "map_{$name}" ) ) {
-			return $method;
-		}
-
-		return false;
-	}
-
-	/**
-	 * Checks whether the attribute has a compute method.
-	 *
-	 * This is used to determine if the attribute should be computed
-	 * from other attributes.
-	 *
-	 * @param string $name
-	 *
-	 * @return false|string
-	 */
-	protected function has_compute_method( $name ) {
-		if ( method_exists( $this, $method = "compute_{$name}" ) ) {
-			return $method;
-		}
-
-		return false;
-	}
-
-	/**
-	 * Checks whether the attribute has a compute method.
-	 *
-	 * This is used to determine if the attribute should be computed
-	 * from other attributes.
-	 *
-	 * @param string $name
-	 *
-	 * @return false|string
-	 */
-	protected function has_related_method( $name ) {
-		if ( method_exists( $this, $method = "related_{$name}" ) ) {
-			return $method;
-		}
-
-		return false;
-	}
-
-	/**
-	 * Clears all the current attributes from the model.
-	 *
-	 * This does not touch the model's original attributes, and will
-	 * only clear fillable attributes, unless the model is unguarded.
-	 *
-	 * @return $this
-	 */
-	public function clear() {
-		$keys = $this->get_attribute_keys();
-
-		foreach ( $keys as $key ) {
-			try {
-				$this->set_attribute( $key, null );
-			} catch ( GuardedPropertyException $e ) {
-				// We won't clear out guarded attributes.
-			}
-		}
-
-		return $this;
-	}
-
-	/**
-	 * Unguards the model.
-	 *
-	 * Sets the model to be unguarded, allowing the filling of
-	 * guarded attributes.
-	 */
-	public function unguard() {
-		$this->is_guarded = false;
-	}
-
-	/**
-	 * Reguards the model.
-	 *
-	 * Sets the model to be guarded, preventing filling of
-	 * guarded attributes.
-	 */
-	public function reguard() {
-		$this->is_guarded = true;
-	}
-
-	/**
-	 * Retrieves all the compute methods on the model.
-	 *
-	 * @return array
-	 */
-	protected function get_compute_methods() {
-		$methods = get_class_methods( get_called_class() );
-		$methods = array_filter( $methods, function ( $method ) {
-			return strrpos( $method, 'compute_', - strlen( $method ) ) !== false;
-		} );
-		$methods = array_map( function ( $method ) {
-			return substr( $method, strlen( 'compute_' ) );
-		}, $methods );
-
-		return $methods;
-	}
-
-	/**
-	 * Retrieves all the related methods on the model.
-	 *
-	 * @return array
-	 */
-	protected function get_related_methods() {
-		$methods = get_class_methods( get_called_class() );
-		$methods = array_filter( $methods, function ( $method ) {
-			return strrpos( $method, 'related_', - strlen( $method ) ) !== false;
-		} );
-		$methods = array_map( function ( $method ) {
-			return substr( $method, strlen( 'related_' ) );
-		}, $methods );
-
-		return $methods;
-	}
-
-	/**
-	 * Returns whether this relation has already been filled on the model.
-	 *
-	 * @param string $relation
-	 *
-	 * @return bool
-	 */
-	public function relation_is_filled( $relation ) {
-		$sha = $this
-			->{$this->has_related_method( $relation )}()
-			->get_sha();
-
-		return isset( $this->related[ $sha ] );
-	}
-
-	/**
-	 * Returns whether the Model is currently having
-	 * its relationships filled.
-	 *
-	 * @return bool
-	 */
-	public function is_filling() {
-		return $this->filling;
-	}
-
-	/**
-	 * Sets whether the Model is having its relationships filled.
-	 *
-	 * @param bool $is_filling
-	 */
-	public function set_filling( $is_filling ) {
-		$this->filling = $is_filling;
-	}
-
-	/**
-	 * Returns a HasMany relationship for the model.
-	 *
-	 * @param string $class
-	 * @param string $type
-	 * @param string $foreign_key
-	 *
-	 * @return HasMany
-	 */
-	protected function has_many( $class, $type, $foreign_key ) {
-		return new HasMany( $this, $class, $type, $foreign_key );
-	}
-
-	/**
-	 * Returns a BelongsToOne relationship for the model.
-	 *
-	 * @param string $class
-	 * @param string $type
-	 * @param string $local_key
-	 *
-	 * @return HasMany
-	 */
-	protected function belongs_to_one( $class, $type, $local_key = '' ) {
-		return new BelongsToOne( $this, $class, $type, $local_key );
-	}
-
-	/**
-	 * Sets up the memo array for the creating model.
-	 */
-	private function maybe_boot() {
-		if ( ! isset( self::$memo[ get_called_class() ] ) ) {
-			self::$memo[ get_called_class() ] = array();
-		}
-	}
-
-	/**
-	 * Whether this Model uses an underlying WordPress object.
-	 *
-	 * @return bool
-	 */
-	protected function uses_wp_object() {
-		return $this instanceof UsesWordPressPost ||
-			$this instanceof UsesWordPressTerm;
-	}
+    /**
+     * Memoized values for class methods.
+     *
+     * @var array
+     */
+    private static $memo = array();
+
+    /**
+     * Model attributes.
+     *
+     * @var array
+     */
+    private $attributes = array(
+        'table'  => array(),
+        'object' => null,
+    );
+
+    /**
+     * Model's original attributes.
+     *
+     * @var array
+     */
+    private $original = array(
+        'table'  => array(),
+        'object' => null,
+    );
+
+    /**
+     * Properties which are allowed to be set on the model.
+     *
+     * If this array is empty, any attributes can be set on the model.
+     *
+     * @var string[]
+     */
+    protected $fillable = array();
+
+    /**
+     * Properties which cannot be automatically filled on the model.
+     *
+     * If the model is unguarded, these properties can be filled.
+     *
+     * @var array
+     */
+    protected $guarded = array();
+
+    /**
+     * Properties which should not be serialized.
+     *
+     * @var array
+     */
+    protected $hidden = array();
+
+    /**
+     * Properties which should be serialized.
+     *
+     * @var array
+     */
+    protected $visible = array();
+
+    /**
+     * Relations saved on the Model.
+     *
+     * @var array
+     */
+    protected $related = array();
+
+    /**
+     * Whether the model's properties are guarded.
+     *
+     * When false, allows guarded properties to be filled.
+     *
+     * @var bool
+     */
+    protected $is_guarded = true;
+
+    /**
+     * Whether the Model is having its relations filled.
+     *
+     * @var bool
+     */
+    protected $filling = false;
+
+    /**
+     * Constructs a new model with provided attributes.
+     *
+     * If 'object' is passed as one of the attributes, the underlying post
+     * will be overwritten.
+     *
+     * @param array <string, mixed> $attributes
+     */
+    public function __construct( array $attributes = array() ) {
+        $this->maybe_boot();
+        $this->sync_original();
+
+        if ( $this->uses_wp_object() ) {
+            $this->create_wp_object();
+        }
+
+        $this->refresh( $attributes );
+    }
+
+    /**
+     * Refreshes the model's current attributes with the provided array.
+     *
+     * The model's attributes will match what was provided in the array,
+     * and any attributes not passed
+     *
+     * @param array $attributes
+     */
+    public function refresh( array $attributes ) {
+        $this->clear();
+
+        foreach ( $attributes as $name => $value ) {
+            $this->set_attribute( $name, $value );
+        }
+    }
+
+    /**
+     * Get the model's table attributes.
+     *
+     * Returns the array of for the model that will either need to be
+     * saved in postmeta or a separate table.
+     *
+     * @return array
+     */
+    public function get_table_attributes() {
+        return $this->attributes['table'];
+    }
+
+    /**
+     * Get the model's original attributes.
+     *
+     * @return array
+     */
+    public function get_original_table_attributes() {
+        return $this->original['table'];
+    }
+
+    /**
+     * Retrieve an array of the attributes on the model
+     * that have changed compared to the model's
+     * original data.
+     *
+     * @return array
+     */
+    public function get_changed_table_attributes() {
+        $changed = array();
+
+        foreach ( $this->get_table_attributes() as $attribute ) {
+            if ( $this->get_attribute( $attribute ) !==
+                    $this->get_original_attribute( $attribute )
+            ) {
+                $changed[ $attribute ] = $this->get_attribute( $attribute );
+            }
+        }
+
+        return $changed;
+    }
+
+    /**
+     * Get the model's underlying post.
+     *
+     * Returns the underlying WP_Post object for the model, representing
+     * the data that will be save in the wp_posts table.
+     *
+     * @return false|WP_Post|WP_Term
+     */
+    public function get_underlying_wp_object() {
+        if ( isset( $this->attributes['object'] ) ) {
+            return $this->attributes['object'];
+        }
+
+        return false;
+    }
+
+    /**
+     * Get the model's original underlying post.
+     *
+     * @return WP_Post
+     */
+    public function get_original_underlying_wp_object() {
+        return $this->original['object'];
+    }
+
+    /**
+     * Get the model attributes on the WordPress object
+     * that have changed compared to the model's
+     * original attributes.
+     *
+     * @return array
+     */
+    public function get_changed_wp_object_attributes() {
+        $changed = array();
+
+        foreach ( $this->get_wp_object_keys() as $key ) {
+            if ( $this->get_attribute( $key ) !==
+                    $this->get_original_attribute( $key )
+            ) {
+                $changed[ $key ] = $this->get_attribute( $key );
+            }
+        }
+
+        return $changed;
+    }
+
+    /**
+     * Magic __set method.
+     *
+     * Passes the name and value to set_attribute, which is where the magic happens.
+     *
+     * @param string $name
+     * @param mixed  $value
+     */
+    public function __set( $name, $value ) {
+        $this->set_attribute( $name, $value );
+    }
+
+    /**
+     * Sets the model attributes.
+     *
+     * Checks whether the model attribute can be set, check if it
+     * maps to the WP_Post property, otherwise, assigns it to the
+     * table attribute array.
+     *
+     * @param string $name
+     * @param mixed  $value
+     *
+     * @return $this
+     *
+     * @throws GuardedPropertyException
+     */
+    public function set_attribute( $name, $value ) {
+        if ( 'object' === $name ) {
+            return $this->override_wp_object( $value );
+        }
+
+        if ( ! $this->is_fillable( $name ) ) {
+            throw new GuardedPropertyException;
+        }
+
+        if ( $method = $this->has_map_method( $name ) ) {
+            $this->attributes['object']->{$this->{$method}()} = $value;
+        } else {
+            $this->attributes['table'][ $name ] = $value;
+        }
+
+        return $this;
+    }
+
+    /**
+     * Retrieves all the attribute keys for the model.
+     *
+     * @return array
+     */
+    public function get_attribute_keys() {
+        if ( isset( self::$memo[ get_called_class() ][ __METHOD__ ] ) ) {
+            return self::$memo[ get_called_class() ][ __METHOD__ ];
+        }
+
+        return self::$memo[ get_called_class() ][ __METHOD__ ]
+            = array_merge(
+                $this->fillable,
+                $this->guarded,
+                $this->get_compute_methods(),
+                $this->get_related_methods()
+            );
+    }
+
+    /**
+     * Retrieves the attribute keys that aren't mapped to a post.
+     *
+     * @return array
+     */
+    public function get_table_keys() {
+        if ( isset( self::$memo[ get_called_class() ][ __METHOD__ ] ) ) {
+            return self::$memo[ get_called_class() ][ __METHOD__ ];
+        }
+
+        $keys = array();
+
+        foreach ( $this->get_attribute_keys() as $key ) {
+            if ( ! $this->has_map_method( $key ) &&
+                 ! $this->has_compute_method( $key ) &&
+                 ! $this->has_related_method( $key )
+            ) {
+                $keys[] = $key;
+            }
+        }
+
+        return self::$memo[ get_called_class() ][ __METHOD__ ] = $keys;
+    }
+
+    /**
+     * Retrieves the attribute keys that are mapped to a post.
+     *
+     * @return array
+     */
+    public function get_wp_object_keys() {
+        if ( isset( self::$memo[ get_called_class() ][ __METHOD__ ] ) ) {
+            return self::$memo[ get_called_class() ][ __METHOD__ ];
+        }
+
+        $keys = array();
+
+        foreach ( $this->get_attribute_keys() as $key ) {
+            if ( $this->has_map_method( $key ) ) {
+                $keys[] = $key;
+            }
+        }
+
+        return self::$memo[ get_called_class() ][ __METHOD__ ] = $keys;
+    }
+
+    /**
+     * Returns the model's keys that are computed at call time.
+     *
+     * @return array
+     */
+    public function get_computed_keys() {
+        if ( isset( self::$memo[ get_called_class() ][ __METHOD__ ] ) ) {
+            return self::$memo[ get_called_class() ][ __METHOD__ ];
+        }
+
+        $keys = array();
+
+        foreach ( $this->get_attribute_keys() as $key ) {
+            if ( $this->has_compute_method( $key ) ) {
+                $keys[] = $key;
+            }
+        }
+
+        return self::$memo[ get_called_class() ][ __METHOD__ ] = $keys;
+    }
+
+    /**
+     * Returns the model's keys that are related to other Models.
+     *
+     * @return array
+     */
+    public function get_related_keys() {
+        if ( isset( self::$memo[ get_called_class() ][ __METHOD__ ] ) ) {
+            return self::$memo[ get_called_class() ][ __METHOD__ ];
+        }
+
+        $keys = array();
+
+        foreach ( $this->get_attribute_keys() as $key ) {
+            if ( $this->has_related_method( $key ) ) {
+                $keys[] = $key;
+            }
+        }
+
+        return self::$memo[ get_called_class() ][ __METHOD__ ] = $keys;
+    }
+
+    /**
+     * Serializes the model's public data into an array.
+     *
+     * @return array
+     */
+    public function serialize() {
+        $attributes = array();
+
+        if ( $this->visible ) {
+            // If visible attributes are set, we'll only reveal those.
+            foreach ( $this->visible as $key ) {
+                $attributes[ $key ] = $this->get_attribute( $key );
+            }
+        } elseif ( $this->hidden ) {
+            // If hidden attributes are set, we'll grab everything and hide those.
+            foreach ( $this->get_attribute_keys() as $key ) {
+                if ( ! in_array( $key, $this->hidden ) ) {
+                    $attributes[ $key ] = $this->get_attribute( $key );
+                }
+            }
+        } else {
+            // If nothing is hidden/visible, we'll grab and reveal everything.
+            foreach ( $this->get_attribute_keys() as $key ) {
+                $attributes[ $key ] = $this->get_attribute( $key );
+            }
+        }
+
+        return array_map( function ( $attribute ) {
+            if ( $attribute instanceof Serializes ) {
+                return $attribute->serialize();
+            }
+
+            return $attribute;
+        }, $attributes );
+    }
+
+    /**
+     * Syncs the current attributes to the model's original.
+     *
+     * @return $this
+     */
+    public function sync_original() {
+        $this->original = $this->attributes;
+
+        if ( $this->attributes['object'] ) {
+            $this->original['object'] = clone $this->attributes['object'];
+        }
+
+        return $this;
+    }
+
+    /**
+     * Checks if a given attribute is mass-fillable.
+     *
+     * Returns true if the attribute can be filled, false if it can't.
+     *
+     * @param string $name
+     *
+     * @return bool
+     */
+    private function is_fillable( $name ) {
+        // If this model isn't guarded, everything is fillable.
+        if ( ! $this->is_guarded ) {
+            return true;
+        }
+
+        // If it's in the fillable array, then it's fillable.
+        if ( in_array( $name, $this->fillable ) ) {
+            return true;
+        }
+
+        // If it's explicitly guarded, then it's not fillable.
+        if ( in_array( $name, $this->guarded ) ) {
+            return false;
+        }
+
+        // If fillable hasn't been defined, then everything else fillable.
+        return ! $this->fillable;
+    }
+
+    /**
+     * Overrides the current WP_Post with a provided one.
+     *
+     * Resets the post's default values and stores it in the attributes.
+     *
+     * @param WP_Post $value
+     *
+     * @return $this
+     */
+    private function override_wp_object( $value ) {
+        $this->attributes['object'] = $this->set_wp_object_constants( $value );
+
+        return $this;
+    }
+
+    /**
+     * Create and set with a new blank post.
+     *
+     * Creates a new WP_Post object, assigns it the default attributes,
+     * and stores it in the attributes.
+     *
+     * @throws LogicException
+     */
+    private function create_wp_object() {
+        switch ( true ) {
+            case $this instanceof UsesWordPressPost:
+                $object = new WP_Post( (object) array() );
+                break;
+            case $this instanceof UsesWordPressTerm:
+                $object = new WP_Term( (object) array() );
+                break;
+            default:
+                throw new LogicException;
+                break;
+        }
+
+        $this->attributes['object'] = $this->set_wp_object_constants( $object );
+    }
+
+    /**
+     * Enforces values on the post that can't change.
+     *
+     * Primarily, this is used to make sure the post_type always maps
+     * to the model's "$type" property, but this can all be overridden
+     * by the developer to enforce other values in the model.
+     *
+     * @param object $object
+     *
+     * @return object
+     */
+    protected function set_wp_object_constants( $object ) {
+        if ( $this instanceof UsesWordPressPost ) {
+            $object->post_type = $this::get_post_type();
+        }
+
+        if ( $this instanceof UsesWordPressTerm ) {
+            $object->taxonomy = $this::get_taxonomy();
+        }
+
+        return $object;
+    }
+
+    /**
+     * Magic __get method.
+     *
+     * Passes the name and value to get_attribute, which is where the magic happens.
+     *
+     * @param string $name
+     *
+     * @return mixed
+     */
+    public function __get( $name ) {
+        return $this->get_attribute( $name );
+    }
+
+    /**
+     * Retrieves the model attribute.
+     *
+     * @param string $name
+     *
+     * @return mixed
+     *
+     * @throws PropertyDoesNotExistException If property isn't found.
+     */
+    public function get_attribute( $name ) {
+        if ( $method = $this->has_map_method( $name ) ) {
+            $value = $this->attributes['object']->{$this->{$method}()};
+        } elseif ( $method = $this->has_compute_method( $name ) ) {
+            $value = $this->{$method}();
+        } else if ( $method = $this->has_related_method( $name ) ) {
+            $value = $this->get_related( $this->{$method}()->get_sha() );
+        } else {
+            if ( ! isset( $this->attributes['table'][ $name ] ) ) {
+                throw new PropertyDoesNotExistException;
+            }
+
+            $value = $this->attributes['table'][ $name ];
+        }
+
+        return $value;
+    }
+
+    /**
+     * Retrieve the model's original attribute value.
+     *
+     * @param string $name
+     *
+     * @return mixed
+     *
+     * @throws PropertyDoesNotExistException If property isn't found.
+     */
+    public function get_original_attribute( $name ) {
+        $original = new static( $this->original );
+
+        return $original->get_attribute( $name );
+    }
+
+    /**
+     * Fetches the Model's primary ID, depending on the model
+     * implementation.
+     *
+     * @return int
+     *
+     * @throws LogicException
+     */
+    public function get_primary_id() {
+        if ( $this instanceof UsesWordPressPost ) {
+            return $this->get_underlying_wp_object()->ID;
+        }
+
+        if ( $this instanceof UsesWordPressTerm ) {
+            return $this->get_underlying_wp_object()->term_id;
+        }
+
+        // Model w/o wp_object not yet supported.
+        throw new LogicException;
+    }
+
+    /**
+     * Generates the table foreign key, depending on the model
+     * implementation.
+     *
+     * @return string
+     *
+     * @throws LogicException
+     */
+    public function get_foreign_key() {
+        if ( $this instanceof UsesWordPressPost ) {
+            return 'post_id';
+        }
+
+        // Model w/o wp_object not yet supported.
+        throw new LogicException;
+    }
+
+    /**
+     * Gets the related Model or Collection for the given sha.
+     *
+     * @param string $sha
+     *
+     * @return Model|Collection
+     */
+    public function get_related( $sha ) {
+        return $this->related[ $sha ];
+    }
+
+    /**
+     * Sets the related Model or Collection for the given sha.
+     *
+     * @param string           $sha
+     * @param Model|Collection $target
+     *
+     * @throws RuntimeException
+     */
+    public function set_related( $sha, $target ) {
+        if ( ! ( $target instanceof Model ) && ! ( $target instanceof Collection ) ) {
+            throw new RuntimeException;
+        }
+
+        $this->related[ $sha ] = $target;
+    }
+
+    /**
+     * Checks whether the attribute has a map method.
+     *
+     * This is used to determine whether the attribute maps to a
+     * property on the underlying WP_Post object. Returns the
+     * method if one exists, returns false if it doesn't.
+     *
+     * @param string $name
+     *
+     * @return false|string
+     */
+    protected function has_map_method( $name ) {
+        if ( method_exists( $this, $method = "map_{$name}" ) ) {
+            return $method;
+        }
+
+        return false;
+    }
+
+    /**
+     * Checks whether the attribute has a compute method.
+     *
+     * This is used to determine if the attribute should be computed
+     * from other attributes.
+     *
+     * @param string $name
+     *
+     * @return false|string
+     */
+    protected function has_compute_method( $name ) {
+        if ( method_exists( $this, $method = "compute_{$name}" ) ) {
+            return $method;
+        }
+
+        return false;
+    }
+
+    /**
+     * Checks whether the attribute has a compute method.
+     *
+     * This is used to determine if the attribute should be computed
+     * from other attributes.
+     *
+     * @param string $name
+     *
+     * @return false|string
+     */
+    protected function has_related_method( $name ) {
+        if ( method_exists( $this, $method = "related_{$name}" ) ) {
+            return $method;
+        }
+
+        return false;
+    }
+
+    /**
+     * Clears all the current attributes from the model.
+     *
+     * This does not touch the model's original attributes, and will
+     * only clear fillable attributes, unless the model is unguarded.
+     *
+     * @return $this
+     */
+    public function clear() {
+        $keys = $this->get_attribute_keys();
+
+        foreach ( $keys as $key ) {
+            try {
+                $this->set_attribute( $key, null );
+            } catch ( GuardedPropertyException $e ) {
+                // We won't clear out guarded attributes.
+            }
+        }
+
+        return $this;
+    }
+
+    /**
+     * Unguards the model.
+     *
+     * Sets the model to be unguarded, allowing the filling of
+     * guarded attributes.
+     */
+    public function unguard() {
+        $this->is_guarded = false;
+    }
+
+    /**
+     * Reguards the model.
+     *
+     * Sets the model to be guarded, preventing filling of
+     * guarded attributes.
+     */
+    public function reguard() {
+        $this->is_guarded = true;
+    }
+
+    /**
+     * Retrieves all the compute methods on the model.
+     *
+     * @return array
+     */
+    protected function get_compute_methods() {
+        $methods = get_class_methods( get_called_class() );
+        $methods = array_filter( $methods, function ( $method ) {
+            return strrpos( $method, 'compute_', - strlen( $method ) ) !== false;
+        } );
+        $methods = array_map( function ( $method ) {
+            return substr( $method, strlen( 'compute_' ) );
+        }, $methods );
+
+        return $methods;
+    }
+
+    /**
+     * Retrieves all the related methods on the model.
+     *
+     * @return array
+     */
+    protected function get_related_methods() {
+        $methods = get_class_methods( get_called_class() );
+        $methods = array_filter( $methods, function ( $method ) {
+            return strrpos( $method, 'related_', - strlen( $method ) ) !== false;
+        } );
+        $methods = array_map( function ( $method ) {
+            return substr( $method, strlen( 'related_' ) );
+        }, $methods );
+
+        return $methods;
+    }
+
+    /**
+     * Returns whether this relation has already been filled on the model.
+     *
+     * @param string $relation
+     *
+     * @return bool
+     */
+    public function relation_is_filled( $relation ) {
+        $sha = $this
+            ->{$this->has_related_method( $relation )}()
+            ->get_sha();
+
+        return isset( $this->related[ $sha ] );
+    }
+
+    /**
+     * Returns whether the Model is currently having
+     * its relationships filled.
+     *
+     * @return bool
+     */
+    public function is_filling() {
+        return $this->filling;
+    }
+
+    /**
+     * Sets whether the Model is having its relationships filled.
+     *
+     * @param bool $is_filling
+     */
+    public function set_filling( $is_filling ) {
+        $this->filling = $is_filling;
+    }
+
+    /**
+     * Returns a HasMany relationship for the model.
+     *
+     * @param string $class
+     * @param string $type
+     * @param string $foreign_key
+     *
+     * @return HasMany
+     */
+    protected function has_many( $class, $type, $foreign_key ) {
+        return new HasMany( $this, $class, $type, $foreign_key );
+    }
+
+    /**
+     * Returns a BelongsToOne relationship for the model.
+     *
+     * @param string $class
+     * @param string $type
+     * @param string $local_key
+     *
+     * @return HasMany
+     */
+    protected function belongs_to_one( $class, $type, $local_key = '' ) {
+        return new BelongsToOne( $this, $class, $type, $local_key );
+    }
+
+    /**
+     * Sets up the memo array for the creating model.
+     */
+    private function maybe_boot() {
+        if ( ! isset( self::$memo[ get_called_class() ] ) ) {
+            self::$memo[ get_called_class() ] = array();
+        }
+    }
+
+    /**
+     * Whether this Model uses an underlying WordPress object.
+     *
+     * @return bool
+     */
+    protected function uses_wp_object() {
+        return $this instanceof UsesWordPressPost ||
+            $this instanceof UsesWordPressTerm;
+    }
 }

src/Axolotl/Collection.php

Indentation +148 added lines, -148 removed lines

@@ -14,152 +14,152 @@
  * @subpackage Axolotl
  */
 class Collection implements Countable, Iterator, Serializes {
-	/**
-	 * Collection elements.
-	 *
-	 * @var array
-	 */
-	protected $elements = array();
-
-	/**
-	 * Models registered to the collection.
-	 *
-	 * @var string
-	 */
-	protected $model;
-
-	/**
-	 * Where Collection is in loop.
-	 *
-	 * @var int
-	 */
-	protected $position = 0;
-
-	/**
-	 * Collection constructor.
-	 *
-	 * @param array $elements
-	 * @param array $config
-	 */
-	public function __construct( array $elements = array(), array $config = array() ) {
-		$this->parse_config( $config );
-
-		foreach ( $elements as $element ) {
-			$this->add( $element );
-		}
-	}
-
-	/**
-	 * Adds a new element to the Collection.
-	 *
-	 * @param mixed $element
-	 *
-	 * @throws RuntimeException
-	 */
-	public function add( $element ) {
-		if ( $this->model && is_array( $element ) ) {
-			$element = new $this->model( $element );
-		}
-
-		if ( $this->model && ! ( $element instanceof $this->model ) ) {
-			throw new RuntimeException;
-		}
-
-		$this->elements[] = $element;
-	}
-
-	/**
-	 * Fetches the element at the provided index.
-	 *
-	 * @param int $index
-	 *
-	 * @return mixed
-	 */
-	public function at( $index ) {
-		return $this->elements[ $index ];
-	}
-
-	/**
-	 * Return the current element.
-	 *
-	 * @return mixed
-	 */
-	public function current() {
-		return $this->at( $this->position );
-	}
-
-	/**
-	 * Move forward to next element.
-	 */
-	public function next() {
-		$this->position ++;
-	}
-
-	/**
-	 * Return the key of the current element.
-	 *
-	 * @return mixed
-	 */
-	public function key() {
-		return $this->position;
-	}
-
-	/**
-	 * Checks if current position is valid.
-	 *
-	 * @return bool
-	 */
-	public function valid() {
-		return isset( $this->elements[ $this->position ] );
-	}
-
-	/**
-	 * Rewind the Iterator to the first element.
-	 */
-	public function rewind() {
-		$this->position = 0;
-	}
-
-	/**
-	 * Count elements of an object.
-	 *
-	 * @return int
-	 */
-	public function count() {
-		return count( $this->elements );
-	}
-
-	/**
-	 * Parses the Collection config to set its properties.
-	 *
-	 * @param array $config
-	 *
-	 * @throws LogicException
-	 */
-	protected function parse_config( array $config ) {
-		if ( isset( $config['model'] ) ) {
-			$model = $config['model'];
-
-			if ( ! is_subclass_of( $model, 'Intraxia\Jaxion\Axolotl\Model' ) ) {
-				throw new LogicException;
-			}
-
-			$this->model = $model;
-		}
-	}
-
-	/**
-	 * {@inheritDoc}
-	 *
-	 * @return array
-	 */
-	public function serialize() {
-		return array_map(function( $element ) {
-			if ( $element instanceof Serializes ) {
-				return $element->serialize();
-			}
-
-			return $element;
-		}, $this->elements);
-	}
+    /**
+     * Collection elements.
+     *
+     * @var array
+     */
+    protected $elements = array();
+
+    /**
+     * Models registered to the collection.
+     *
+     * @var string
+     */
+    protected $model;
+
+    /**
+     * Where Collection is in loop.
+     *
+     * @var int
+     */
+    protected $position = 0;
+
+    /**
+     * Collection constructor.
+     *
+     * @param array $elements
+     * @param array $config
+     */
+    public function __construct( array $elements = array(), array $config = array() ) {
+        $this->parse_config( $config );
+
+        foreach ( $elements as $element ) {
+            $this->add( $element );
+        }
+    }
+
+    /**
+     * Adds a new element to the Collection.
+     *
+     * @param mixed $element
+     *
+     * @throws RuntimeException
+     */
+    public function add( $element ) {
+        if ( $this->model && is_array( $element ) ) {
+            $element = new $this->model( $element );
+        }
+
+        if ( $this->model && ! ( $element instanceof $this->model ) ) {
+            throw new RuntimeException;
+        }
+
+        $this->elements[] = $element;
+    }
+
+    /**
+     * Fetches the element at the provided index.
+     *
+     * @param int $index
+     *
+     * @return mixed
+     */
+    public function at( $index ) {
+        return $this->elements[ $index ];
+    }
+
+    /**
+     * Return the current element.
+     *
+     * @return mixed
+     */
+    public function current() {
+        return $this->at( $this->position );
+    }
+
+    /**
+     * Move forward to next element.
+     */
+    public function next() {
+        $this->position ++;
+    }
+
+    /**
+     * Return the key of the current element.
+     *
+     * @return mixed
+     */
+    public function key() {
+        return $this->position;
+    }
+
+    /**
+     * Checks if current position is valid.
+     *
+     * @return bool
+     */
+    public function valid() {
+        return isset( $this->elements[ $this->position ] );
+    }
+
+    /**
+     * Rewind the Iterator to the first element.
+     */
+    public function rewind() {
+        $this->position = 0;
+    }
+
+    /**
+     * Count elements of an object.
+     *
+     * @return int
+     */
+    public function count() {
+        return count( $this->elements );
+    }
+
+    /**
+     * Parses the Collection config to set its properties.
+     *
+     * @param array $config
+     *
+     * @throws LogicException
+     */
+    protected function parse_config( array $config ) {
+        if ( isset( $config['model'] ) ) {
+            $model = $config['model'];
+
+            if ( ! is_subclass_of( $model, 'Intraxia\Jaxion\Axolotl\Model' ) ) {
+                throw new LogicException;
+            }
+
+            $this->model = $model;
+        }
+    }
+
+    /**
+     * {@inheritDoc}
+     *
+     * @return array
+     */
+    public function serialize() {
+        return array_map(function( $element ) {
+            if ( $element instanceof Serializes ) {
+                return $element->serialize();
+            }
+
+            return $element;
+        }, $this->elements);
+    }
 }

src/Axolotl/ServiceProvider.php

Indentation +13 added lines, -13 removed lines

@@ -12,17 +12,17 @@
  * @subpackage Axolotl
  */
 class ServiceProvider implements ServiceProviderContract {
-	/**
-	 * {@inheritdoc}
-	 *
-	 * @param Container $container
-	 */
-	public function register( Container $container ) {
-		$container->define(
-			array( 'database' => 'Intraxia\Jaxion\Contract\Axolotl\EntityManager' ),
-			function ( $app ) {
-				return new EntityManager( new WP_Query, $app->fetch( 'slug' ) );
-			}
-		);
-	}
+    /**
+     * {@inheritdoc}
+     *
+     * @param Container $container
+     */
+    public function register( Container $container ) {
+        $container->define(
+            array( 'database' => 'Intraxia\Jaxion\Contract\Axolotl\EntityManager' ),
+            function ( $app ) {
+                return new EntityManager( new WP_Query, $app->fetch( 'slug' ) );
+            }
+        );
+    }
 }

src/Utility/Str.php

Indentation +32 added lines, -32 removed lines

@@ -11,39 +11,39 @@
  * @subpackage Utility
  */
 class Str {
-	/**
-	 * Determine if a given string starts with a given substring.
-	 *
-	 * @param  string       $haystack
-	 * @param  string|array $needles
-	 *
-	 * @return bool
-	 */
-	public static function starts_with( $haystack, $needles ) {
-		foreach ( (array) $needles as $needle ) {
-			if ( '' !== $needle && 0 === strpos( $haystack, $needle ) ) {
-				return true;
-			}
-		}
+    /**
+     * Determine if a given string starts with a given substring.
+     *
+     * @param  string       $haystack
+     * @param  string|array $needles
+     *
+     * @return bool
+     */
+    public static function starts_with( $haystack, $needles ) {
+        foreach ( (array) $needles as $needle ) {
+            if ( '' !== $needle && 0 === strpos( $haystack, $needle ) ) {
+                return true;
+            }
+        }
 
-		return false;
-	}
+        return false;
+    }
 
-	/**
-	 * Determine if a given string ends with a given substring.
-	 *
-	 * @param  string       $haystack
-	 * @param  string|array $needles
-	 *
-	 * @return bool
-	 */
-	public static function ends_with( $haystack, $needles ) {
-		foreach ( (array) $needles as $needle ) {
-			if ( substr( $haystack, - strlen( $needle ) ) === (string) $needle ) {
-				return true;
-			}
-		}
+    /**
+     * Determine if a given string ends with a given substring.
+     *
+     * @param  string       $haystack
+     * @param  string|array $needles
+     *
+     * @return bool
+     */
+    public static function ends_with( $haystack, $needles ) {
+        foreach ( (array) $needles as $needle ) {
+            if ( substr( $haystack, - strlen( $needle ) ) === (string) $needle ) {
+                return true;
+            }
+        }
 
-		return false;
-	}
+        return false;
+    }
 }