You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
			
				
					128 lines
				
				2.9 KiB
			
		
		
			
		
	
	
					128 lines
				
				2.9 KiB
			| 
								 
											12 years ago
										 
									 | 
							
								<?php
							 | 
						||
| 
								 | 
							
								/**
							 | 
						||
| 
								 | 
							
								 * @link http://www.yiiframework.com/
							 | 
						||
| 
								 | 
							
								 * @copyright Copyright (c) 2008 Yii Software LLC
							 | 
						||
| 
								 | 
							
								 * @license http://www.yiiframework.com/license/
							 | 
						||
| 
								 | 
							
								 */
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								namespace yii\gii;
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								use Yii;
							 | 
						||
| 
								 | 
							
								use yii\base\Object;
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								/**
							 | 
						||
| 
								 | 
							
								 *
							 | 
						||
| 
								 | 
							
								 * @author Qiang Xue <qiang.xue@gmail.com>
							 | 
						||
| 
								 | 
							
								 * @since 2.0
							 | 
						||
| 
								 | 
							
								 */
							 | 
						||
| 
								 | 
							
								class CodeFile extends Object
							 | 
						||
| 
								 | 
							
								{
							 | 
						||
| 
								 | 
							
									const OP_NEW = 'new';
							 | 
						||
| 
								 | 
							
									const OP_OVERWRITE = 'overwrite';
							 | 
						||
| 
								 | 
							
									const OP_SKIP = 'skip';
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
									/**
							 | 
						||
| 
								 | 
							
									 * @var string the file path that the new code should be saved to.
							 | 
						||
| 
								 | 
							
									 */
							 | 
						||
| 
								 | 
							
									public $path;
							 | 
						||
| 
								 | 
							
									/**
							 | 
						||
| 
								 | 
							
									 * @var mixed the newly generated code. If this is null, it means {@link path}
							 | 
						||
| 
								 | 
							
									 * should be treated as a directory.
							 | 
						||
| 
								 | 
							
									 */
							 | 
						||
| 
								 | 
							
									public $content;
							 | 
						||
| 
								 | 
							
									/**
							 | 
						||
| 
								 | 
							
									 * @var string the operation to be performed
							 | 
						||
| 
								 | 
							
									 */
							 | 
						||
| 
								 | 
							
									public $operation;
							 | 
						||
| 
								 | 
							
									/**
							 | 
						||
| 
								 | 
							
									 * @var string the error occurred when saving the code into a file
							 | 
						||
| 
								 | 
							
									 */
							 | 
						||
| 
								 | 
							
									public $error;
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
									/**
							 | 
						||
| 
								 | 
							
									 * Constructor.
							 | 
						||
| 
								 | 
							
									 * @param string $path the file path that the new code should be saved to.
							 | 
						||
| 
								 | 
							
									 * @param string $content the newly generated code
							 | 
						||
| 
								 | 
							
									 */
							 | 
						||
| 
								 | 
							
									public function __construct($path, $content)
							 | 
						||
| 
								 | 
							
									{
							 | 
						||
| 
								 | 
							
										$this->path = strtr($path, array('/' => DIRECTORY_SEPARATOR, '\\' => DIRECTORY_SEPARATOR));
							 | 
						||
| 
								 | 
							
										$this->content = $content;
							 | 
						||
| 
								 | 
							
										if (is_file($path)) {
							 | 
						||
| 
								 | 
							
											$this->operation = file_get_contents($path) === $content ? self::OP_SKIP : self::OP_OVERWRITE;
							 | 
						||
| 
								 | 
							
										} elseif ($content === null) // is dir
							 | 
						||
| 
								 | 
							
										{
							 | 
						||
| 
								 | 
							
											$this->operation = is_dir($path) ? self::OP_SKIP : self::OP_NEW;
							 | 
						||
| 
								 | 
							
										} else {
							 | 
						||
| 
								 | 
							
											$this->operation = self::OP_NEW;
							 | 
						||
| 
								 | 
							
										}
							 | 
						||
| 
								 | 
							
									}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
									/**
							 | 
						||
| 
								 | 
							
									 * Saves the code into the file {@link path}.
							 | 
						||
| 
								 | 
							
									 */
							 | 
						||
| 
								 | 
							
									public function save()
							 | 
						||
| 
								 | 
							
									{
							 | 
						||
| 
								 | 
							
										$module = Yii::$app->controller->module;
							 | 
						||
| 
								 | 
							
										if ($this->content === null) // a directory
							 | 
						||
| 
								 | 
							
										{
							 | 
						||
| 
								 | 
							
											if (!is_dir($this->path)) {
							 | 
						||
| 
								 | 
							
												$oldmask = @umask(0);
							 | 
						||
| 
								 | 
							
												$result = @mkdir($this->path, $module->newDirMode, true);
							 | 
						||
| 
								 | 
							
												@umask($oldmask);
							 | 
						||
| 
								 | 
							
												if (!$result) {
							 | 
						||
| 
								 | 
							
													$this->error = "Unable to create the directory '{$this->path}'.";
							 | 
						||
| 
								 | 
							
													return false;
							 | 
						||
| 
								 | 
							
												}
							 | 
						||
| 
								 | 
							
											}
							 | 
						||
| 
								 | 
							
											return true;
							 | 
						||
| 
								 | 
							
										}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
										if ($this->operation === self::OP_NEW) {
							 | 
						||
| 
								 | 
							
											$dir = dirname($this->path);
							 | 
						||
| 
								 | 
							
											if (!is_dir($dir)) {
							 | 
						||
| 
								 | 
							
												$oldmask = @umask(0);
							 | 
						||
| 
								 | 
							
												$result = @mkdir($dir, $module->newDirMode, true);
							 | 
						||
| 
								 | 
							
												@umask($oldmask);
							 | 
						||
| 
								 | 
							
												if (!$result) {
							 | 
						||
| 
								 | 
							
													$this->error = "Unable to create the directory '$dir'.";
							 | 
						||
| 
								 | 
							
													return false;
							 | 
						||
| 
								 | 
							
												}
							 | 
						||
| 
								 | 
							
											}
							 | 
						||
| 
								 | 
							
										}
							 | 
						||
| 
								 | 
							
										if (@file_put_contents($this->path, $this->content) === false) {
							 | 
						||
| 
								 | 
							
											$this->error = "Unable to write the file '{$this->path}'.";
							 | 
						||
| 
								 | 
							
											return false;
							 | 
						||
| 
								 | 
							
										} else {
							 | 
						||
| 
								 | 
							
											$oldmask = @umask(0);
							 | 
						||
| 
								 | 
							
											@chmod($this->path, $module->newFileMode);
							 | 
						||
| 
								 | 
							
											@umask($oldmask);
							 | 
						||
| 
								 | 
							
										}
							 | 
						||
| 
								 | 
							
										return true;
							 | 
						||
| 
								 | 
							
									}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
									/**
							 | 
						||
| 
								 | 
							
									 * @return string the code file path relative to the application base path.
							 | 
						||
| 
								 | 
							
									 */
							 | 
						||
| 
								 | 
							
									public function getRelativePath()
							 | 
						||
| 
								 | 
							
									{
							 | 
						||
| 
								 | 
							
										if (strpos($this->path, Yii::$app->basePath) === 0) {
							 | 
						||
| 
								 | 
							
											return substr($this->path, strlen(Yii::$app->basePath) + 1);
							 | 
						||
| 
								 | 
							
										} else {
							 | 
						||
| 
								 | 
							
											return $this->path;
							 | 
						||
| 
								 | 
							
										}
							 | 
						||
| 
								 | 
							
									}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
									/**
							 | 
						||
| 
								 | 
							
									 * @return string the code file extension (e.g. php, txt)
							 | 
						||
| 
								 | 
							
									 */
							 | 
						||
| 
								 | 
							
									public function getType()
							 | 
						||
| 
								 | 
							
									{
							 | 
						||
| 
								 | 
							
										if (($pos = strrpos($this->path, '.')) !== false) {
							 | 
						||
| 
								 | 
							
											return substr($this->path, $pos + 1);
							 | 
						||
| 
								 | 
							
										} else {
							 | 
						||
| 
								 | 
							
											return 'unknown';
							 | 
						||
| 
								 | 
							
										}
							 | 
						||
| 
								 | 
							
									}
							 | 
						||
| 
								 | 
							
								}
							 |