AR WIP

framework/db/ar/ActiveQuery.php

@@ -95,7 +95,9 @@ class ActiveQuery extends BaseQuery
 			$class = $this->modelClass;
 			$model = $class::create($row);
 			if (!empty($this->with)) {
-				$this->fetchRelatedModels(array($model), $this->with);
+				$models = array($model);
+				$this->fetchRelatedModels($models, $this->with);
+				$model = $models[0];
 			}
 			return $model;
 		}

framework/db/ar/ActiveRelation.php

@@ -39,9 +39,25 @@ class ActiveRelation extends ActiveQuery
 	 */
 	public $link;
 	/**
-	 * @var ActiveRelation
+	 * @var array
 	 */
 	public $via;
+	/**
+	 * @var array
+	 */
+	public $viaTable;
+
+	public function via($modelClass, $properties = array())
+	{
+		$this->via = array($modelClass, $properties);
+		return $this;
+	}
+
+	public function viaTable($tableName, $link, $properties = array())
+	{
+		$this->viaTable = array($tableName, $link, $properties);
+		return $this;
+	}
 
 	public function createCommand($db = null)
 	{
@@ -53,9 +69,13 @@ class ActiveRelation extends ActiveQuery
 
 	public function findWith($name, &$primaryModels)
 	{
-		if (empty($this->link) || !is_array($this->link)) {
+		if (!is_array($this->link)) {
 			throw new \yii\base\Exception('invalid link');
 		}
+
+		if ($this->via !== null) {
+		}
+
 		$this->filterByPrimaryModels($primaryModels);
 
 		if (count($primaryModels) === 1 && !$this->multiple) {
@@ -64,7 +84,12 @@ class ActiveRelation extends ActiveQuery
 			}
 		} else {
 			$models = $this->all();
-			// distribute models into buckets which are indexed by the link keys
+			$this->bindModels($name, $primaryModels, $models);
+		}
+	}
+
+	protected function bindModels($name, &$primaryModels, $models)
+	{
 		$buckets = array();
 		foreach ($models as $i => $model) {
 			$key = $this->getModelKey($model, array_keys($this->link));
@@ -88,7 +113,6 @@ class ActiveRelation extends ActiveQuery
 			}
 		}
 	}
-	}
 
 	protected function getModelKey($model, $attributes)
 	{

tests/unit/data/ar/Order.php

@@ -29,7 +29,7 @@ class Order extends ActiveRecord
 	public function books()
 	{
 		return $this->hasMany('Item', array('id' => 'item_id'))
-			->via('tbl_order_item', array('order_id' => 'id'))
+			->viaTable('tbl_order_item', array('order_id' => 'id'))
 			->where(array('category_id' => 1));
 	}
 }