<?php namespace explosivebit\arangodb\console\controllers; use explosivebit\arangodb\Connection; use explosivebit\arangodb\Exception; use explosivebit\arangodb\Migration; use explosivebit\arangodb\Query; use yii; use yii\console\controllers\BaseMigrateController; use yii\helpers\ArrayHelper; class MigrateController extends BaseMigrateController { /** * @var string the name of the collection for keeping applied migration information. */ public $migrationCollection = 'migration'; /** * @var string the directory storing the migration classes. This can be either * a path alias or a directory. */ public $migrationPath = '@app/migrations/arangodb'; /** * @inheritdoc */ public $templateFile = '@explosivebit/arangodb/views/migration.php'; /** * @var Connection|string the DB connection object or the application * component ID of the DB connection. */ public $db = 'arangodb'; /** * @inheritdoc */ public function options($actionId) { return array_merge( parent::options($actionId), ['migrationCollection', 'db'] // global for all actions ); } /** * This method is invoked right before an action is to be executed (after all possible filters.) * It checks the existence of the [[migrationPath]]. * @param yii\base\Action $action the action to be executed. * @throws Exception if db component isn't configured * @return boolean whether the action should continue to be executed. */ public function beforeAction($action) { if (parent::beforeAction($action)) { if ($action->id !== 'create') { if (is_string($this->db)) { $this->db = \Yii::$app->get($this->db); } if (!$this->db instanceof Connection) { throw new Exception("The 'db' option must refer to the application component ID of a ArangoDB connection."); } } return true; } else { return false; } } /** * Creates a new migration instance. * @param string $class the migration class name * @return Migration the migration instance */ protected function createMigration($class) { $file = $this->migrationPath . DIRECTORY_SEPARATOR . $class . '.php'; require_once($file); return new $class(['db' => $this->db]); } /** * @inheritdoc */ protected function getMigrationHistory($limit) { try { $history = $this->getHistory($limit); } catch (\Exception $ex) { if ($ex->getPrevious()->getServerCode() == 1203) { $this->createMigrationHistoryCollection(); $history = $this->getHistory($limit); } else { throw $ex; } } unset($history[self::BASE_MIGRATION]); return $history; } private function getHistory($limit) { $query = new Query; $rows = $query->select(['version' => 'version', 'apply_time' => 'apply_time']) ->from($this->migrationCollection) ->orderBy('version DESC') ->limit($limit) ->all($this->db); $history = ArrayHelper::map($rows, 'version', 'apply_time'); unset($history[self::BASE_MIGRATION]); return $history; } protected function createMigrationHistoryCollection() { echo "Creating migration history collection \"$this->migrationCollection\"..."; $this->db->getCollectionHandler()->create($this->migrationCollection); $this->db->getDocumentHandler()->save( $this->migrationCollection, [ 'version' => self::BASE_MIGRATION, 'apply_time' => time(), ] ); echo "done.\n"; } /** * @inheritdoc */ protected function addMigrationHistory($version) { $this->db->getDocumentHandler()->save( $this->migrationCollection, [ 'version' => $version, 'apply_time' => time(), ] ); } /** * @inheritdoc */ protected function removeMigrationHistory($version) { $this->db->getCollectionHandler()->removeByExample( $this->migrationCollection, [ 'version' => $version, ] ); } }