]>
git.saurik.com Git - cycript.git/blob - Output.cpp
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/>.
22 #include "cycript.hpp"
28 void CYOutput::Terminate() {
33 CYOutput
&CYOutput::operator <<(char rhs
) {
34 if (rhs
== ' ' || rhs
== '\n')
40 for (unsigned i(0); i
!= indent_
; ++i
)
43 else if (rhs
== '\r') {
55 if (mode_
== Terminated
&& rhs
!= '}') {
67 } else if (rhs
== '+') {
71 } else if (rhs
== '-') {
72 if (mode_
== NoHyphen
)
75 } else if (WordEndRange_
[rhs
]) {
76 if (mode_
== NoLetter
)
88 CYOutput
&CYOutput::operator <<(const char *rhs
) {
89 size_t size(strlen(rhs
));
94 if (mode_
== Terminated
)
97 mode_
== NoPlus
&& *rhs
== '+' ||
98 mode_
== NoHyphen
&& *rhs
== '-' ||
99 mode_
== NoLetter
&& WordEndRange_
[*rhs
]
103 char last(rhs
[size
- 1]);
104 if (WordEndRange_
[last
] || last
== '/')
110 operator ()(rhs
, size
);
114 void CYArgument::Output(CYOutput
&out
) const {
121 value_
->Output(out
, CYAssign::Precedence_
, CYNoFlags
);
124 out
<< ' ' << *next_
;
128 void CYArray::Output(CYOutput
&out
, CYFlags flags
) const {
129 out
<< '[' << elements_
<< ']';
132 void CYArrayComprehension::Output(CYOutput
&out
, CYFlags flags
) const {
133 out
<< '[' << *expression_
<< ' ' << *comprehensions_
<< ']';
136 void CYAssignment::Output(CYOutput
&out
, CYFlags flags
) const {
137 lhs_
->Output(out
, Precedence() - 1, CYLeft(flags
) | CYNoRightHand
);
138 out
<< ' ' << Operator() << ' ';
139 rhs_
->Output(out
, Precedence(), CYRight(flags
));
142 void CYBlock::Output(CYOutput
&out
, CYFlags flags
) const {
150 void CYBoolean::Output(CYOutput
&out
, CYFlags flags
) const {
151 out
<< '!' << (Value() ? "0" : "1");
152 if ((flags
& CYNoInteger
) != 0)
156 void CYBreak::Output(CYOutput
&out
, CYFlags flags
) const {
159 out
<< ' ' << *label_
;
163 void CYCall::Output(CYOutput
&out
, CYFlags flags
) const {
164 bool protect((flags
& CYNoCall
) != 0);
167 function_
->Output(out
, Precedence(), protect
? CYNoFlags
: flags
);
168 out
<< '(' << arguments_
<< ')';
176 void Catch::Output(CYOutput
&out
) const {
177 out
<< ' ' << "catch" << ' ' << '(' << *name_
<< ')' << ' ';
187 void CYClassExpression::Output(CYOutput
&out
, CYFlags flags
) const {
188 bool protect((flags
& CYNoClass
) != 0);
193 out
<< ' ' << *name_
;
199 void CYClassStatement::Output(CYOutput
&out
, CYFlags flags
) const {
200 out
<< "class" << ' ' << *name_
<< *tail_
;
203 void CYClassTail::Output(CYOutput
&out
) const {
204 if (extends_
== NULL
)
209 out
<< "extends" << ' ';
210 extends_
->Output(out
, CYAssign::Precedence_
- 1, CYNoFlags
);
222 void CYCompound::Output(CYOutput
&out
, CYFlags flags
) const {
224 expression_
->Output(out
, flags
);
226 expression_
->Output(out
, CYLeft(flags
));
228 next_
->Output(out
, CYRight(flags
));
232 void CYComputed::PropertyName(CYOutput
&out
) const {
234 expression_
->Output(out
, CYAssign::Precedence_
, CYNoFlags
);
238 void CYCondition::Output(CYOutput
&out
, CYFlags flags
) const {
239 test_
->Output(out
, Precedence() - 1, CYLeft(flags
));
240 out
<< ' ' << '?' << ' ';
242 true_
->Output(out
, CYAssign::Precedence_
, CYNoFlags
);
243 out
<< ' ' << ':' << ' ';
244 false_
->Output(out
, CYAssign::Precedence_
, CYRight(flags
));
247 void CYContinue::Output(CYOutput
&out
, CYFlags flags
) const {
250 out
<< ' ' << *label_
;
254 void CYClause::Output(CYOutput
&out
) const {
257 out
<< "case" << ' ' << *case_
;
267 void CYDebugger::Output(CYOutput
&out
, CYFlags flags
) const {
268 out
<< "debugger" << ';';
271 void CYDeclaration::Output(CYOutput
&out
, CYFlags flags
) const {
273 //out.out_ << ':' << identifier_->usage_ << '#' << identifier_->offset_;
274 if (initialiser_
!= NULL
) {
275 out
<< ' ' << '=' << ' ';
276 initialiser_
->Output(out
, CYAssign::Precedence_
, CYRight(flags
));
280 void CYDeclarations::Output(CYOutput
&out
) const {
281 Output(out
, CYNoFlags
);
284 void CYDeclarations::Output(CYOutput
&out
, CYFlags flags
) const {
285 const CYDeclarations
*declaration(this);
289 CYDeclarations
*next(declaration
->next_
);
291 CYFlags
jacks(first
? CYLeft(flags
) : next
== NULL
? CYRight(flags
) : CYCenter(flags
));
293 declaration
->declaration_
->Output(out
, jacks
);
303 void CYDirectMember::Output(CYOutput
&out
, CYFlags flags
) const {
304 object_
->Output(out
, Precedence(), CYLeft(flags
) | CYNoInteger
);
305 if (const char *word
= property_
->Word())
308 out
<< '[' << *property_
<< ']';
311 void CYDoWhile::Output(CYOutput
&out
, CYFlags flags
) const {
314 unsigned line(out
.position_
.line
);
315 unsigned indent(out
.indent_
);
316 code_
->Single(out
, CYCenter(flags
), CYCompactLong
);
318 if (out
.position_
.line
!= line
&& out
.recent_
== indent
)
323 out
<< "while" << ' ' << '(' << *test_
<< ')';
326 void CYElementSpread::Output(CYOutput
&out
) const {
327 out
<< "..." << value_
;
330 void CYElementValue::Output(CYOutput
&out
) const {
332 value_
->Output(out
, CYAssign::Precedence_
, CYNoFlags
);
333 if (next_
!= NULL
|| value_
== NULL
) {
335 if (next_
!= NULL
&& !next_
->Elision())
342 void CYEmpty::Output(CYOutput
&out
, CYFlags flags
) const {
346 void CYEval::Output(CYOutput
&out
, CYFlags flags
) const {
350 void CYExpress::Output(CYOutput
&out
, CYFlags flags
) const {
351 expression_
->Output(out
, flags
| CYNoBFC
);
355 void CYExpression::Output(CYOutput
&out
) const {
356 Output(out
, CYNoFlags
);
359 void CYExpression::Output(CYOutput
&out
, int precedence
, CYFlags flags
) const {
360 if (precedence
< Precedence() || (flags
& CYNoRightHand
) != 0 && RightHand())
361 out
<< '(' << *this << ')';
366 void CYExternal::Output(CYOutput
&out
, CYFlags flags
) const {
367 out
<< "extern" << abi_
<< typed_
<< ';';
370 void CYFatArrow::Output(CYOutput
&out
, CYFlags flags
) const {
371 out
<< '(' << parameters_
<< ')' << ' ' << "=>" << ' ' << '{' << code_
<< '}';
374 void CYFinally::Output(CYOutput
&out
) const {
375 out
<< ' ' << "finally" << ' ';
383 void CYFor::Output(CYOutput
&out
, CYFlags flags
) const {
384 out
<< "for" << ' ' << '(';
385 if (initialiser_
!= NULL
)
386 initialiser_
->Output(out
, CYNoIn
);
392 if (increment_
!= NULL
)
396 code_
->Single(out
, CYRight(flags
), CYCompactShort
);
399 void CYForLexical::Output(CYOutput
&out
, CYFlags flags
) const {
400 out
<< (constant_
? "const" : "let") << ' ';
401 declaration_
->Output(out
, CYRight(flags
));
404 void CYForIn::Output(CYOutput
&out
, CYFlags flags
) const {
405 out
<< "for" << ' ' << '(';
406 initialiser_
->Output(out
, CYNoIn
| CYNoRightHand
);
407 out
<< ' ' << "in" << ' ' << *set_
<< ')';
408 code_
->Single(out
, CYRight(flags
), CYCompactShort
);
411 void CYForInitialized::Output(CYOutput
&out
, CYFlags flags
) const {
412 out
<< "for" << ' ' << '(' << "var" << ' ';
413 declaration_
->Output(out
, CYNoIn
| CYNoRightHand
);
414 out
<< ' ' << "in" << ' ' << *set_
<< ')';
415 code_
->Single(out
, CYRight(flags
), CYCompactShort
);
418 void CYForInComprehension::Output(CYOutput
&out
) const {
419 out
<< "for" << ' ' << '(';
420 declaration_
->Output(out
, CYNoIn
| CYNoRightHand
);
421 out
<< ' ' << "in" << ' ' << *set_
<< ')';
424 void CYForOf::Output(CYOutput
&out
, CYFlags flags
) const {
425 out
<< "for" << ' ' << '(';
426 initialiser_
->Output(out
, CYNoRightHand
);
427 out
<< ' ' << "of" << ' ' << *set_
<< ')';
428 code_
->Single(out
, CYRight(flags
), CYCompactShort
);
431 void CYForOfComprehension::Output(CYOutput
&out
) const {
432 out
<< "for" << ' ' << '(';
433 declaration_
->Output(out
, CYNoRightHand
);
434 out
<< ' ' << "of" << ' ' << *set_
<< ')' << next_
;
437 void CYForVariable::Output(CYOutput
&out
, CYFlags flags
) const {
439 declaration_
->Output(out
, CYRight(flags
));
442 void CYFunction::Output(CYOutput
&out
) const {
443 out
<< '(' << parameters_
<< ')' << ' ';
451 void CYFunctionExpression::Output(CYOutput
&out
, CYFlags flags
) const {
452 // XXX: one could imagine using + here to save a byte
453 bool protect((flags
& CYNoFunction
) != 0);
458 out
<< ' ' << *name_
;
459 CYFunction::Output(out
);
464 void CYFunctionStatement::Output(CYOutput
&out
, CYFlags flags
) const {
465 out
<< "function" << ' ' << *name_
;
466 CYFunction::Output(out
);
469 void CYFunctionParameter::Output(CYOutput
&out
) const {
470 initialiser_
->Output(out
, CYNoFlags
);
472 out
<< ',' << ' ' << *next_
;
475 const char *CYIdentifier::Word() const {
476 return next_
== NULL
|| next_
== this ? CYWord::Word() : next_
->Word();
479 void CYIf::Output(CYOutput
&out
, CYFlags flags
) const {
481 if (false_
== NULL
&& (flags
& CYNoDangle
) != 0) {
486 out
<< "if" << ' ' << '(' << *test_
<< ')';
488 CYFlags
right(protect
? CYNoFlags
: CYRight(flags
));
490 CYFlags
jacks(CYNoDangle
);
494 jacks
|= protect
? CYNoFlags
: CYCenter(flags
);
496 unsigned line(out
.position_
.line
);
497 unsigned indent(out
.indent_
);
498 true_
->Single(out
, jacks
, CYCompactShort
);
500 if (false_
!= NULL
) {
501 if (out
.position_
.line
!= line
&& out
.recent_
== indent
)
507 false_
->Single(out
, right
, CYCompactLong
);
514 void CYIfComprehension::Output(CYOutput
&out
) const {
515 out
<< "if" << ' ' << '(' << *test_
<< ')' << next_
;
518 void CYImport::Output(CYOutput
&out
, CYFlags flags
) const {
522 void CYIndirect::Output(CYOutput
&out
, CYFlags flags
) const {
524 rhs_
->Output(out
, Precedence(), CYRight(flags
));
527 void CYIndirectMember::Output(CYOutput
&out
, CYFlags flags
) const {
528 object_
->Output(out
, Precedence(), CYLeft(flags
));
529 if (const char *word
= property_
->Word())
532 out
<< "->" << '[' << *property_
<< ']';
535 void CYInfix::Output(CYOutput
&out
, CYFlags flags
) const {
536 const char *name(Operator());
537 bool protect((flags
& CYNoIn
) != 0 && strcmp(name
, "in") == 0);
540 CYFlags
left(protect
? CYNoFlags
: CYLeft(flags
));
541 lhs_
->Output(out
, Precedence(), left
);
542 out
<< ' ' << name
<< ' ';
543 CYFlags
right(protect
? CYNoFlags
: CYRight(flags
));
544 rhs_
->Output(out
, Precedence() - 1, right
);
549 void CYLabel::Output(CYOutput
&out
, CYFlags flags
) const {
550 out
<< *name_
<< ':';
551 statement_
->Single(out
, CYRight(flags
), CYCompactShort
);
554 void CYParenthetical::Output(CYOutput
&out
, CYFlags flags
) const {
556 expression_
->Output(out
, CYCompound::Precedence_
, CYNoFlags
);
560 void CYStatement::Output(CYOutput
&out
) const {
564 void CYTemplate::Output(CYOutput
&out
, CYFlags flags
) const {
568 void CYTypeArrayOf::Output(CYOutput
&out
, CYIdentifier
*identifier
) const {
569 next_
->Output(out
, Precedence(), identifier
);
575 void CYTypeBlockWith::Output(CYOutput
&out
, CYIdentifier
*identifier
) const {
577 next_
->Output(out
, Precedence(), identifier
);
578 out
<< ')' << '(' << parameters_
<< ')';
581 void CYTypeConstant::Output(CYOutput
&out
, CYIdentifier
*identifier
) const {
582 out
<< "const" << ' ';
583 next_
->Output(out
, Precedence(), identifier
);
586 void CYTypeFunctionWith::Output(CYOutput
&out
, CYIdentifier
*identifier
) const {
587 next_
->Output(out
, Precedence(), identifier
);
588 out
<< '(' << parameters_
<< ')';
591 void CYTypePointerTo::Output(CYOutput
&out
, CYIdentifier
*identifier
) const {
593 next_
->Output(out
, Precedence(), identifier
);
596 void CYTypeVolatile::Output(CYOutput
&out
, CYIdentifier
*identifier
) const {
598 next_
->Output(out
, Precedence(), identifier
);
601 void CYTypeModifier::Output(CYOutput
&out
, int precedence
, CYIdentifier
*identifier
) const {
607 bool protect(precedence
> Precedence());
611 Output(out
, identifier
);
616 void CYTypedIdentifier::Output(CYOutput
&out
) const {
617 specifier_
->Output(out
);
618 modifier_
->Output(out
, 0, identifier_
);
621 void CYEncodedType::Output(CYOutput
&out
, CYFlags flags
) const {
622 out
<< "@encode(" << typed_
<< ")";
625 void CYTypedParameter::Output(CYOutput
&out
) const {
628 out
<< ',' << ' ' << next_
;
631 void CYLambda::Output(CYOutput
&out
, CYFlags flags
) const {
632 // XXX: this is seriously wrong
639 void CYTypeDefinition::Output(CYOutput
&out
, CYFlags flags
) const {
640 out
<< "typedef" << ' ' << *typed_
;
643 void CYLet::Output(CYOutput
&out
, CYFlags flags
) const {
645 declarations_
->Output(out
, flags
); // XXX: flags
649 void CYModule::Output(CYOutput
&out
) const {
658 void New::Output(CYOutput
&out
, CYFlags flags
) const {
660 CYFlags
jacks(CYNoCall
| CYCenter(flags
));
661 constructor_
->Output(out
, Precedence(), jacks
);
662 if (arguments_
!= NULL
)
663 out
<< '(' << *arguments_
<< ')';
668 void CYNull::Output(CYOutput
&out
, CYFlags flags
) const {
672 void CYNumber::Output(CYOutput
&out
, CYFlags flags
) const {
673 std::ostringstream str
;
674 CYNumerify(str
, Value());
675 std::string
value(str
.str());
676 out
<< value
.c_str();
677 // XXX: this should probably also handle hex conversions and exponents
678 if ((flags
& CYNoInteger
) != 0 && value
.find('.') == std::string::npos
)
682 void CYNumber::PropertyName(CYOutput
&out
) const {
683 Output(out
, CYNoFlags
);
686 void CYObject::Output(CYOutput
&out
, CYFlags flags
) const {
687 bool protect((flags
& CYNoBrace
) != 0);
699 void CYPostfix::Output(CYOutput
&out
, CYFlags flags
) const {
700 lhs_
->Output(out
, Precedence(), CYLeft(flags
));
704 void CYPrefix::Output(CYOutput
&out
, CYFlags flags
) const {
705 const char *name(Operator());
709 rhs_
->Output(out
, Precedence(), CYRight(flags
));
712 void CYScript::Output(CYOutput
&out
) const {
716 void CYProperty::Output(CYOutput
&out
) const {
717 if (next_
!= NULL
|| out
.pretty_
)
719 out
<< '\n' << next_
;
722 void CYPropertyGetter::Output(CYOutput
&out
) const {
724 name_
->PropertyName(out
);
725 CYFunction::Output(out
);
726 CYProperty::Output(out
);
729 void CYPropertyMethod::Output(CYOutput
&out
) const {
730 name_
->PropertyName(out
);
731 CYFunction::Output(out
);
732 CYProperty::Output(out
);
735 void CYPropertySetter::Output(CYOutput
&out
) const {
737 name_
->PropertyName(out
);
738 CYFunction::Output(out
);
739 CYProperty::Output(out
);
742 void CYPropertyValue::Output(CYOutput
&out
) const {
744 name_
->PropertyName(out
);
746 value_
->Output(out
, CYAssign::Precedence_
, CYNoFlags
);
747 CYProperty::Output(out
);
750 void CYRegEx::Output(CYOutput
&out
, CYFlags flags
) const {
754 void CYReturn::Output(CYOutput
&out
, CYFlags flags
) const {
757 out
<< ' ' << *value_
;
761 void CYRubyBlock::Output(CYOutput
&out
, CYFlags flags
) const {
762 call_
->Output(out
, CYLeft(flags
));
764 proc_
->Output(out
, CYRight(flags
));
767 void CYRubyProc::Output(CYOutput
&out
, CYFlags flags
) const {
768 out
<< '{' << ' ' << '|' << parameters_
<< '|' << '\n';
775 void CYStatement::Multiple(CYOutput
&out
, CYFlags flags
) const {
777 CYForEach (next
, this) {
778 bool last(next
->next_
== NULL
);
779 CYFlags
jacks(first
? last
? flags
: CYLeft(flags
) : last
? CYRight(flags
) : CYCenter(flags
));
782 next
->Output(out
, jacks
);
787 void CYStatement::Single(CYOutput
&out
, CYFlags flags
, CYCompactType request
) const {
789 return out
.Terminate();
791 _assert(next_
== NULL
);
793 CYCompactType
compact(Compact());
795 if (compact
>= request
)
805 if (compact
< request
)
809 void CYString::Output(CYOutput
&out
, CYFlags flags
) const {
810 std::ostringstream str
;
811 CYStringify(str
, value_
, size_
);
812 out
<< str
.str().c_str();
815 void CYString::PropertyName(CYOutput
&out
) const {
816 if (const char *word
= Word())
822 static const char *Reserved_
[] = {
823 "false", "null", "true",
825 "break", "case", "catch", "continue", "default",
826 "delete", "do", "else", "finally", "for", "function",
827 "if", "in", "instanceof", "new", "return", "switch",
828 "this", "throw", "try", "typeof", "var", "void",
833 "class", "enum", "export", "extends", "import", "super",
835 "abstract", "boolean", "byte", "char", "double", "final",
836 "float", "goto", "int", "long", "native", "short",
837 "synchronized", "throws", "transient", "volatile",
844 const char *CYString::Word() const {
845 if (size_
== 0 || !WordStartRange_
[value_
[0]])
847 for (size_t i(1); i
!= size_
; ++i
)
848 if (!WordEndRange_
[value_
[i
]])
850 const char *value(Value());
851 for (const char **reserved(Reserved_
); *reserved
!= NULL
; ++reserved
)
852 if (strcmp(*reserved
, value
) == 0)
857 void CYSuperAccess::Output(CYOutput
&out
, CYFlags flags
) const {
859 if (const char *word
= property_
->Word())
862 out
<< '[' << *property_
<< ']';
865 void CYSuperCall::Output(CYOutput
&out
, CYFlags flags
) const {
866 out
<< "super" << '(' << arguments_
<< ')';
869 void CYSwitch::Output(CYOutput
&out
, CYFlags flags
) const {
870 out
<< "switch" << ' ' << '(' << *value_
<< ')' << ' ' << '{' << '\n';
877 void CYThis::Output(CYOutput
&out
, CYFlags flags
) const {
884 void Throw::Output(CYOutput
&out
, CYFlags flags
) const {
887 out
<< ' ' << *value_
;
891 void Try::Output(CYOutput
&out
, CYFlags flags
) const {
898 out
<< catch_
<< finally_
;
903 void CYTypeError::Output(CYOutput
&out
) const {
907 void CYTypeLong::Output(CYOutput
&out
) const {
908 out
<< "long" << specifier_
;
911 void CYTypeShort::Output(CYOutput
&out
) const {
912 out
<< "short" << specifier_
;
915 void CYTypeSigned::Output(CYOutput
&out
) const {
916 out
<< "signed" << specifier_
;
919 void CYTypeUnsigned::Output(CYOutput
&out
) const {
920 out
<< "unsigned" << specifier_
;
923 void CYTypeVariable::Output(CYOutput
&out
) const {
927 void CYTypeVoid::Output(CYOutput
&out
) const {
931 void CYVar::Output(CYOutput
&out
, CYFlags flags
) const {
933 declarations_
->Output(out
, flags
); // XXX: flags
937 void CYVariable::Output(CYOutput
&out
, CYFlags flags
) const {
941 void CYWhile::Output(CYOutput
&out
, CYFlags flags
) const {
942 out
<< "while" << ' ' << '(' << *test_
<< ')';
943 code_
->Single(out
, CYRight(flags
), CYCompactShort
);
946 void CYWith::Output(CYOutput
&out
, CYFlags flags
) const {
947 out
<< "with" << ' ' << '(' << *scope_
<< ')';
948 code_
->Single(out
, CYRight(flags
), CYCompactShort
);
951 void CYWord::Output(CYOutput
&out
) const {
953 if (out
.options_
.verbose_
) {
956 sprintf(number
, "%p", this);
961 void CYWord::PropertyName(CYOutput
&out
) const {
965 const char *CYWord::Word() const {