1 /* Cycript - Optimizing JavaScript Compiler/Runtime
2 * Copyright (C) 2009-2015 Jay Freeman (saurik)
5 /* GNU Affero General Public License, Version 3 {{{ */
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.
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.
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/>.
25 #include "Replace.hpp"
28 CYFunctionExpression
*CYNonLocalize(CYContext
&context
, CYFunctionExpression
*function
) {
29 function
->nonlocal_
= context
.nextlocal_
;
33 CYFunctionExpression
*CYSuperize(CYContext
&context
, CYFunctionExpression
*function
) {
34 function
->super_
= context
.super_
;
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
)
45 static void CYImplicitReturn(CYStatement
*&code
) {
46 if (CYStatement
*&last
= CYGetLast(code
))
47 last
= last
->Return();
50 CYExpression
*CYAdd::Replace(CYContext
&context
) {
51 CYInfix::Replace(context
);
53 CYString
*lhs(dynamic_cast<CYString
*>(lhs_
));
54 CYString
*rhs(dynamic_cast<CYString
*>(rhs_
));
56 if (lhs
!= NULL
|| rhs
!= NULL
) {
58 lhs
= lhs_
->String(context
);
61 } else if (rhs
== NULL
) {
62 rhs
= rhs_
->String(context
);
67 return lhs
->Concat(context
, rhs
);
70 if (CYNumber
*lhn
= lhs_
->Number(context
))
71 if (CYNumber
*rhn
= rhs_
->Number(context
))
72 return $
D(lhn
->Value() + rhn
->Value());
77 CYExpression
*CYAddressOf::Replace(CYContext
&context
) {
78 return $
C0($
M(rhs_
, $
S("$cya")));
81 CYTarget
*CYApply::AddArgument(CYContext
&context
, CYExpression
*value
) {
82 CYArgument
**argument(&arguments_
);
83 while (*argument
!= NULL
)
84 argument
= &(*argument
)->next_
;
85 *argument
= $
CYArgument(value
);
89 CYArgument
*CYArgument::Replace(CYContext
&context
) { $
T(NULL
)
90 context
.Replace(value_
);
91 next_
= next_
->Replace(context
);
103 CYTarget
*CYArray::Replace(CYContext
&context
) {
104 CYForEach (element
, elements_
)
105 element
->Replace(context
);
109 CYTarget
*CYArrayComprehension::Replace(CYContext
&context
) {
110 CYIdentifier
*cyv(context
.Unique());
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
))
119 CYExpression
*CYAssignment::Replace(CYContext
&context
) {
120 context
.Replace(lhs_
);
121 context
.Replace(rhs_
);
125 CYStatement
*CYBlock::Return() {
126 CYImplicitReturn(code_
);
130 CYStatement
*CYBlock::Replace(CYContext
&context
) {
131 CYScope
scope(true, context
);
132 context
.ReplaceAll(code_
);
133 scope
.Close(context
);
140 CYStatement
*CYBreak::Replace(CYContext
&context
) {
144 CYTarget
*CYCall::Replace(CYContext
&context
) {
145 context
.Replace(function_
);
146 arguments_
->Replace(context
);
153 void Catch::Replace(CYContext
&context
) { $
T()
154 CYScope
scope(true, context
);
156 name_
= name_
->Replace(context
, CYIdentifierCatch
);
158 context
.ReplaceAll(code_
);
159 scope
.Close(context
);
164 CYTarget
*CYClassExpression::Replace(CYContext
&context
) {
167 CYIdentifier
*super(context
.Unique());
169 CYIdentifier
*old(context
.super_
);
170 context
.super_
= super
;
172 CYIdentifier
*constructor(context
.Unique());
173 CYForEach (member
, tail_
->static_
)
174 member
->Replace(context
, builder
, $
V(constructor
), true);
176 CYIdentifier
*prototype(context
.Unique());
177 CYForEach (member
, tail_
->instance_
)
178 member
->Replace(context
, builder
, $
V(prototype
), true);
180 if (tail_
->constructor_
== NULL
)
181 tail_
->constructor_
= $
CYFunctionExpression(NULL
, NULL
, NULL
);
182 tail_
->constructor_
= CYSuperize(context
, tail_
->constructor_
);
184 context
.super_
= old
;
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")));
199 CYStatement
*CYClassStatement::Replace(CYContext
&context
) {
200 return $
CYVar($
B1($
B(name_
, $
CYClassExpression(name_
, tail_
))));
203 void CYClause::Replace(CYContext
&context
) { $
T()
204 context
.Replace(value_
);
205 context
.ReplaceAll(code_
);
206 next_
->Replace(context
);
209 CYExpression
*CYCompound::Replace(CYContext
&context
) {
210 context
.Replace(expression_
);
211 context
.Replace(next_
);
213 if (CYCompound
*compound
= dynamic_cast<CYCompound
*>(expression_
)) {
214 expression_
= compound
->expression_
;
215 compound
->expression_
= compound
->next_
;
216 compound
->next_
= next_
;
223 CYFunctionParameter
*CYCompound::Parameter() const {
224 CYFunctionParameter
*next(next_
->Parameter());
228 CYFunctionParameter
*parameter(expression_
->Parameter());
229 if (parameter
== NULL
)
232 parameter
->SetNext(next
);
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
);
245 CYStatement
*CYComprehension::Replace(CYContext
&context
, CYStatement
*statement
) const {
246 return next_
== NULL
? statement
: next_
->Replace(context
, statement
);
249 CYExpression
*CYComputed::PropertyName(CYContext
&context
) {
253 CYExpression
*CYCondition::Replace(CYContext
&context
) {
254 context
.Replace(test_
);
255 context
.Replace(true_
);
256 context
.Replace(false_
);
260 void CYContext::NonLocal(CYStatement
*&statements
) {
261 CYContext
&context(*this);
263 if (nextlocal_
!= NULL
&& nextlocal_
->identifier_
!= NULL
) {
264 CYIdentifier
*cye($
I("$cye")->Replace(context
, CYIdentifierGlobal
));
265 CYIdentifier
*unique(nextlocal_
->identifier_
->Replace(context
, CYIdentifierGlobal
));
267 CYStatement
*declare(
268 $
CYVar($
B1($
B(unique
, $
CYObject()))));
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
))));
276 context
.Replace(declare
);
277 rescue
->Replace(context
);
281 ->* $
cy::Syntax::Try(statements
, rescue
, NULL
);
285 CYIdentifier
*CYContext::Unique() {
286 return $
CYIdentifier($pool
.strcat("$cy", $pool
.itoa(unique_
++), NULL
));
289 CYStatement
*CYContinue::Replace(CYContext
&context
) {
293 CYStatement
*CYDebugger::Replace(CYContext
&context
) {
297 CYTarget
*CYBinding::Target(CYContext
&context
) {
298 return $
V(identifier_
);
301 CYAssignment
*CYBinding::Replace(CYContext
&context
, CYIdentifierKind kind
) {
302 identifier_
= identifier_
->Replace(context
, kind
);
304 if (initializer_
== NULL
)
307 CYAssignment
*value($
CYAssign(Target(context
), initializer_
));
312 CYExpression
*CYBindings::Replace(CYContext
&context
, CYIdentifierKind kind
) { $
T(NULL
)
313 CYAssignment
*assignment(binding_
->Replace(context
, kind
));
314 CYExpression
*compound(next_
->Replace(context
, kind
));
316 if (assignment
!= NULL
)
317 if (compound
== NULL
)
318 compound
= assignment
;
320 compound
= $
CYCompound(assignment
, compound
);
324 CYFunctionParameter
*CYBindings::Parameter(CYContext
&context
) { $
T(NULL
)
325 return $
CYFunctionParameter($
CYBinding(binding_
->identifier_
), next_
->Parameter(context
));
328 CYArgument
*CYBindings::Argument(CYContext
&context
) { $
T(NULL
)
329 return $
CYArgument(binding_
->initializer_
, next_
->Argument(context
));
332 CYTarget
*CYDirectMember::Replace(CYContext
&context
) {
333 context
.Replace(object_
);
334 context
.Replace(property_
);
338 CYStatement
*CYDoWhile::Replace(CYContext
&context
) {
339 context
.Replace(test_
);
340 context
.ReplaceAll(code_
);
344 void CYElementSpread::Replace(CYContext
&context
) {
345 context
.Replace(value_
);
348 void CYElementValue::Replace(CYContext
&context
) {
349 context
.Replace(value_
);
352 CYForInitializer
*CYEmpty::Replace(CYContext
&context
) {
356 CYTarget
*CYEncodedType::Replace(CYContext
&context
) {
357 return typed_
->Replace(context
);
360 CYTarget
*CYEval::Replace(CYContext
&context
) {
361 context
.scope_
->Damage();
362 if (arguments_
!= NULL
)
363 arguments_
->value_
= $
C1($
M($
V("Cycript"), $
S("compile")), arguments_
->value_
);
364 return $
C($
V("eval"), arguments_
);
367 CYStatement
*CYExpress::Return() {
368 return $
CYReturn(expression_
);
371 CYForInitializer
*CYExpress::Replace(CYContext
&context
) {
372 context
.Replace(expression_
);
376 CYTarget
*CYExpression::AddArgument(CYContext
&context
, CYExpression
*value
) {
377 return $
C1(this, value
);
380 CYFunctionParameter
*CYExpression::Parameter() const {
384 CYTarget
*CYExtend::Replace(CYContext
&context
) {
385 return object_
.Replace(context
, lhs_
);
388 CYStatement
*CYExternal::Replace(CYContext
&context
) {
389 return $
E($
CYAssign($
V(typed_
->identifier_
), $
C1(typed_
->Replace(context
), $
C2($
V("dlsym"), $
V("RTLD_DEFAULT"), $
S(typed_
->identifier_
->Word())))));
392 CYNumber
*CYFalse::Number(CYContext
&context
) {
396 CYString
*CYFalse::String(CYContext
&context
) {
400 CYExpression
*CYFatArrow::Replace(CYContext
&context
) {
401 CYFunctionExpression
*function($
CYFunctionExpression(NULL
, parameters_
, code_
));
402 function
->this_
.SetNext(context
.this_
);
406 void CYFinally::Replace(CYContext
&context
) { $
T()
407 CYScope
scope(true, context
);
408 context
.ReplaceAll(code_
);
409 scope
.Close(context
);
412 CYStatement
*CYFor::Replace(CYContext
&context
) {
413 CYScope
outer(true, context
);
414 context
.Replace(initializer_
);
416 context
.Replace(test_
);
419 CYScope
inner(true, context
);
420 context
.ReplaceAll(code_
);
421 inner
.Close(context
);
424 context
.Replace(increment_
);
426 outer
.Close(context
);
430 CYStatement
*CYForLexical::Initialize(CYContext
&context
, CYExpression
*value
) {
432 if (binding_
->initializer_
== NULL
)
434 value
= binding_
->initializer_
;
437 return $
CYLexical(constant_
, $
B1($
CYBinding(binding_
->identifier_
, value
)));
440 CYTarget
*CYForLexical::Replace(CYContext
&context
) {
441 _assert(binding_
->Replace(context
, CYIdentifierLexical
) == NULL
);
442 return binding_
->Target(context
);
445 CYStatement
*CYForIn::Replace(CYContext
&context
) {
446 CYScope
scope(true, context
);
447 context
.Replace(initializer_
);
448 context
.Replace(iterable_
);
449 context
.ReplaceAll(code_
);
450 scope
.Close(context
);
454 CYStatement
*CYForInitialized::Replace(CYContext
&context
) {
455 CYAssignment
*assignment(binding_
->Replace(context
, CYIdentifierVariable
));
457 ->* (assignment
== NULL
? NULL
: $
CYExpress(assignment
))
458 ->* $
CYForIn(binding_
->Target(context
), iterable_
, code_
));
461 CYFunctionParameter
*CYForInComprehension::Parameter(CYContext
&context
) const {
462 return $
CYFunctionParameter(binding_
);
465 CYStatement
*CYForInComprehension::Replace(CYContext
&context
, CYStatement
*statement
) const {
466 return $
CYForIn(binding_
->Target(context
), iterable_
, CYComprehension::Replace(context
, statement
));
469 CYStatement
*CYForOf::Replace(CYContext
&context
) {
470 CYIdentifier
*item(context
.Unique()), *list(context
.Unique());
473 ->* initializer_
->Initialize(context
, NULL
)
474 ->* $
CYLexical(false, $
B2($
B(list
, iterable_
), $
B(item
)))
475 ->* $
CYForIn($
V(item
), $
V(list
), $
CYBlock($$
476 ->* initializer_
->Initialize(context
, $
M($
V(list
), $
V(item
)))
481 CYFunctionParameter
*CYForOfComprehension::Parameter(CYContext
&context
) const {
482 return $
CYFunctionParameter(binding_
);
485 CYStatement
*CYForOfComprehension::Replace(CYContext
&context
, CYStatement
*statement
) const {
486 CYIdentifier
*cys(context
.Unique());
489 ->* $
CYLexical(false, $
B1($
B(cys
, iterable_
)))
490 ->* $
CYForIn(binding_
->Target(context
), $
V(cys
), $
CYBlock($$
491 ->* $
E($
CYAssign(binding_
->Target(context
), $
M($
V(cys
), binding_
->Target(context
))))
492 ->* CYComprehension::Replace(context
, statement
)
496 CYStatement
*CYForVariable::Initialize(CYContext
&context
, CYExpression
*value
) {
498 if (binding_
->initializer_
== NULL
)
500 value
= binding_
->initializer_
;
503 return $
CYVar($
B1($
CYBinding(binding_
->identifier_
, value
)));
506 CYTarget
*CYForVariable::Replace(CYContext
&context
) {
507 _assert(binding_
->Replace(context
, CYIdentifierVariable
) == NULL
);
508 return binding_
->Target(context
);
511 // XXX: this is evil evil black magic. don't ask, don't tell... don't believe!
512 #define MappingSet "0etnirsoalfucdphmgyvbxTwSNECAFjDLkMOIBPqzRH$_WXUVGYKQJZ"
513 //#define MappingSet "0abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ$_"
515 void CYFunction::Replace(CYContext
&context
) {
516 CYThisScope
*_this(context
.this_
);
517 context
.this_
= &this_
;
518 context
.this_
= CYGetLast(context
.this_
);
520 CYIdentifier
*super(context
.super_
);
521 context
.super_
= super_
;
523 CYNonLocal
*nonlocal(context
.nonlocal_
);
524 CYNonLocal
*nextlocal(context
.nextlocal_
);
527 if (nonlocal_
!= NULL
) {
529 context
.nonlocal_
= nonlocal_
;
532 nonlocal_
= $
CYNonLocal();
533 context
.nextlocal_
= nonlocal_
;
536 CYScope
scope(!localize
, context
);
538 $
I("arguments")->Replace(context
, CYIdentifierMagic
);
540 parameters_
->Replace(context
, code_
);
542 context
.ReplaceAll(code_
);
545 CYImplicitReturn(code_
);
547 if (CYIdentifier
*identifier
= this_
.identifier_
) {
548 context
.scope_
->Declare(context
, identifier
, CYIdentifierVariable
);
550 ->* $
E($
CYAssign($
V(identifier
), $
CYThis()))
555 context
.NonLocal(code_
);
557 context
.nextlocal_
= nextlocal
;
558 context
.nonlocal_
= nonlocal
;
560 context
.super_
= super
;
561 context
.this_
= _this
;
563 scope
.Close(context
, code_
);
566 CYTarget
*CYFunctionExpression::Replace(CYContext
&context
) {
567 CYScope
scope(false, context
);
569 name_
= name_
->Replace(context
, CYIdentifierOther
);
571 CYFunction::Replace(context
);
572 scope
.Close(context
);
576 void CYFunctionParameter::Replace(CYContext
&context
, CYStatement
*&statements
) { $
T()
577 CYAssignment
*assignment(binding_
->Replace(context
, CYIdentifierArgument
));
579 next_
->Replace(context
, statements
);
581 if (assignment
!= NULL
)
583 ->* $
CYIf($
CYIdentical($
CYTypeOf(binding_
->Target(context
)), $
S("undefined")), $$
588 CYStatement
*CYFunctionStatement::Replace(CYContext
&context
) {
589 name_
= name_
->Replace(context
, CYIdentifierOther
);
590 CYFunction::Replace(context
);
594 CYIdentifier
*CYIdentifier::Replace(CYContext
&context
, CYIdentifierKind kind
) {
598 return next_
->Replace(context
, kind
);
599 next_
= context
.scope_
->Declare(context
, this, kind
)->identifier_
;
603 CYStatement
*CYIf::Return() {
604 CYImplicitReturn(true_
);
605 CYImplicitReturn(false_
);
609 CYStatement
*CYIf::Replace(CYContext
&context
) {
610 context
.Replace(test_
);
611 context
.ReplaceAll(true_
);
612 context
.ReplaceAll(false_
);
616 CYFunctionParameter
*CYIfComprehension::Parameter(CYContext
&context
) const {
620 CYStatement
*CYIfComprehension::Replace(CYContext
&context
, CYStatement
*statement
) const {
621 return $
CYIf(test_
, CYComprehension::Replace(context
, statement
));
624 CYStatement
*CYImport::Replace(CYContext
&context
) {
625 return $
CYVar($
B1($
B($
I(module_
->part_
->Word()), $
C1($
V("require"), module_
->Replace(context
, "/")))));
628 CYStatement
*CYImportDeclaration::Replace(CYContext
&context
) {
629 CYIdentifier
*module(context
.Unique());
631 CYList
<CYStatement
> statements
;
632 CYForEach (specifier
, specifiers_
)
633 statements
->*specifier
->Replace(context
, module);
636 ->* $
CYLexical(false, $
B1($
B(module, $
C1($
V("require"), module_
))))
640 CYStatement
*CYImportSpecifier::Replace(CYContext
&context
, CYIdentifier
*module) {
641 binding_
= binding_
->Replace(context
, CYIdentifierLexical
);
643 CYExpression
*import($
V(module));
645 import = $
M(import, $
S(name_
));
646 return $
E($
CYAssign($
V(binding_
), import));
649 CYTarget
*CYIndirect::Replace(CYContext
&context
) {
650 return $
M(rhs_
, $
S("$cyi"));
653 CYTarget
*CYIndirectMember::Replace(CYContext
&context
) {
654 return $
M($
CYIndirect(object_
), property_
);
657 CYExpression
*CYInfix::Replace(CYContext
&context
) {
658 context
.Replace(lhs_
);
659 context
.Replace(rhs_
);
663 CYStatement
*CYLabel::Replace(CYContext
&context
) {
664 context
.Replace(statement_
);
668 CYTarget
*CYLambda::Replace(CYContext
&context
) {
669 return $
N2($
V("Functor"), $
CYFunctionExpression(NULL
, parameters_
->Parameters(context
), code_
), parameters_
->TypeSignature(context
, typed_
->Replace(context
)));
672 CYForInitializer
*CYLexical::Replace(CYContext
&context
) {
673 if (CYExpression
*expression
= bindings_
->Replace(context
, CYIdentifierLexical
))
674 return $
E(expression
);
678 CYFunctionExpression
*CYMethod::Constructor() {
682 void CYMethod::Replace(CYContext
&context
) {
683 CYFunction::Replace(context
);
686 CYString
*CYModule::Replace(CYContext
&context
, const char *separator
) const {
688 return $
CYString(part_
);
689 return $
CYString($pool
.strcat(next_
->Replace(context
, separator
)->Value(), separator
, part_
->Word(), NULL
));
692 CYExpression
*CYMultiply::Replace(CYContext
&context
) {
693 CYInfix::Replace(context
);
695 if (CYNumber
*lhn
= lhs_
->Number(context
))
696 if (CYNumber
*rhn
= rhs_
->Number(context
))
697 return $
D(lhn
->Value() * rhn
->Value());
705 CYTarget
*New::AddArgument(CYContext
&context
, CYExpression
*value
) {
706 CYSetLast(arguments_
) = $
CYArgument(value
);
710 CYTarget
*New::Replace(CYContext
&context
) {
711 context
.Replace(constructor_
);
712 arguments_
->Replace(context
);
718 CYNumber
*CYNull::Number(CYContext
&context
) {
722 CYString
*CYNull::String(CYContext
&context
) {
726 CYNumber
*CYNumber::Number(CYContext
&context
) {
730 CYString
*CYNumber::String(CYContext
&context
) {
731 // XXX: there is a precise algorithm for this
732 return $
S($pool
.sprintf(24, "%.17g", Value()));
735 CYExpression
*CYNumber::PropertyName(CYContext
&context
) {
736 return String(context
);
739 CYTarget
*CYObject::Replace(CYContext
&context
, CYTarget
*seed
) {
741 if (properties_
!= NULL
)
742 properties_
= properties_
->ReplaceAll(context
, builder
, $
CYThis(), seed
!= this);
745 return $
C1($
M($
CYFunctionExpression(NULL
, builder
.bindings_
->Parameter(context
),
747 ->* $
CYReturn($
CYThis())
748 ), $
S("call")), seed
, builder
.bindings_
->Argument(context
));
751 CYForEach (property
, properties_
)
752 property
->Replace(context
);
756 CYTarget
*CYObject::Replace(CYContext
&context
) {
757 return Replace(context
, this);
760 CYTarget
*CYParenthetical::Replace(CYContext
&context
) {
761 // XXX: return expression_;
762 context
.Replace(expression_
);
766 CYExpression
*CYPostfix::Replace(CYContext
&context
) {
767 context
.Replace(lhs_
);
771 CYExpression
*CYPrefix::Replace(CYContext
&context
) {
772 context
.Replace(rhs_
);
776 CYProperty
*CYProperty::ReplaceAll(CYContext
&context
, CYBuilder
&builder
, CYExpression
*self
, bool update
) {
779 Replace(context
, builder
, self
, false);
781 next_
= next_
->ReplaceAll(context
, builder
, self
, update
);
782 return update
? next_
: this;
785 void CYProperty::Replace(CYContext
&context
, CYBuilder
&builder
, CYExpression
*self
, bool protect
) {
786 CYExpression
*name(name_
->PropertyName(context
));
787 if (name_
->Computed()) {
788 CYIdentifier
*unique(context
.Unique());
790 ->* $
B1($
B(unique
, name
));
794 Replace(context
, builder
, self
, name
, protect
);
797 bool CYProperty::Update() const {
798 return name_
->Computed();
801 void CYPropertyGetter::Replace(CYContext
&context
, CYBuilder
&builder
, CYExpression
*self
, CYExpression
*name
, bool protect
) {
802 CYIdentifier
*unique(context
.Unique());
804 ->* $
B1($
B(unique
, CYSuperize(context
, $
CYFunctionExpression(NULL
, parameters_
, code_
))));
806 ->* CYDefineProperty(self
, name
, true, !protect
, $
CYPropertyValue($
S("get"), $
V(unique
)));
809 CYFunctionExpression
*CYPropertyMethod::Constructor() {
810 return name_
->Constructor() ? $
CYFunctionExpression(NULL
, parameters_
, code_
) : NULL
;
813 void CYPropertyMethod::Replace(CYContext
&context
, CYBuilder
&builder
, CYExpression
*self
, CYExpression
*name
, bool protect
) {
814 CYIdentifier
*unique(context
.Unique());
816 ->* $
B1($
B(unique
, CYSuperize(context
, $
CYFunctionExpression(NULL
, parameters_
, code_
))));
818 ->* (!protect
? $
E($
CYAssign($
M(self
, name
), $
V(unique
))) :
819 CYDefineProperty(self
, name
, true, !protect
, $
CYPropertyValue($
S("value"), $
V(unique
), $
CYPropertyValue($
S("writable"), $
CYTrue()))));
822 bool CYPropertyMethod::Update() const {
826 void CYPropertySetter::Replace(CYContext
&context
, CYBuilder
&builder
, CYExpression
*self
, CYExpression
*name
, bool protect
) {
827 CYIdentifier
*unique(context
.Unique());
829 ->* $
B1($
B(unique
, CYSuperize(context
, $
CYFunctionExpression(NULL
, parameters_
, code_
))));
831 ->* CYDefineProperty(self
, name
, true, !protect
, $
CYPropertyValue($
S("set"), $
V(unique
)));
834 void CYPropertyValue::Replace(CYContext
&context
, CYBuilder
&builder
, CYExpression
*self
, CYExpression
*name
, bool protect
) {
836 CYIdentifier
*unique(context
.Unique());
838 ->* $
B1($
B(unique
, value_
));
840 ->* $
E($
CYAssign($
M(self
, name
), $
V(unique
)));
843 void CYPropertyValue::Replace(CYContext
&context
) {
844 context
.Replace(value_
);
847 void CYScript::Replace(CYContext
&context
) {
848 CYScope
scope(false, context
);
849 context
.scope_
->Damage();
851 context
.nextlocal_
= $
CYNonLocal();
852 context
.ReplaceAll(code_
);
853 context
.NonLocal(code_
);
855 scope
.Close(context
, code_
);
859 for (std::vector
<CYIdentifier
*>::const_iterator
i(context
.replace_
.begin()); i
!= context
.replace_
.end(); ++i
) {
861 if (context
.options_
.verbose_
)
862 name
= $pool
.strcat("$", $pool
.itoa(offset
++), NULL
);
868 unsigned position(7), local(offset
++ + 1);
871 unsigned index(local
% (sizeof(MappingSet
) - 1));
872 local
/= sizeof(MappingSet
) - 1;
873 id
[--position
] = MappingSet
[index
];
874 } while (local
!= 0);
876 if (scope
.Lookup(context
, id
+ position
) != NULL
)
878 // XXX: at some point, this could become a keyword
880 name
= $pool
.strmemdup(id
+ position
, 7 - position
);
883 CYIdentifier
*identifier(*i
);
884 _assert(identifier
->next_
== identifier
);
885 identifier
->next_
= $
I(name
);
889 CYTarget
*CYResolveMember::Replace(CYContext
&context
) {
890 return $
M($
M(object_
, $
S("$cyr")), property_
);
893 CYStatement
*CYReturn::Replace(CYContext
&context
) {
894 if (context
.nonlocal_
!= NULL
) {
895 CYProperty
*value(value_
== NULL
? NULL
: $
CYPropertyValue($
S("$cyv"), value_
));
896 return $
cy::Syntax::Throw($
CYObject(
897 $
CYPropertyValue($
S("$cyk"), $
V(context
.nonlocal_
->Target(context
)), value
)
901 context
.Replace(value_
);
905 CYTarget
*CYRubyBlock::Replace(CYContext
&context
) {
906 return lhs_
->AddArgument(context
, proc_
->Replace(context
));
909 CYTarget
*CYRubyBlock::AddArgument(CYContext
&context
, CYExpression
*value
) {
910 return Replace(context
)->AddArgument(context
, value
);
913 CYTarget
*CYRubyProc::Replace(CYContext
&context
) {
914 CYFunctionExpression
*function($
CYFunctionExpression(NULL
, parameters_
, code_
));
915 function
= CYNonLocalize(context
, function
);
916 function
->implicit_
= true;
920 CYScope::CYScope(bool transparent
, CYContext
&context
) :
921 transparent_(transparent
),
922 parent_(context
.scope_
),
927 _assert(!transparent_
|| parent_
!= NULL
);
928 context
.scope_
= this;
931 void CYScope::Damage() {
937 CYIdentifierFlags
*CYScope::Lookup(CYContext
&context
, const char *word
) {
938 CYForEach (i
, internal_
)
939 if (strcmp(i
->identifier_
->Word(), word
) == 0)
944 CYIdentifierFlags
*CYScope::Lookup(CYContext
&context
, CYIdentifier
*identifier
) {
945 return Lookup(context
, identifier
->Word());
948 CYIdentifierFlags
*CYScope::Declare(CYContext
&context
, CYIdentifier
*identifier
, CYIdentifierKind kind
) {
949 _assert(identifier
->next_
== NULL
|| identifier
->next_
== identifier
);
951 CYIdentifierFlags
*existing(Lookup(context
, identifier
));
952 if (existing
== NULL
)
953 internal_
= $
CYIdentifierFlags(identifier
, kind
, internal_
);
955 if (existing
== NULL
)
958 if (kind
== CYIdentifierGlobal
);
959 else if (existing
->kind_
== CYIdentifierGlobal
|| existing
->kind_
== CYIdentifierMagic
)
960 existing
->kind_
= kind
;
961 else if (existing
->kind_
== CYIdentifierLexical
|| kind
== CYIdentifierLexical
)
962 _assert(false); // XXX: throw new SyntaxError()
967 void CYScope::Merge(CYContext
&context
, const CYIdentifierFlags
*flags
) {
968 _assert(flags
->identifier_
->next_
== flags
->identifier_
);
969 CYIdentifierFlags
*existing(Declare(context
, flags
->identifier_
, flags
->kind_
));
970 flags
->identifier_
->next_
= existing
->identifier_
;
972 existing
->count_
+= flags
->count_
;
973 if (existing
->offset_
< flags
->offset_
)
974 existing
->offset_
= flags
->offset_
;
977 void CYScope::Close(CYContext
&context
, CYStatement
*&statements
) {
980 CYList
<CYBindings
> bindings
;
982 CYForEach (i
, internal_
)
983 if (i
->kind_
== CYIdentifierVariable
)
985 ->* $
CYBindings($
CYBinding(i
->identifier_
));
988 CYVar
*var($
CYVar(bindings
));
989 var
->SetNext(statements
);
994 void CYScope::Close(CYContext
&context
) {
995 context
.scope_
= parent_
;
997 CYForEach (i
, internal_
) {
998 _assert(i
->identifier_
->next_
== i
->identifier_
);
1000 case CYIdentifierArgument
: {
1001 _assert(!transparent_
);
1004 case CYIdentifierLexical
: {
1006 CYIdentifier
*replace(context
.Unique());
1007 replace
->next_
= replace
;
1008 i
->identifier_
->next_
= replace
;
1009 i
->identifier_
= replace
;
1013 i
->kind_
= CYIdentifierVariable
;
1015 parent_
->Declare(context
, i
->identifier_
, CYIdentifierVariable
);
1018 case CYIdentifierVariable
: {
1020 parent_
->Declare(context
, i
->identifier_
, i
->kind_
);
1021 i
->kind_
= CYIdentifierGlobal
;
1029 typedef std::multimap
<unsigned, CYIdentifier
*> CYIdentifierOffsetMap
;
1030 CYIdentifierOffsetMap offsets
;
1032 CYForEach (i
, internal_
) {
1033 _assert(i
->identifier_
->next_
== i
->identifier_
);
1035 case CYIdentifierArgument
:
1036 case CYIdentifierVariable
:
1037 offsets
.insert(CYIdentifierOffsetMap::value_type(i
->offset_
, i
->identifier_
));
1043 for (CYIdentifierOffsetMap::const_iterator
i(offsets
.begin()); i
!= offsets
.end(); ++i
) {
1044 if (offset
< i
->first
)
1046 CYIdentifier
*identifier(i
->second
);
1048 if (offset
>= context
.replace_
.size())
1049 context
.replace_
.resize(offset
+ 1, NULL
);
1050 CYIdentifier
*&replace(context
.replace_
[offset
++]);
1052 if (replace
== NULL
)
1053 replace
= identifier
;
1055 _assert(replace
->next_
== replace
);
1056 identifier
->next_
= replace
;
1060 if (parent_
== NULL
)
1063 CYForEach (i
, internal_
) {
1065 case CYIdentifierGlobal
: {
1066 if (i
->offset_
< offset
)
1067 i
->offset_
= offset
;
1068 parent_
->Merge(context
, i
);
1073 CYElementValue
*CYSpan::Replace(CYContext
&context
) { $
T(NULL
)
1074 return $
CYElementValue(expression_
, $
CYElementValue(string_
, next_
->Replace(context
)));
1077 CYStatement
*CYStatement::Return() {
1081 CYString
*CYString::Concat(CYContext
&context
, CYString
*rhs
) const {
1082 size_t size(size_
+ rhs
->size_
);
1083 char *value($
char[size
+ 1]);
1084 memcpy(value
, value_
, size_
);
1085 memcpy(value
+ size_
, rhs
->value_
, rhs
->size_
);
1087 return $
S(value
, size
);
1090 CYNumber
*CYString::Number(CYContext
&context
) {
1091 // XXX: there is a precise algorithm for this
1095 CYExpression
*CYString::PropertyName(CYContext
&context
) {
1099 CYString
*CYString::String(CYContext
&context
) {
1103 CYStatement
*CYStructDefinition::Replace(CYContext
&context
) {
1104 CYTarget
*target(tail_
->Replace(context
));
1106 target
= $
C1($
M(target
, $
S("withName")), $
S(name_
->Word()));
1107 return $
CYLexical(false, $
B1($
B($
I($pool
.strcat(name_
->Word(), "$cy", NULL
)), target
)));
1110 CYTarget
*CYStructTail::Replace(CYContext
&context
) {
1111 CYList
<CYElementValue
> types
;
1112 CYList
<CYElementValue
> names
;
1114 CYForEach (field
, fields_
) {
1115 CYTypedIdentifier
*typed(field
->typed_
);
1116 types
->*$
CYElementValue(typed
->Replace(context
));
1119 if (typed
->identifier_
== NULL
)
1122 name
= $
S(typed
->identifier_
->Word());
1123 names
->*$
CYElementValue(name
);
1126 return $
N2($
V("Type"), $
CYArray(types
), $
CYArray(names
));
1129 CYTarget
*CYSuperAccess::Replace(CYContext
&context
) {
1130 return $
C1($
M($
M($
M($
V(context
.super_
), $
S("prototype")), property_
), $
S("bind")), $
CYThis());
1133 CYTarget
*CYSuperCall::Replace(CYContext
&context
) {
1134 return $
C($
C1($
M($
V(context
.super_
), $
S("bind")), $
CYThis()), arguments_
);
1137 CYTarget
*CYSymbol::Replace(CYContext
&context
) {
1138 return $
C1($
M($
V("Symbol"), $
S("for")), $
S(name_
));
1141 CYStatement
*CYSwitch::Replace(CYContext
&context
) {
1142 context
.Replace(value_
);
1143 clauses_
->Replace(context
);
1147 CYStatement
*CYTarget::Initialize(CYContext
&context
, CYExpression
*value
) {
1150 return $
E($
CYAssign(this, value
));
1153 CYTarget
*CYTemplate::Replace(CYContext
&context
) {
1154 return $
C2($
M($
M($
M($
V("String"), $
S("prototype")), $
S("concat")), $
S("apply")), $
S(""), $
CYArray($
CYElementValue(string_
, spans_
->Replace(context
))));
1157 CYTarget
*CYThis::Replace(CYContext
&context
) {
1158 if (context
.this_
!= NULL
)
1159 return $
V(context
.this_
->Identifier(context
));
1166 CYStatement
*Throw::Replace(CYContext
&context
) {
1167 context
.Replace(value_
);
1173 CYTarget
*CYTrivial::Replace(CYContext
&context
) {
1177 CYNumber
*CYTrue::Number(CYContext
&context
) {
1181 CYString
*CYTrue::String(CYContext
&context
) {
1188 CYStatement
*Try::Replace(CYContext
&context
) {
1189 CYScope
scope(true, context
);
1190 context
.ReplaceAll(code_
);
1191 scope
.Close(context
);
1193 catch_
->Replace(context
);
1194 finally_
->Replace(context
);
1200 CYTarget
*CYTypeArrayOf::Replace_(CYContext
&context
, CYTarget
*type
) {
1201 return next_
->Replace(context
, $
CYCall($
CYDirectMember(type
, $
CYString("arrayOf")), $
CYArgument(size_
)));
1204 CYTarget
*CYTypeBlockWith::Replace_(CYContext
&context
, CYTarget
*type
) {
1205 return next_
->Replace(context
, $
CYCall($
CYDirectMember(type
, $
CYString("blockWith")), parameters_
->Argument(context
)));
1208 CYTarget
*CYTypeCharacter::Replace(CYContext
&context
) {
1210 case CYTypeNeutral
: return $
V("char");
1211 case CYTypeSigned
: return $
V("schar");
1212 case CYTypeUnsigned
: return $
V("uchar");
1213 default: _assert(false);
1217 CYTarget
*CYTypeConstant::Replace_(CYContext
&context
, CYTarget
*type
) {
1218 return next_
->Replace(context
, $
CYCall($
CYDirectMember(type
, $
CYString("constant"))));
1221 CYStatement
*CYTypeDefinition::Replace(CYContext
&context
) {
1222 CYIdentifier
*identifier(typed_
->identifier_
);
1223 typed_
->identifier_
= NULL
;
1224 return $
CYLexical(false, $
B1($
B(identifier
, $
CYTypeExpression(typed_
))));
1227 CYTarget
*CYTypeError::Replace(CYContext
&context
) {
1232 CYTarget
*CYTypeExpression::Replace(CYContext
&context
) {
1233 return typed_
->Replace(context
);
1236 CYTarget
*CYTypeIntegral::Replace(CYContext
&context
) {
1237 bool u(signing_
== CYTypeUnsigned
);
1239 case 0: return $
V(u
? "ushort" : "short");
1240 case 1: return $
V(u
? "uint" : "int");
1241 case 2: return $
V(u
? "ulong" : "long");
1242 case 3: return $
V(u
? "ulonglong" : "longlong");
1243 default: _assert(false);
1247 CYTarget
*CYTypeModifier::Replace(CYContext
&context
, CYTarget
*type
) { $
T(type
)
1248 return Replace_(context
, type
);
1251 CYTarget
*CYTypeFunctionWith::Replace_(CYContext
&context
, CYTarget
*type
) {
1252 CYList
<CYArgument
> arguments(parameters_
->Argument(context
));
1254 arguments
->*$
C_($
CYNull());
1255 return next_
->Replace(context
, $
CYCall($
CYDirectMember(type
, $
CYString("functionWith")), arguments
));
1258 CYTarget
*CYTypePointerTo::Replace_(CYContext
&context
, CYTarget
*type
) {
1259 return next_
->Replace(context
, $
CYCall($
CYDirectMember(type
, $
CYString("pointerTo"))));
1262 CYTarget
*CYTypeReference::Replace(CYContext
&context
) {
1263 return $
V($pool
.strcat(name_
->Word(), "$cy", NULL
));
1266 CYTarget
*CYTypeStruct::Replace(CYContext
&context
) {
1267 CYTarget
*target(tail_
->Replace(context
));
1269 target
= $
C1($
M(target
, $
S("withName")), $
S(name_
->Word()));
1273 CYTarget
*CYTypeVariable::Replace(CYContext
&context
) {
1277 CYTarget
*CYTypeVoid::Replace(CYContext
&context
) {
1278 return $
N1($
V("Type"), $
CYString("v"));
1281 CYTarget
*CYTypeVolatile::Replace_(CYContext
&context
, CYTarget
*type
) {
1282 return next_
->Replace(context
, $
CYCall($
CYDirectMember(type
, $
CYString("volatile"))));
1285 CYTarget
*CYTypedIdentifier::Replace(CYContext
&context
) {
1286 return modifier_
->Replace(context
, specifier_
->Replace(context
));
1289 CYTypeFunctionWith
*CYTypedIdentifier::Function() {
1290 CYTypeModifier
*&modifier(CYGetLast(modifier_
));
1291 if (modifier
== NULL
)
1294 CYTypeFunctionWith
*function(modifier
->Function());
1295 if (function
== NULL
)
1302 CYArgument
*CYTypedParameter::Argument(CYContext
&context
) { $
T(NULL
)
1303 return $
CYArgument(typed_
->Replace(context
), next_
->Argument(context
));
1306 CYFunctionParameter
*CYTypedParameter::Parameters(CYContext
&context
) { $
T(NULL
)
1307 return $
CYFunctionParameter($
CYBinding(typed_
->identifier_
?: context
.Unique()), next_
->Parameters(context
));
1310 CYExpression
*CYTypedParameter::TypeSignature(CYContext
&context
, CYExpression
*prefix
) { $
T(prefix
)
1311 return next_
->TypeSignature(context
, $
CYAdd(prefix
, typed_
->Replace(context
)));
1314 CYForInitializer
*CYVar::Replace(CYContext
&context
) {
1315 if (CYExpression
*expression
= bindings_
->Replace(context
, CYIdentifierVariable
))
1316 return $
E(expression
);
1320 CYTarget
*CYVariable::Replace(CYContext
&context
) {
1321 name_
= name_
->Replace(context
, CYIdentifierGlobal
);
1325 CYFunctionParameter
*CYVariable::Parameter() const {
1326 return $
CYFunctionParameter($
CYBinding(name_
));
1329 CYStatement
*CYWhile::Replace(CYContext
&context
) {
1330 context
.Replace(test_
);
1331 context
.ReplaceAll(code_
);
1335 CYStatement
*CYWith::Replace(CYContext
&context
) {
1336 context
.Replace(scope_
);
1337 CYScope
scope(true, context
);
1339 context
.ReplaceAll(code_
);
1340 scope
.Close(context
);
1344 CYExpression
*CYWord::PropertyName(CYContext
&context
) {