X-Git-Url: https://git.saurik.com/cycript.git/blobdiff_plain/04ea132d27de7cfd2d61ad674ee1d4873e500b88..b0385401f63fe31f3cc1459ed159747a445852d6:/Replace.cpp diff --git a/Replace.cpp b/Replace.cpp index 5be4e01..74f04d9 100644 --- a/Replace.cpp +++ b/Replace.cpp @@ -1,21 +1,21 @@ /* Cycript - Optimizing JavaScript Compiler/Runtime - * Copyright (C) 2009-2010 Jay Freeman (saurik) + * Copyright (C) 2009-2015 Jay Freeman (saurik) */ -/* GNU Lesser General Public License, Version 3 {{{ */ +/* GNU Affero General Public License, Version 3 {{{ */ /* - * Cycript is free software: you can redistribute it and/or modify it under - * the terms of the GNU Lesser General Public License as published by the - * Free Software Foundation, either version 3 of the License, or (at your - * option) any later version. - * - * Cycript is distributed in the hope that it will be useful, but WITHOUT - * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or - * FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public - * License for more details. - * - * You should have received a copy of the GNU Lesser General Public License - * along with Cycript. If not, see . + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU Affero General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU Affero General Public License for more details. + + * You should have received a copy of the GNU Affero General Public License + * along with this program. If not, see . **/ /* }}} */ @@ -32,19 +32,16 @@ CYFunctionExpression *CYNonLocalize(CYContext &context, CYFunctionExpression *fu CYExpression *CYAdd::Replace(CYContext &context) { CYInfix::Replace(context); - CYExpression *lhp(lhs_->Primitive(context)); - CYExpression *rhp(rhs_->Primitive(context)); - - CYString *lhs(dynamic_cast(lhp)); - CYString *rhs(dynamic_cast(rhp)); + CYString *lhs(dynamic_cast(lhs_)); + CYString *rhs(dynamic_cast(rhs_)); if (lhs != NULL || rhs != NULL) { if (lhs == NULL) { - lhs = lhp->String(context); + lhs = lhs_->String(context); if (lhs == NULL) return this; } else if (rhs == NULL) { - rhs = rhp->String(context); + rhs = rhs_->String(context); if (rhs == NULL) return this; } @@ -52,8 +49,8 @@ CYExpression *CYAdd::Replace(CYContext &context) { return lhs->Concat(context, rhs); } - if (CYNumber *lhn = lhp->Number(context)) - if (CYNumber *rhn = rhp->Number(context)) + if (CYNumber *lhn = lhs_->Number(context)) + if (CYNumber *rhn = rhs_->Number(context)) return $D(lhn->Value() + rhn->Value()); return this; @@ -85,7 +82,7 @@ CYExpression *CYArray::Replace(CYContext &context) { CYExpression *CYArrayComprehension::Replace(CYContext &context) { CYVariable *cyv($V("$cyv")); - return $C0($F(NULL, $P1("$cyv", comprehensions_->Parameters(context)), $$->* + return $C0($F(NULL, $P1($L("$cyv"), comprehensions_->Parameters(context)), $$->* $E($ CYAssign(cyv, $ CYArray()))->* comprehensions_->Replace(context, $E($C1($M(cyv, $S("push")), expression_)))->* $ CYReturn(cyv) @@ -99,8 +96,8 @@ CYExpression *CYAssignment::Replace(CYContext &context) { } CYStatement *CYBlock::Replace(CYContext &context) { - context.ReplaceAll(statements_); - if (statements_ == NULL) + context.ReplaceAll(code_); + if (code_ == NULL) return $ CYEmpty(); return this; } @@ -127,20 +124,20 @@ namespace cy { namespace Syntax { void Catch::Replace(CYContext &context) { $T() - CYScope scope(true, context, code_.statements_); + CYScope scope(true, context); context.Replace(name_); context.scope_->Declare(context, name_, CYIdentifierCatch); - code_.Replace(context); - scope.Close(); + context.ReplaceAll(code_); + scope.Close(context, code_); } } } void CYClause::Replace(CYContext &context) { $T() context.Replace(case_); - context.ReplaceAll(statements_); + context.ReplaceAll(code_); next_->Replace(context); } @@ -149,12 +146,32 @@ CYStatement *CYComment::Replace(CYContext &context) { } CYExpression *CYCompound::Replace(CYContext &context) { - context.ReplaceAll(expressions_); - if (expressions_ == NULL) - return NULL; + context.Replace(expression_); + context.Replace(next_); + + if (CYCompound *compound = dynamic_cast(expression_)) { + expression_ = compound->expression_; + compound->expression_ = compound->next_; + compound->next_ = next_; + next_ = compound; + } + return this; } +CYFunctionParameter *CYCompound::Parameter() const { + CYFunctionParameter *next(next_->Parameter()); + if (next == NULL) + return NULL; + + CYFunctionParameter *parameter(expression_->Parameter()); + if (parameter == NULL) + return NULL; + + parameter->SetNext(next); + return parameter; +} + CYFunctionParameter *CYComprehension::Parameters(CYContext &context) const { $T(NULL) CYFunctionParameter *next(next_->Parameters(context)); if (CYFunctionParameter *parameter = Parameter(context)) { @@ -183,7 +200,7 @@ void CYContext::NonLocal(CYStatement *&statements) { CYIdentifier *unique(nextlocal_->identifier_->Replace(context)); CYStatement *declare( - $ CYVar($L1($L(unique, $ CYObject())))); + $ CYVar($L1($ CYDeclaration(unique, $ CYObject())))); cy::Syntax::Catch *rescue( $ cy::Syntax::Catch(cye, $$->* @@ -191,6 +208,7 @@ void CYContext::NonLocal(CYStatement *&statements) { $ CYReturn($M($V(cye), $S("$cyv"))))->* $ cy::Syntax::Throw($V(cye)))); + // XXX: I don't understand any of this context.Replace(declare); rescue->Replace(context); @@ -201,13 +219,17 @@ void CYContext::NonLocal(CYStatement *&statements) { } CYIdentifier *CYContext::Unique() { - return $ CYIdentifier(apr_psprintf($pool, "$cy%u", unique_++)); + return $ CYIdentifier($pool.strcat("$cy", $pool.itoa(unique_++), NULL)); } CYStatement *CYContinue::Replace(CYContext &context) { return this; } +CYStatement *CYDebugger::Replace(CYContext &context) { + return this; +} + CYAssignment *CYDeclaration::Assignment(CYContext &context) { if (initialiser_ == NULL) return NULL; @@ -222,7 +244,7 @@ CYVariable *CYDeclaration::Variable(CYContext &context) { } CYStatement *CYDeclaration::ForEachIn(CYContext &context, CYExpression *value) { - return $ CYVar($L1($L(identifier_, value))); + return $ CYVar($L1($ CYDeclaration(identifier_, value))); } CYExpression *CYDeclaration::Replace(CYContext &context) { @@ -241,17 +263,20 @@ CYProperty *CYDeclarations::Property(CYContext &context) { $T(NULL) } CYFunctionParameter *CYDeclarations::Parameter(CYContext &context) { $T(NULL) - return $ CYFunctionParameter(declaration_->identifier_, next_->Parameter(context)); + return $ CYFunctionParameter($ CYDeclaration(declaration_->identifier_), next_->Parameter(context)); } CYArgument *CYDeclarations::Argument(CYContext &context) { $T(NULL) return $ CYArgument(declaration_->initialiser_, next_->Argument(context)); } -CYCompound *CYDeclarations::Compound(CYContext &context) { $T(NULL) - CYCompound *compound(next_->Compound(context) ?: $ CYCompound()); +CYExpression *CYDeclarations::Expression(CYContext &context) { $T(NULL) + CYExpression *compound(next_->Expression(context)); if (CYAssignment *assignment = declaration_->Assignment(context)) - compound->AddPrev(assignment); + if (compound == NULL) + compound = assignment; + else + compound = $ CYCompound(assignment, compound); return compound; } @@ -263,7 +288,7 @@ CYExpression *CYDirectMember::Replace(CYContext &context) { CYStatement *CYDoWhile::Replace(CYContext &context) { context.Replace(test_); - context.Replace(code_); + context.ReplaceAll(code_); return this; } @@ -276,20 +301,12 @@ CYStatement *CYEmpty::Replace(CYContext &context) { return NULL; } -CYStatement *CYExpress::Replace(CYContext &context) { - while (CYExpress *express = dynamic_cast(next_)) { - CYCompound *compound(dynamic_cast(express->expression_)); - if (compound == NULL) - compound = $ CYCompound(express->expression_); - compound->AddPrev(expression_); - expression_ = compound; - SetNext(express->next_); - } +CYExpression *CYEncodedType::Replace(CYContext &context) { + return typed_->Replace(context); +} +CYStatement *CYExpress::Replace(CYContext &context) { context.Replace(expression_); - if (expression_ == NULL) - return $ CYEmpty(); - return this; } @@ -309,6 +326,14 @@ CYAssignment *CYExpression::Assignment(CYContext &context) { return NULL; } +CYFunctionParameter *CYExpression::Parameter() const { + return NULL; +} + +CYStatement *CYExternal::Replace(CYContext &context) { + return $E($ CYAssign($V(typed_->identifier_), $C1(typed_->Replace(context), $C2($V("dlsym"), $V("RTLD_DEFAULT"), $S(typed_->identifier_->Word()))))); +} + CYNumber *CYFalse::Number(CYContext &context) { return $D(0); } @@ -317,21 +342,27 @@ CYString *CYFalse::String(CYContext &context) { return $S("false"); } +CYExpression *CYFatArrow::Replace(CYContext &context) { + CYFunctionExpression *function($ CYFunctionExpression(NULL, parameters_, code_)); + function->this_.SetNext(context.this_); + return function; +} + void CYFinally::Replace(CYContext &context) { $T() - code_.Replace(context); + context.ReplaceAll(code_); } CYStatement *CYFor::Replace(CYContext &context) { context.Replace(initialiser_); context.Replace(test_); context.Replace(increment_); - context.Replace(code_); + context.ReplaceAll(code_); return this; } -CYCompound *CYForDeclarations::Replace(CYContext &context) { +CYExpression *CYForDeclarations::Replace(CYContext &context) { declarations_->Replace(context); - return declarations_->Compound(context); + return declarations_->Expression(context); } // XXX: this still feels highly suboptimal @@ -344,19 +375,19 @@ CYStatement *CYForIn::Replace(CYContext &context) { context.Replace(initialiser_); context.Replace(set_); - context.Replace(code_); + context.ReplaceAll(code_); return this; } CYFunctionParameter *CYForInComprehension::Parameter(CYContext &context) const { - return $ CYFunctionParameter(name_); + return $ CYFunctionParameter($ CYDeclaration(name_)); } CYStatement *CYForInComprehension::Replace(CYContext &context, CYStatement *statement) const { return $ CYForIn($V(name_), set_, CYComprehension::Replace(context, statement)); } -CYStatement *CYForEachIn::Replace(CYContext &context) { +CYStatement *CYForOf::Replace(CYContext &context) { if (CYAssignment *assignment = initialiser_->Assignment(context)) return $ CYBlock($$->* $E(assignment)->* @@ -365,7 +396,7 @@ CYStatement *CYForEachIn::Replace(CYContext &context) { CYIdentifier *cys($I("$cys")), *cyt($I("$cyt")); - return $ CYLet($L2($L(cys, set_), $L(cyt)), $$->* + return $ CYLetStatement($L2($ CYDeclaration(cys, set_), $ CYDeclaration(cyt)), $$->* $ CYForIn($V(cyt), $V(cys), $ CYBlock($$->* initialiser_->ForEachIn(context, $M($V(cys), $V(cyt)))->* code_ @@ -373,14 +404,14 @@ CYStatement *CYForEachIn::Replace(CYContext &context) { ); } -CYFunctionParameter *CYForEachInComprehension::Parameter(CYContext &context) const { - return $ CYFunctionParameter(name_); +CYFunctionParameter *CYForOfComprehension::Parameter(CYContext &context) const { + return $ CYFunctionParameter($ CYDeclaration(name_)); } -CYStatement *CYForEachInComprehension::Replace(CYContext &context, CYStatement *statement) const { - CYIdentifier *cys($I("cys")); +CYStatement *CYForOfComprehension::Replace(CYContext &context, CYStatement *statement) const { + CYIdentifier *cys($I("$cys")); - return $E($C0($F(NULL, $P1("$cys"), $$->* + return $E($C0($F(NULL, $P1($L("$cys")), $$->* $E($ CYAssign($V(cys), set_))->* $ CYForIn($V(name_), $V(cys), $ CYBlock($$->* $E($ CYAssign($V(name_), $M($V(cys), $V(name_))))->* @@ -398,6 +429,9 @@ void CYFunction::Replace_(CYContext &context, bool outer) { if (outer) Inject(context); + CYThisScope *_this(context.this_); + context.this_ = CYGetLast(&this_); + CYNonLocal *nonlocal(context.nonlocal_); CYNonLocal *nextlocal(context.nextlocal_); @@ -411,23 +445,28 @@ void CYFunction::Replace_(CYContext &context, bool outer) { context.nextlocal_ = nonlocal_; } - CYScope scope(!localize, context, code_.statements_); + CYScope scope(!localize, context); if (!outer && name_ != NULL) Inject(context); - if (parameters_ != NULL) - parameters_ = parameters_->Replace(context, code_); + parameters_->Replace(context, code_); + context.ReplaceAll(code_); - code_.Replace(context); + if (CYIdentifier *identifier = this_.identifier_) + code_ = $$->* + $ CYVar($L1($ CYDeclaration(identifier, $ CYThis())))->* + code_; if (localize) - context.NonLocal(code_.statements_); + context.NonLocal(code_); context.nextlocal_ = nextlocal; context.nonlocal_ = nonlocal; - scope.Close(); + context.this_ = _this; + + scope.Close(context, code_); } CYExpression *CYFunctionExpression::Replace(CYContext &context) { @@ -435,12 +474,19 @@ CYExpression *CYFunctionExpression::Replace(CYContext &context) { return this; } -CYFunctionParameter *CYFunctionParameter::Replace(CYContext &context, CYBlock &code) { - context.Replace(name_); - context.scope_->Declare(context, name_, CYIdentifierArgument); - if (next_ != NULL) - next_ = next_->Replace(context, code); - return this; +void CYFunctionParameter::Replace(CYContext &context, CYStatement *&statements) { $T() + CYAssignment *assignment(initialiser_->Assignment(context)); + context.Replace(initialiser_); + + next_->Replace(context, statements); + + if (assignment != NULL) + statements = $$->* + // XXX: this cast is quite incorrect + $ CYIf($ CYIdentical($ CYTypeOf(dynamic_cast(initialiser_)), $S("undefined")), $$->* + $E(assignment) + )->* + statements; } CYStatement *CYFunctionStatement::Replace(CYContext &context) { @@ -457,8 +503,8 @@ CYIdentifier *CYIdentifier::Replace(CYContext &context) { CYStatement *CYIf::Replace(CYContext &context) { context.Replace(test_); - context.Replace(true_); - context.Replace(false_); + context.ReplaceAll(true_); + context.ReplaceAll(false_); return this; } @@ -470,6 +516,10 @@ CYStatement *CYIfComprehension::Replace(CYContext &context, CYStatement *stateme return $ CYIf(test_, CYComprehension::Replace(context, statement)); } +CYStatement *CYImport::Replace(CYContext &context) { + return $ CYVar($L1($L(module_->part_->Word(), $C1($V("require"), module_->Replace(context, "/"))))); +} + CYExpression *CYIndirect::Replace(CYContext &context) { return $M(rhs_, $S("$cyi")); } @@ -489,15 +539,35 @@ CYStatement *CYLabel::Replace(CYContext &context) { return this; } -CYStatement *CYLet::Replace(CYContext &context) { +CYExpression *CYLambda::Replace(CYContext &context) { + return $N2($V("Functor"), $ CYFunctionExpression(NULL, parameters_->Parameters(context), code_), parameters_->TypeSignature(context, typed_->Replace(context))); +} + +CYStatement *CYLetStatement::Replace(CYContext &context) { return $E($ CYCall(CYNonLocalize(context, $ CYFunctionExpression(NULL, declarations_->Parameter(context), code_)), declarations_->Argument(context))); } +CYString *CYModule::Replace(CYContext &context, const char *separator) const { + if (next_ == NULL) + return $ CYString(part_); + return $ CYString($pool.strcat(next_->Replace(context, separator)->Value(), separator, part_->Word(), NULL)); +} + +CYExpression *CYMultiply::Replace(CYContext &context) { + CYInfix::Replace(context); + + if (CYNumber *lhn = lhs_->Number(context)) + if (CYNumber *rhn = rhs_->Number(context)) + return $D(lhn->Value() * rhn->Value()); + + return this; +} + namespace cy { namespace Syntax { CYExpression *New::AddArgument(CYContext &context, CYExpression *value) { - CYSetLast(arguments_, $ CYArgument(value)); + CYSetLast(arguments_) = $ CYArgument(value); return this; } @@ -523,7 +593,7 @@ CYNumber *CYNumber::Number(CYContext &context) { CYString *CYNumber::String(CYContext &context) { // XXX: there is a precise algorithm for this - return $S(apr_psprintf($pool, "%.17g", Value())); + return $S($pool.sprintf(24, "%.17g", Value())); } CYExpression *CYObject::Replace(CYContext &context) { @@ -531,17 +601,8 @@ CYExpression *CYObject::Replace(CYContext &context) { return this; } -CYFunctionParameter *CYOptionalFunctionParameter::Replace(CYContext &context, CYBlock &code) { - CYFunctionParameter *parameter($ CYFunctionParameter(name_, next_)); - parameter = parameter->Replace(context, code); - context.Replace(initializer_); - - CYVariable *name($V(name_)); - code.AddPrev($ CYIf($ CYIdentical($ CYTypeOf(name), $S("undefined")), $$->* - $E($ CYAssign(name, initializer_)) - )); - - return parameter; +CYExpression *CYParenthetical::Replace(CYContext &context) { + return expression_; } CYExpression *CYPostfix::Replace(CYContext &context) { @@ -573,13 +634,13 @@ namespace { } void CYProgram::Replace(CYContext &context) { - CYScope scope(true, context, statements_); + CYScope scope(true, context); context.nextlocal_ = $ CYNonLocal(); - context.ReplaceAll(statements_); - context.NonLocal(statements_); + context.ReplaceAll(code_); + context.NonLocal(code_); - scope.Close(); + scope.Close(context, code_); size_t offset(0); @@ -600,7 +661,7 @@ void CYProgram::Replace(CYContext &context) { const char *name; if (context.options_.verbose_) - name = apr_psprintf($pool, "$%"APR_SIZE_T_FMT"", offset); + name = $pool.strcat("$", $pool.itoa(offset), NULL); else { char id[8]; id[7] = '\0'; @@ -619,7 +680,7 @@ void CYProgram::Replace(CYContext &context) { goto id; } - name = apr_pstrmemdup($pool, id + position, 7 - position); + name = $pool.strmemdup(id + position, 7 - position); // XXX: at some point, this could become a keyword } @@ -656,21 +717,11 @@ CYExpression *CYRubyProc::Replace(CYContext &context) { return CYNonLocalize(context, $ CYFunctionExpression(NULL, parameters_, code_)); } -CYScope::CYScope(bool transparent, CYContext &context, CYStatement *&statements) : +CYScope::CYScope(bool transparent, CYContext &context) : transparent_(transparent), - context_(context), - statements_(statements), parent_(context.scope_) { - context_.scope_ = this; -} - -CYScope::~CYScope() { -} - -void CYScope::Close() { - context_.scope_ = parent_; - Scope(context_, statements_); + context.scope_ = this; } void CYScope::Declare(CYContext &context, CYIdentifier *identifier, CYIdentifierFlags flags) { @@ -728,7 +779,9 @@ namespace { typedef std::set IdentifierOffsets; } -void CYScope::Scope(CYContext &context, CYStatement *&statements) { +void CYScope::Close(CYContext &context, CYStatement *&statements) { + context.scope_ = parent_; + if (parent_ == NULL) return; @@ -803,6 +856,8 @@ CYStatement *CYSwitch::Replace(CYContext &context) { } CYExpression *CYThis::Replace(CYContext &context) { + if (context.this_ != NULL) + return $V(context.this_->Identifier(context)); return this; } @@ -832,7 +887,7 @@ namespace cy { namespace Syntax { CYStatement *Try::Replace(CYContext &context) { - code_.Replace(context); + context.ReplaceAll(code_); catch_->Replace(context); finally_->Replace(context); return this; @@ -840,9 +895,101 @@ CYStatement *Try::Replace(CYContext &context) { } } +CYExpression *CYTypeArrayOf::Replace_(CYContext &context, CYExpression *type) { + return next_->Replace(context, $ CYCall($ CYDirectMember(type, $ CYString("arrayOf")), $ CYArgument(size_))); +} + +CYExpression *CYTypeBlockWith::Replace_(CYContext &context, CYExpression *type) { + return next_->Replace(context, $ CYCall($ CYDirectMember(type, $ CYString("blockWith")), parameters_->Argument(context))); +} + +CYExpression *CYTypeConstant::Replace_(CYContext &context, CYExpression *type) { + return next_->Replace(context, $ CYCall($ CYDirectMember(type, $ CYString("constant")))); +} + +CYStatement *CYTypeDefinition::Replace(CYContext &context) { + return $E($ CYAssign($V(typed_->identifier_), typed_->Replace(context))); +} + +CYExpression *CYTypeError::Replace(CYContext &context) { + _assert(false); + return NULL; +} + +CYExpression *CYTypeModifier::Replace(CYContext &context, CYExpression *type) { $T(type) + return Replace_(context, type); +} + +CYExpression *CYTypeFunctionWith::Replace_(CYContext &context, CYExpression *type) { + return next_->Replace(context, $ CYCall($ CYDirectMember(type, $ CYString("functionWith")), parameters_->Argument(context))); +} + +CYExpression *CYTypeLong::Replace(CYContext &context) { + return $ CYCall($ CYDirectMember(specifier_->Replace(context), $ CYString("long"))); +} + +CYExpression *CYTypePointerTo::Replace_(CYContext &context, CYExpression *type) { + return next_->Replace(context, $ CYCall($ CYDirectMember(type, $ CYString("pointerTo")))); +} + +CYExpression *CYTypeShort::Replace(CYContext &context) { + return $ CYCall($ CYDirectMember(specifier_->Replace(context), $ CYString("short"))); +} + +CYExpression *CYTypeSigned::Replace(CYContext &context) { + return $ CYCall($ CYDirectMember(specifier_->Replace(context), $ CYString("signed"))); +} + +CYExpression *CYTypeUnsigned::Replace(CYContext &context) { + return $ CYCall($ CYDirectMember(specifier_->Replace(context), $ CYString("unsigned"))); +} + +CYExpression *CYTypeVariable::Replace(CYContext &context) { + return $V(name_); +} + +CYExpression *CYTypeVoid::Replace(CYContext &context) { + return $N1($V("Type"), $ CYString("v")); +} + +CYExpression *CYTypeVolatile::Replace_(CYContext &context, CYExpression *type) { + return next_->Replace(context, $ CYCall($ CYDirectMember(type, $ CYString("volatile")))); +} + +CYExpression *CYTypedIdentifier::Replace(CYContext &context) { + return modifier_->Replace(context, specifier_->Replace(context)); +} + +CYTypeFunctionWith *CYTypedIdentifier::Function() { + CYTypeModifier **modifier(&modifier_); + if (*modifier == NULL) + return NULL; + while ((*modifier)->next_ != NULL) + modifier = &(*modifier)->next_; + CYTypeFunctionWith *function((*modifier)->Function()); + if (function == NULL) + return NULL; + *modifier = NULL; + return function; +} + +CYArgument *CYTypedParameter::Argument(CYContext &context) { $T(NULL) + return $ CYArgument(typed_->Replace(context), next_->Argument(context)); +} + +CYFunctionParameter *CYTypedParameter::Parameters(CYContext &context) { $T(NULL) + return $ CYFunctionParameter($ CYDeclaration(typed_->identifier_ ?: context.Unique()), next_->Parameters(context)); +} + +CYExpression *CYTypedParameter::TypeSignature(CYContext &context, CYExpression *prefix) { $T(prefix) + return next_->TypeSignature(context, $ CYAdd(prefix, typed_->Replace(context))); +} + CYStatement *CYVar::Replace(CYContext &context) { declarations_->Replace(context); - return $E(declarations_->Compound(context)); + if (CYExpression *expression = declarations_->Expression(context)) + return $E(expression); + return $ CYEmpty(); } CYExpression *CYVariable::Replace(CYContext &context) { @@ -850,15 +997,19 @@ CYExpression *CYVariable::Replace(CYContext &context) { return this; } +CYFunctionParameter *CYVariable::Parameter() const { + return $ CYFunctionParameter($ CYDeclaration(name_)); +} + CYStatement *CYWhile::Replace(CYContext &context) { context.Replace(test_); - context.Replace(code_); + context.ReplaceAll(code_); return this; } CYStatement *CYWith::Replace(CYContext &context) { context.Replace(scope_); - context.Replace(code_); + context.ReplaceAll(code_); return this; }