TypePHP 编译器 https://swoole.com/aot/
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.
 
 

851 lines
33 KiB

<?php
/**
* This file is part of TypePHP.
*
* @link https://www.swoole.com/
* @contact service@swoole.com
*/
namespace TypePhp\Optimizer;
use TypePhp\Type;
use TypePhp\Resolver\Reflection;
use PhpParser\Node;
/**
* Config-driven optimizer for built-in function calls.
*
* Used as a trait by CompilerBase. All method calls are direct (no __call/reflection).
*/
trait FuncCallOptimizer
{
protected const string ARG_TYPE_VAR = 'v';
protected const string ARG_TYPE_STR = 's';
protected const string ARG_TYPE_INT = 'i';
protected const string ARG_TYPE_FLOAT = 'f';
protected const string ARG_TYPE_BOOL = 'b';
protected const string ARG_TYPE_REF = 'R';
protected const string ARG_TYPE_ARRAY = 'A';
protected const string ARG_OPTIONAL = '?';
protected const int FOLD_STRING_LEN = 1;
protected const int FOLD_STRING_CASE = 2;
protected const int FOLD_CMP2 = 3;
protected const int FOLD_CMP3 = 4;
protected const int FOLD_COUNT_LITERAL = 5;
protected const int FOLD_KNOWN_CLASS = 6;
protected const int FOLD_KNOWN_CONSTANT = 7;
protected const int FOLD_SSA_TYPE = 8;
/** @var array<string,string|array>|null */
protected ?array $_funcCallConfig = null;
/** @var array<string,array> Cache for auto-detected arg reflection info */
protected array $_autoArgTypes = [];
// =========================================================================
// Config
// =========================================================================
protected function getFuncCallConfig(): array
{
if ($this->_funcCallConfig !== null) {
return $this->_funcCallConfig;
}
return $this->_funcCallConfig = $this->buildFuncCallConfig();
}
protected function buildFuncCallConfig(): array
{
$simple = [
'method_exists', 'property_exists',
'number_format',
'implode',
'array_key_first', 'array_key_last',
'array_values',
'version_compare', 'gettype',
'is_array', 'is_string', 'is_object', 'is_resource',
'is_scalar', 'is_numeric', 'is_countable', 'is_iterable',
'array_is_list', 'is_dir', 'is_file', 'file_exists', 'realpath', 'time',
'in_array', 'array_search',
'date', 'strtotime', 'md5', 'sha1', 'hash', 'print_r',
'base64_encode', 'base64_decode',
'urlencode', 'urldecode', 'rawurlencode', 'rawurldecode',
'json_encode', 'json_decode', 'serialize', 'unserialize',
'random_int', 'random_bytes', 'mt_rand', 'rand',
'strstr', 'strrpos', 'is_a', 'is_subclass_of',
'uniqid',
'dirname', 'basename',
// Math: trig, hyperbolic, exp/log, misc
'sin', 'cos', 'tan', 'asin', 'acos', 'atan', 'atan2',
'sinh', 'cosh', 'tanh', 'asinh', 'acosh', 'atanh',
'pi', 'exp', 'expm1', 'log', 'log10', 'log1p',
'hypot', 'deg2rad', 'rad2deg', 'fmod', 'fdiv', 'fpow', 'intdiv',
// Math: is_* checks
'is_finite', 'is_infinite', 'is_nan',
// Math: base conversion
'decbin', 'decoct', 'dechex', 'bindec', 'hexdec', 'octdec', 'base_convert',
];
$extra = [
// Aliases (PHP function name → C++ target name)
'join' => 'implode',
'stristr' => 'stristr',
'strlen' => ['constFold' => self::FOLD_STRING_LEN],
'ord' => [],
'ucfirst' => [],
'lcfirst' => [],
'strtolower' => ['constFold' => self::FOLD_STRING_CASE],
'strtoupper' => ['constFold' => self::FOLD_STRING_CASE],
'crc32' => [],
'chr' => [],
'strcmp' => ['constFold' => self::FOLD_CMP2],
'strcasecmp' => ['constFold' => self::FOLD_CMP2],
'str_starts_with' => [],
'str_ends_with' => [],
'str_contains' => [],
'strncmp' => ['constFold' => self::FOLD_CMP3],
'strncasecmp' => ['constFold' => self::FOLD_CMP3],
'explode' => [],
'strpos' => [],
'stripos' => [],
'substr' => [],
'str_repeat' => [],
'array_fill' => [],
// Variadic
'array_merge' => ['variadic' => true],
// Compile-time fold with defaults
'class_exists' => ['constFold' => self::FOLD_KNOWN_CLASS, 'defaults' => [1 => 'true']],
'interface_exists' => ['defaults' => [1 => 'true']],
'trait_exists' => ['defaults' => [1 => 'true']],
'enum_exists' => ['defaults' => [1 => 'true']],
'defined' => ['constFold' => self::FOLD_KNOWN_CONSTANT],
// Big* dispatch
'abs' => ['bigDispatch' => [
Type::BIGINT => 'php::BigInt::abs',
Type::BIGFLOAT => 'php::BigFloat::abs',
Type::DECIMAL => 'php::Decimal::abs',
'fallback' => 'php::fn::abs',
]],
'pow' => ['bigDispatch' => [
Type::BIGINT => 'php::BigInt::pow',
Type::DECIMAL => 'php::Decimal::pow',
'fallback' => 'php::fn::pow',
]],
'sqrt' => ['bigDispatch' => [
Type::BIGINT => 'php::BigInt::sqrt',
Type::DECIMAL => 'php::Decimal::sqrt',
Type::BIGFLOAT => 'php::BigFloat::sqrt',
'fallback' => 'php::fn::sqrt',
]],
'floor' => ['bigDispatch' => [
Type::DECIMAL => 'php::Decimal::floor',
'fallback' => 'php::fn::floor',
]],
'ceil' => ['bigDispatch' => [
Type::DECIMAL => 'php::Decimal::ceil',
'fallback' => 'php::fn::ceil',
]],
// Type conversions
'strval' => ['conversion' => self::ARG_TYPE_STR],
'intval' => ['conversion' => self::ARG_TYPE_INT],
'floatval' => ['conversion' => self::ARG_TYPE_FLOAT],
'boolval' => ['conversion' => self::ARG_TYPE_BOOL],
// SSA compile-time type checks
'is_int' => ['constFold' => self::FOLD_SSA_TYPE, 'constFoldExtra' => Type::INT],
'is_float' => ['constFold' => self::FOLD_SSA_TYPE, 'constFoldExtra' => Type::FLOAT],
'is_bool' => ['constFold' => self::FOLD_SSA_TYPE, 'constFoldExtra' => Type::BOOL],
// Custom handlers
'is_null' => ['handler' => 'genIsNull'],
'get_class' => ['handler' => 'genGetClassOptimized'],
'get_parent_class' => ['handler' => 'genGetParentClass'],
'function_exists' => ['handler' => 'genFunctionExistsOptimized'],
'func_get_arg' => ['handler' => 'genFuncGetArgOptimized'],
'func_get_args' => ['handler' => 'genFuncGetArgsOptimized'],
'func_num_args' => ['handler' => 'genFuncNumArgsOptimized'],
'compact' => ['handler' => 'genCompactOptimized'],
'array_keys' => ['handler' => 'genArrayKeys'],
'array_key_exists' => ['handler' => 'genArrayKeyExists'],
'round' => ['handler' => 'genRound'],
'count' => ['handler' => 'genCount'],
'define' => ['handler' => 'genDefine'],
'is_callable' => ['handler' => 'genIsCallable'],
];
$config = $extra;
foreach ($simple as $name) {
if (!isset($config[$name])) {
$config[$name] = [];
}
}
return $config;
}
// =========================================================================
// Main entry point
// =========================================================================
protected function parseFuncCallWithOptimizer(string $name, Node\Expr\FuncCall $expr): string|false
{
foreach ($expr->args as $arg) {
if ($this->isPlaceholderExpr($arg)) {
return false;
}
if ($arg instanceof Node\Arg && $arg->name !== null) {
return false;
}
}
$config = $this->getFuncCallConfig()[$name] ?? null;
if ($config === null) {
return false;
}
// 检测参数中使用的变量是否已定义,若变量不存在则回退到动态调用路径
// 动态路径中的 parseCallArgs() 会给出明确的错误信息
foreach ($expr->args as $arg) {
if (!$arg instanceof Node\Arg) {
continue;
}
if ($this->isVarExpr($arg->value) && is_string($arg->value->name) && !$this->hasVar($arg->value->name)) {
return false;
}
}
if (is_string($config)) {
$targetName = $config;
$config = ['target' => $targetName];
$name = $targetName;
}
if (isset($config['handler'])) {
return $this->{$config['handler']}($name, $expr, $config);
}
if (isset($config['bigDispatch'])) {
return $this->dispatchBigType($expr, $config['bigDispatch']);
}
if (isset($config['conversion'])) {
return $this->dispatchConversion($expr, $config['conversion']);
}
return $this->dispatchFuncCall($name, $expr, $config);
}
// =========================================================================
// Generic dispatcher
// =========================================================================
protected function dispatchFuncCall(string $name, Node\Expr\FuncCall $expr, array $config): string|false
{
// 命名参数 / unpack(...)展开需要运行时处理,回退到动态调用路径
foreach ($expr->args as $arg) {
if ($arg->name !== null || $arg->unpack) {
return false;
}
}
$target = $config['target'] ?? null;
if ($target === null) {
$target = 'php::fn::' . $name;
} elseif (!str_starts_with($target, 'php::')) {
$target = 'php::fn::' . $target;
}
$refInfo = $this->getArgReflectionInfo($name);
$argTypeStr = $config['args'] ?? ($refInfo['args'] ?? '');
$defaults = $config['defaults'] ?? [];
if (!empty($config['variadic']) || ($refInfo['variadic'] ?? false)) {
$variadicType = $config['variadicType'] ?? $refInfo['variadicType'] ?? '';
return $this->genVariadicCall($target, $expr, $variadicType);
}
if (isset($config['constFold'])) {
$folded = $this->tryConstFold($config['constFold'], $config['constFoldExtra'] ?? null, $expr);
if ($folded !== false) {
return $folded;
}
}
$nullables = $refInfo['nullables'] ?? [];
$args = $this->buildArgList($expr, $argTypeStr, $defaults, $nullables);
return $target . '(' . implode(', ', $args) . ')';
}
// =========================================================================
// Auto-detect argument types from PHP reflection
// =========================================================================
protected function getArgReflectionInfo(string $funcName): array
{
if (isset($this->_autoArgTypes[$funcName])) {
return $this->_autoArgTypes[$funcName];
}
$ref = Reflection::getFunction($funcName);
if (!$ref) {
return $this->_autoArgTypes[$funcName] = ['args' => '', 'variadic' => false, 'variadicType' => '', 'minArgs' => 0, 'maxArgs' => 0, 'nullables' => []];
}
$types = [];
$nullables = [];
$variadic = false;
$variadicType = '';
foreach ($ref->getParameters() as $param) {
if ($param->isVariadic()) {
$variadic = true;
$variadicType = $this->phpParamToArgChar($param);
continue;
}
$char = $this->phpParamToArgChar($param);
if ($param->isOptional()) {
$char = self::ARG_OPTIONAL . $char;
}
$types[] = $char;
$nullables[] = $param->allowsNull();
}
return $this->_autoArgTypes[$funcName] = [
'args' => implode('_', $types),
'variadic' => $variadic,
'variadicType' => $variadicType,
'minArgs' => $ref->getNumberOfRequiredParameters(),
'maxArgs' => $ref->getNumberOfParameters(),
'nullables' => $nullables,
];
}
protected function phpParamToArgChar(\ReflectionParameter $param): string
{
if ($param->isPassedByReference()) {
return self::ARG_TYPE_REF;
}
$type = $param->getType();
if ($type instanceof \ReflectionNamedType) {
return match ($type->getName()) {
'string' => self::ARG_TYPE_STR,
'int' => self::ARG_TYPE_INT,
'float' => self::ARG_TYPE_FLOAT,
'bool' => self::ARG_TYPE_BOOL,
'array' => self::ARG_TYPE_ARRAY,
default => self::ARG_TYPE_VAR,
};
}
return self::ARG_TYPE_VAR;
}
// =========================================================================
// Arg helpers
// =========================================================================
protected function getArg(Node\Expr\FuncCall $expr, int $i): string
{
$arg = $expr->args[$i]->value;
if ($this->isVarExpr($arg) and $arg->name === 'GLOBALS') {
return 'php_globals_array()';
}
return $this->parseOrderedOperand($arg, false);
}
protected function getRefArg(Node\Expr\FuncCall $expr, int $i): string
{
$arg = $expr->args[$i]->value;
if ($this->isArrayDimFetch($arg) and $this->isVarExpr($arg->var)) {
$array = $this->parseIdentifier($arg->var);
if ($arg->dim !== null) {
$tmpRef = $this->genTmpVarName();
$this->context->beforeStmtLines[] = 'auto&& ' . $tmpRef . ' = ' . $array . '.item(' . $this->identifierToStr($arg->dim) . ', true);';
return $tmpRef;
}
}
if ($this->isPropertyFetch($arg) and $this->isVarExpr($arg->var)) {
$obj = $this->parseIdentifier($arg->var);
$tmpRef = $this->genTmpVarName();
$this->context->beforeStmtLines[] = 'auto&& ' . $tmpRef . ' = ' . $obj . '.attr(' . $this->identifierToStr($arg->name) . ', true);';
return $tmpRef;
}
return $this->getArg($expr, $i);
}
protected function resolveArg(Node\Expr\FuncCall $expr, int $index, string $type): string
{
$base = ($type[0] ?? '') === self::ARG_OPTIONAL ? substr($type, 1) : $type;
if ($base === self::ARG_TYPE_ARRAY) {
return $this->convertStdContainerArrayExpr($expr, $index, $this->getArg($expr, $index));
}
$raw = ($base === self::ARG_TYPE_REF) ? $this->getRefArg($expr, $index) : $this->getArg($expr, $index);
return $this->applyArgConversion($raw, $type);
}
protected function applyArgConversion(string $cxxExpr, string $type): string
{
$base = ($type[0] ?? '') === self::ARG_OPTIONAL ? substr($type, 1) : $type;
return match ($base) {
self::ARG_TYPE_STR => $this->convertStringExpr($cxxExpr),
self::ARG_TYPE_INT => $this->convertIntExpr($cxxExpr),
self::ARG_TYPE_FLOAT => $this->convertFloatExpr($cxxExpr),
self::ARG_TYPE_BOOL => $this->convertBoolExpr($cxxExpr),
self::ARG_TYPE_ARRAY => $this->convertArrayExpr($cxxExpr),
default => $cxxExpr,
};
}
protected function convertStdContainerArrayExpr(Node\Expr\FuncCall $expr, int $index, string $raw): string
{
$arg = $expr->args[$index]->value;
if ($this->isVarExpr($arg) and $this->isStdContainer($arg->name)) {
return $this->convertArrayExpr($raw . '_ref');
}
return $this->convertArrayExpr($raw);
}
protected function buildArgList(Node\Expr\FuncCall $expr, string $argTypeStr, array $defaults = [], array $nullables = []): array
{
if ($argTypeStr === '') {
return [];
}
$types = explode('_', $argTypeStr);
$argCount = count($expr->args);
$args = [];
foreach ($types as $i => $type) {
$optional = ($type[0] ?? '') === self::ARG_OPTIONAL;
$nullable = $nullables[$i] ?? false;
// Missing optional arg — use configured default or skip (C++ default handles it)
if ($optional && $argCount <= $i) {
if (isset($defaults[$i])) {
$args[] = $defaults[$i];
}
continue;
}
// Nullable param — pass raw Variant; C++ function checks isNull() at runtime
if ($nullable) {
$args[] = $this->getArg($expr, $i);
continue;
}
$args[] = $this->resolveArg($expr, $i, $type);
}
return $args;
}
// =========================================================================
// Variadic, conversion, Big* dispatch
// =========================================================================
protected function genVariadicCall(string $target, Node\Expr\FuncCall $expr, string $variadicType = ''): string
{
$base = ($variadicType !== '' && ($variadicType[0] ?? '') === self::ARG_OPTIONAL) ? substr($variadicType, 1) : $variadicType;
$args = [];
foreach ($expr->args as $index => $arg) {
$raw = $this->parseOrderedOperand($arg->value, false);
$args[] = match ($base) {
self::ARG_TYPE_STR => $this->convertStringExpr($raw),
self::ARG_TYPE_INT => $this->convertIntExpr($raw),
self::ARG_TYPE_FLOAT => $this->convertFloatExpr($raw),
self::ARG_TYPE_BOOL => $this->convertBoolExpr($raw),
self::ARG_TYPE_ARRAY => $this->convertStdContainerArrayExpr($expr, $index, $raw),
default => $raw,
};
}
return $target . '(' . implode(', ', $args) . ')';
}
protected function dispatchConversion(Node\Expr\FuncCall $expr, string $convType): string
{
$arg = $expr->args[0]->value;
$type = $this->detectTypeOfExpr($arg);
$parsed = $this->parseExpr($arg);
if ($convType === self::ARG_TYPE_STR) {
return match ($type) {
Type::BIGINT => 'php::BigInt::toString(' . $parsed . ')',
Type::BIGFLOAT => 'php::BigFloat::toString(' . $parsed . ')',
Type::DECIMAL => 'php::Decimal::toString(' . $parsed . ')',
default => $this->convertStringExpr($parsed),
};
}
return match ($convType) {
self::ARG_TYPE_INT => $this->convertIntExpr($parsed),
self::ARG_TYPE_FLOAT => $this->convertFloatExpr($parsed),
self::ARG_TYPE_BOOL => $this->convertBoolExpr($parsed),
default => $parsed,
};
}
protected function dispatchBigType(Node\Expr\FuncCall $expr, array $dispatch): string|false
{
$type = $this->detectTypeOfExpr($expr->args[0]->value);
$target = $dispatch[$type] ?? $dispatch['fallback'] ?? null;
if (!$target) {
return false;
}
$args = [$this->parseOrderedOperand($expr->args[0]->value, false)];
if (count($expr->args) >= 2) {
$args[] = $this->parseOrderedOperand($expr->args[1]->value, false);
}
return $target . '(' . implode(', ', $args) . ')';
}
// =========================================================================
// Constant folding
// =========================================================================
protected function tryConstFold(int $rule, mixed $extra, Node\Expr\FuncCall $expr): string|false
{
return match ($rule) {
self::FOLD_STRING_LEN => $this->doFoldStringLen($expr),
self::FOLD_STRING_CASE => $this->doFoldStringCase($expr),
self::FOLD_CMP2 => $this->doFoldCmp2($expr),
self::FOLD_CMP3 => $this->doFoldCmp3($expr),
self::FOLD_COUNT_LITERAL => $this->doFoldCountLiteral($expr),
self::FOLD_KNOWN_CLASS => $this->doFoldKnownClass($expr),
self::FOLD_KNOWN_CONSTANT => $this->doFoldKnownConstant($expr),
self::FOLD_SSA_TYPE => $this->doFoldSsaType($expr, $extra),
default => false,
};
}
protected function doFoldStringLen(Node\Expr\FuncCall $expr): string|false
{
$arg = $expr->args[0]->value;
return ($arg instanceof Node\Scalar\String_)
? strlen($arg->value) . $this->getPlatform()->getIntegerLiteralSuffix()
: false;
}
protected function doFoldStringCase(Node\Expr\FuncCall $expr): string|false
{
$arg = $expr->args[0]->value;
if (!$this->isScalarString($arg)) {
return false;
}
$func = $expr->name instanceof Node\Name ? $expr->name->toLowerString() : '';
$val = $func === 'strtoupper' ? strtoupper($arg->value) : strtolower($arg->value);
return $this->getLiteralString($val);
}
protected function doFoldCmp2(Node\Expr\FuncCall $expr): string|false
{
$a0 = $expr->args[0]->value;
$a1 = $expr->args[1]->value;
if (!$this->isScalarString($a0) || !$this->isScalarString($a1)) {
return false;
}
$func = $expr->name instanceof Node\Name ? $expr->name->toLowerString() : '';
$result = $func === 'strcasecmp'
? strcasecmp($a0->value, $a1->value)
: strcmp($a0->value, $a1->value);
return $result . $this->getPlatform()->getIntegerLiteralSuffix();
}
protected function doFoldCmp3(Node\Expr\FuncCall $expr): string|false
{
$a0 = $expr->args[0]->value;
$a1 = $expr->args[1]->value;
$a2 = $expr->args[2]->value;
if (!$this->isScalarString($a0) || !$this->isScalarString($a1) || !$this->isScalarInt($a2)) {
return false;
}
$func = $expr->name instanceof Node\Name ? $expr->name->toLowerString() : '';
$result = $func === 'strncasecmp'
? strncasecmp($a0->value, $a1->value, (int) $a2->value)
: strncmp($a0->value, $a1->value, (int) $a2->value);
return $result . $this->getPlatform()->getIntegerLiteralSuffix();
}
protected function doFoldCountLiteral(Node\Expr\FuncCall $expr): string|false
{
if (count($expr->args) !== 1 || !($expr->args[0] instanceof Node\Arg)) {
return false;
}
$arg = $expr->args[0]->value;
if ($arg instanceof Node\Expr\Array_) {
return count($arg->items) . $this->getPlatform()->getIntegerLiteralSuffix();
}
return $this->genStdContainerCount($arg);
}
protected function doFoldKnownClass(Node\Expr\FuncCall $expr): string|false
{
$cn = $expr->args[0]->value;
return ($this->isScalarString($cn) && $this->hasClass($cn->value)) ? 'true' : false;
}
protected function doFoldKnownConstant(Node\Expr\FuncCall $expr): string|false
{
$cn = $expr->args[0]->value;
return ($this->isScalarString($cn) && $this->hasConstant($cn->value)) ? 'true' : false;
}
protected function doFoldSsaType(Node\Expr\FuncCall $expr, mixed $expectType): string|false
{
if (count($expr->args) !== 1 || !($expr->args[0] instanceof Node\Arg)) {
return false;
}
return ($this->detectTypeOfExpr($expr->args[0]->value) === $expectType) ? 'true' : false;
}
// =========================================================================
// Custom handlers
// =========================================================================
protected function genIsNull(string $n, Node\Expr\FuncCall $e, array $c): string
{
return $this->parseIdentifier($e->args[0]->value) . '.isNull()';
}
protected function genIsCallable(string $n, Node\Expr\FuncCall $e, array $c): string|false
{
if (count($e->args) >= 3) {
return false;
}
return $this->dispatchFuncCall('is_callable', $e, ['target' => 'php::fn::is_callable']);
}
protected function genGetClassOptimized(string $n, Node\Expr\FuncCall $e, array $c): string
{
$obj = $e->args[0]->value;
if ($this->isVarExpr($obj) && $this->isTypedObject($obj->name)) {
return $this->getLiteralString($this->getObjectType($obj->name));
}
return 'php::fn::get_class(' . $this->parseIdentifier($obj) . ')';
}
protected function genGetParentClass(string $n, Node\Expr\FuncCall $e, array $c): string
{
if (count($e->args) === 0) {
if ($this->classDef && $this->classDef->extends) {
return $this->getLiteralString($this->classDef->extends);
}
return 'false';
}
$arg = $e->args[0]->value;
if ($this->isScalarString($arg)) {
$cls = $this->getClass($arg->value);
if ($cls && $cls->extends) return $this->getLiteralString($cls->extends);
if ($cls && !$cls->extends) return 'false';
}
return 'php::fn::get_parent_class(' . $this->parseIdentifier($arg) . ')';
}
protected function genFunctionExistsOptimized(string $n, Node\Expr\FuncCall $e, array $c): string
{
return $this->genFunctionExists($n, $e);
}
protected function genFuncGetArgOptimized(string $n, Node\Expr\FuncCall $e, array $c): string
{
return $this->genFuncGetArg($n, $e);
}
protected function genFuncGetArgsOptimized(string $n, Node\Expr\FuncCall $e, array $c): string
{
return $this->genFuncGetArgs($n, $e);
}
protected function genFuncNumArgsOptimized(string $n, Node\Expr\FuncCall $e, array $c): string
{
return $this->genFuncNumArgs($n, $e);
}
protected function genCompactOptimized(string $n, Node\Expr\FuncCall $e, array $c): string
{
return $this->genCompactOrig($e);
}
protected function genArrayKeys(string $n, Node\Expr\FuncCall $e, array $c): string
{
$cnt = count($e->args);
if ($cnt >= 3) {
return 'php::fn::array_keys_filter(' . $this->getArg($e, 0) . ', ' . $this->getArg($e, 1) . ', ' . $this->getArg($e, 2) . ')';
}
if ($cnt >= 2) {
return 'php::fn::array_keys_filter(' . $this->getArg($e, 0) . ', ' . $this->getArg($e, 1) . ', false)';
}
return 'php::fn::array_keys(' . $this->getArg($e, 0) . ')';
}
protected function genArrayKeyExists(string $n, Node\Expr\FuncCall $e, array $c): string
{
return $this->getArg($e, 1) . '.offsetExists(' . $this->getArg($e, 0) . ')';
}
protected function genRound(string $n, Node\Expr\FuncCall $e, array $c): string
{
$type = $this->detectTypeOfExpr($e->args[0]->value);
if ($type === Type::DECIMAL) {
$a0 = $this->parseExpr($e->args[0]->value);
if (count($e->args) >= 2) {
return 'php::Decimal::round(' . $a0 . ', ' . $this->parseExpr($e->args[1]->value) . ')';
}
return 'php::Decimal::round(' . $a0 . ')';
}
$args = count($e->args);
if ($args >= 3) {
return 'php::fn::round(' . $this->getArg($e, 0) . ', ' . $this->convertIntExpr($this->getArg($e, 1)) . ', ' . $this->convertIntExpr($this->getArg($e, 2)) . ')';
}
if ($args >= 2) {
return 'php::fn::round(' . $this->getArg($e, 0) . ', ' . $this->convertIntExpr($this->getArg($e, 1)) . ')';
}
return 'php::fn::round(' . $this->getArg($e, 0) . ')';
}
protected function genCount(string $n, Node\Expr\FuncCall $e, array $c): string
{
$folded = $this->doFoldCountLiteral($e);
if ($folded !== false) return $folded;
if (count($e->args) >= 2) {
return 'php::fn::count(' . $this->getArg($e, 0) . ', ' . $this->convertIntExpr($this->getArg($e, 1)) . ')';
}
return 'php::fn::count(' . $this->getArg($e, 0) . ')';
}
protected function genDefine(string $n, Node\Expr\FuncCall $e, array $c): string|false
{
$arg = $e->args[0]->value;
if ($this->isScalarString($arg) && !$this->isValidDefineName($arg->value)) {
$this->fatalError($e, 'Invalid define name `' . $arg->value . '`');
}
$args = count($e->args) >= 3 ? 3 : 2;
if ($args == 3) {
return 'php::fn::define(' . $this->getArg($e, 0) . ', ' . $this->getArg($e, 1) . ', ' . $this->getArg($e, 2) . ')';
}
return 'php::fn::define(' . $this->getArg($e, 0) . ', ' . $this->getArg($e, 1) . ')';
}
// =========================================================================
// Legacy helpers
// =========================================================================
protected function genFuncGetArgs(string $name, Node\Expr\FuncCall $expr): string
{
$this->warningUndefinedBehavior($expr);
$funcDef = $this->functionDef;
$list = [];
foreach ($funcDef->argInfoList as $i => $argInfo) {
if ($argInfo->variadic) {
$tmpVar = $this->addTmpVar(Type::ARRAY);
$this->context->beforeStmtLines[] = $this->genArray($list) . ';';
$this->context->beforeStmtLines[] = $tmpVar . '.merge(' . $argInfo->name . ');';
return $tmpVar;
}
$list[] = $argInfo->name;
}
return $this->genArray($list);
}
protected function genFuncGetArg(string $name, Node\Expr\FuncCall $expr)
{
$this->warningUndefinedBehavior($expr);
$position = $expr->args[0]->value;
if ($this->isScalarInt($position)) {
$funcDef = $this->functionDef;
$posInt = intval($position->value);
foreach ($funcDef->argInfoList as $i => $argInfo) {
if ($argInfo->variadic) {
return $argInfo->name . '.offsetGet(' . ($posInt - $i) . ')';
}
if ($i == $posInt) {
return $argInfo->name;
}
}
$this->fatalError($expr, 'wrong parameter position `' . $posInt . '`');
} else {
$this->fatalError($expr, 'func_get_arg() only support scalar int');
}
}
protected function genFuncNumArgs(string $name, Node\Expr\FuncCall $expr): string
{
$this->warningUndefinedBehavior($expr);
$funcDef = $this->functionDef;
foreach ($funcDef->argInfoList as $i => $argInfo) {
if ($argInfo->variadic) {
return '(' . $argInfo->name . '.count() + ' . $i . ')';
}
}
return count($funcDef->argInfoList);
}
protected function genFunctionExists(string $name, Node\Expr\FuncCall $expr): string
{
$funcName = $expr->args[0]->value;
if ($this->isScalarString($funcName)) {
$nameLower = strtolower(trim($funcName->value, '\\'));
if ($this->findNativeFunction($nameLower)) {
return 'true';
}
$funcName = $this->getLiteralString($nameLower);
return 'php::fn::function_exists(' . $funcName . ')';
}
return 'php::fn::function_exists(' . $this->parseIdentifier($funcName) . ')';
}
protected function genGetClass(Node\Expr\FuncCall $expr): string
{
$object = $expr->args[0]->value;
if ($this->isVarExpr($object) and $this->isTypedObject($object->name)) {
return $this->getLiteralString($this->getObjectType($object->name));
}
return 'php::fn::get_class(' . $this->parseIdentifier($object) . ')';
}
protected function genCompactOrig(Node\Expr\FuncCall $expr): string
{
$list = [];
foreach ($expr->args as $arg) {
if (!$this->isScalarString($arg->value)) {
$this->fatalError($expr, 'The argument of compact function can only be literal string');
}
$var = $arg->value->value;
if (!$this->hasVar($var) && $var !== 'this') {
$this->fatalError($arg->value, "Undefined variable `{$var}` in compact()");
}
if ($this->isSuperGlobal($var)) {
$this->fatalError($expr, 'Cannot use super global variable `' . $var . '` in compact function');
}
$key = $this->getLiteralString($var);
if ($var === 'this') {
if (empty($this->class)) {
$this->fatalError($expr, 'Cannot use compact("this") outside of class method');
}
}
$cVar = $this->escapeVarName($var);
$list[] = '{' . $key . '.str(), php::Var(' . $cVar . ')}';
}
return 'php::Array{' . implode(', ', $list) . '}';
}
// =========================================================================
// Utility
// =========================================================================
protected function isValidDefineName(string $name): bool
{
return preg_match('/^(?!\d)[\p{L}_][\p{L}\p{N}_]*$/u', $name) === 1;
}
}