|
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468 |
- <?php
-
-
- namespace yii\faker;
-
- use Yii;
- use yii\console\Exception;
- use yii\helpers\Console;
- use yii\helpers\FileHelper;
- use yii\helpers\VarDumper;
-
-
- class FixtureController extends \yii\console\controllers\FixtureController
- {
-
-
- public $templatePath = '@tests/unit/templates/fixtures';
-
-
- public $fixtureDataPath = '@tests/unit/fixtures/data';
-
-
- public $language;
-
-
- public $count = 2;
-
-
- public $providers = [];
-
-
-
- private $_generator;
-
-
-
-
- public function options($actionID)
- {
- return array_merge(parent::options($actionID), [
- 'templatePath', 'language', 'fixtureDataPath', 'count'
- ]);
- }
-
- public function beforeAction($action)
- {
- if (parent::beforeAction($action)) {
- $this->checkPaths();
- $this->addProviders();
- return true;
- } else {
- return false;
- }
- }
-
-
-
- public function actionTemplates()
- {
- $foundTemplates = $this->findTemplatesFiles();
-
- if (!$foundTemplates) {
- $this->notifyNoTemplatesFound();
- } else {
- $this->notifyTemplatesCanBeGenerated($foundTemplates);
- }
- }
-
-
-
- public function actionGenerate()
- {
- $templatesInput = func_get_args();
-
- if (empty($templatesInput)) {
- throw new Exception('You should specify input fixtures template files');
- }
-
- $foundTemplates = $this->findTemplatesFiles($templatesInput);
-
- $notFoundTemplates = array_diff($templatesInput, $foundTemplates);
-
- if ($notFoundTemplates) {
- $this->notifyNotFoundTemplates($notFoundTemplates);
- }
-
- if (!$foundTemplates) {
- $this->notifyNoTemplatesFound();
- return static::EXIT_CODE_NORMAL;
- }
-
- if (!$this->confirmGeneration($foundTemplates)) {
- return static::EXIT_CODE_NORMAL;
- }
-
- $templatePath = Yii::getAlias($this->templatePath);
- $fixtureDataPath = Yii::getAlias($this->fixtureDataPath);
-
- FileHelper::createDirectory($fixtureDataPath);
-
- $generatedTemplates = [];
-
- foreach ($foundTemplates as $templateName) {
- $this->generateFixtureFile($templateName, $templatePath, $fixtureDataPath);
- $generatedTemplates[] = $templateName;
- }
-
- $this->notifyTemplatesGenerated($generatedTemplates);
- }
-
-
-
- public function actionGenerateAll()
- {
- $foundTemplates = $this->findTemplatesFiles();
-
- if (!$foundTemplates) {
- $this->notifyNoTemplatesFound();
- return static::EXIT_CODE_NORMAL;
- }
-
- if (!$this->confirmGeneration($foundTemplates)) {
- return static::EXIT_CODE_NORMAL;
- }
-
- $templatePath = Yii::getAlias($this->templatePath);
- $fixtureDataPath = Yii::getAlias($this->fixtureDataPath);
-
- FileHelper::createDirectory($fixtureDataPath);
-
- $generatedTemplates = [];
-
- foreach ($foundTemplates as $templateName) {
- $this->generateFixtureFile($templateName, $templatePath, $fixtureDataPath);
- $generatedTemplates[] = $templateName;
- }
-
- $this->notifyTemplatesGenerated($generatedTemplates);
- }
-
-
-
- private function notifyNotFoundTemplates($templatesNames)
- {
- $this->stdout("The following fixtures templates were NOT found:\n\n", Console::FG_RED);
-
- foreach ($templatesNames as $name) {
- $this->stdout("\t * $name \n", Console::FG_GREEN);
- }
-
- $this->stdout("\n");
- }
-
-
-
- private function notifyNoTemplatesFound()
- {
- $this->stdout("No fixtures template files matching input conditions were found under the path:\n\n", Console::FG_RED);
- $this->stdout("\t " . Yii::getAlias($this->templatePath) . " \n\n", Console::FG_GREEN);
- }
-
-
-
- private function notifyTemplatesGenerated($templatesNames)
- {
- $this->stdout("The following fixtures template files were generated:\n\n", Console::FG_YELLOW);
-
- foreach ($templatesNames as $name) {
- $this->stdout("\t* " . $name . "\n", Console::FG_GREEN);
- }
-
- $this->stdout("\n");
- }
-
- private function notifyTemplatesCanBeGenerated($templatesNames)
- {
- $this->stdout("Template files path: ", Console::FG_YELLOW);
- $this->stdout(Yii::getAlias($this->templatePath) . "\n\n", Console::FG_GREEN);
-
- foreach ($templatesNames as $name) {
- $this->stdout("\t* " . $name . "\n", Console::FG_GREEN);
- }
-
- $this->stdout("\n");
- }
-
-
-
- private function findTemplatesFiles(array $templatesNames = [])
- {
- $findAll = ($templatesNames == []);
-
- if ($findAll) {
- $files = FileHelper::findFiles(Yii::getAlias($this->templatePath), ['only' => ['*.php']]);
- } else {
- $filesToSearch = [];
-
- foreach ($templatesNames as $fileName) {
- $filesToSearch[] = $fileName . '.php';
- }
-
- $files = FileHelper::findFiles(Yii::getAlias($this->templatePath), ['only' => $filesToSearch]);
- }
-
- $foundTemplates = [];
-
- foreach ($files as $fileName) {
- $foundTemplates[] = basename($fileName, '.php');
- }
-
- return $foundTemplates;
- }
-
-
-
- public function getGenerator()
- {
- if ($this->_generator === null) {
- $language = $this->language === null ? Yii::$app->language : $this->language;
- $this->_generator = \Faker\Factory::create(str_replace('-', '_', $language));
- }
- return $this->_generator;
- }
-
-
-
- public function checkPaths()
- {
- $path = Yii::getAlias($this->templatePath, false);
-
- if (!$path || !is_dir($path)) {
- throw new Exception("The template path \"{$this->templatePath}\" does not exist");
- }
- }
-
-
-
- public function addProviders()
- {
- foreach ($this->providers as $provider) {
- $this->generator->addProvider(new $provider($this->generator));
- }
- }
-
-
-
- public function exportFixtures($fixtures)
- {
- return "<?php\n\nreturn " . VarDumper::export($fixtures) . ";\n";
- }
-
-
-
- public function generateFixture($_template_, $index)
- {
-
- $faker = $this->getGenerator();
- return require($_template_);
- }
-
-
-
- public function generateFixtureFile($templateName, $templatePath, $fixtureDataPath)
- {
- $fixtures = [];
-
- for ($i = 0; $i < $this->count; $i++) {
- $fixtures[$i] = $this->generateFixture($templatePath . '/' . $templateName . '.php', $i);
- }
-
- $content = $this->exportFixtures($fixtures);
-
- file_put_contents($fixtureDataPath . '/'. $templateName . '.php', $content);
- }
-
-
-
- public function confirmGeneration($files)
- {
- $this->stdout("Fixtures will be generated under the path: \n", Console::FG_YELLOW);
- $this->stdout("\t" . Yii::getAlias($this->fixtureDataPath) . "\n\n", Console::FG_GREEN);
- $this->stdout("Templates will be taken from path: \n", Console::FG_YELLOW);
- $this->stdout("\t" . Yii::getAlias($this->templatePath) . "\n\n", Console::FG_GREEN);
-
- foreach ($files as $fileName) {
- $this->stdout("\t* " . $fileName . "\n", Console::FG_GREEN);
- }
-
- return $this->confirm('Generate above fixtures?');
- }
-
- }
|