<?php error_reporting(E_ALL);

$outCodeFile = __DIR__ . '/../ast_data.c';
$strDefsFile = __DIR__ . '/../ast_str_defs.h';

$code = <<<EOC
#include "php_ast.h"

const zend_ast_kind ast_kinds[] = {
{KINDS}
};

const size_t ast_kinds_count = sizeof(ast_kinds) / sizeof(ast_kinds[0]);

const char *ast_kind_to_name(zend_ast_kind kind) {
\tswitch (kind) {
{STRS}
\t}

\treturn NULL;
}

zend_string *ast_kind_child_name(zend_ast_kind kind, uint32_t child) {
\tswitch (kind) {
{CHILD_NAMES}
\t}

\treturn NULL;
}

void ast_register_kind_constants(INIT_FUNC_ARGS) {
{CONSTS}
}

EOC;

$strDefsHeader = <<<EOC
#ifndef AST_STR_DEFS_H
#define AST_STR_DEFS_H

#define AST_STR_DEFS \
{STR_DEFS}

#endif

EOC;

$funcNames = ['params', 'uses', 'stmts', 'returnType', 'attributes'];

$names = [
    /* special nodes */
    'AST_NAME' => ['name'],
    'AST_CLOSURE_VAR' => ['name'],
    'AST_NULLABLE_TYPE' => ['type'],

    /* declaration nodes */
    'ZEND_AST_FUNC_DECL' => $funcNames,
    'ZEND_AST_CLOSURE' => $funcNames,
    'ZEND_AST_METHOD' => $funcNames,
    'ZEND_AST_ARROW_FUNC' => $funcNames,
    'ZEND_AST_CLASS' => ['extends', 'implements', 'stmts', 'attributes', 'type'],
    'ZEND_AST_PROPERTY_HOOK' => $funcNames, // only params/stmts are used.

    /* 0 child nodes */
    'ZEND_AST_MAGIC_CONST' => [],
    'ZEND_AST_TYPE' => [],
    'ZEND_AST_CALLABLE_CONVERT' => [],

    /* 1 child node */
    'ZEND_AST_VAR' => ['name'],
    'ZEND_AST_CONST' => ['name'],
    'ZEND_AST_UNPACK' => ['expr'],
    'ZEND_AST_CAST' => ['expr'],
    'ZEND_AST_EMPTY' => ['expr'],
    'ZEND_AST_ISSET' => ['var'],
    'ZEND_AST_SHELL_EXEC' => ['expr'],
    'ZEND_AST_CLONE' => ['expr'],
    'ZEND_AST_EXIT' => ['expr'],
    'ZEND_AST_PRINT' => ['expr'],
    'ZEND_AST_INCLUDE_OR_EVAL' => ['expr'],
    'ZEND_AST_UNARY_OP' => ['expr'],
    'ZEND_AST_PRE_INC' => ['var'],
    'ZEND_AST_PRE_DEC' => ['var'],
    'ZEND_AST_POST_INC' => ['var'],
    'ZEND_AST_POST_DEC' => ['var'],
    'ZEND_AST_YIELD_FROM' => ['expr'],

    'ZEND_AST_GLOBAL' => ['var'],
    'ZEND_AST_UNSET' => ['var'],
    'ZEND_AST_RETURN' => ['expr'],
    'ZEND_AST_LABEL' => ['name'],
    'ZEND_AST_REF' => ['var'],
    'ZEND_AST_HALT_COMPILER' => ['offset'],
    'ZEND_AST_ECHO' => ['expr'],
    'ZEND_AST_THROW' => ['expr'],
    'ZEND_AST_GOTO' => ['label'],
    'ZEND_AST_BREAK' => ['depth'],
    'ZEND_AST_CONTINUE' => ['depth'],
    'ZEND_AST_CLASS_NAME' => ['class'],
    'ZEND_AST_PROPERTY_HOOK_SHORT_BODY' => ['expr'],

    /* 2 child nodes */
    'ZEND_AST_CLASS_CONST_GROUP' => ['const', 'attributes', 'type'],
    'ZEND_AST_DIM' => ['expr', 'dim'],
    'ZEND_AST_PROP' => ['expr', 'prop'],
    'ZEND_AST_NULLSAFE_PROP' => ['expr', 'prop'],
    'ZEND_AST_STATIC_PROP' => ['class', 'prop'],
    'ZEND_AST_CALL' => ['expr', 'args'],
    'ZEND_AST_CLASS_CONST' => ['class', 'const'],
    'ZEND_AST_ASSIGN' => ['var', 'expr'],
    'ZEND_AST_ASSIGN_REF' => ['var', 'expr'],
    'ZEND_AST_ASSIGN_OP' => ['var', 'expr'],
    'ZEND_AST_BINARY_OP' => ['left', 'right'],
    'ZEND_AST_ARRAY_ELEM' => ['value', 'key'],
    'ZEND_AST_NEW' => ['class', 'args'],
    'ZEND_AST_INSTANCEOF' => ['expr', 'class'],
    'ZEND_AST_YIELD' => ['value', 'key'],

    'ZEND_AST_STATIC' => ['var', 'default'],
    'ZEND_AST_WHILE' => ['cond', 'stmts'],
    'ZEND_AST_DO_WHILE' => ['stmts', 'cond'],
    'ZEND_AST_IF_ELEM' => ['cond', 'stmts'],
    'ZEND_AST_SWITCH' => ['cond', 'stmts'],
    'ZEND_AST_SWITCH_CASE' => ['cond', 'stmts'],
    'ZEND_AST_DECLARE' => ['declares', 'stmts'],
    'ZEND_AST_PROP_ELEM' => ['name', 'default', 'docComment', 'hooks'],
    'ZEND_AST_PROP_GROUP' => ['type', 'props', 'attributes'],
    'ZEND_AST_CONST_ELEM' => ['name', 'value', 'docComment'],
    'ZEND_AST_USE_TRAIT' => ['traits', 'adaptations'],
    'ZEND_AST_TRAIT_PRECEDENCE' => ['method', 'insteadof'],
    'ZEND_AST_METHOD_REFERENCE' => ['class', 'method'],
    'ZEND_AST_NAMESPACE' => ['name', 'stmts'],
    'ZEND_AST_USE_ELEM' => ['name', 'alias'],
    'ZEND_AST_TRAIT_ALIAS' => ['method', 'alias'],
    'ZEND_AST_GROUP_USE' => ['prefix', 'uses'],
    'ZEND_AST_ATTRIBUTE' => ['class', 'args'],
    'ZEND_AST_MATCH' => ['cond', 'stmts'],
    'ZEND_AST_MATCH_ARM' => ['cond', 'expr'],
    'ZEND_AST_NAMED_ARG' => ['name', 'expr'],

    /* 3 child nodes */
    'ZEND_AST_METHOD_CALL' => ['expr', 'method', 'args'],
    'ZEND_AST_NULLSAFE_METHOD_CALL' => ['expr', 'method', 'args'],
    'ZEND_AST_STATIC_CALL' => ['class', 'method', 'args'],
    'ZEND_AST_CONDITIONAL' => ['cond', 'true', 'false'],

    'ZEND_AST_TRY' => ['try', 'catches', 'finally'],
    'ZEND_AST_CATCH' => ['class', 'var', 'stmts'],

    /* 4 child nodes */
    'ZEND_AST_FOR' => ['init', 'cond', 'loop', 'stmts'],
    'ZEND_AST_FOREACH' => ['expr', 'value', 'key', 'stmts'],
    'ZEND_AST_ENUM_CASE' => ['name', 'expr', 'docComment', 'attributes'],

    /* 6 child nodes */
    'ZEND_AST_PARAM' => ['type', 'name', 'default', 'attributes', 'docComment', 'hooks'],
];

$listNodes = [
	'ZEND_AST_ARG_LIST',
	'ZEND_AST_LIST',
	'ZEND_AST_ARRAY',
	'ZEND_AST_ENCAPS_LIST',
	'ZEND_AST_EXPR_LIST',
	'ZEND_AST_STMT_LIST',
	'ZEND_AST_IF',
	'ZEND_AST_SWITCH_LIST',
	'ZEND_AST_CATCH_LIST',
	'ZEND_AST_PARAM_LIST',
	'ZEND_AST_CLOSURE_USES',
	'ZEND_AST_PROP_DECL',
	'ZEND_AST_CONST_DECL',
	'ZEND_AST_CLASS_CONST_DECL',
	'ZEND_AST_NAME_LIST',
	'ZEND_AST_TRAIT_ADAPTATIONS',
	'ZEND_AST_USE',
	'ZEND_AST_TYPE_UNION',
	'ZEND_AST_TYPE_INTERSECTION',
	'ZEND_AST_ATTRIBUTE_LIST',
	'ZEND_AST_ATTRIBUTE_GROUP',
	'ZEND_AST_MATCH_ARM_LIST',
];

$data = [];
foreach ($listNodes as $name) {
    $shortName = str_replace('ZEND_', '', $name);
    $data[$name] = $shortName;
}
foreach ($names as $name => $_) {
    $shortName = str_replace('ZEND_', '', $name);
    $data[$name] = $shortName;
}

if ($argc > 1) {
    /* Optional: Check against zend_ast.h */
    $inFile = $argv[1];

    if (!is_readable($inFile)) {
	die("Input file not readable\n");
    }

    $inCode = file_get_contents($inFile);
    if (!preg_match('/enum _zend_ast_kind \{(.*?)\};/s', $inCode, $matches)) {
	die("Malformed input file\n");
    }

    $lines = explode("\n", $matches[1]);
    foreach ($lines as $line) {
	if (!preg_match('/\s*(ZEND_([A-Z_]+))/', $line, $matches)) {
	    continue;
	}

	list(, $zend_name, $name) = $matches;
	if ($name == 'AST_ZNODE' || $name == 'AST_ZVAL') {
	    continue;
	}

	if (!isset($data[$zend_name])) {
	    echo "Missing $zend_name\n";
	}
    }
}

$kinds = [];
$strs = [];
$consts = [];
foreach ($data as $zend_name => $name) {
    $kinds[] = "\t$zend_name,";
    $strs[] = "\t\tcase $zend_name: return \"$name\";";
    $consts[] = "\tREGISTER_NS_LONG_CONSTANT(\"ast\", \"$name\", $zend_name,"
        . " CONST_CS | CONST_PERSISTENT);";
}

$code = str_replace('{KINDS}', implode("\n", $kinds), $code);
$code = str_replace('{STRS}', implode("\n", $strs), $code);
$code = str_replace('{CONSTS}', implode("\n", $consts), $code);

$implementations = [];
foreach ($names as $kind => $children) {
    if (empty($children)) {
        $implementations["\t\t\treturn NULL;"][] = $kind;
        continue;
    }

    $kindChildNames = [];
    foreach ($children as $index => $name) {
        $kindChildNames[] = "\t\t\t\tcase $index: return AST_STR(str_$name);";
    }
    $body = "\t\t\tswitch (child) {\n"
        . implode("\n", $kindChildNames) . "\n\t\t\t}\n\t\t\treturn NULL;";
    $implementations[$body][] = $kind;
}
$childNames = [];
foreach ($implementations as $body => $kindList) {
    $codeForGroup = '';
    foreach ($kindList as $kind) {
        $codeForGroup .= "\t\tcase $kind:\n";
    }
    $codeForGroup .= $body;
    $childNames[] = $codeForGroup;
}

$code = str_replace('{CHILD_NAMES}', implode("\n", $childNames), $code);

file_put_contents($outCodeFile, $code);

$strings = get_possible_strings($names);
$strDefs = [];
foreach ($strings as $name) {
    $strDefs[] = "\tX($name, \"$name\") \\";
}
$strDefs[] = "\tX(bracketed_closure, \"{closure}\") \\";

$strDefsHeader = str_replace('{STR_DEFS}', implode("\n", $strDefs), $strDefsHeader);
file_put_contents($strDefsFile, $strDefsHeader);

function get_possible_strings(array $spec) {
    $strings = array_fill_keys([
        'kind', 'flags', 'lineno', 'children',
        'name', 'docComment', 'endLineno', '__declId',
        'flagsCombinable',
    ], true);

    foreach ($spec as $kind => $children) {
        foreach ($children as $childName) {
            $strings[$childName] = true;
        }
    }
    return array_keys($strings);
}