]> git.saurik.com Git - cycript.git/blame_incremental - Replace.cpp
Forgot to include license header in libcycript.cy.
[cycript.git] / Replace.cpp
... / ...
CommitLineData
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
28CYFunctionExpression *CYNonLocalize(CYContext &context, CYFunctionExpression *function) {
29 function->nonlocal_ = context.nextlocal_;
30 return function;
31}
32
33CYFunctionExpression *CYSuperize(CYContext &context, CYFunctionExpression *function) {
34 function->super_ = context.super_;
35 return function;
36}
37
38CYStatement *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
45static void CYImplicitReturn(CYStatement *&code) {
46 if (CYStatement *&last = CYGetLast(code))
47 last = last->Return();
48}
49
50CYExpression *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
77CYExpression *CYAddressOf::Replace(CYContext &context) {
78 return $C0($M(rhs_, $S("$cya")));
79}
80
81CYTarget *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
89CYArgument *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
103CYTarget *CYArray::Replace(CYContext &context) {
104 if (elements_ != NULL)
105 elements_->Replace(context);
106 return this;
107}
108
109CYTarget *CYArrayComprehension::Replace(CYContext &context) {
110 CYIdentifier *cyv(context.Unique());
111
112 return $C0($F(NULL, $P1($B(cyv), comprehensions_->Parameters(context)), $$
113 ->* $E($ CYAssign($V(cyv), $ CYArray()))
114 ->* comprehensions_->Replace(context, $E($C1($M($V(cyv), $S("push")), expression_)))
115 ->* $ CYReturn($V(cyv))
116 ));
117}
118
119CYExpression *CYAssignment::Replace(CYContext &context) {
120 context.Replace(lhs_);
121 context.Replace(rhs_);
122 return this;
123}
124
125CYStatement *CYBlock::Return() {
126 CYImplicitReturn(code_);
127 return this;
128}
129
130CYStatement *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
140CYStatement *CYBreak::Replace(CYContext &context) {
141 return this;
142}
143
144CYTarget *CYCall::Replace(CYContext &context) {
145 context.Replace(function_);
146 arguments_->Replace(context);
147 return this;
148}
149
150namespace cy {
151namespace Syntax {
152
153void 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
164CYTarget *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($B(super)), $$
187 ->* $ CYVar($B1($B(constructor, tail_->constructor_)))
188 ->* $ CYVar($B1($B(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.bindings_)
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
199CYStatement *CYClassStatement::Replace(CYContext &context) {
200 return $ CYVar($B1($B(name_, $ CYClassExpression(name_, tail_))));
201}
202
203void CYClause::Replace(CYContext &context) { $T()
204 context.Replace(value_);
205 context.ReplaceAll(code_);
206 next_->Replace(context);
207}
208
209CYExpression *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
223CYFunctionParameter *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
236CYFunctionParameter *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
245CYStatement *CYComprehension::Replace(CYContext &context, CYStatement *statement) const {
246 return next_ == NULL ? statement : next_->Replace(context, statement);
247}
248
249CYExpression *CYComputed::PropertyName(CYContext &context) {
250 return expression_;
251}
252
253CYExpression *CYCondition::Replace(CYContext &context) {
254 context.Replace(test_);
255 context.Replace(true_);
256 context.Replace(false_);
257 return this;
258}
259
260void 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($B1($B(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
285CYIdentifier *CYContext::Unique() {
286 return $ CYIdentifier($pool.strcat("$cy", $pool.itoa(unique_++), NULL));
287}
288
289CYStatement *CYContinue::Replace(CYContext &context) {
290 return this;
291}
292
293CYStatement *CYDebugger::Replace(CYContext &context) {
294 return this;
295}
296
297CYTarget *CYBinding::Target(CYContext &context) {
298 return $V(identifier_);
299}
300
301CYAssignment *CYBinding::Replace(CYContext &context, CYIdentifierKind kind) {
302 identifier_ = identifier_->Replace(context, kind);
303
304 if (initializer_ == NULL)
305 return NULL;
306
307 CYAssignment *value($ CYAssign(Target(context), initializer_));
308 initializer_ = NULL;
309 return value;
310}
311
312CYExpression *CYBindings::Replace(CYContext &context, CYIdentifierKind kind) { $T(NULL)
313 CYAssignment *assignment(binding_->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
324CYFunctionParameter *CYBindings::Parameter(CYContext &context) { $T(NULL)
325 return $ CYFunctionParameter($ CYBinding(binding_->identifier_), next_->Parameter(context));
326}
327
328CYArgument *CYBindings::Argument(CYContext &context) { $T(NULL)
329 return $ CYArgument(binding_->initializer_, next_->Argument(context));
330}
331
332CYTarget *CYDirectMember::Replace(CYContext &context) {
333 context.Replace(object_);
334 context.Replace(property_);
335 return this;
336}
337
338CYStatement *CYDoWhile::Replace(CYContext &context) {
339 context.Replace(test_);
340 context.ReplaceAll(code_);
341 return this;
342}
343
344void CYElementSpread::Replace(CYContext &context) {
345 context.Replace(value_);
346}
347
348void CYElementValue::Replace(CYContext &context) {
349 context.Replace(value_);
350 if (next_ != NULL)
351 next_->Replace(context);
352}
353
354CYForInitializer *CYEmpty::Replace(CYContext &context) {
355 return NULL;
356}
357
358CYTarget *CYEncodedType::Replace(CYContext &context) {
359 return typed_->Replace(context);
360}
361
362CYTarget *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
369CYStatement *CYExpress::Return() {
370 return $ CYReturn(expression_);
371}
372
373CYForInitializer *CYExpress::Replace(CYContext &context) {
374 context.Replace(expression_);
375 return this;
376}
377
378CYTarget *CYExpression::AddArgument(CYContext &context, CYExpression *value) {
379 return $C1(this, value);
380}
381
382CYFunctionParameter *CYExpression::Parameter() const {
383 return NULL;
384}
385
386CYStatement *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
390CYNumber *CYFalse::Number(CYContext &context) {
391 return $D(0);
392}
393
394CYString *CYFalse::String(CYContext &context) {
395 return $S("false");
396}
397
398CYExpression *CYFatArrow::Replace(CYContext &context) {
399 CYFunctionExpression *function($ CYFunctionExpression(NULL, parameters_, code_));
400 function->this_.SetNext(context.this_);
401 return function;
402}
403
404void CYFinally::Replace(CYContext &context) { $T()
405 CYScope scope(true, context);
406 context.ReplaceAll(code_);
407 scope.Close(context);
408}
409
410CYStatement *CYFor::Replace(CYContext &context) {
411 CYScope outer(true, context);
412 context.Replace(initializer_);
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
428CYStatement *CYForLexical::Initialize(CYContext &context, CYExpression *value) {
429 if (value == NULL) {
430 if (binding_->initializer_ == NULL)
431 return NULL;
432 value = binding_->initializer_;
433 }
434
435 return $ CYLexical(constant_, $B1($ CYBinding(binding_->identifier_, value)));
436}
437
438CYTarget *CYForLexical::Replace(CYContext &context) {
439 _assert(binding_->Replace(context, CYIdentifierLexical) == NULL);
440 return binding_->Target(context);
441}
442
443CYStatement *CYForIn::Replace(CYContext &context) {
444 CYScope scope(true, context);
445 context.Replace(initializer_);
446 context.Replace(iterable_);
447 context.ReplaceAll(code_);
448 scope.Close(context);
449 return this;
450}
451
452CYStatement *CYForInitialized::Replace(CYContext &context) {
453 CYAssignment *assignment(binding_->Replace(context, CYIdentifierVariable));
454 return $ CYBlock($$
455 ->* (assignment == NULL ? NULL : $ CYExpress(assignment))
456 ->* $ CYForIn(binding_->Target(context), iterable_, code_));
457}
458
459CYFunctionParameter *CYForInComprehension::Parameter(CYContext &context) const {
460 return $ CYFunctionParameter(binding_);
461}
462
463CYStatement *CYForInComprehension::Replace(CYContext &context, CYStatement *statement) const {
464 return $ CYForIn(binding_->Target(context), iterable_, CYComprehension::Replace(context, statement));
465}
466
467CYStatement *CYForOf::Replace(CYContext &context) {
468 CYIdentifier *item(context.Unique()), *list(context.Unique());
469
470 return $ CYBlock($$
471 ->* initializer_->Initialize(context, NULL)
472 ->* $ CYLexical(false, $B2($B(list, iterable_), $B(item)))
473 ->* $ CYForIn($V(item), $V(list), $ CYBlock($$
474 ->* initializer_->Initialize(context, $M($V(list), $V(item)))
475 ->* code_
476 )));
477}
478
479CYFunctionParameter *CYForOfComprehension::Parameter(CYContext &context) const {
480 return $ CYFunctionParameter(binding_);
481}
482
483CYStatement *CYForOfComprehension::Replace(CYContext &context, CYStatement *statement) const {
484 CYIdentifier *cys(context.Unique());
485
486 return $ CYBlock($$
487 ->* $ CYLexical(false, $B1($B(cys, iterable_)))
488 ->* $ CYForIn(binding_->Target(context), $V(cys), $ CYBlock($$
489 ->* $E($ CYAssign(binding_->Target(context), $M($V(cys), binding_->Target(context))))
490 ->* CYComprehension::Replace(context, statement)
491 )));
492}
493
494CYStatement *CYForVariable::Initialize(CYContext &context, CYExpression *value) {
495 if (value == NULL) {
496 if (binding_->initializer_ == NULL)
497 return NULL;
498 value = binding_->initializer_;
499 }
500
501 return $ CYVar($B1($ CYBinding(binding_->identifier_, value)));
502}
503
504CYTarget *CYForVariable::Replace(CYContext &context) {
505 _assert(binding_->Replace(context, CYIdentifierVariable) == NULL);
506 return binding_->Target(context);
507}
508
509// XXX: this is evil evil black magic. don't ask, don't tell... don't believe!
510#define MappingSet "0etnirsoalfucdphmgyvbxTwSNECAFjDLkMOIBPqzRH$_WXUVGYKQJZ"
511//#define MappingSet "0abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ$_"
512
513void CYFunction::Replace(CYContext &context) {
514 CYThisScope *_this(context.this_);
515 context.this_ = &this_;
516 context.this_ = CYGetLast(context.this_);
517
518 CYIdentifier *super(context.super_);
519 context.super_ = super_;
520
521 CYNonLocal *nonlocal(context.nonlocal_);
522 CYNonLocal *nextlocal(context.nextlocal_);
523
524 bool localize;
525 if (nonlocal_ != NULL) {
526 localize = false;
527 context.nonlocal_ = nonlocal_;
528 } else {
529 localize = true;
530 nonlocal_ = $ CYNonLocal();
531 context.nextlocal_ = nonlocal_;
532 }
533
534 CYScope scope(!localize, context);
535
536 $I("arguments")->Replace(context, CYIdentifierMagic);
537
538 parameters_->Replace(context, code_);
539
540 context.ReplaceAll(code_);
541
542 if (implicit_)
543 CYImplicitReturn(code_);
544
545 if (CYIdentifier *identifier = this_.identifier_) {
546 context.scope_->Declare(context, identifier, CYIdentifierVariable);
547 code_ = $$
548 ->* $E($ CYAssign($V(identifier), $ CYThis()))
549 ->* code_;
550 }
551
552 if (localize)
553 context.NonLocal(code_);
554
555 context.nextlocal_ = nextlocal;
556 context.nonlocal_ = nonlocal;
557
558 context.super_ = super;
559 context.this_ = _this;
560
561 scope.Close(context, code_);
562}
563
564CYTarget *CYFunctionExpression::Replace(CYContext &context) {
565 CYScope scope(false, context);
566 if (name_ != NULL)
567 name_ = name_->Replace(context, CYIdentifierOther);
568
569 CYFunction::Replace(context);
570 scope.Close(context);
571 return this;
572}
573
574void CYFunctionParameter::Replace(CYContext &context, CYStatement *&statements) { $T()
575 CYAssignment *assignment(binding_->Replace(context, CYIdentifierArgument));
576
577 next_->Replace(context, statements);
578
579 if (assignment != NULL)
580 statements = $$
581 ->* $ CYIf($ CYIdentical($ CYTypeOf(binding_->Target(context)), $S("undefined")), $$
582 ->* $E(assignment))
583 ->* statements;
584}
585
586CYStatement *CYFunctionStatement::Replace(CYContext &context) {
587 name_ = name_->Replace(context, CYIdentifierOther);
588 CYFunction::Replace(context);
589 return this;
590}
591
592CYIdentifier *CYIdentifier::Replace(CYContext &context, CYIdentifierKind kind) {
593 if (next_ == this)
594 return this;
595 if (next_ != NULL)
596 return next_->Replace(context, kind);
597 next_ = context.scope_->Declare(context, this, kind)->identifier_;
598 return next_;
599}
600
601CYStatement *CYIf::Return() {
602 CYImplicitReturn(true_);
603 CYImplicitReturn(false_);
604 return this;
605}
606
607CYStatement *CYIf::Replace(CYContext &context) {
608 context.Replace(test_);
609 context.ReplaceAll(true_);
610 context.ReplaceAll(false_);
611 return this;
612}
613
614CYFunctionParameter *CYIfComprehension::Parameter(CYContext &context) const {
615 return NULL;
616}
617
618CYStatement *CYIfComprehension::Replace(CYContext &context, CYStatement *statement) const {
619 return $ CYIf(test_, CYComprehension::Replace(context, statement));
620}
621
622CYStatement *CYImport::Replace(CYContext &context) {
623 return $ CYVar($B1($B($I(module_->part_->Word()), $C1($V("require"), module_->Replace(context, "/")))));
624}
625
626CYStatement *CYImportDeclaration::Replace(CYContext &context) {
627 CYIdentifier *module(context.Unique());
628
629 CYList<CYStatement> statements;
630 CYForEach (specifier, specifiers_)
631 statements->*specifier->Replace(context, module);
632
633 return $ CYBlock($$
634 ->* $ CYLexical(false, $B1($B(module, $C1($V("require"), module_))))
635 ->* statements);
636}
637
638CYStatement *CYImportSpecifier::Replace(CYContext &context, CYIdentifier *module) {
639 binding_ = binding_->Replace(context, CYIdentifierLexical);
640
641 CYExpression *import($V(module));
642 if (name_ != NULL)
643 import = $M(import, $S(name_));
644 return $E($ CYAssign($V(binding_), import));
645}
646
647CYTarget *CYIndirect::Replace(CYContext &context) {
648 return $M(rhs_, $S("$cyi"));
649}
650
651CYTarget *CYIndirectMember::Replace(CYContext &context) {
652 return $M($ CYIndirect(object_), property_);
653}
654
655CYExpression *CYInfix::Replace(CYContext &context) {
656 context.Replace(lhs_);
657 context.Replace(rhs_);
658 return this;
659}
660
661CYStatement *CYLabel::Replace(CYContext &context) {
662 context.Replace(statement_);
663 return this;
664}
665
666CYTarget *CYLambda::Replace(CYContext &context) {
667 return $N2($V("Functor"), $ CYFunctionExpression(NULL, parameters_->Parameters(context), code_), parameters_->TypeSignature(context, typed_->Replace(context)));
668}
669
670CYForInitializer *CYLexical::Replace(CYContext &context) {
671 if (CYExpression *expression = bindings_->Replace(context, CYIdentifierLexical))
672 return $E(expression);
673 return $ CYEmpty();
674}
675
676CYFunctionExpression *CYMethod::Constructor() {
677 return NULL;
678}
679
680void CYMethod::Replace(CYContext &context) {
681 CYFunction::Replace(context);
682}
683
684CYString *CYModule::Replace(CYContext &context, const char *separator) const {
685 if (next_ == NULL)
686 return $ CYString(part_);
687 return $ CYString($pool.strcat(next_->Replace(context, separator)->Value(), separator, part_->Word(), NULL));
688}
689
690CYExpression *CYMultiply::Replace(CYContext &context) {
691 CYInfix::Replace(context);
692
693 if (CYNumber *lhn = lhs_->Number(context))
694 if (CYNumber *rhn = rhs_->Number(context))
695 return $D(lhn->Value() * rhn->Value());
696
697 return this;
698}
699
700namespace cy {
701namespace Syntax {
702
703CYTarget *New::AddArgument(CYContext &context, CYExpression *value) {
704 CYSetLast(arguments_) = $ CYArgument(value);
705 return this;
706}
707
708CYTarget *New::Replace(CYContext &context) {
709 context.Replace(constructor_);
710 arguments_->Replace(context);
711 return this;
712}
713
714} }
715
716CYNumber *CYNull::Number(CYContext &context) {
717 return $D(0);
718}
719
720CYString *CYNull::String(CYContext &context) {
721 return $S("null");
722}
723
724CYNumber *CYNumber::Number(CYContext &context) {
725 return this;
726}
727
728CYString *CYNumber::String(CYContext &context) {
729 // XXX: there is a precise algorithm for this
730 return $S($pool.sprintf(24, "%.17g", Value()));
731}
732
733CYExpression *CYNumber::PropertyName(CYContext &context) {
734 return String(context);
735}
736
737CYTarget *CYObject::Replace(CYContext &context) {
738 CYBuilder builder;
739 if (properties_ != NULL)
740 properties_ = properties_->ReplaceAll(context, builder, $ CYThis(), false);
741
742 if (builder) {
743 return $C1($M($ CYFunctionExpression(NULL, builder.bindings_->Parameter(context),
744 builder.statements_
745 ->* $ CYReturn($ CYThis())
746 ), $S("call")), this, builder.bindings_->Argument(context));
747 }
748
749 CYForEach (property, properties_)
750 property->Replace(context);
751 return this;
752}
753
754CYTarget *CYParenthetical::Replace(CYContext &context) {
755 // XXX: return expression_;
756 context.Replace(expression_);
757 return this;
758}
759
760CYExpression *CYPostfix::Replace(CYContext &context) {
761 context.Replace(lhs_);
762 return this;
763}
764
765CYExpression *CYPrefix::Replace(CYContext &context) {
766 context.Replace(rhs_);
767 return this;
768}
769
770CYProperty *CYProperty::ReplaceAll(CYContext &context, CYBuilder &builder, CYExpression *self, bool update) {
771 update |= Update();
772 if (update)
773 Replace(context, builder, self, false);
774 if (next_ != NULL)
775 next_ = next_->ReplaceAll(context, builder, self, update);
776 return update ? next_ : this;
777}
778
779void CYProperty::Replace(CYContext &context, CYBuilder &builder, CYExpression *self, bool protect) {
780 CYExpression *name(name_->PropertyName(context));
781 if (name_->Computed()) {
782 CYIdentifier *unique(context.Unique());
783 builder.bindings_
784 ->* $B1($B(unique, name));
785 name = $V(unique);
786 }
787
788 Replace(context, builder, self, name, protect);
789}
790
791bool CYProperty::Update() const {
792 return name_->Computed();
793}
794
795void CYPropertyGetter::Replace(CYContext &context, CYBuilder &builder, CYExpression *self, CYExpression *name, bool protect) {
796 CYIdentifier *unique(context.Unique());
797 builder.bindings_
798 ->* $B1($B(unique, CYSuperize(context, $ CYFunctionExpression(NULL, parameters_, code_))));
799 builder.statements_
800 ->* CYDefineProperty(self, name, true, !protect, $ CYPropertyValue($S("get"), $V(unique)));
801}
802
803CYFunctionExpression *CYPropertyMethod::Constructor() {
804 return name_->Constructor() ? $ CYFunctionExpression(NULL, parameters_, code_) : NULL;
805}
806
807void CYPropertyMethod::Replace(CYContext &context, CYBuilder &builder, CYExpression *self, CYExpression *name, bool protect) {
808 CYIdentifier *unique(context.Unique());
809 builder.bindings_
810 ->* $B1($B(unique, CYSuperize(context, $ CYFunctionExpression(NULL, parameters_, code_))));
811 builder.statements_
812 ->* (!protect ? $E($ CYAssign($M(self, name), $V(unique))) :
813 CYDefineProperty(self, name, true, !protect, $ CYPropertyValue($S("value"), $V(unique), $ CYPropertyValue($S("writable"), $ CYTrue()))));
814}
815
816bool CYPropertyMethod::Update() const {
817 return true;
818}
819
820void CYPropertySetter::Replace(CYContext &context, CYBuilder &builder, CYExpression *self, CYExpression *name, bool protect) {
821 CYIdentifier *unique(context.Unique());
822 builder.bindings_
823 ->* $B1($B(unique, CYSuperize(context, $ CYFunctionExpression(NULL, parameters_, code_))));
824 builder.statements_
825 ->* CYDefineProperty(self, name, true, !protect, $ CYPropertyValue($S("set"), $V(unique)));
826}
827
828void CYPropertyValue::Replace(CYContext &context, CYBuilder &builder, CYExpression *self, CYExpression *name, bool protect) {
829 _assert(!protect);
830 CYIdentifier *unique(context.Unique());
831 builder.bindings_
832 ->* $B1($B(unique, value_));
833 builder.statements_
834 ->* $E($ CYAssign($M(self, name), $V(unique)));
835}
836
837void CYPropertyValue::Replace(CYContext &context) {
838 context.Replace(value_);
839}
840
841void CYScript::Replace(CYContext &context) {
842 CYScope scope(false, context);
843 context.scope_->Damage();
844
845 context.nextlocal_ = $ CYNonLocal();
846 context.ReplaceAll(code_);
847 context.NonLocal(code_);
848
849 scope.Close(context, code_);
850
851 unsigned offset(0);
852
853 for (std::vector<CYIdentifier *>::const_iterator i(context.replace_.begin()); i != context.replace_.end(); ++i) {
854 const char *name;
855 if (context.options_.verbose_)
856 name = $pool.strcat("$", $pool.itoa(offset++), NULL);
857 else {
858 char id[8];
859 id[7] = '\0';
860
861 id:
862 unsigned position(7), local(offset++ + 1);
863
864 do {
865 unsigned index(local % (sizeof(MappingSet) - 1));
866 local /= sizeof(MappingSet) - 1;
867 id[--position] = MappingSet[index];
868 } while (local != 0);
869
870 if (scope.Lookup(context, id + position) != NULL)
871 goto id;
872 // XXX: at some point, this could become a keyword
873
874 name = $pool.strmemdup(id + position, 7 - position);
875 }
876
877 CYIdentifier *identifier(*i);
878 _assert(identifier->next_ == identifier);
879 identifier->next_ = $I(name);
880 }
881}
882
883CYStatement *CYReturn::Replace(CYContext &context) {
884 if (context.nonlocal_ != NULL) {
885 CYProperty *value(value_ == NULL ? NULL : $ CYPropertyValue($S("$cyv"), value_));
886 return $ cy::Syntax::Throw($ CYObject(
887 $ CYPropertyValue($S("$cyk"), $V(context.nonlocal_->Target(context)), value)
888 ));
889 }
890
891 context.Replace(value_);
892 return this;
893}
894
895CYTarget *CYRubyBlock::Replace(CYContext &context) {
896 return call_->AddArgument(context, proc_->Replace(context));
897}
898
899CYTarget *CYRubyBlock::AddArgument(CYContext &context, CYExpression *value) {
900 return Replace(context)->AddArgument(context, value);
901}
902
903CYTarget *CYRubyProc::Replace(CYContext &context) {
904 CYFunctionExpression *function($ CYFunctionExpression(NULL, parameters_, code_));
905 function = CYNonLocalize(context, function);
906 function->implicit_ = true;
907 return function;
908}
909
910CYScope::CYScope(bool transparent, CYContext &context) :
911 transparent_(transparent),
912 parent_(context.scope_),
913 damaged_(false),
914 shadow_(NULL),
915 internal_(NULL)
916{
917 _assert(!transparent_ || parent_ != NULL);
918 context.scope_ = this;
919}
920
921void CYScope::Damage() {
922 damaged_ = true;
923 if (parent_ != NULL)
924 parent_->Damage();
925}
926
927CYIdentifierFlags *CYScope::Lookup(CYContext &context, const char *word) {
928 CYForEach (i, internal_)
929 if (strcmp(i->identifier_->Word(), word) == 0)
930 return i;
931 return NULL;
932}
933
934CYIdentifierFlags *CYScope::Lookup(CYContext &context, CYIdentifier *identifier) {
935 return Lookup(context, identifier->Word());
936}
937
938CYIdentifierFlags *CYScope::Declare(CYContext &context, CYIdentifier *identifier, CYIdentifierKind kind) {
939 _assert(identifier->next_ == NULL || identifier->next_ == identifier);
940
941 CYIdentifierFlags *existing(Lookup(context, identifier));
942 if (existing == NULL)
943 internal_ = $ CYIdentifierFlags(identifier, kind, internal_);
944 ++internal_->count_;
945 if (existing == NULL)
946 return internal_;
947
948 if (kind == CYIdentifierGlobal);
949 else if (existing->kind_ == CYIdentifierGlobal || existing->kind_ == CYIdentifierMagic)
950 existing->kind_ = kind;
951 else if (existing->kind_ == CYIdentifierLexical || kind == CYIdentifierLexical)
952 _assert(false); // XXX: throw new SyntaxError()
953
954 return existing;
955}
956
957void CYScope::Merge(CYContext &context, const CYIdentifierFlags *flags) {
958 _assert(flags->identifier_->next_ == flags->identifier_);
959 CYIdentifierFlags *existing(Declare(context, flags->identifier_, flags->kind_));
960 flags->identifier_->next_ = existing->identifier_;
961
962 existing->count_ += flags->count_;
963 if (existing->offset_ < flags->offset_)
964 existing->offset_ = flags->offset_;
965}
966
967void CYScope::Close(CYContext &context, CYStatement *&statements) {
968 Close(context);
969
970 CYList<CYBindings> bindings;
971
972 CYForEach (i, internal_)
973 if (i->kind_ == CYIdentifierVariable)
974 bindings
975 ->* $ CYBindings($ CYBinding(i->identifier_));
976
977 if (bindings) {
978 CYVar *var($ CYVar(bindings));
979 var->SetNext(statements);
980 statements = var;
981 }
982}
983
984void CYScope::Close(CYContext &context) {
985 context.scope_ = parent_;
986
987 CYForEach (i, internal_) {
988 _assert(i->identifier_->next_ == i->identifier_);
989 switch (i->kind_) {
990 case CYIdentifierArgument: {
991 _assert(!transparent_);
992 } break;
993
994 case CYIdentifierLexical: {
995 if (!damaged_) {
996 CYIdentifier *replace(context.Unique());
997 replace->next_ = replace;
998 i->identifier_->next_ = replace;
999 i->identifier_ = replace;
1000 }
1001
1002 if (!transparent_)
1003 i->kind_ = CYIdentifierVariable;
1004 else
1005 parent_->Declare(context, i->identifier_, CYIdentifierVariable);
1006 } break;
1007
1008 case CYIdentifierVariable: {
1009 if (transparent_) {
1010 parent_->Declare(context, i->identifier_, i->kind_);
1011 i->kind_ = CYIdentifierGlobal;
1012 }
1013 } break;
1014 default:; } }
1015
1016 if (damaged_)
1017 return;
1018
1019 typedef std::multimap<unsigned, CYIdentifier *> CYIdentifierOffsetMap;
1020 CYIdentifierOffsetMap offsets;
1021
1022 CYForEach (i, internal_) {
1023 _assert(i->identifier_->next_ == i->identifier_);
1024 switch (i->kind_) {
1025 case CYIdentifierArgument:
1026 case CYIdentifierVariable:
1027 offsets.insert(CYIdentifierOffsetMap::value_type(i->offset_, i->identifier_));
1028 break;
1029 default:; } }
1030
1031 unsigned offset(0);
1032
1033 for (CYIdentifierOffsetMap::const_iterator i(offsets.begin()); i != offsets.end(); ++i) {
1034 if (offset < i->first)
1035 offset = i->first;
1036 CYIdentifier *identifier(i->second);
1037
1038 if (offset >= context.replace_.size())
1039 context.replace_.resize(offset + 1, NULL);
1040 CYIdentifier *&replace(context.replace_[offset++]);
1041
1042 if (replace == NULL)
1043 replace = identifier;
1044 else {
1045 _assert(replace->next_ == replace);
1046 identifier->next_ = replace;
1047 }
1048 }
1049
1050 if (parent_ == NULL)
1051 return;
1052
1053 CYForEach (i, internal_) {
1054 switch (i->kind_) {
1055 case CYIdentifierGlobal: {
1056 if (i->offset_ < offset)
1057 i->offset_ = offset;
1058 parent_->Merge(context, i);
1059 } break;
1060 default:; } }
1061}
1062
1063CYElementValue *CYSpan::Replace(CYContext &context) { $T(NULL)
1064 return $ CYElementValue(expression_, $ CYElementValue(string_, next_->Replace(context)));
1065}
1066
1067CYStatement *CYStatement::Return() {
1068 return this;
1069}
1070
1071CYString *CYString::Concat(CYContext &context, CYString *rhs) const {
1072 size_t size(size_ + rhs->size_);
1073 char *value($ char[size + 1]);
1074 memcpy(value, value_, size_);
1075 memcpy(value + size_, rhs->value_, rhs->size_);
1076 value[size] = '\0';
1077 return $S(value, size);
1078}
1079
1080CYNumber *CYString::Number(CYContext &context) {
1081 // XXX: there is a precise algorithm for this
1082 return NULL;
1083}
1084
1085CYExpression *CYString::PropertyName(CYContext &context) {
1086 return this;
1087}
1088
1089CYString *CYString::String(CYContext &context) {
1090 return this;
1091}
1092
1093CYTarget *CYSuperAccess::Replace(CYContext &context) {
1094 return $C1($M($M($M($V(context.super_), $S("prototype")), property_), $S("bind")), $ CYThis());
1095}
1096
1097CYTarget *CYSuperCall::Replace(CYContext &context) {
1098 return $C($C1($M($V(context.super_), $S("bind")), $ CYThis()), arguments_);
1099}
1100
1101CYStatement *CYSwitch::Replace(CYContext &context) {
1102 context.Replace(value_);
1103 clauses_->Replace(context);
1104 return this;
1105}
1106
1107CYStatement *CYTarget::Initialize(CYContext &context, CYExpression *value) {
1108 if (value == NULL)
1109 return NULL;
1110 return $E($ CYAssign(this, value));
1111}
1112
1113CYTarget *CYTemplate::Replace(CYContext &context) {
1114 return $C2($M($M($M($V("String"), $S("prototype")), $S("concat")), $S("apply")), $S(""), $ CYArray($ CYElementValue(string_, spans_->Replace(context))));
1115}
1116
1117CYTarget *CYThis::Replace(CYContext &context) {
1118 if (context.this_ != NULL)
1119 return $V(context.this_->Identifier(context));
1120 return this;
1121}
1122
1123namespace cy {
1124namespace Syntax {
1125
1126CYStatement *Throw::Replace(CYContext &context) {
1127 context.Replace(value_);
1128 return this;
1129}
1130
1131} }
1132
1133CYTarget *CYTrivial::Replace(CYContext &context) {
1134 return this;
1135}
1136
1137CYNumber *CYTrue::Number(CYContext &context) {
1138 return $D(1);
1139}
1140
1141CYString *CYTrue::String(CYContext &context) {
1142 return $S("true");
1143}
1144
1145namespace cy {
1146namespace Syntax {
1147
1148CYStatement *Try::Replace(CYContext &context) {
1149 CYScope scope(true, context);
1150 context.ReplaceAll(code_);
1151 scope.Close(context);
1152
1153 catch_->Replace(context);
1154 finally_->Replace(context);
1155 return this;
1156}
1157
1158} }
1159
1160CYTarget *CYTypeArrayOf::Replace_(CYContext &context, CYTarget *type) {
1161 return next_->Replace(context, $ CYCall($ CYDirectMember(type, $ CYString("arrayOf")), $ CYArgument(size_)));
1162}
1163
1164CYTarget *CYTypeBlockWith::Replace_(CYContext &context, CYTarget *type) {
1165 return next_->Replace(context, $ CYCall($ CYDirectMember(type, $ CYString("blockWith")), parameters_->Argument(context)));
1166}
1167
1168CYTarget *CYTypeConstant::Replace_(CYContext &context, CYTarget *type) {
1169 return next_->Replace(context, $ CYCall($ CYDirectMember(type, $ CYString("constant"))));
1170}
1171
1172CYStatement *CYTypeDefinition::Replace(CYContext &context) {
1173 CYIdentifier *identifier(typed_->identifier_);
1174 typed_->identifier_ = NULL;
1175 return $ CYLexical(false, $B1($B(identifier, $ CYTypeExpression(typed_))));
1176}
1177
1178CYTarget *CYTypeError::Replace(CYContext &context) {
1179 _assert(false);
1180 return NULL;
1181}
1182
1183CYTarget *CYTypeExpression::Replace(CYContext &context) {
1184 return typed_->Replace(context);
1185}
1186
1187CYTarget *CYTypeModifier::Replace(CYContext &context, CYTarget *type) { $T(type)
1188 return Replace_(context, type);
1189}
1190
1191CYTarget *CYTypeFunctionWith::Replace_(CYContext &context, CYTarget *type) {
1192 return next_->Replace(context, $ CYCall($ CYDirectMember(type, $ CYString("functionWith")), parameters_->Argument(context)));
1193}
1194
1195CYTarget *CYTypeLong::Replace(CYContext &context) {
1196 return $ CYCall($ CYDirectMember(specifier_->Replace(context), $ CYString("long")));
1197}
1198
1199CYTarget *CYTypePointerTo::Replace_(CYContext &context, CYTarget *type) {
1200 return next_->Replace(context, $ CYCall($ CYDirectMember(type, $ CYString("pointerTo"))));
1201}
1202
1203CYTarget *CYTypeShort::Replace(CYContext &context) {
1204 return $ CYCall($ CYDirectMember(specifier_->Replace(context), $ CYString("short")));
1205}
1206
1207CYTarget *CYTypeSigned::Replace(CYContext &context) {
1208 return $ CYCall($ CYDirectMember(specifier_->Replace(context), $ CYString("signed")));
1209}
1210
1211CYTarget *CYTypeStruct::Replace(CYContext &context) {
1212 CYList<CYElementValue> types;
1213 CYList<CYElementValue> names;
1214
1215 CYForEach (field, fields_) {
1216 CYTypedIdentifier *typed(field->typed_);
1217 types->*$ CYElementValue(typed->Replace(context));
1218
1219 CYExpression *name;
1220 if (typed->identifier_ == NULL)
1221 name = NULL;
1222 else
1223 name = $S(typed->identifier_->Word());
1224 names->*$ CYElementValue(name);
1225 }
1226
1227 CYTarget *target($N2($V("Type"), $ CYArray(types), $ CYArray(names)));
1228 if (name_ != NULL)
1229 target = $C1($M(target, $S("withName")), $S(name_->Word()));
1230 return target;
1231}
1232
1233CYTarget *CYTypeUnsigned::Replace(CYContext &context) {
1234 return $ CYCall($ CYDirectMember(specifier_->Replace(context), $ CYString("unsigned")));
1235}
1236
1237CYTarget *CYTypeVariable::Replace(CYContext &context) {
1238 return $V(name_);
1239}
1240
1241CYTarget *CYTypeVoid::Replace(CYContext &context) {
1242 return $N1($V("Type"), $ CYString("v"));
1243}
1244
1245CYTarget *CYTypeVolatile::Replace_(CYContext &context, CYTarget *type) {
1246 return next_->Replace(context, $ CYCall($ CYDirectMember(type, $ CYString("volatile"))));
1247}
1248
1249CYTarget *CYTypedIdentifier::Replace(CYContext &context) {
1250 return modifier_->Replace(context, specifier_->Replace(context));
1251}
1252
1253CYTypeFunctionWith *CYTypedIdentifier::Function() {
1254 CYTypeModifier **modifier(&modifier_);
1255 if (*modifier == NULL)
1256 return NULL;
1257 while ((*modifier)->next_ != NULL)
1258 modifier = &(*modifier)->next_;
1259 CYTypeFunctionWith *function((*modifier)->Function());
1260 if (function == NULL)
1261 return NULL;
1262 *modifier = NULL;
1263 return function;
1264}
1265
1266CYArgument *CYTypedParameter::Argument(CYContext &context) { $T(NULL)
1267 return $ CYArgument(typed_->Replace(context), next_->Argument(context));
1268}
1269
1270CYFunctionParameter *CYTypedParameter::Parameters(CYContext &context) { $T(NULL)
1271 return $ CYFunctionParameter($ CYBinding(typed_->identifier_ ?: context.Unique()), next_->Parameters(context));
1272}
1273
1274CYExpression *CYTypedParameter::TypeSignature(CYContext &context, CYExpression *prefix) { $T(prefix)
1275 return next_->TypeSignature(context, $ CYAdd(prefix, typed_->Replace(context)));
1276}
1277
1278CYForInitializer *CYVar::Replace(CYContext &context) {
1279 if (CYExpression *expression = bindings_->Replace(context, CYIdentifierVariable))
1280 return $E(expression);
1281 return $ CYEmpty();
1282}
1283
1284CYTarget *CYVariable::Replace(CYContext &context) {
1285 name_ = name_->Replace(context, CYIdentifierGlobal);
1286 return this;
1287}
1288
1289CYFunctionParameter *CYVariable::Parameter() const {
1290 return $ CYFunctionParameter($ CYBinding(name_));
1291}
1292
1293CYStatement *CYWhile::Replace(CYContext &context) {
1294 context.Replace(test_);
1295 context.ReplaceAll(code_);
1296 return this;
1297}
1298
1299CYStatement *CYWith::Replace(CYContext &context) {
1300 context.Replace(scope_);
1301 CYScope scope(true, context);
1302 scope.Damage();
1303 context.ReplaceAll(code_);
1304 scope.Close(context);
1305 return this;
1306}
1307
1308CYExpression *CYWord::PropertyName(CYContext &context) {
1309 return $S(this);
1310}