FixtureController.php 8.85 KB
Newer Older
Mark committed
1 2 3 4 5 6 7 8 9 10 11
<?php
/**
 * @link http://www.yiiframework.com/
 * @copyright Copyright (c) 2008 Yii Software LLC
 * @license http://www.yiiframework.com/license/
 */

namespace yii\console\controllers;

use Yii;
use yii\console\Controller;
12
use yii\console\Exception;
Mark committed
13
use yii\helpers\FileHelper;
Mark committed
14
use yii\helpers\Console;
Mark committed
15 16 17 18 19

/**
 * This command manages fixtures load to the database tables.
 * You can specify different options of this command to point fixture manager
 * to the specific tables of the different database connections.
Qiang Xue committed
20
 *
Mark committed
21
 * To use this command simply configure your console.php config like this:
Qiang Xue committed
22
 *
Mark committed
23
 * ~~~
Qiang Xue committed
24 25 26 27 28 29 30 31 32 33
 * 'db' => [
 *     'class' => 'yii\db\Connection',
 *     'dsn' => 'mysql:host=localhost;dbname={your_database}',
 *     'username' => '{your_db_user}',
 *     'password' => '',
 *     'charset' => 'utf8',
 * ],
 * 'fixture' => [
 *     'class' => 'yii\test\DbFixtureManager',
 * ],
Mark committed
34
 * ~~~
Qiang Xue committed
35
 *
Mark committed
36
 * ~~~
Qiang Xue committed
37 38 39
 * #load fixtures under $fixturePath to the "users" table
 * yii fixture/apply users
 *
Mark committed
40
 * #also a short version of this command (generate action is default)
Qiang Xue committed
41 42 43
 * yii fixture users
 *
 * #load fixtures under $fixturePath to the "users" table to the different connection
Qiang Xue committed
44
 * yii fixture/apply users --db=someOtherDbConneciton
Qiang Xue committed
45 46
 *
 * #load fixtures under different $fixturePath to the "users" table.
Qiang Xue committed
47
 * yii fixture/apply users --fixturePath=@app/some/other/path/to/fixtures
Mark committed
48
 * ~~~
Qiang Xue committed
49
 *
Mark committed
50 51 52 53 54
 * @author Mark Jebri <mark.github@yandex.ru>
 * @since 2.0
 */
class FixtureController extends Controller
{
Qiang Xue committed
55
	use DbTestTrait;
Mark committed
56 57 58 59 60
	
	/**
	 * type of fixture apply to database
	 */
	const APPLY_ALL = 'all';
Mark committed
61 62 63 64 65 66 67 68 69

	/**
	 * @var string controller default action ID.
	 */
	public $defaultAction = 'apply';
	/**
	 * Alias to the path, where all fixtures are stored.
	 * @var string
	 */
Qiang Xue committed
70
	public $fixturePath = '@tests/unit/fixtures';
Mark committed
71 72
	/**
	 * Id of the database connection component of the application.
Qiang Xue committed
73
	 * @var string
Mark committed
74 75 76
	 */
	public $db = 'db';

Carsten Brandt committed
77

Mark committed
78 79 80 81 82 83 84
	/**
	 * Returns the names of the global options for this command.
	 * @return array the names of the global options for this command.
	 */
	public function globalOptions()
	{
		return array_merge(parent::globalOptions(), [
Qiang Xue committed
85
			'db', 'fixturePath'
Mark committed
86 87 88 89 90 91
		]);
	}

	/**
	 * This method is invoked right before an action is to be executed (after all possible filters.)
	 * It checks that fixtures path and database connection are available.
Qiang Xue committed
92
	 * @param \yii\base\Action $action
Mark committed
93 94 95 96 97 98 99 100 101 102 103 104 105
	 * @return boolean
	 */
	public function beforeAction($action)
	{
		if (parent::beforeAction($action)) {
			$this->checkRequirements();
			return true;
		} else {
			return false;
		}
	}

	/**
106 107 108
	 * Apply given fixture to the table. You can load several fixtures specifying
	 * their names separated with commas, like: tbl_user,tbl_profile. Be sure there is no
	 * whitespace between tables names.
Carsten Brandt committed
109 110
	 * @param array $fixtures
	 * @throws \yii\console\Exception
Mark committed
111
	 */
Mark committed
112
	public function actionApply(array $fixtures, array $except = [])
Mark committed
113
	{
114 115
		if ($this->getFixtureManager() === null) {
			throw new Exception('Fixture manager is not configured properly. Please refer to official documentation for this purposes.');
116 117
		}

Mark committed
118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133
		$foundFixtures = $this->findFixtures($fixtures);

		if (!$this->needToApplyAll($fixtures[0])) {
			$notFoundFixtures = array_diff($fixtures, $foundFixtures);

			if ($notFoundFixtures) {
				$this->notifyNotFound($notFoundFixtures);
			}
		}

		if (!$foundFixtures) {
			throw new Exception("No files were found by name: \"" . implode(', ', $fixtures) . "\".\n"
				. "Check that fixtures with these name exists, under fixtures path: \n\"" . Yii::getAlias($this->fixturePath) . "\"."
			);			
		}

Mark committed
134
		if (!$this->confirmApply($foundFixtures, $except)) {
Mark committed
135 136 137
			return;
		}

Mark committed
138
		$fixtures = array_diff($foundFixtures, $except);
Mark committed
139

Qiang Xue committed
140 141
		$this->getFixtureManager()->basePath = $this->fixturePath;
		$this->getFixtureManager()->db = $this->db;
Mark committed
142 143 144

		$transaction = Yii::$app->db->beginTransaction();

Mark committed
145
		try {
Mark committed
146 147 148
			$this->loadFixtures($foundFixtures);
			$transaction->commit();

Mark committed
149
		} catch (\Exception $e) {
Mark committed
150 151 152 153 154
			$transaction->rollback();
			$this->stdout("Exception occured, transaction rollback. Tables will be in same state.\n", Console::BG_RED);
			throw $e;
		}
		$this->notifySuccess($foundFixtures);
Mark committed
155 156 157
	}

	/**
158 159 160
	 * Truncate given table and clear all fixtures from it. You can clear several tables specifying
	 * their names separated with commas, like: tbl_user,tbl_profile. Be sure there is no
	 * whitespace between tables names.
Carsten Brandt committed
161
	 * @param array|string $tables
Mark committed
162
	 */
Mark committed
163
	public function actionClear(array $tables, array $except = ['tbl_migration'])
Mark committed
164 165 166 167 168
	{		
		if ($this->needToApplyAll($tables[0])) {
			$tables = $this->getDbConnection()->schema->getTableNames();
		}

Mark committed
169
		if (!$this->confirmClear($tables, $except)) {
Mark committed
170 171 172
			return;
		}

Mark committed
173 174
		$tables = array_diff($tables, $except);

Mark committed
175 176
		$transaction = Yii::$app->db->beginTransaction();

Mark committed
177
		try {
Mark committed
178 179 180
			$this->getDbConnection()->createCommand()->checkIntegrity(false)->execute();

			foreach($tables as $table) {
Mark committed
181
				$this->getDbConnection()->createCommand()->delete($table)->execute();
Mark committed
182 183 184 185 186 187 188
				$this->getDbConnection()->createCommand()->resetSequence($table)->execute();
				$this->stdout("    Table \"{$table}\" was successfully cleared. \n", Console::FG_GREEN);
			}

			$this->getDbConnection()->createCommand()->checkIntegrity(true)->execute();
			$transaction->commit();

Mark committed
189
		} catch (\Exception $e) {
Mark committed
190 191 192
			$transaction->rollback();
			$this->stdout("Exception occured, transaction rollback. Tables will be in same state.\n", Console::BG_RED);
			throw $e;
Mark committed
193
		}
Mark committed
194 195 196 197
	}

	/**
	 * Checks if the database and fixtures path are available.
198
	 * @throws Exception
Mark committed
199 200 201
	 */
	public function checkRequirements()
	{
Qiang Xue committed
202
		$path = Yii::getAlias($this->fixturePath, false);
Mark committed
203 204

		if (!is_dir($path) || !is_writable($path)) {
205
			throw new Exception("The fixtures path \"{$this->fixturePath}\" not exist or is not writable.");
Mark committed
206 207 208 209 210 211
		}

	}

	/**
	 * Returns database connection component
Qiang Xue committed
212
	 * @return \yii\db\Connection
213
	 * @throws Exception if [[db]] is invalid.
Mark committed
214 215 216 217 218
	 */
	public function getDbConnection()
	{
		$db = Yii::$app->getComponent($this->db);

Carsten Brandt committed
219
		if ($db === null) {
220
			throw new Exception("There is no database connection component with id \"{$this->db}\".");
Mark committed
221 222 223 224 225
		}

		return $db;
	}

226 227 228 229 230 231
	/**
	 * Notifies user that fixtures were successfully loaded.
	 * @param array $fixtures
	 */
	private function notifySuccess($fixtures)
	{
Carsten Brandt committed
232 233
		$this->stdout("Fixtures were successfully loaded from path:\n", Console::FG_YELLOW);
		$this->stdout(Yii::getAlias($this->fixturePath) . "\n\n", Console::FG_GREEN);
Mark committed
234 235
		$this->outputList($fixtures);
	}
236

Mark committed
237 238 239 240 241 242 243 244 245 246 247 248
	/**
	 * Notifies user that fixtures were not found under fixtures path.
	 * @param array $fixtures
	 */
	private function notifyNotFound($fixtures)
	{
		$this->stdout("Some fixtures were not found under path:\n", Console::BG_RED);
		$this->stdout(Yii::getAlias($this->fixturePath) . "\n\n", Console::FG_GREEN);
		$this->outputList($fixtures);
		$this->stdout("\n");
	}

Mark committed
249 250 251
	/**
	 * Prompts user with confirmation if fixtures should be loaded.
	 * @param array $fixtures
Mark committed
252
	 * @param array $except
Mark committed
253 254
	 * @return boolean
	 */
Mark committed
255
	private function confirmApply($fixtures, $except)
Mark committed
256 257
	{
		$this->stdout("Fixtures will be loaded from path: \n", Console::FG_YELLOW);
Carsten Brandt committed
258
		$this->stdout(Yii::getAlias($this->fixturePath) . "\n\n", Console::FG_GREEN);
Mark committed
259
		$this->outputList($fixtures);
Mark committed
260 261 262 263 264 265 266

		if (count($except)) {
			$this->stdout("\nFixtures that will NOT be loaded: \n\n", Console::FG_YELLOW);
			$this->outputList($except);
		}

		return $this->confirm("\nLoad to database above fixtures?");
Mark committed
267 268 269 270 271
	}

	/**
	 * Prompts user with confirmation for tables that should be cleared.
	 * @param array $tables
Mark committed
272
	 * @param array $except
Mark committed
273 274
	 * @return boolean
	 */
Mark committed
275
	private function confirmClear($tables, $except)
Mark committed
276
	{
Carsten Brandt committed
277
		$this->stdout("Tables below will be cleared:\n\n", Console::FG_YELLOW);
Mark committed
278
		$this->outputList($tables);
Mark committed
279 280 281 282 283 284 285

		if (count($except)) {
			$this->stdout("\nTables that will NOT be cleared:\n\n", Console::FG_YELLOW);
			$this->outputList($except);
		}

		return $this->confirm("\nClear tables?");
Mark committed
286 287 288 289 290 291 292 293 294
	}

	/**
	 * Outputs data to the console as a list.
	 * @param array $data
	 */
	private function outputList($data)
	{
		foreach($data as $index => $item) {
Mark committed
295
			$this->stdout("    " . ($index + 1) . ". {$item}\n", Console::FG_GREEN);
296 297
		}
	}
Mark committed
298 299 300 301 302 303 304 305 306 307 308 309 310

	/**
	 * Checks if needed to apply all fixtures.
	 * @param string $fixture
	 * @return bool
	 */
	public function needToApplyAll($fixture)
	{
		return $fixture == self::APPLY_ALL;
	}

	/**
	 * @param array $fixtures
Qiang Xue committed
311
	 * @return array Array of found fixtures. These may differ from input parameter as not all fixtures may exists.
Mark committed
312 313 314 315
	 */
	private function findFixtures(array $fixtures)
	{
		$fixturesPath = Yii::getAlias($this->fixturePath);
316

317
		$filesToSearch = ['*.php'];
318 319
		if (!$this->needToApplyAll($fixtures[0])) {
			$filesToSearch = [];
Mark committed
320 321 322 323
			foreach ($fixtures as $fileName) {
				$filesToSearch[] = $fileName . '.php';
			}
		}
Alexander Makarov committed
324 325

		$files = FileHelper::findFiles($fixturesPath, ['only' => $filesToSearch]);
Mark committed
326 327
		$foundFixtures = [];

Alexander Makarov committed
328
		foreach ($files as $fixture) {
Mark committed
329 330 331 332 333 334
			$foundFixtures[] = basename($fixture , '.php');
		}

		return $foundFixtures;
	}

Mark committed
335
}