Replace.cpp

   1 /* Cycript - Optimizing JavaScript Compiler/Runtime
   2  * Copyright (C) 2009-2015  Jay Freeman (saurik)
   3 */
   4
   5 /* GNU Affero General Public License, Version 3 {{{ */
   6 /*
   7  * This program is free software: you can redistribute it and/or modify
   8  * it under the terms of the GNU Affero General Public License as published by
   9  * the Free Software Foundation, either version 3 of the License, or
  10  * (at your option) any later version.
  11
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU Affero General Public License for more details.
  16
  17  * You should have received a copy of the GNU Affero General Public License
  18  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 **/
  20 /* }}} */
  21
  22 #include <iomanip>
  23 #include <map>
  24
  25 #include "Replace.hpp"
  26 #include "Syntax.hpp"
  27
  28 CYFunctionExpression *CYNonLocalize(CYContext &context, CYFunctionExpression *function) {
  29     function->nonlocal_ = context.nextlocal_;
  30     return function;
  31 }
  32
  33 CYFunctionExpression *CYSuperize(CYContext &context, CYFunctionExpression *function) {
  34     function->super_ = context.super_;
  35     return function;
  36 }
  37
  38 CYStatement *CYDefineProperty(CYExpression *object, CYExpression *name, bool configurable, bool enumerable, CYProperty *descriptor) {
  39     return $E($C3($M($V("Object"), $S("defineProperty")), object, name, $ CYObject(CYList<CYProperty>()
  40         ->* (configurable ? $ CYPropertyValue($S("configurable"), $ CYTrue()) : NULL)
  41         ->* (enumerable ? $ CYPropertyValue($S("enumerable"), $ CYTrue()) : NULL)
  42         ->* descriptor)));
  43 }
  44
  45 static void CYImplicitReturn(CYStatement *&code) {
  46     if (CYStatement *&last = CYGetLast(code))
  47         last = last->Return();
  48 }
  49
  50 CYExpression *CYAdd::Replace(CYContext &context) {
  51     CYInfix::Replace(context);
  52
  53     CYString *lhs(dynamic_cast<CYString *>(lhs_));
  54     CYString *rhs(dynamic_cast<CYString *>(rhs_));
  55
  56     if (lhs != NULL || rhs != NULL) {
  57         if (lhs == NULL) {
  58             lhs = lhs_->String(context);
  59             if (lhs == NULL)
  60                 return this;
  61         } else if (rhs == NULL) {
  62             rhs = rhs_->String(context);
  63             if (rhs == NULL)
  64                 return this;
  65         }
  66
  67         return lhs->Concat(context, rhs);
  68     }
  69
  70     if (CYNumber *lhn = lhs_->Number(context))
  71         if (CYNumber *rhn = rhs_->Number(context))
  72             return $D(lhn->Value() + rhn->Value());
  73
  74     return this;
  75 }
  76
  77 CYExpression *CYAddressOf::Replace(CYContext &context) {
  78     return $C0($M(rhs_, $S("$cya")));
  79 }
  80
  81 CYTarget *CYApply::AddArgument(CYContext &context, CYExpression *value) {
  82     CYArgument **argument(&arguments_);
  83     while (*argument != NULL)
  84         argument = &(*argument)->next_;
  85     *argument = $ CYArgument(value);
  86     return this;
  87 }
  88
  89 CYArgument *CYArgument::Replace(CYContext &context) { $T(NULL)
  90     context.Replace(value_);
  91     next_ = next_->Replace(context);
  92
  93     if (value_ == NULL) {
  94         if (next_ == NULL)
  95             return NULL;
  96         else
  97             value_ = $U;
  98     }
  99
 100     return this;
 101 }
 102
 103 CYTarget *CYArray::Replace(CYContext &context) {
 104     if (elements_ != NULL)
 105         elements_->Replace(context);
 106     return this;
 107 }
 108
 109 CYTarget *CYArrayComprehension::Replace(CYContext &context) {
 110     CYVariable *cyv($V("$cyv"));
 111
 112     return $C0($F(NULL, $P1($L($I("$cyv")), comprehensions_->Parameters(context)), $$
 113         ->* $E($ CYAssign(cyv, $ CYArray()))
 114         ->* comprehensions_->Replace(context, $E($C1($M(cyv, $S("push")), expression_)))
 115         ->* $ CYReturn(cyv)
 116     ));
 117 }
 118
 119 CYExpression *CYAssignment::Replace(CYContext &context) {
 120     context.Replace(lhs_);
 121     context.Replace(rhs_);
 122     return this;
 123 }
 124
 125 CYStatement *CYBlock::Return() {
 126     CYImplicitReturn(code_);
 127     return this;
 128 }
 129
 130 CYStatement *CYBlock::Replace(CYContext &context) {
 131     CYScope scope(true, context);
 132     context.ReplaceAll(code_);
 133     scope.Close(context);
 134
 135     if (code_ == NULL)
 136         return $ CYEmpty();
 137     return this;
 138 }
 139
 140 CYStatement *CYBreak::Replace(CYContext &context) {
 141     return this;
 142 }
 143
 144 CYTarget *CYCall::Replace(CYContext &context) {
 145     context.Replace(function_);
 146     arguments_->Replace(context);
 147     return this;
 148 }
 149
 150 namespace cy {
 151 namespace Syntax {
 152
 153 void Catch::Replace(CYContext &context) { $T()
 154     CYScope scope(true, context);
 155
 156     name_ = name_->Replace(context, CYIdentifierCatch);
 157
 158     context.ReplaceAll(code_);
 159     scope.Close(context);
 160 }
 161
 162 } }
 163
 164 CYTarget *CYClassExpression::Replace(CYContext &context) {
 165     CYBuilder builder;
 166
 167     CYIdentifier *super(context.Unique());
 168
 169     CYIdentifier *old(context.super_);
 170     context.super_ = super;
 171
 172     CYIdentifier *constructor(context.Unique());
 173     CYForEach (member, tail_->static_)
 174         member->Replace(context, builder, $V(constructor), true);
 175
 176     CYIdentifier *prototype(context.Unique());
 177     CYForEach (member, tail_->instance_)
 178         member->Replace(context, builder, $V(prototype), true);
 179
 180     if (tail_->constructor_ == NULL)
 181         tail_->constructor_ = $ CYFunctionExpression(NULL, NULL, NULL);
 182     tail_->constructor_ = CYSuperize(context, tail_->constructor_);
 183
 184     context.super_ = old;
 185
 186     return $C1($ CYFunctionExpression(NULL, $P($L(super)), $$
 187         ->* $ CYVar($L1($L(constructor, tail_->constructor_)))
 188         ->* $ CYVar($L1($L(prototype, $ CYFunctionExpression(NULL, NULL, NULL))))
 189         ->* $E($ CYAssign($M($V(prototype), $S("prototype")), $M($V(super), $S("prototype"))))
 190         ->* $E($ CYAssign($V(prototype), $N($V(prototype))))
 191         ->* CYDefineProperty($V(prototype), $S("constructor"), false, false, $ CYPropertyValue($S("value"), $V(constructor)))
 192         ->* $ CYVar(builder.declarations_)
 193         ->* builder.statements_
 194         ->* CYDefineProperty($V(constructor), $S("prototype"), false, false, $ CYPropertyValue($S("value"), $V(prototype)))
 195         ->* $ CYReturn($V(constructor))
 196     ), tail_->extends_ ?: $V($I("Object")));
 197 }
 198
 199 CYStatement *CYClassStatement::Replace(CYContext &context) {
 200     return $ CYVar($L1($L(name_, $ CYClassExpression(name_, tail_))));
 201 }
 202
 203 void CYClause::Replace(CYContext &context) { $T()
 204     context.Replace(case_);
 205     context.ReplaceAll(code_);
 206     next_->Replace(context);
 207 }
 208
 209 CYExpression *CYCompound::Replace(CYContext &context) {
 210     context.Replace(expression_);
 211     context.Replace(next_);
 212
 213     if (CYCompound *compound = dynamic_cast<CYCompound *>(expression_)) {
 214         expression_ = compound->expression_;
 215         compound->expression_ = compound->next_;
 216         compound->next_ = next_;
 217         next_ = compound;
 218     }
 219
 220     return this;
 221 }
 222
 223 CYFunctionParameter *CYCompound::Parameter() const {
 224     CYFunctionParameter *next(next_->Parameter());
 225     if (next == NULL)
 226         return NULL;
 227
 228     CYFunctionParameter *parameter(expression_->Parameter());
 229     if (parameter == NULL)
 230         return NULL;
 231
 232     parameter->SetNext(next);
 233     return parameter;
 234 }
 235
 236 CYFunctionParameter *CYComprehension::Parameters(CYContext &context) const { $T(NULL)
 237     CYFunctionParameter *next(next_->Parameters(context));
 238     if (CYFunctionParameter *parameter = Parameter(context)) {
 239         parameter->SetNext(next);
 240         return parameter;
 241     } else
 242         return next;
 243 }
 244
 245 CYStatement *CYComprehension::Replace(CYContext &context, CYStatement *statement) const {
 246     return next_ == NULL ? statement : next_->Replace(context, statement);
 247 }
 248
 249 CYExpression *CYComputed::PropertyName(CYContext &context) {
 250     return expression_;
 251 }
 252
 253 CYExpression *CYCondition::Replace(CYContext &context) {
 254     context.Replace(test_);
 255     context.Replace(true_);
 256     context.Replace(false_);
 257     return this;
 258 }
 259
 260 void CYContext::NonLocal(CYStatement *&statements) {
 261     CYContext &context(*this);
 262
 263     if (nextlocal_ != NULL && nextlocal_->identifier_ != NULL) {
 264         CYIdentifier *cye($I("$cye")->Replace(context, CYIdentifierGlobal));
 265         CYIdentifier *unique(nextlocal_->identifier_->Replace(context, CYIdentifierGlobal));
 266
 267         CYStatement *declare(
 268             $ CYVar($L1($L(unique, $ CYObject()))));
 269
 270         cy::Syntax::Catch *rescue(
 271             $ cy::Syntax::Catch(cye, $$
 272                 ->* $ CYIf($ CYIdentical($M($V(cye), $S("$cyk")), $V(unique)), $$
 273                     ->* $ CYReturn($M($V(cye), $S("$cyv"))))
 274                 ->* $ cy::Syntax::Throw($V(cye))));
 275
 276         context.Replace(declare);
 277         rescue->Replace(context);
 278
 279         statements = $$
 280             ->* declare
 281             ->* $ cy::Syntax::Try(statements, rescue, NULL);
 282     }
 283 }
 284
 285 CYIdentifier *CYContext::Unique() {
 286     return $ CYIdentifier($pool.strcat("$cy", $pool.itoa(unique_++), NULL));
 287 }
 288
 289 CYStatement *CYContinue::Replace(CYContext &context) {
 290     return this;
 291 }
 292
 293 CYStatement *CYDebugger::Replace(CYContext &context) {
 294     return this;
 295 }
 296
 297 CYTarget *CYDeclaration::Target(CYContext &context) {
 298     return $V(identifier_);
 299 }
 300
 301 CYAssignment *CYDeclaration::Replace(CYContext &context, CYIdentifierKind kind) {
 302     identifier_ = identifier_->Replace(context, kind);
 303
 304     if (initialiser_ == NULL)
 305         return NULL;
 306
 307     CYAssignment *value($ CYAssign(Target(context), initialiser_));
 308     initialiser_ = NULL;
 309     return value;
 310 }
 311
 312 CYExpression *CYDeclarations::Replace(CYContext &context, CYIdentifierKind kind) { $T(NULL)
 313     CYAssignment *assignment(declaration_->Replace(context, kind));
 314     CYExpression *compound(next_->Replace(context, kind));
 315
 316     if (assignment != NULL)
 317         if (compound == NULL)
 318             compound = assignment;
 319         else
 320             compound = $ CYCompound(assignment, compound);
 321     return compound;
 322 }
 323
 324 CYFunctionParameter *CYDeclarations::Parameter(CYContext &context) { $T(NULL)
 325     return $ CYFunctionParameter($ CYDeclaration(declaration_->identifier_), next_->Parameter(context));
 326 }
 327
 328 CYArgument *CYDeclarations::Argument(CYContext &context) { $T(NULL)
 329     return $ CYArgument(declaration_->initialiser_, next_->Argument(context));
 330 }
 331
 332 CYTarget *CYDirectMember::Replace(CYContext &context) {
 333     context.Replace(object_);
 334     context.Replace(property_);
 335     return this;
 336 }
 337
 338 CYStatement *CYDoWhile::Replace(CYContext &context) {
 339     context.Replace(test_);
 340     context.ReplaceAll(code_);
 341     return this;
 342 }
 343
 344 void CYElementSpread::Replace(CYContext &context) {
 345     context.Replace(value_);
 346 }
 347
 348 void CYElementValue::Replace(CYContext &context) {
 349     context.Replace(value_);
 350     if (next_ != NULL)
 351         next_->Replace(context);
 352 }
 353
 354 CYForInitializer *CYEmpty::Replace(CYContext &context) {
 355     return NULL;
 356 }
 357
 358 CYTarget *CYEncodedType::Replace(CYContext &context) {
 359     return typed_->Replace(context);
 360 }
 361
 362 CYTarget *CYEval::Replace(CYContext &context) {
 363     context.scope_->Damage();
 364     if (arguments_ != NULL)
 365         arguments_->value_ = $C1($M($V("Cycript"), $S("compile")), arguments_->value_);
 366     return $C($V("eval"), arguments_);
 367 }
 368
 369 CYStatement *CYExpress::Return() {
 370     return $ CYReturn(expression_);
 371 }
 372
 373 CYForInitializer *CYExpress::Replace(CYContext &context) {
 374     context.Replace(expression_);
 375     return this;
 376 }
 377
 378 CYTarget *CYExpression::AddArgument(CYContext &context, CYExpression *value) {
 379     return $C1(this, value);
 380 }
 381
 382 CYFunctionParameter *CYExpression::Parameter() const {
 383     return NULL;
 384 }
 385
 386 CYStatement *CYExternal::Replace(CYContext &context) {
 387     return $E($ CYAssign($V(typed_->identifier_), $C1(typed_->Replace(context), $C2($V("dlsym"), $V("RTLD_DEFAULT"), $S(typed_->identifier_->Word())))));
 388 }
 389
 390 CYNumber *CYFalse::Number(CYContext &context) {
 391     return $D(0);
 392 }
 393
 394 CYString *CYFalse::String(CYContext &context) {
 395     return $S("false");
 396 }
 397
 398 CYExpression *CYFatArrow::Replace(CYContext &context) {
 399     CYFunctionExpression *function($ CYFunctionExpression(NULL, parameters_, code_));
 400     function->this_.SetNext(context.this_);
 401     return function;
 402 }
 403
 404 void CYFinally::Replace(CYContext &context) { $T()
 405     CYScope scope(true, context);
 406     context.ReplaceAll(code_);
 407     scope.Close(context);
 408 }
 409
 410 CYStatement *CYFor::Replace(CYContext &context) {
 411     CYScope outer(true, context);
 412     context.Replace(initialiser_);
 413
 414     context.Replace(test_);
 415
 416     {
 417         CYScope inner(true, context);
 418         context.ReplaceAll(code_);
 419         inner.Close(context);
 420     }
 421
 422     context.Replace(increment_);
 423
 424     outer.Close(context);
 425     return this;
 426 }
 427
 428 CYStatement *CYForLexical::Initialize(CYContext &context, CYExpression *value) {
 429     if (value == NULL) {
 430         if (declaration_->initialiser_ == NULL)
 431             return NULL;
 432         value = declaration_->initialiser_;
 433     }
 434
 435     return $ CYLet(constant_, $L1($ CYDeclaration(declaration_->identifier_, value)));
 436 }
 437
 438 CYTarget *CYForLexical::Replace(CYContext &context) {
 439     _assert(declaration_->Replace(context, CYIdentifierLexical) == NULL);
 440     return declaration_->Target(context);
 441 }
 442
 443 CYStatement *CYForIn::Replace(CYContext &context) {
 444     CYScope scope(true, context);
 445     context.Replace(initialiser_);
 446     context.Replace(set_);
 447     context.ReplaceAll(code_);
 448     scope.Close(context);
 449     return this;
 450 }
 451
 452 CYFunctionParameter *CYForInComprehension::Parameter(CYContext &context) const {
 453     return $ CYFunctionParameter(declaration_);
 454 }
 455
 456 CYStatement *CYForInComprehension::Replace(CYContext &context, CYStatement *statement) const {
 457     return $ CYForIn(declaration_->Target(context), set_, CYComprehension::Replace(context, statement));
 458 }
 459
 460 CYStatement *CYForOf::Replace(CYContext &context) {
 461     CYIdentifier *item(context.Unique()), *list(context.Unique());
 462
 463     return $ CYBlock($$
 464         ->* initialiser_->Initialize(context, NULL)
 465         ->* $ CYLet(false, $L2($L(list, set_), $L(item)))
 466         ->* $ CYForIn($V(item), $V(list), $ CYBlock($$
 467             ->* initialiser_->Initialize(context, $M($V(list), $V(item)))
 468             ->* code_
 469     )));
 470 }
 471
 472 CYFunctionParameter *CYForOfComprehension::Parameter(CYContext &context) const {
 473     return $ CYFunctionParameter(declaration_);
 474 }
 475
 476 CYStatement *CYForOfComprehension::Replace(CYContext &context, CYStatement *statement) const {
 477     CYIdentifier *cys($I("$cys"));
 478
 479     return $E($C0($F(NULL, $P1($L($I("$cys"))), $$
 480         ->* $E($ CYAssign($V(cys), set_))
 481         ->* $ CYForIn(declaration_->Target(context), $V(cys), $ CYBlock($$
 482             ->* $E($ CYAssign(declaration_->Target(context), $M($V(cys), declaration_->Target(context))))
 483             ->* CYComprehension::Replace(context, statement)
 484         ))
 485     )));
 486 }
 487
 488 CYStatement *CYForVariable::Initialize(CYContext &context, CYExpression *value) {
 489     if (value == NULL) {
 490         if (declaration_->initialiser_ == NULL)
 491             return NULL;
 492         value = declaration_->initialiser_;
 493     }
 494
 495     return $ CYVar($L1($ CYDeclaration(declaration_->identifier_, value)));
 496 }
 497
 498 CYTarget *CYForVariable::Replace(CYContext &context) {
 499     _assert(declaration_->Replace(context, CYIdentifierVariable) == NULL);
 500     return declaration_->Target(context);
 501 }
 502
 503 // XXX: this is evil evil black magic. don't ask, don't tell... don't believe!
 504 #define MappingSet "0etnirsoalfucdphmgyvbxTwSNECAFjDLkMOIBPqzRH$_WXUVGYKQJZ"
 505 //#define MappingSet "0abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ$_"
 506
 507 void CYFunction::Replace(CYContext &context) {
 508     CYThisScope *_this(context.this_);
 509     context.this_ = &this_;
 510     context.this_ = CYGetLast(context.this_);
 511
 512     CYIdentifier *super(context.super_);
 513     context.super_ = super_;
 514
 515     CYNonLocal *nonlocal(context.nonlocal_);
 516     CYNonLocal *nextlocal(context.nextlocal_);
 517
 518     bool localize;
 519     if (nonlocal_ != NULL) {
 520         localize = false;
 521         context.nonlocal_ = nonlocal_;
 522     } else {
 523         localize = true;
 524         nonlocal_ = $ CYNonLocal();
 525         context.nextlocal_ = nonlocal_;
 526     }
 527
 528     CYScope scope(!localize, context);
 529
 530     parameters_->Replace(context, code_);
 531
 532     context.ReplaceAll(code_);
 533
 534     if (implicit_)
 535         CYImplicitReturn(code_);
 536
 537     if (CYIdentifier *identifier = this_.identifier_) {
 538         context.scope_->Declare(context, identifier, CYIdentifierVariable);
 539         code_ = $$
 540             ->* $E($ CYAssign($V(identifier), $ CYThis()))
 541             ->* code_;
 542     }
 543
 544     if (localize)
 545         context.NonLocal(code_);
 546
 547     context.nextlocal_ = nextlocal;
 548     context.nonlocal_ = nonlocal;
 549
 550     context.super_ = super;
 551     context.this_ = _this;
 552
 553     scope.Close(context, code_);
 554 }
 555
 556 CYTarget *CYFunctionExpression::Replace(CYContext &context) {
 557     CYScope scope(false, context);
 558     if (name_ != NULL)
 559         name_ = name_->Replace(context, CYIdentifierOther);
 560
 561     CYFunction::Replace(context);
 562     scope.Close(context);
 563     return this;
 564 }
 565
 566 void CYFunctionParameter::Replace(CYContext &context, CYStatement *&statements) { $T()
 567     CYAssignment *assignment(initialiser_->Replace(context, CYIdentifierArgument));
 568
 569     next_->Replace(context, statements);
 570
 571     if (assignment != NULL)
 572         statements = $$
 573             ->* $ CYIf($ CYIdentical($ CYTypeOf(initialiser_->Target(context)), $S("undefined")), $$
 574                 ->* $E(assignment))
 575             ->* statements;
 576 }
 577
 578 CYStatement *CYFunctionStatement::Replace(CYContext &context) {
 579     name_ = name_->Replace(context, CYIdentifierOther);
 580     CYFunction::Replace(context);
 581     return this;
 582 }
 583
 584 CYIdentifier *CYIdentifier::Replace(CYContext &context, CYIdentifierKind kind) {
 585     if (next_ == this)
 586         return this;
 587     if (next_ != NULL)
 588         return next_->Replace(context, kind);
 589     next_ = context.scope_->Declare(context, this, kind)->identifier_;
 590     return next_;
 591 }
 592
 593 CYStatement *CYIf::Return() {
 594     CYImplicitReturn(true_);
 595     CYImplicitReturn(false_);
 596     return this;
 597 }
 598
 599 CYStatement *CYIf::Replace(CYContext &context) {
 600     context.Replace(test_);
 601     context.ReplaceAll(true_);
 602     context.ReplaceAll(false_);
 603     return this;
 604 }
 605
 606 CYFunctionParameter *CYIfComprehension::Parameter(CYContext &context) const {
 607     return NULL;
 608 }
 609
 610 CYStatement *CYIfComprehension::Replace(CYContext &context, CYStatement *statement) const {
 611     return $ CYIf(test_, CYComprehension::Replace(context, statement));
 612 }
 613
 614 CYStatement *CYImport::Replace(CYContext &context) {
 615     return $ CYVar($L1($L($I(module_->part_->Word()), $C1($V("require"), module_->Replace(context, "/")))));
 616 }
 617
 618 CYTarget *CYIndirect::Replace(CYContext &context) {
 619     return $M(rhs_, $S("$cyi"));
 620 }
 621
 622 CYTarget *CYIndirectMember::Replace(CYContext &context) {
 623     return $M($ CYIndirect(object_), property_);
 624 }
 625
 626 CYExpression *CYInfix::Replace(CYContext &context) {
 627     context.Replace(lhs_);
 628     context.Replace(rhs_);
 629     return this;
 630 }
 631
 632 CYStatement *CYLabel::Replace(CYContext &context) {
 633     context.Replace(statement_);
 634     return this;
 635 }
 636
 637 CYTarget *CYLambda::Replace(CYContext &context) {
 638     return $N2($V("Functor"), $ CYFunctionExpression(NULL, parameters_->Parameters(context), code_), parameters_->TypeSignature(context, typed_->Replace(context)));
 639 }
 640
 641 CYForInitializer *CYLet::Replace(CYContext &context) {
 642     if (CYExpression *expression = declarations_->Replace(context, CYIdentifierLexical))
 643         return $E(expression);
 644     return $ CYEmpty();
 645 }
 646
 647 CYFunctionExpression *CYMethod::Constructor() {
 648     return NULL;
 649 }
 650
 651 void CYMethod::Replace(CYContext &context) {
 652     CYFunction::Replace(context);
 653 }
 654
 655 CYString *CYModule::Replace(CYContext &context, const char *separator) const {
 656     if (next_ == NULL)
 657         return $ CYString(part_);
 658     return $ CYString($pool.strcat(next_->Replace(context, separator)->Value(), separator, part_->Word(), NULL));
 659 }
 660
 661 CYExpression *CYMultiply::Replace(CYContext &context) {
 662     CYInfix::Replace(context);
 663
 664     if (CYNumber *lhn = lhs_->Number(context))
 665         if (CYNumber *rhn = rhs_->Number(context))
 666             return $D(lhn->Value() * rhn->Value());
 667
 668     return this;
 669 }
 670
 671 namespace cy {
 672 namespace Syntax {
 673
 674 CYTarget *New::AddArgument(CYContext &context, CYExpression *value) {
 675     CYSetLast(arguments_) = $ CYArgument(value);
 676     return this;
 677 }
 678
 679 CYTarget *New::Replace(CYContext &context) {
 680     context.Replace(constructor_);
 681     arguments_->Replace(context);
 682     return this;
 683 }
 684
 685 } }
 686
 687 CYNumber *CYNull::Number(CYContext &context) {
 688     return $D(0);
 689 }
 690
 691 CYString *CYNull::String(CYContext &context) {
 692     return $S("null");
 693 }
 694
 695 CYNumber *CYNumber::Number(CYContext &context) {
 696     return this;
 697 }
 698
 699 CYString *CYNumber::String(CYContext &context) {
 700     // XXX: there is a precise algorithm for this
 701     return $S($pool.sprintf(24, "%.17g", Value()));
 702 }
 703
 704 CYExpression *CYNumber::PropertyName(CYContext &context) {
 705     return String(context);
 706 }
 707
 708 CYTarget *CYObject::Replace(CYContext &context) {
 709     CYBuilder builder;
 710     if (properties_ != NULL)
 711         properties_ = properties_->ReplaceAll(context, builder, $ CYThis(), false);
 712
 713     if (builder) {
 714         return $C1($M($ CYFunctionExpression(NULL, builder.declarations_->Parameter(context),
 715             builder.statements_
 716                 ->* $ CYReturn($ CYThis())
 717         ), $S("call")), this, builder.declarations_->Argument(context));
 718     }
 719
 720     CYForEach (property, properties_)
 721         property->Replace(context);
 722     return this;
 723 }
 724
 725 CYTarget *CYParenthetical::Replace(CYContext &context) {
 726     // XXX: return expression_;
 727     context.Replace(expression_);
 728     return this;
 729 }
 730
 731 CYExpression *CYPostfix::Replace(CYContext &context) {
 732     context.Replace(lhs_);
 733     return this;
 734 }
 735
 736 CYExpression *CYPrefix::Replace(CYContext &context) {
 737     context.Replace(rhs_);
 738     return this;
 739 }
 740
 741 CYProperty *CYProperty::ReplaceAll(CYContext &context, CYBuilder &builder, CYExpression *self, bool update) {
 742     update |= name_->Computed();
 743     if (update)
 744         Replace(context, builder, self, false);
 745     if (next_ != NULL)
 746         next_ = next_->ReplaceAll(context, builder, self, update);
 747     return update ? next_ : this;
 748 }
 749
 750 void CYProperty::Replace(CYContext &context, CYBuilder &builder, CYExpression *self, bool protect) {
 751     CYExpression *name(name_->PropertyName(context));
 752     if (name_->Computed()) {
 753         CYIdentifier *unique(context.Unique());
 754         builder.declarations_
 755             ->* $L1($L(unique, name));
 756         name = $V(unique);
 757     }
 758
 759     Replace(context, builder, self, name, protect);
 760 }
 761
 762 void CYPropertyGetter::Replace(CYContext &context, CYBuilder &builder, CYExpression *self, CYExpression *name, bool protect) {
 763     CYIdentifier *unique(context.Unique());
 764     builder.declarations_
 765         ->* $L1($L(unique, CYSuperize(context, $ CYFunctionExpression(NULL, parameters_, code_))));
 766     builder.statements_
 767         ->* CYDefineProperty(self, name, true, !protect, $ CYPropertyValue($S("get"), $V(unique)));
 768 }
 769
 770 CYFunctionExpression *CYPropertyMethod::Constructor() {
 771     return name_->Constructor() ? $ CYFunctionExpression(NULL, parameters_, code_) : NULL;
 772 }
 773
 774 void CYPropertyMethod::Replace(CYContext &context, CYBuilder &builder, CYExpression *self, CYExpression *name, bool protect) {
 775     CYIdentifier *unique(context.Unique());
 776     builder.declarations_
 777         ->* $L1($L(unique, CYSuperize(context, $ CYFunctionExpression(NULL, parameters_, code_))));
 778     builder.statements_
 779         ->* (!protect ? $E($ CYAssign($M(self, name), $V(unique))) :
 780             CYDefineProperty(self, name, true, !protect, $ CYPropertyValue($S("value"), $V(unique), $ CYPropertyValue($S("writable"), $ CYTrue()))));
 781 }
 782
 783 void CYPropertySetter::Replace(CYContext &context, CYBuilder &builder, CYExpression *self, CYExpression *name, bool protect) {
 784     CYIdentifier *unique(context.Unique());
 785     builder.declarations_
 786         ->* $L1($L(unique, CYSuperize(context, $ CYFunctionExpression(NULL, parameters_, code_))));
 787     builder.statements_
 788         ->* CYDefineProperty(self, name, true, !protect, $ CYPropertyValue($S("set"), $V(unique)));
 789 }
 790
 791 void CYPropertyValue::Replace(CYContext &context, CYBuilder &builder, CYExpression *self, CYExpression *name, bool protect) {
 792     _assert(!protect);
 793     CYIdentifier *unique(context.Unique());
 794     builder.declarations_
 795         ->* $L1($L(unique, value_));
 796     builder.statements_
 797         ->* $E($ CYAssign($M(self, name), $V(unique)));
 798 }
 799
 800 void CYPropertyValue::Replace(CYContext &context) {
 801     context.Replace(value_);
 802 }
 803
 804 void CYScript::Replace(CYContext &context) {
 805     CYScope scope(false, context);
 806     context.scope_->Damage();
 807
 808     context.nextlocal_ = $ CYNonLocal();
 809     context.ReplaceAll(code_);
 810     context.NonLocal(code_);
 811
 812     scope.Close(context, code_);
 813
 814     unsigned offset(0);
 815
 816     for (std::vector<CYIdentifier *>::const_iterator i(context.replace_.begin()); i != context.replace_.end(); ++i) {
 817         const char *name;
 818         if (context.options_.verbose_)
 819             name = $pool.strcat("$", $pool.itoa(offset++), NULL);
 820         else {
 821             char id[8];
 822             id[7] = '\0';
 823
 824           id:
 825             unsigned position(7), local(offset++ + 1);
 826
 827             do {
 828                 unsigned index(local % (sizeof(MappingSet) - 1));
 829                 local /= sizeof(MappingSet) - 1;
 830                 id[--position] = MappingSet[index];
 831             } while (local != 0);
 832
 833             if (scope.Lookup(context, id + position) != NULL)
 834                 goto id;
 835             // XXX: at some point, this could become a keyword
 836
 837             name = $pool.strmemdup(id + position, 7 - position);
 838         }
 839
 840         CYIdentifier *identifier(*i);
 841         _assert(identifier->next_ == identifier);
 842         identifier->next_ = $I(name);
 843     }
 844 }
 845
 846 CYStatement *CYReturn::Replace(CYContext &context) {
 847     if (context.nonlocal_ != NULL) {
 848         CYProperty *value(value_ == NULL ? NULL : $ CYPropertyValue($S("$cyv"), value_));
 849         return $ cy::Syntax::Throw($ CYObject(
 850             $ CYPropertyValue($S("$cyk"), $V(context.nonlocal_->Target(context)), value)
 851         ));
 852     }
 853
 854     context.Replace(value_);
 855     return this;
 856 }
 857
 858 CYTarget *CYRubyBlock::Replace(CYContext &context) {
 859     return call_->AddArgument(context, proc_->Replace(context));
 860 }
 861
 862 CYTarget *CYRubyBlock::AddArgument(CYContext &context, CYExpression *value) {
 863     return Replace(context)->AddArgument(context, value);
 864 }
 865
 866 CYTarget *CYRubyProc::Replace(CYContext &context) {
 867     CYFunctionExpression *function($ CYFunctionExpression(NULL, parameters_, code_));
 868     function = CYNonLocalize(context, function);
 869     function->implicit_ = true;
 870     return function;
 871 }
 872
 873 CYScope::CYScope(bool transparent, CYContext &context) :
 874     transparent_(transparent),
 875     parent_(context.scope_),
 876     damaged_(false),
 877     shadow_(NULL),
 878     internal_(NULL)
 879 {
 880     _assert(!transparent_ || parent_ != NULL);
 881     context.scope_ = this;
 882 }
 883
 884 void CYScope::Damage() {
 885     damaged_ = true;
 886     if (parent_ != NULL)
 887         parent_->Damage();
 888 }
 889
 890 CYIdentifierFlags *CYScope::Lookup(CYContext &context, const char *word) {
 891     CYForEach (i, internal_)
 892         if (strcmp(i->identifier_->Word(), word) == 0)
 893             return i;
 894     return NULL;
 895 }
 896
 897 CYIdentifierFlags *CYScope::Lookup(CYContext &context, CYIdentifier *identifier) {
 898     return Lookup(context, identifier->Word());
 899 }
 900
 901 CYIdentifierFlags *CYScope::Declare(CYContext &context, CYIdentifier *identifier, CYIdentifierKind kind) {
 902     _assert(identifier->next_ == NULL || identifier->next_ == identifier);
 903
 904     CYIdentifierFlags *existing(Lookup(context, identifier));
 905     if (existing == NULL)
 906         internal_ = $ CYIdentifierFlags(identifier, kind, internal_);
 907     ++internal_->count_;
 908     if (existing == NULL)
 909         return internal_;
 910
 911     switch (kind) {
 912         case CYIdentifierArgument:
 913         case CYIdentifierCatch:
 914         case CYIdentifierMagic:
 915             _assert(false);
 916         default:
 917             break;
 918     }
 919
 920     if (existing->kind_ == CYIdentifierGlobal)
 921         existing->kind_ = kind;
 922     else if (kind == CYIdentifierGlobal)
 923         ;
 924     else if (existing->kind_ == CYIdentifierLexical || kind == CYIdentifierLexical)
 925         _assert(false); // XXX: throw new SyntaxError()
 926
 927     return existing;
 928 }
 929
 930 void CYScope::Merge(CYContext &context, const CYIdentifierFlags *flags) {
 931     _assert(flags->identifier_->next_ == flags->identifier_);
 932     CYIdentifierFlags *existing(Declare(context, flags->identifier_, flags->kind_));
 933     flags->identifier_->next_ = existing->identifier_;
 934
 935     existing->count_ += flags->count_;
 936     if (existing->offset_ < flags->offset_)
 937         existing->offset_ = flags->offset_;
 938 }
 939
 940 void CYScope::Close(CYContext &context, CYStatement *&statements) {
 941     Close(context);
 942
 943     CYList<CYDeclarations> declarations;
 944
 945     CYForEach (i, internal_)
 946         if (i->kind_ == CYIdentifierVariable)
 947             declarations
 948                 ->* $ CYDeclarations($ CYDeclaration(i->identifier_));
 949
 950     if (declarations) {
 951         CYVar *var($ CYVar(declarations));
 952         var->SetNext(statements);
 953         statements = var;
 954     }
 955 }
 956
 957 void CYScope::Close(CYContext &context) {
 958     context.scope_ = parent_;
 959
 960     CYForEach (i, internal_) {
 961         _assert(i->identifier_->next_ == i->identifier_);
 962     switch (i->kind_) {
 963         case CYIdentifierArgument: {
 964             _assert(!transparent_);
 965         } break;
 966
 967         case CYIdentifierLexical: {
 968             if (!damaged_) {
 969                 CYIdentifier *replace(context.Unique());
 970                 replace->next_ = replace;
 971                 i->identifier_->next_ = replace;
 972                 i->identifier_ = replace;
 973             }
 974
 975             if (!transparent_)
 976                 i->kind_ = CYIdentifierVariable;
 977             else
 978                 parent_->Declare(context, i->identifier_, CYIdentifierVariable);
 979         } break;
 980
 981         case CYIdentifierVariable: {
 982             if (transparent_) {
 983                 parent_->Declare(context, i->identifier_, i->kind_);
 984                 i->kind_ = CYIdentifierGlobal;
 985             }
 986         } break;
 987     default:; } }
 988
 989     if (damaged_)
 990         return;
 991
 992     typedef std::multimap<unsigned, CYIdentifier *> CYIdentifierOffsetMap;
 993     CYIdentifierOffsetMap offsets;
 994
 995     CYForEach (i, internal_) {
 996         _assert(i->identifier_->next_ == i->identifier_);
 997     switch (i->kind_) {
 998         case CYIdentifierArgument:
 999         case CYIdentifierVariable:
1000             offsets.insert(CYIdentifierOffsetMap::value_type(i->offset_, i->identifier_));
1001         break;
1002     default:; } }
1003
1004     unsigned offset(0);
1005
1006     for (CYIdentifierOffsetMap::const_iterator i(offsets.begin()); i != offsets.end(); ++i) {
1007         if (offset < i->first)
1008             offset = i->first;
1009         CYIdentifier *identifier(i->second);
1010
1011         if (offset >= context.replace_.size())
1012             context.replace_.resize(offset + 1, NULL);
1013         CYIdentifier *&replace(context.replace_[offset++]);
1014
1015         if (replace == NULL)
1016             replace = identifier;
1017         else {
1018             _assert(replace->next_ == replace);
1019             identifier->next_ = replace;
1020         }
1021     }
1022
1023     if (parent_ == NULL)
1024         return;
1025
1026     CYForEach (i, internal_) {
1027     switch (i->kind_) {
1028         case CYIdentifierGlobal: {
1029             if (i->offset_ < offset)
1030                 i->offset_ = offset;
1031             parent_->Merge(context, i);
1032         } break;
1033     default:; } }
1034 }
1035
1036 CYElementValue *CYSpan::Replace(CYContext &context) { $T(NULL)
1037     return $ CYElementValue(expression_, $ CYElementValue(string_, next_->Replace(context)));
1038 }
1039
1040 CYStatement *CYStatement::Return() {
1041     return this;
1042 }
1043
1044 CYString *CYString::Concat(CYContext &context, CYString *rhs) const {
1045     size_t size(size_ + rhs->size_);
1046     char *value($ char[size + 1]);
1047     memcpy(value, value_, size_);
1048     memcpy(value + size_, rhs->value_, rhs->size_);
1049     value[size] = '\0';
1050     return $S(value, size);
1051 }
1052
1053 CYNumber *CYString::Number(CYContext &context) {
1054     // XXX: there is a precise algorithm for this
1055     return NULL;
1056 }
1057
1058 CYExpression *CYString::PropertyName(CYContext &context) {
1059     return this;
1060 }
1061
1062 CYString *CYString::String(CYContext &context) {
1063     return this;
1064 }
1065
1066 CYTarget *CYSuperAccess::Replace(CYContext &context) {
1067     return $C1($M($M($M($V(context.super_), $S("prototype")), property_), $S("bind")), $ CYThis());
1068 }
1069
1070 CYTarget *CYSuperCall::Replace(CYContext &context) {
1071     return $C($C1($M($V(context.super_), $S("bind")), $ CYThis()), arguments_);
1072 }
1073
1074 CYStatement *CYSwitch::Replace(CYContext &context) {
1075     context.Replace(value_);
1076     clauses_->Replace(context);
1077     return this;
1078 }
1079
1080 CYStatement *CYTarget::Initialize(CYContext &context, CYExpression *value) {
1081     if (value == NULL)
1082         return NULL;
1083     return $E($ CYAssign(this, value));
1084 }
1085
1086 CYTarget *CYTemplate::Replace(CYContext &context) {
1087     return $C2($M($M($M($V("String"), $S("prototype")), $S("concat")), $S("apply")), $S(""), $ CYArray($ CYElementValue(string_, spans_->Replace(context))));
1088 }
1089
1090 CYTarget *CYThis::Replace(CYContext &context) {
1091     if (context.this_ != NULL)
1092         return $V(context.this_->Identifier(context));
1093     return this;
1094 }
1095
1096 namespace cy {
1097 namespace Syntax {
1098
1099 CYStatement *Throw::Replace(CYContext &context) {
1100     context.Replace(value_);
1101     return this;
1102 }
1103
1104 } }
1105
1106 CYTarget *CYTrivial::Replace(CYContext &context) {
1107     return this;
1108 }
1109
1110 CYNumber *CYTrue::Number(CYContext &context) {
1111     return $D(1);
1112 }
1113
1114 CYString *CYTrue::String(CYContext &context) {
1115     return $S("true");
1116 }
1117
1118 namespace cy {
1119 namespace Syntax {
1120
1121 CYStatement *Try::Replace(CYContext &context) {
1122     CYScope scope(true, context);
1123     context.ReplaceAll(code_);
1124     scope.Close(context);
1125
1126     catch_->Replace(context);
1127     finally_->Replace(context);
1128     return this;
1129 }
1130
1131 } }
1132
1133 CYTarget *CYTypeArrayOf::Replace_(CYContext &context, CYTarget *type) {
1134     return next_->Replace(context, $ CYCall($ CYDirectMember(type, $ CYString("arrayOf")), $ CYArgument(size_)));
1135 }
1136
1137 CYTarget *CYTypeBlockWith::Replace_(CYContext &context, CYTarget *type) {
1138     return next_->Replace(context, $ CYCall($ CYDirectMember(type, $ CYString("blockWith")), parameters_->Argument(context)));
1139 }
1140
1141 CYTarget *CYTypeConstant::Replace_(CYContext &context, CYTarget *type) {
1142     return next_->Replace(context, $ CYCall($ CYDirectMember(type, $ CYString("constant"))));
1143 }
1144
1145 CYStatement *CYTypeDefinition::Replace(CYContext &context) {
1146     return $E($ CYAssign($V(typed_->identifier_), typed_->Replace(context)));
1147 }
1148
1149 CYTarget *CYTypeError::Replace(CYContext &context) {
1150     _assert(false);
1151     return NULL;
1152 }
1153
1154 CYTarget *CYTypeModifier::Replace(CYContext &context, CYTarget *type) { $T(type)
1155     return Replace_(context, type);
1156 }
1157
1158 CYTarget *CYTypeFunctionWith::Replace_(CYContext &context, CYTarget *type) {
1159     return next_->Replace(context, $ CYCall($ CYDirectMember(type, $ CYString("functionWith")), parameters_->Argument(context)));
1160 }
1161
1162 CYTarget *CYTypeLong::Replace(CYContext &context) {
1163     return $ CYCall($ CYDirectMember(specifier_->Replace(context), $ CYString("long")));
1164 }
1165
1166 CYTarget *CYTypePointerTo::Replace_(CYContext &context, CYTarget *type) {
1167     return next_->Replace(context, $ CYCall($ CYDirectMember(type, $ CYString("pointerTo"))));
1168 }
1169
1170 CYTarget *CYTypeShort::Replace(CYContext &context) {
1171     return $ CYCall($ CYDirectMember(specifier_->Replace(context), $ CYString("short")));
1172 }
1173
1174 CYTarget *CYTypeSigned::Replace(CYContext &context) {
1175     return $ CYCall($ CYDirectMember(specifier_->Replace(context), $ CYString("signed")));
1176 }
1177
1178 CYTarget *CYTypeUnsigned::Replace(CYContext &context) {
1179     return $ CYCall($ CYDirectMember(specifier_->Replace(context), $ CYString("unsigned")));
1180 }
1181
1182 CYTarget *CYTypeVariable::Replace(CYContext &context) {
1183     return $V(name_);
1184 }
1185
1186 CYTarget *CYTypeVoid::Replace(CYContext &context) {
1187     return $N1($V("Type"), $ CYString("v"));
1188 }
1189
1190 CYTarget *CYTypeVolatile::Replace_(CYContext &context, CYTarget *type) {
1191     return next_->Replace(context, $ CYCall($ CYDirectMember(type, $ CYString("volatile"))));
1192 }
1193
1194 CYTarget *CYTypedIdentifier::Replace(CYContext &context) {
1195     return modifier_->Replace(context, specifier_->Replace(context));
1196 }
1197
1198 CYTypeFunctionWith *CYTypedIdentifier::Function() {
1199     CYTypeModifier **modifier(&modifier_);
1200     if (*modifier == NULL)
1201         return NULL;
1202     while ((*modifier)->next_ != NULL)
1203         modifier = &(*modifier)->next_;
1204     CYTypeFunctionWith *function((*modifier)->Function());
1205     if (function == NULL)
1206         return NULL;
1207     *modifier = NULL;
1208     return function;
1209 }
1210
1211 CYArgument *CYTypedParameter::Argument(CYContext &context) { $T(NULL)
1212     return $ CYArgument(typed_->Replace(context), next_->Argument(context));
1213 }
1214
1215 CYFunctionParameter *CYTypedParameter::Parameters(CYContext &context) { $T(NULL)
1216     return $ CYFunctionParameter($ CYDeclaration(typed_->identifier_ ?: context.Unique()), next_->Parameters(context));
1217 }
1218
1219 CYExpression *CYTypedParameter::TypeSignature(CYContext &context, CYExpression *prefix) { $T(prefix)
1220     return next_->TypeSignature(context, $ CYAdd(prefix, typed_->Replace(context)));
1221 }
1222
1223 CYForInitializer *CYVar::Replace(CYContext &context) {
1224     if (CYExpression *expression = declarations_->Replace(context, CYIdentifierVariable))
1225         return $E(expression);
1226     return $ CYEmpty();
1227 }
1228
1229 CYTarget *CYVariable::Replace(CYContext &context) {
1230     name_ = name_->Replace(context, CYIdentifierGlobal);
1231     return this;
1232 }
1233
1234 CYFunctionParameter *CYVariable::Parameter() const {
1235     return $ CYFunctionParameter($ CYDeclaration(name_));
1236 }
1237
1238 CYStatement *CYWhile::Replace(CYContext &context) {
1239     context.Replace(test_);
1240     context.ReplaceAll(code_);
1241     return this;
1242 }
1243
1244 CYStatement *CYWith::Replace(CYContext &context) {
1245     context.Replace(scope_);
1246     context.ReplaceAll(code_);
1247     return this;
1248 }
1249
1250 CYExpression *CYWord::PropertyName(CYContext &context) {
1251     return $S(this);
1252 }