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 if (elements_
!= NULL
)
105 elements_
->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_
);
351 next_
->Replace(context
);
354 CYForInitializer
*CYEmpty::Replace(CYContext
&context
) {
358 CYTarget
*CYEncodedType::Replace(CYContext
&context
) {
359 return typed_
->Replace(context
);
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_
);
369 CYStatement
*CYExpress::Return() {
370 return $
CYReturn(expression_
);
373 CYForInitializer
*CYExpress::Replace(CYContext
&context
) {
374 context
.Replace(expression_
);
378 CYTarget
*CYExpression::AddArgument(CYContext
&context
, CYExpression
*value
) {
379 return $
C1(this, value
);
382 CYFunctionParameter
*CYExpression::Parameter() const {
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())))));
390 CYNumber
*CYFalse::Number(CYContext
&context
) {
394 CYString
*CYFalse::String(CYContext
&context
) {
398 CYExpression
*CYFatArrow::Replace(CYContext
&context
) {
399 CYFunctionExpression
*function($
CYFunctionExpression(NULL
, parameters_
, code_
));
400 function
->this_
.SetNext(context
.this_
);
404 void CYFinally::Replace(CYContext
&context
) { $
T()
405 CYScope
scope(true, context
);
406 context
.ReplaceAll(code_
);
407 scope
.Close(context
);
410 CYStatement
*CYFor::Replace(CYContext
&context
) {
411 CYScope
outer(true, context
);
412 context
.Replace(initializer_
);
414 context
.Replace(test_
);
417 CYScope
inner(true, context
);
418 context
.ReplaceAll(code_
);
419 inner
.Close(context
);
422 context
.Replace(increment_
);
424 outer
.Close(context
);
428 CYStatement
*CYForLexical::Initialize(CYContext
&context
, CYExpression
*value
) {
430 if (binding_
->initializer_
== NULL
)
432 value
= binding_
->initializer_
;
435 return $
CYLexical(constant_
, $
B1($
CYBinding(binding_
->identifier_
, value
)));
438 CYTarget
*CYForLexical::Replace(CYContext
&context
) {
439 _assert(binding_
->Replace(context
, CYIdentifierLexical
) == NULL
);
440 return binding_
->Target(context
);
443 CYStatement
*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
);
452 CYStatement
*CYForInitialized::Replace(CYContext
&context
) {
453 CYAssignment
*assignment(binding_
->Replace(context
, CYIdentifierVariable
));
455 ->* (assignment
== NULL
? NULL
: $
CYExpress(assignment
))
456 ->* $
CYForIn(binding_
->Target(context
), iterable_
, code_
));
459 CYFunctionParameter
*CYForInComprehension::Parameter(CYContext
&context
) const {
460 return $
CYFunctionParameter(binding_
);
463 CYStatement
*CYForInComprehension::Replace(CYContext
&context
, CYStatement
*statement
) const {
464 return $
CYForIn(binding_
->Target(context
), iterable_
, CYComprehension::Replace(context
, statement
));
467 CYStatement
*CYForOf::Replace(CYContext
&context
) {
468 CYIdentifier
*item(context
.Unique()), *list(context
.Unique());
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
)))
479 CYFunctionParameter
*CYForOfComprehension::Parameter(CYContext
&context
) const {
480 return $
CYFunctionParameter(binding_
);
483 CYStatement
*CYForOfComprehension::Replace(CYContext
&context
, CYStatement
*statement
) const {
484 CYIdentifier
*cys(context
.Unique());
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
)
494 CYStatement
*CYForVariable::Initialize(CYContext
&context
, CYExpression
*value
) {
496 if (binding_
->initializer_
== NULL
)
498 value
= binding_
->initializer_
;
501 return $
CYVar($
B1($
CYBinding(binding_
->identifier_
, value
)));
504 CYTarget
*CYForVariable::Replace(CYContext
&context
) {
505 _assert(binding_
->Replace(context
, CYIdentifierVariable
) == NULL
);
506 return binding_
->Target(context
);
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$_"
513 void CYFunction::Replace(CYContext
&context
) {
514 CYThisScope
*_this(context
.this_
);
515 context
.this_
= &this_
;
516 context
.this_
= CYGetLast(context
.this_
);
518 CYIdentifier
*super(context
.super_
);
519 context
.super_
= super_
;
521 CYNonLocal
*nonlocal(context
.nonlocal_
);
522 CYNonLocal
*nextlocal(context
.nextlocal_
);
525 if (nonlocal_
!= NULL
) {
527 context
.nonlocal_
= nonlocal_
;
530 nonlocal_
= $
CYNonLocal();
531 context
.nextlocal_
= nonlocal_
;
534 CYScope
scope(!localize
, context
);
536 $
I("arguments")->Replace(context
, CYIdentifierMagic
);
538 parameters_
->Replace(context
, code_
);
540 context
.ReplaceAll(code_
);
543 CYImplicitReturn(code_
);
545 if (CYIdentifier
*identifier
= this_
.identifier_
) {
546 context
.scope_
->Declare(context
, identifier
, CYIdentifierVariable
);
548 ->* $
E($
CYAssign($
V(identifier
), $
CYThis()))
553 context
.NonLocal(code_
);
555 context
.nextlocal_
= nextlocal
;
556 context
.nonlocal_
= nonlocal
;
558 context
.super_
= super
;
559 context
.this_
= _this
;
561 scope
.Close(context
, code_
);
564 CYTarget
*CYFunctionExpression::Replace(CYContext
&context
) {
565 CYScope
scope(false, context
);
567 name_
= name_
->Replace(context
, CYIdentifierOther
);
569 CYFunction::Replace(context
);
570 scope
.Close(context
);
574 void CYFunctionParameter::Replace(CYContext
&context
, CYStatement
*&statements
) { $
T()
575 CYAssignment
*assignment(binding_
->Replace(context
, CYIdentifierArgument
));
577 next_
->Replace(context
, statements
);
579 if (assignment
!= NULL
)
581 ->* $
CYIf($
CYIdentical($
CYTypeOf(binding_
->Target(context
)), $
S("undefined")), $$
586 CYStatement
*CYFunctionStatement::Replace(CYContext
&context
) {
587 name_
= name_
->Replace(context
, CYIdentifierOther
);
588 CYFunction::Replace(context
);
592 CYIdentifier
*CYIdentifier::Replace(CYContext
&context
, CYIdentifierKind kind
) {
596 return next_
->Replace(context
, kind
);
597 next_
= context
.scope_
->Declare(context
, this, kind
)->identifier_
;
601 CYStatement
*CYIf::Return() {
602 CYImplicitReturn(true_
);
603 CYImplicitReturn(false_
);
607 CYStatement
*CYIf::Replace(CYContext
&context
) {
608 context
.Replace(test_
);
609 context
.ReplaceAll(true_
);
610 context
.ReplaceAll(false_
);
614 CYFunctionParameter
*CYIfComprehension::Parameter(CYContext
&context
) const {
618 CYStatement
*CYIfComprehension::Replace(CYContext
&context
, CYStatement
*statement
) const {
619 return $
CYIf(test_
, CYComprehension::Replace(context
, statement
));
622 CYStatement
*CYImport::Replace(CYContext
&context
) {
623 return $
CYVar($
B1($
B($
I(module_
->part_
->Word()), $
C1($
V("require"), module_
->Replace(context
, "/")))));
626 CYStatement
*CYImportDeclaration::Replace(CYContext
&context
) {
627 CYIdentifier
*module(context
.Unique());
629 CYList
<CYStatement
> statements
;
630 CYForEach (specifier
, specifiers_
)
631 statements
->*specifier
->Replace(context
, module);
634 ->* $
CYLexical(false, $
B1($
B(module, $
C1($
V("require"), module_
))))
638 CYStatement
*CYImportSpecifier::Replace(CYContext
&context
, CYIdentifier
*module) {
639 binding_
= binding_
->Replace(context
, CYIdentifierLexical
);
641 CYExpression
*import($
V(module));
643 import = $
M(import, $
S(name_
));
644 return $
E($
CYAssign($
V(binding_
), import));
647 CYTarget
*CYIndirect::Replace(CYContext
&context
) {
648 return $
M(rhs_
, $
S("$cyi"));
651 CYTarget
*CYIndirectMember::Replace(CYContext
&context
) {
652 return $
M($
CYIndirect(object_
), property_
);
655 CYExpression
*CYInfix::Replace(CYContext
&context
) {
656 context
.Replace(lhs_
);
657 context
.Replace(rhs_
);
661 CYStatement
*CYLabel::Replace(CYContext
&context
) {
662 context
.Replace(statement_
);
666 CYTarget
*CYLambda::Replace(CYContext
&context
) {
667 return $
N2($
V("Functor"), $
CYFunctionExpression(NULL
, parameters_
->Parameters(context
), code_
), parameters_
->TypeSignature(context
, typed_
->Replace(context
)));
670 CYForInitializer
*CYLexical::Replace(CYContext
&context
) {
671 if (CYExpression
*expression
= bindings_
->Replace(context
, CYIdentifierLexical
))
672 return $
E(expression
);
676 CYFunctionExpression
*CYMethod::Constructor() {
680 void CYMethod::Replace(CYContext
&context
) {
681 CYFunction::Replace(context
);
684 CYString
*CYModule::Replace(CYContext
&context
, const char *separator
) const {
686 return $
CYString(part_
);
687 return $
CYString($pool
.strcat(next_
->Replace(context
, separator
)->Value(), separator
, part_
->Word(), NULL
));
690 CYExpression
*CYMultiply::Replace(CYContext
&context
) {
691 CYInfix::Replace(context
);
693 if (CYNumber
*lhn
= lhs_
->Number(context
))
694 if (CYNumber
*rhn
= rhs_
->Number(context
))
695 return $
D(lhn
->Value() * rhn
->Value());
703 CYTarget
*New::AddArgument(CYContext
&context
, CYExpression
*value
) {
704 CYSetLast(arguments_
) = $
CYArgument(value
);
708 CYTarget
*New::Replace(CYContext
&context
) {
709 context
.Replace(constructor_
);
710 arguments_
->Replace(context
);
716 CYNumber
*CYNull::Number(CYContext
&context
) {
720 CYString
*CYNull::String(CYContext
&context
) {
724 CYNumber
*CYNumber::Number(CYContext
&context
) {
728 CYString
*CYNumber::String(CYContext
&context
) {
729 // XXX: there is a precise algorithm for this
730 return $
S($pool
.sprintf(24, "%.17g", Value()));
733 CYExpression
*CYNumber::PropertyName(CYContext
&context
) {
734 return String(context
);
737 CYTarget
*CYObject::Replace(CYContext
&context
) {
739 if (properties_
!= NULL
)
740 properties_
= properties_
->ReplaceAll(context
, builder
, $
CYThis(), false);
743 return $
C1($
M($
CYFunctionExpression(NULL
, builder
.bindings_
->Parameter(context
),
745 ->* $
CYReturn($
CYThis())
746 ), $
S("call")), this, builder
.bindings_
->Argument(context
));
749 CYForEach (property
, properties_
)
750 property
->Replace(context
);
754 CYTarget
*CYParenthetical::Replace(CYContext
&context
) {
755 // XXX: return expression_;
756 context
.Replace(expression_
);
760 CYExpression
*CYPostfix::Replace(CYContext
&context
) {
761 context
.Replace(lhs_
);
765 CYExpression
*CYPrefix::Replace(CYContext
&context
) {
766 context
.Replace(rhs_
);
770 CYProperty
*CYProperty::ReplaceAll(CYContext
&context
, CYBuilder
&builder
, CYExpression
*self
, bool update
) {
773 Replace(context
, builder
, self
, false);
775 next_
= next_
->ReplaceAll(context
, builder
, self
, update
);
776 return update
? next_
: this;
779 void 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());
784 ->* $
B1($
B(unique
, name
));
788 Replace(context
, builder
, self
, name
, protect
);
791 bool CYProperty::Update() const {
792 return name_
->Computed();
795 void CYPropertyGetter::Replace(CYContext
&context
, CYBuilder
&builder
, CYExpression
*self
, CYExpression
*name
, bool protect
) {
796 CYIdentifier
*unique(context
.Unique());
798 ->* $
B1($
B(unique
, CYSuperize(context
, $
CYFunctionExpression(NULL
, parameters_
, code_
))));
800 ->* CYDefineProperty(self
, name
, true, !protect
, $
CYPropertyValue($
S("get"), $
V(unique
)));
803 CYFunctionExpression
*CYPropertyMethod::Constructor() {
804 return name_
->Constructor() ? $
CYFunctionExpression(NULL
, parameters_
, code_
) : NULL
;
807 void CYPropertyMethod::Replace(CYContext
&context
, CYBuilder
&builder
, CYExpression
*self
, CYExpression
*name
, bool protect
) {
808 CYIdentifier
*unique(context
.Unique());
810 ->* $
B1($
B(unique
, CYSuperize(context
, $
CYFunctionExpression(NULL
, parameters_
, code_
))));
812 ->* (!protect
? $
E($
CYAssign($
M(self
, name
), $
V(unique
))) :
813 CYDefineProperty(self
, name
, true, !protect
, $
CYPropertyValue($
S("value"), $
V(unique
), $
CYPropertyValue($
S("writable"), $
CYTrue()))));
816 bool CYPropertyMethod::Update() const {
820 void CYPropertySetter::Replace(CYContext
&context
, CYBuilder
&builder
, CYExpression
*self
, CYExpression
*name
, bool protect
) {
821 CYIdentifier
*unique(context
.Unique());
823 ->* $
B1($
B(unique
, CYSuperize(context
, $
CYFunctionExpression(NULL
, parameters_
, code_
))));
825 ->* CYDefineProperty(self
, name
, true, !protect
, $
CYPropertyValue($
S("set"), $
V(unique
)));
828 void CYPropertyValue::Replace(CYContext
&context
, CYBuilder
&builder
, CYExpression
*self
, CYExpression
*name
, bool protect
) {
830 CYIdentifier
*unique(context
.Unique());
832 ->* $
B1($
B(unique
, value_
));
834 ->* $
E($
CYAssign($
M(self
, name
), $
V(unique
)));
837 void CYPropertyValue::Replace(CYContext
&context
) {
838 context
.Replace(value_
);
841 void CYScript::Replace(CYContext
&context
) {
842 CYScope
scope(false, context
);
843 context
.scope_
->Damage();
845 context
.nextlocal_
= $
CYNonLocal();
846 context
.ReplaceAll(code_
);
847 context
.NonLocal(code_
);
849 scope
.Close(context
, code_
);
853 for (std::vector
<CYIdentifier
*>::const_iterator
i(context
.replace_
.begin()); i
!= context
.replace_
.end(); ++i
) {
855 if (context
.options_
.verbose_
)
856 name
= $pool
.strcat("$", $pool
.itoa(offset
++), NULL
);
862 unsigned position(7), local(offset
++ + 1);
865 unsigned index(local
% (sizeof(MappingSet
) - 1));
866 local
/= sizeof(MappingSet
) - 1;
867 id
[--position
] = MappingSet
[index
];
868 } while (local
!= 0);
870 if (scope
.Lookup(context
, id
+ position
) != NULL
)
872 // XXX: at some point, this could become a keyword
874 name
= $pool
.strmemdup(id
+ position
, 7 - position
);
877 CYIdentifier
*identifier(*i
);
878 _assert(identifier
->next_
== identifier
);
879 identifier
->next_
= $
I(name
);
883 CYStatement
*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
)
891 context
.Replace(value_
);
895 CYTarget
*CYRubyBlock::Replace(CYContext
&context
) {
896 return call_
->AddArgument(context
, proc_
->Replace(context
));
899 CYTarget
*CYRubyBlock::AddArgument(CYContext
&context
, CYExpression
*value
) {
900 return Replace(context
)->AddArgument(context
, value
);
903 CYTarget
*CYRubyProc::Replace(CYContext
&context
) {
904 CYFunctionExpression
*function($
CYFunctionExpression(NULL
, parameters_
, code_
));
905 function
= CYNonLocalize(context
, function
);
906 function
->implicit_
= true;
910 CYScope::CYScope(bool transparent
, CYContext
&context
) :
911 transparent_(transparent
),
912 parent_(context
.scope_
),
917 _assert(!transparent_
|| parent_
!= NULL
);
918 context
.scope_
= this;
921 void CYScope::Damage() {
927 CYIdentifierFlags
*CYScope::Lookup(CYContext
&context
, const char *word
) {
928 CYForEach (i
, internal_
)
929 if (strcmp(i
->identifier_
->Word(), word
) == 0)
934 CYIdentifierFlags
*CYScope::Lookup(CYContext
&context
, CYIdentifier
*identifier
) {
935 return Lookup(context
, identifier
->Word());
938 CYIdentifierFlags
*CYScope::Declare(CYContext
&context
, CYIdentifier
*identifier
, CYIdentifierKind kind
) {
939 _assert(identifier
->next_
== NULL
|| identifier
->next_
== identifier
);
941 CYIdentifierFlags
*existing(Lookup(context
, identifier
));
942 if (existing
== NULL
)
943 internal_
= $
CYIdentifierFlags(identifier
, kind
, internal_
);
945 if (existing
== NULL
)
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()
957 void 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_
;
962 existing
->count_
+= flags
->count_
;
963 if (existing
->offset_
< flags
->offset_
)
964 existing
->offset_
= flags
->offset_
;
967 void CYScope::Close(CYContext
&context
, CYStatement
*&statements
) {
970 CYList
<CYBindings
> bindings
;
972 CYForEach (i
, internal_
)
973 if (i
->kind_
== CYIdentifierVariable
)
975 ->* $
CYBindings($
CYBinding(i
->identifier_
));
978 CYVar
*var($
CYVar(bindings
));
979 var
->SetNext(statements
);
984 void CYScope::Close(CYContext
&context
) {
985 context
.scope_
= parent_
;
987 CYForEach (i
, internal_
) {
988 _assert(i
->identifier_
->next_
== i
->identifier_
);
990 case CYIdentifierArgument
: {
991 _assert(!transparent_
);
994 case CYIdentifierLexical
: {
996 CYIdentifier
*replace(context
.Unique());
997 replace
->next_
= replace
;
998 i
->identifier_
->next_
= replace
;
999 i
->identifier_
= replace
;
1003 i
->kind_
= CYIdentifierVariable
;
1005 parent_
->Declare(context
, i
->identifier_
, CYIdentifierVariable
);
1008 case CYIdentifierVariable
: {
1010 parent_
->Declare(context
, i
->identifier_
, i
->kind_
);
1011 i
->kind_
= CYIdentifierGlobal
;
1019 typedef std::multimap
<unsigned, CYIdentifier
*> CYIdentifierOffsetMap
;
1020 CYIdentifierOffsetMap offsets
;
1022 CYForEach (i
, internal_
) {
1023 _assert(i
->identifier_
->next_
== i
->identifier_
);
1025 case CYIdentifierArgument
:
1026 case CYIdentifierVariable
:
1027 offsets
.insert(CYIdentifierOffsetMap::value_type(i
->offset_
, i
->identifier_
));
1033 for (CYIdentifierOffsetMap::const_iterator
i(offsets
.begin()); i
!= offsets
.end(); ++i
) {
1034 if (offset
< i
->first
)
1036 CYIdentifier
*identifier(i
->second
);
1038 if (offset
>= context
.replace_
.size())
1039 context
.replace_
.resize(offset
+ 1, NULL
);
1040 CYIdentifier
*&replace(context
.replace_
[offset
++]);
1042 if (replace
== NULL
)
1043 replace
= identifier
;
1045 _assert(replace
->next_
== replace
);
1046 identifier
->next_
= replace
;
1050 if (parent_
== NULL
)
1053 CYForEach (i
, internal_
) {
1055 case CYIdentifierGlobal
: {
1056 if (i
->offset_
< offset
)
1057 i
->offset_
= offset
;
1058 parent_
->Merge(context
, i
);
1063 CYElementValue
*CYSpan::Replace(CYContext
&context
) { $
T(NULL
)
1064 return $
CYElementValue(expression_
, $
CYElementValue(string_
, next_
->Replace(context
)));
1067 CYStatement
*CYStatement::Return() {
1071 CYString
*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_
);
1077 return $
S(value
, size
);
1080 CYNumber
*CYString::Number(CYContext
&context
) {
1081 // XXX: there is a precise algorithm for this
1085 CYExpression
*CYString::PropertyName(CYContext
&context
) {
1089 CYString
*CYString::String(CYContext
&context
) {
1093 CYStatement
*CYStructDefinition::Replace(CYContext
&context
) {
1094 CYTarget
*target(tail_
->Replace(context
));
1096 target
= $
C1($
M(target
, $
S("withName")), $
S(name_
->Word()));
1097 return $
CYLexical(false, $
B1($
B($
I($pool
.strcat(name_
->Word(), "$cy", NULL
)), target
)));
1100 CYTarget
*CYStructTail::Replace(CYContext
&context
) {
1101 CYList
<CYElementValue
> types
;
1102 CYList
<CYElementValue
> names
;
1104 CYForEach (field
, fields_
) {
1105 CYTypedIdentifier
*typed(field
->typed_
);
1106 types
->*$
CYElementValue(typed
->Replace(context
));
1109 if (typed
->identifier_
== NULL
)
1112 name
= $
S(typed
->identifier_
->Word());
1113 names
->*$
CYElementValue(name
);
1116 return $
N2($
V("Type"), $
CYArray(types
), $
CYArray(names
));
1119 CYTarget
*CYSuperAccess::Replace(CYContext
&context
) {
1120 return $
C1($
M($
M($
M($
V(context
.super_
), $
S("prototype")), property_
), $
S("bind")), $
CYThis());
1123 CYTarget
*CYSuperCall::Replace(CYContext
&context
) {
1124 return $
C($
C1($
M($
V(context
.super_
), $
S("bind")), $
CYThis()), arguments_
);
1127 CYStatement
*CYSwitch::Replace(CYContext
&context
) {
1128 context
.Replace(value_
);
1129 clauses_
->Replace(context
);
1133 CYStatement
*CYTarget::Initialize(CYContext
&context
, CYExpression
*value
) {
1136 return $
E($
CYAssign(this, value
));
1139 CYTarget
*CYTemplate::Replace(CYContext
&context
) {
1140 return $
C2($
M($
M($
M($
V("String"), $
S("prototype")), $
S("concat")), $
S("apply")), $
S(""), $
CYArray($
CYElementValue(string_
, spans_
->Replace(context
))));
1143 CYTarget
*CYThis::Replace(CYContext
&context
) {
1144 if (context
.this_
!= NULL
)
1145 return $
V(context
.this_
->Identifier(context
));
1152 CYStatement
*Throw::Replace(CYContext
&context
) {
1153 context
.Replace(value_
);
1159 CYTarget
*CYTrivial::Replace(CYContext
&context
) {
1163 CYNumber
*CYTrue::Number(CYContext
&context
) {
1167 CYString
*CYTrue::String(CYContext
&context
) {
1174 CYStatement
*Try::Replace(CYContext
&context
) {
1175 CYScope
scope(true, context
);
1176 context
.ReplaceAll(code_
);
1177 scope
.Close(context
);
1179 catch_
->Replace(context
);
1180 finally_
->Replace(context
);
1186 CYTarget
*CYTypeArrayOf::Replace_(CYContext
&context
, CYTarget
*type
) {
1187 return next_
->Replace(context
, $
CYCall($
CYDirectMember(type
, $
CYString("arrayOf")), $
CYArgument(size_
)));
1190 CYTarget
*CYTypeBlockWith::Replace_(CYContext
&context
, CYTarget
*type
) {
1191 return next_
->Replace(context
, $
CYCall($
CYDirectMember(type
, $
CYString("blockWith")), parameters_
->Argument(context
)));
1194 CYTarget
*CYTypeConstant::Replace_(CYContext
&context
, CYTarget
*type
) {
1195 return next_
->Replace(context
, $
CYCall($
CYDirectMember(type
, $
CYString("constant"))));
1198 CYStatement
*CYTypeDefinition::Replace(CYContext
&context
) {
1199 CYIdentifier
*identifier(typed_
->identifier_
);
1200 typed_
->identifier_
= NULL
;
1201 return $
CYLexical(false, $
B1($
B(identifier
, $
CYTypeExpression(typed_
))));
1204 CYTarget
*CYTypeError::Replace(CYContext
&context
) {
1209 CYTarget
*CYTypeExpression::Replace(CYContext
&context
) {
1210 return typed_
->Replace(context
);
1213 CYTarget
*CYTypeModifier::Replace(CYContext
&context
, CYTarget
*type
) { $
T(type
)
1214 return Replace_(context
, type
);
1217 CYTarget
*CYTypeFunctionWith::Replace_(CYContext
&context
, CYTarget
*type
) {
1218 return next_
->Replace(context
, $
CYCall($
CYDirectMember(type
, $
CYString("functionWith")), parameters_
->Argument(context
)));
1221 CYTarget
*CYTypeLong::Replace(CYContext
&context
) {
1222 return $
CYCall($
CYDirectMember(specifier_
->Replace(context
), $
CYString("long")));
1225 CYTarget
*CYTypePointerTo::Replace_(CYContext
&context
, CYTarget
*type
) {
1226 return next_
->Replace(context
, $
CYCall($
CYDirectMember(type
, $
CYString("pointerTo"))));
1229 CYTarget
*CYTypeReference::Replace(CYContext
&context
) {
1230 return $
V($pool
.strcat(name_
->Word(), "$cy", NULL
));
1233 CYTarget
*CYTypeShort::Replace(CYContext
&context
) {
1234 return $
CYCall($
CYDirectMember(specifier_
->Replace(context
), $
CYString("short")));
1237 CYTarget
*CYTypeSigned::Replace(CYContext
&context
) {
1238 return $
CYCall($
CYDirectMember(specifier_
->Replace(context
), $
CYString("signed")));
1241 CYTarget
*CYTypeStruct::Replace(CYContext
&context
) {
1242 CYTarget
*target(tail_
->Replace(context
));
1244 target
= $
C1($
M(target
, $
S("withName")), $
S(name_
->Word()));
1248 CYTarget
*CYTypeUnsigned::Replace(CYContext
&context
) {
1249 return $
CYCall($
CYDirectMember(specifier_
->Replace(context
), $
CYString("unsigned")));
1252 CYTarget
*CYTypeVariable::Replace(CYContext
&context
) {
1256 CYTarget
*CYTypeVoid::Replace(CYContext
&context
) {
1257 return $
N1($
V("Type"), $
CYString("v"));
1260 CYTarget
*CYTypeVolatile::Replace_(CYContext
&context
, CYTarget
*type
) {
1261 return next_
->Replace(context
, $
CYCall($
CYDirectMember(type
, $
CYString("volatile"))));
1264 CYTarget
*CYTypedIdentifier::Replace(CYContext
&context
) {
1265 return modifier_
->Replace(context
, specifier_
->Replace(context
));
1268 CYTypeFunctionWith
*CYTypedIdentifier::Function() {
1269 CYTypeModifier
**modifier(&modifier_
);
1270 if (*modifier
== NULL
)
1272 while ((*modifier
)->next_
!= NULL
)
1273 modifier
= &(*modifier
)->next_
;
1274 CYTypeFunctionWith
*function((*modifier
)->Function());
1275 if (function
== NULL
)
1281 CYArgument
*CYTypedParameter::Argument(CYContext
&context
) { $
T(NULL
)
1282 return $
CYArgument(typed_
->Replace(context
), next_
->Argument(context
));
1285 CYFunctionParameter
*CYTypedParameter::Parameters(CYContext
&context
) { $
T(NULL
)
1286 return $
CYFunctionParameter($
CYBinding(typed_
->identifier_
?: context
.Unique()), next_
->Parameters(context
));
1289 CYExpression
*CYTypedParameter::TypeSignature(CYContext
&context
, CYExpression
*prefix
) { $
T(prefix
)
1290 return next_
->TypeSignature(context
, $
CYAdd(prefix
, typed_
->Replace(context
)));
1293 CYForInitializer
*CYVar::Replace(CYContext
&context
) {
1294 if (CYExpression
*expression
= bindings_
->Replace(context
, CYIdentifierVariable
))
1295 return $
E(expression
);
1299 CYTarget
*CYVariable::Replace(CYContext
&context
) {
1300 name_
= name_
->Replace(context
, CYIdentifierGlobal
);
1304 CYFunctionParameter
*CYVariable::Parameter() const {
1305 return $
CYFunctionParameter($
CYBinding(name_
));
1308 CYStatement
*CYWhile::Replace(CYContext
&context
) {
1309 context
.Replace(test_
);
1310 context
.ReplaceAll(code_
);
1314 CYStatement
*CYWith::Replace(CYContext
&context
) {
1315 context
.Replace(scope_
);
1316 CYScope
scope(true, context
);
1318 context
.ReplaceAll(code_
);
1319 scope
.Close(context
);
1323 CYExpression
*CYWord::PropertyName(CYContext
&context
) {