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';
 | ||
|  | 		}
 | ||
|  | 	}
 | ||
|  | }
 |