- Move variable management methods to CompilationStateTrait - Extract class and function handling logic into separate trait - Removepull/17/head
parent
742820dc0f
commit
e9182ed5b6
2 changed files with 173 additions and 154 deletions
@ -0,0 +1,171 @@ |
||||
<?php |
||||
/** |
||||
* This file is part of TypePHP. |
||||
* |
||||
* @link https://www.swoole.com/ |
||||
* @contact service@swoole.com |
||||
*/ |
||||
|
||||
namespace TypePhp\Context; |
||||
|
||||
use PhpParser\Node\Expr\Variable; |
||||
use TypePhp\Entity\ClassDef; |
||||
use TypePhp\Entity\FunctionDef; |
||||
use TypePhp\Entity\InterfaceDef; |
||||
use TypePhp\Exception\Skip; |
||||
|
||||
trait CompilationStateTrait |
||||
{ |
||||
protected function addLocalVar(string $name, string $type): void |
||||
{ |
||||
$this->context->localVars[$name] = $type; |
||||
} |
||||
|
||||
protected function registerStdType(string $key): int |
||||
{ |
||||
if (isset($this->stdTypeMap[$key])) { |
||||
return $this->stdTypeMap[$key]; |
||||
} |
||||
$typeId = count($this->stdTypeMap) + 1; |
||||
$this->stdTypeMap[$key] = $typeId; |
||||
return $typeId; |
||||
} |
||||
|
||||
protected function addTmpVar(string $type): string |
||||
{ |
||||
$var = $this->genTmpVarName(); |
||||
$this->addLocalVar($var, $type); |
||||
return $var; |
||||
} |
||||
|
||||
protected function addStaticVar(Variable $var, string $name, string $type): string |
||||
{ |
||||
if ($this->hasVar($name)) { |
||||
$this->fatalError($var, 'Duplicate variable `$' . $var->name . '`'); |
||||
} |
||||
$this->context->staticVars[$name] = $type; |
||||
// 静态变量实际上是一个全局变量的引用 |
||||
$globalVar = $this->escapeStaticVar($name); |
||||
$this->addGlobalVar($globalVar, $type); |
||||
return $globalVar; |
||||
} |
||||
|
||||
protected function hasArgument(string $name): bool |
||||
{ |
||||
return isset($this->context->arguments[$name]); |
||||
} |
||||
|
||||
protected function addArgument(string $name, string $type): void |
||||
{ |
||||
$this->context->arguments[$name] = $type; |
||||
$this->addLocalVar($name, $type); |
||||
} |
||||
|
||||
protected function addLiteralString(string $value): int |
||||
{ |
||||
$index = $this->literalStringIndex++; |
||||
$this->literalStrings[$value] = $index; |
||||
|
||||
return $index; |
||||
} |
||||
|
||||
protected function addGlobalVar(string $name, string $type): void |
||||
{ |
||||
$this->globalVars[$name] = $type; |
||||
} |
||||
|
||||
protected function addScopeGlobalVar(string $name, string $type): void |
||||
{ |
||||
$this->context->globalVars[$name] = $type; |
||||
} |
||||
|
||||
protected function addObject(string $name, string $class): void |
||||
{ |
||||
// Interfaces have no concrete method body for native calls. Abstract classes may have concrete methods. |
||||
if ($this->isInterface($class)) { |
||||
$this->context->declaredObjects[$name] = $class; |
||||
} elseif ($this->isNativeClass($class) or $this->isInternalClass($class)) { |
||||
$this->context->objects[$name] = $class; |
||||
} |
||||
} |
||||
|
||||
protected function hasVar(string $name): bool |
||||
{ |
||||
return $this->hasLocalVar($name) || $this->hasStaticVar($name) || $this->hasScopeGlobalVar($name) || $this->isSuperGlobal($name); |
||||
} |
||||
|
||||
protected function hasLocalVar(string $name): bool |
||||
{ |
||||
return isset($this->context->localVars[$name]); |
||||
} |
||||
|
||||
protected function hasObjectPropVar(string $name): bool |
||||
{ |
||||
return isset($this->context->objectProps[$name]); |
||||
} |
||||
|
||||
protected function addFunction(string $name, FunctionDef $functionDef): void |
||||
{ |
||||
$this->functions[$this->escapeFunction($name)] = $functionDef; |
||||
} |
||||
|
||||
/** |
||||
* @param string $name 必须传入带有完整命名空间的类名，将会自动转义为 native name |
||||
*/ |
||||
protected function hasFunction(string $name): bool |
||||
{ |
||||
return array_key_exists($this->escapeFunction($name), $this->functions); |
||||
} |
||||
|
||||
protected function getFunction(string $name): FunctionDef |
||||
{ |
||||
return $this->functions[$this->escapeFunction($name)]; |
||||
} |
||||
|
||||
protected function addClass(string $name, ClassDef $classDef): void |
||||
{ |
||||
$this->classes[$this->escapeClass($name)] = $classDef; |
||||
} |
||||
|
||||
protected function getClass(string $name): ClassDef |
||||
{ |
||||
return $this->classes[$this->escapeClass($name)]; |
||||
} |
||||
|
||||
public function getClassDef(string $name): ?ClassDef |
||||
{ |
||||
return $this->classes[$this->escapeClass($name)] ?? null; |
||||
} |
||||
|
||||
public function getParentClass(string $class): string |
||||
{ |
||||
return $this->classExtends[strtolower(ltrim($class, '\\'))] ?? ''; |
||||
} |
||||
|
||||
protected function hasClass(string $name): bool |
||||
{ |
||||
return array_key_exists($this->escapeClass($name), $this->classes); |
||||
} |
||||
|
||||
protected function hasInterface(string $name): bool |
||||
{ |
||||
return array_key_exists($this->escapeClass($name), $this->interfaces); |
||||
} |
||||
|
||||
protected function getInterface(string $name): InterfaceDef |
||||
{ |
||||
return $this->interfaces[$this->escapeClass($name)]; |
||||
} |
||||
|
||||
protected function checkFunction(string $name): void |
||||
{ |
||||
// 在预处理阶段检测到函数声明，但是未定义，说明在当前文件，但是顺序错误 |
||||
// 跳过，稍后再处理 |
||||
if (isset($this->symbolDeclInFile[$name]) |
||||
and $this->symbolDeclInFile[$name] === $this->file |
||||
and !$this->hasFunction($name)) { |
||||
$this->redoAfterDeclare[$name] = true; |
||||
throw new Skip(); |
||||
} |
||||
} |
||||
} |
||||
Loading…
Reference in new issue