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.

152 lines
3.5 KiB

11 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;
11 years ago
use yii\gii\components\TextDiff;
use yii\helpers\Html;
use yii\helpers\StringHelper;
11 years ago
/**
* CodeFile represents a code file to be generated.
11 years ago
*
* @author Qiang Xue <qiang.xue@gmail.com>
* @since 2.0
*/
class CodeFile extends Object
{
/**
* The code file is new.
*/
11 years ago
const OP_NEW = 'new';
/**
* The code file already exists, and the new one may need to overwrite it.
*/
11 years ago
const OP_OVERWRITE = 'overwrite';
/**
* The new code file and the existing one are identical.
*/
11 years ago
const OP_SKIP = 'skip';
/**
* @var string an ID that uniquely identifies this code file.
*/
11 years ago
public $id;
11 years ago
/**
* @var string the file path that the new code should be saved to.
*/
public $path;
/**
* @var string the newly generated code content
11 years ago
*/
public $content;
/**
* @var string the operation to be performed. This can be [[OP_NEW]], [[OP_OVERWRITE]] or [[OP_SKIP]].
11 years ago
*/
public $operation;
/**
* Constructor.
* @param string $path the file path that the new code should be saved to.
* @param string $content the newly generated code content.
11 years ago
*/
public function __construct($path, $content)
{
$this->path = strtr($path, array('/' => DIRECTORY_SEPARATOR, '\\' => DIRECTORY_SEPARATOR));
$this->content = $content;
11 years ago
$this->id = md5($this->path);
11 years ago
if (is_file($path)) {
$this->operation = file_get_contents($path) === $content ? self::OP_SKIP : self::OP_OVERWRITE;
} else {
$this->operation = self::OP_NEW;
}
}
/**
* Saves the code into the file specified by [[path]].
* @return string|boolean the error occurred while saving the code file, or true if no error.
11 years ago
*/
public function save()
{
$module = Yii::$app->controller->module;
if ($this->operation === self::OP_NEW) {
$dir = dirname($this->path);
if (!is_dir($dir)) {
$mask = @umask(0);
11 years ago
$result = @mkdir($dir, $module->newDirMode, true);
@umask($mask);
11 years ago
if (!$result) {
11 years ago
return "Unable to create the directory '$dir'.";
11 years ago
}
}
}
if (@file_put_contents($this->path, $this->content) === false) {
11 years ago
return "Unable to write the file '{$this->path}'.";
11 years ago
} else {
$mask = @umask(0);
11 years ago
@chmod($this->path, $module->newFileMode);
@umask($mask);
11 years ago
}
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';
}
}
11 years ago
public function preview()
{
if (($pos = strrpos($this->path, '.')) !== false) {
$type = substr($this->path, $pos + 1);
} else {
$type = 'unknown';
}
if ($type === 'php') {
return highlight_string($this->content, true);
} elseif (!in_array($type, array('jpg', 'gif', 'png', 'exe'))) {
return nl2br(Html::encode($this->content));
11 years ago
} else {
return false;
11 years ago
}
}
public function diff()
{
$type = strtolower($this->getType());
if (in_array($type, array('jpg', 'gif', 'png', 'exe'))) {
11 years ago
return false;
} elseif ($this->operation === self::OP_OVERWRITE) {
return StringHelper::diff(file($this->path), $this->content);
11 years ago
} else {
return '';
}
}
11 years ago
}