YII 类库手册

public function __construct($name,$runner)
{
    $this->_name=$name;
    $this->_runner=$runner;
}

protected function afterAction($action,$params)
{
}

protected function beforeAction($action,$params)
{
    return true;
}

public function buildFileList($sourceDir, $targetDir, $baseDir='')
{
    $list=array();
    $handle=opendir($sourceDir);
    while(($file=readdir($handle))!==false)
    {
        if($file==='.' || $file==='..' || $file==='.svn' ||$file==='.yii')
            continue;
        $sourcePath=$sourceDir.DIRECTORY_SEPARATOR.$file;
        $targetPath=$targetDir.DIRECTORY_SEPARATOR.$file;
        $name=$baseDir===''?$file : $baseDir.'/'.$file;
        $list[$name]=array('source'=>$sourcePath, 'target'=>$targetPath);
        if(is_dir($sourcePath))
            $list=array_merge($list,$this->buildFileList($sourcePath,$targetPath,$name));
    }
    closedir($handle);
    return $list;
}

public function confirm($message)
{
    echo $message.' [yes|no] ';
    return !strncasecmp(trim(fgets(STDIN)),'y',1);
}

public function copyFiles($fileList)
{
    $overwriteAll=false;
    foreach($fileList as $name=>$file)
    {
        $source=strtr($file['source'],'/\\',DIRECTORY_SEPARATOR);
        $target=strtr($file['target'],'/\\',DIRECTORY_SEPARATOR);
        $callback=isset($file['callback']) ? $file['callback'] : null;
        $params=isset($file['params']) ? $file['params'] : null;

        if(is_dir($source))
        {
            $this->ensureDirectory($target);
            continue;
        }

        if($callback!==null)
            $content=call_user_func($callback,$source,$params);
        else
            $content=file_get_contents($source);
        if(is_file($target))
        {
            if($content===file_get_contents($target))
            {
                echo "  unchanged $name\n";
                continue;
            }
            if($overwriteAll)
                echo "  overwrite $name\n";
            else
            {
                echo "      exist $name\n";
                echo "            ...overwrite? [Yes|No|All|Quit] ";
                $answer=trim(fgets(STDIN));
                if(!strncasecmp($answer,'q',1))
                    return;
                else if(!strncasecmp($answer,'y',1))
                    echo "  overwrite $name\n";
                else if(!strncasecmp($answer,'a',1))
                {
                    echo "  overwrite $name\n";
                    $overwriteAll=true;
                }
                else
                {
                    echo "       skip $name\n";
                    continue;
                }
            }
        }
        else
        {
            $this->ensureDirectory(dirname($target));
            echo "   generate $name\n";
        }
        file_put_contents($target,$content);
    }
}

public function ensureDirectory($directory)
{
    if(!is_dir($directory))
    {
        $this->ensureDirectory(dirname($directory));
        echo "      mkdir ".strtr($directory,'\\','/')."\n";
        mkdir($directory);
    }
}

public function getCommandRunner()
{
    return $this->_runner;
}

public function getHelp()
{
    $help='Usage: '.$this->getCommandRunner()->getScriptName().' '.$this->getName();
    $options=$this->getOptionHelp();
    if(empty($options))
        return $help;
    if(count($options)===1)
        return $help.' '.$options[0];
    $help.=" <action>\nActions:\n";
    foreach($options as $option)
        $help.='    '.$option."\n";
    return $help;
}

public function getName()
{
    return $this->_name;
}

public function getOptionHelp()
{
    $options=array();
    $class=new ReflectionClass(get_class($this));
    foreach($class->getMethods(ReflectionMethod::IS_PUBLIC) as $method)
    {
        $name=$method->getName();
        if(!strncasecmp($name,'action',6) && strlen($name)>6)
        {
            $name=substr($name,6);
            $name[0]=strtolower($name[0]);
            $help=$name;

            foreach($method->getParameters() as $param)
            {
                $optional=$param->isDefaultValueAvailable();
                $defaultValue=$optional ? $param->getDefaultValue() : null;
                $name=$param->getName();
                if($optional)
                    $help.=" [--$name=$defaultValue]";
                else
                    $help.=" --$name=value";
            }
            $options[]=$help;
        }
    }
    return $options;
}

public function init()
{
}

public function pluralize($name)
{
    $rules=array(
        '/move$/i' => 'moves',
        '/foot$/i' => 'feet',
        '/child$/i' => 'children',
        '/human$/i' => 'humans',
        '/man$/i' => 'men',
        '/tooth$/i' => 'teeth',
        '/person$/i' => 'people',
        '/([m|l])ouse$/i' => '\1ice',
        '/(x|ch|ss|sh|us|as|is|os)$/i' => '\1es',
        '/([^aeiouy]|qu)y$/i' => '\1ies',
        '/(?:([^f])fe|([lr])f)$/i' => '\1\2ves',
        '/(shea|lea|loa|thie)f$/i' => '\1ves',
        '/([ti])um$/i' => '\1a',
        '/(tomat|potat|ech|her|vet)o$/i' => '\1oes',
        '/(bu)s$/i' => '\1ses',
        '/(ax|test)is$/i' => '\1es',
        '/s$/' => 's',
    );
    foreach($rules as $rule=>$replacement)
    {
        if(preg_match($rule,$name))
            return preg_replace($rule,$replacement,$name);
    }
    return $name.'s';
}

public function prompt($message)
{
    if(extension_loaded('readline'))
    {
        $input = readline($message.' ');
        readline_add_history($input);
        return $input;
    }
    else
    {
        echo $message.' ';
        return trim(fgets(STDIN));
    }
}

public function renderFile($_viewFile_,$_data_=null,$_return_=false)
{
    if(is_array($_data_))
        extract($_data_,EXTR_PREFIX_SAME,'data');
    else
        $data=$_data_;
    if($_return_)
    {
        ob_start();
        ob_implicit_flush(false);
        require($_viewFile_);
        return ob_get_clean();
    }
    else
        require($_viewFile_);
}

protected function resolveRequest($args)
{
    $options=array();    // named parameters
    $params=array();    // unnamed parameters
    foreach($args as $arg)
    {
        if(preg_match('/^--(\w+)(=(.*))?$/',$arg,$matches))  // an option
        {
            $name=$matches[1];
            $value=isset($matches[3]) ? $matches[3] : true;
            if(isset($options[$name]))
            {
                if(!is_array($options[$name]))
                    $options[$name]=array($options[$name]);
                $options[$name][]=$value;
            }
            else
                $options[$name]=$value;
        }
        else if(isset($action))
            $params[]=$arg;
        else
            $action=$arg;
    }
    if(!isset($action))
        $action=$this->defaultAction;

    return array($action,$options,$params);
}

public function run($args)
{
    list($action, $options, $args)=$this->resolveRequest($args);
    $methodName='action'.$action;
    if(!preg_match('/^\w+$/',$action) || !method_exists($this,$methodName))
        $this->usageError("Unknown action: ".$action);

    $method=new ReflectionMethod($this,$methodName);
    $params=array();
    // named and unnamed options
    foreach($method->getParameters() as $i=>$param)
    {
        $name=$param->getName();
        if(isset($options[$name]))
        {
            if($param->isArray())
                $params[]=is_array($options[$name]) ? $options[$name] : array($options[$name]);
            else if(!is_array($options[$name]))
                $params[]=$options[$name];
            else
                $this->usageError("Option --$name requires a scalar. Array is given.");
        }
        else if($name==='args')
            $params[]=$args;
        else if($param->isDefaultValueAvailable())
            $params[]=$param->getDefaultValue();
        else
            $this->usageError("Missing required option --$name.");
        unset($options[$name]);
    }

    // try global options
    if(!empty($options))
    {
        $class=new ReflectionClass(get_class($this));
        foreach($options as $name=>$value)
        {
            if($class->hasProperty($name))
            {
                $property=$class->getProperty($name);
                if($property->isPublic() && !$property->isStatic())
                {
                    $this->$name=$value;
                    unset($options[$name]);
                }
            }
        }
    }

    if(!empty($options))
        $this->usageError("Unknown options: ".implode(', ',array_keys($options)));

    if($this->beforeAction($action,$params))
    {
        $method->invokeArgs($this,$params);
        $this->afterAction($action,$params);
    }
}

public function usageError($message)
{
    echo "Error: $message\n\n".$this->getHelp()."\n";
    exit(1);
}