]> git.saurik.com Git - cycript.git/blob - Output.cpp
If CydgetMemoryParse fails, it should return NULL.
[cycript.git] / Output.cpp
1 /* Cycript - Optimizing JavaScript Compiler/Runtime
2 * Copyright (C) 2009-2013 Jay Freeman (saurik)
3 */
4
5 /* GNU General Public License, Version 3 {{{ */
6 /*
7 * Cycript is free software: you can redistribute it and/or modify
8 * it under the terms of the GNU General Public License as published
9 * by the Free Software Foundation, either version 3 of the License,
10 * or (at your option) any later version.
11 *
12 * Cycript is distributed in the hope that it will be useful, but
13 * WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 * GNU General Public License for more details.
16 *
17 * You should have received a copy of the GNU General Public License
18 * along with Cycript. If not, see <http://www.gnu.org/licenses/>.
19 **/
20 /* }}} */
21
22 #include "cycript.hpp"
23 #include "Parser.hpp"
24
25 #include <sstream>
26
27 void CYOutput::Terminate() {
28 out_ << ';';
29 mode_ = NoMode;
30 }
31
32 CYOutput &CYOutput::operator <<(char rhs) {
33 if (rhs == ' ' || rhs == '\n')
34 if (pretty_)
35 out_ << rhs;
36 else goto done;
37 else if (rhs == '\t')
38 if (pretty_)
39 for (unsigned i(0); i != indent_; ++i)
40 out_ << " ";
41 else goto done;
42 else if (rhs == '\r') {
43 if (right_) {
44 out_ << '\n';
45 right_ = false;
46 } goto done;
47 } else goto work;
48
49 right_ = true;
50 mode_ = NoMode;
51 goto done;
52
53 work:
54 if (mode_ == Terminated && rhs != '}') {
55 right_ = true;
56 out_ << ';';
57 }
58
59 if (rhs == ';') {
60 if (pretty_)
61 goto none;
62 else {
63 mode_ = Terminated;
64 goto done;
65 }
66 } else if (rhs == '+') {
67 if (mode_ == NoPlus)
68 out_ << ' ';
69 mode_ = NoPlus;
70 } else if (rhs == '-') {
71 if (mode_ == NoHyphen)
72 out_ << ' ';
73 mode_ = NoHyphen;
74 } else if (WordEndRange_[rhs]) {
75 if (mode_ == NoLetter)
76 out_ << ' ';
77 mode_ = NoLetter;
78 } else none:
79 mode_ = NoMode;
80
81 right_ = true;
82 out_ << rhs;
83 done:
84 return *this;
85 }
86
87 CYOutput &CYOutput::operator <<(const char *rhs) {
88 size_t size(strlen(rhs));
89
90 if (size == 1)
91 return *this << *rhs;
92
93 if (mode_ == Terminated)
94 out_ << ';';
95 else if (
96 mode_ == NoPlus && *rhs == '+' ||
97 mode_ == NoHyphen && *rhs == '-' ||
98 mode_ == NoLetter && WordEndRange_[*rhs]
99 )
100 out_ << ' ';
101
102 if (WordEndRange_[rhs[size - 1]])
103 mode_ = NoLetter;
104 else
105 mode_ = NoMode;
106
107 right_ = true;
108 out_ << rhs;
109 return *this;
110 }
111
112 void CYArgument::Output(CYOutput &out) const {
113 if (name_ != NULL) {
114 out << *name_;
115 if (value_ != NULL)
116 out << ':' << ' ';
117 }
118 if (value_ != NULL)
119 value_->Output(out, CYAssign::Precedence_, CYNoFlags);
120 if (next_ != NULL) {
121 if (next_->name_ == NULL)
122 out << ',';
123 out << ' ' << *next_;
124 }
125 }
126
127 void CYArray::Output(CYOutput &out, CYFlags flags) const {
128 out << '[' << elements_ << ']';
129 }
130
131 void CYArrayComprehension::Output(CYOutput &out, CYFlags flags) const {
132 out << '[' << *expression_ << ' ' << *comprehensions_ << ']';
133 }
134
135 void CYAssignment::Output(CYOutput &out, CYFlags flags) const {
136 lhs_->Output(out, Precedence() - 1, CYLeft(flags) | CYNoRightHand);
137 out << ' ' << Operator() << ' ';
138 rhs_->Output(out, Precedence(), CYRight(flags));
139 }
140
141 void CYBlock::Output(CYOutput &out) const {
142 out << '{' << '\n';
143 ++out.indent_;
144 if (statements_ != NULL)
145 statements_->Multiple(out);
146 --out.indent_;
147 out << '\t' << '}';
148 }
149
150 void CYBlock::Output(CYOutput &out, CYFlags flags) const {
151 if (statements_ == NULL)
152 out.Terminate();
153 else if (statements_->next_ == NULL)
154 statements_->Single(out, flags);
155 else
156 Output(out);
157 }
158
159 void CYBoolean::Output(CYOutput &out, CYFlags flags) const {
160 out << (Value() ? "true" : "false");
161 }
162
163 void CYBreak::Output(CYOutput &out, CYFlags flags) const {
164 out << "break";
165 if (label_ != NULL)
166 out << ' ' << *label_;
167 out << ';';
168 }
169
170 void CYCall::Output(CYOutput &out, CYFlags flags) const {
171 bool protect((flags & CYNoCall) != 0);
172 if (protect)
173 out << '(';
174 function_->Output(out, Precedence(), protect ? CYNoFlags : flags);
175 out << '(' << arguments_ << ')';
176 if (protect)
177 out << ')';
178 }
179
180 namespace cy {
181 namespace Syntax {
182
183 void Catch::Output(CYOutput &out) const {
184 out << ' ' << "catch" << ' ' << '(' << *name_ << ')' << ' ' << code_;
185 }
186
187 } }
188
189 void CYComment::Output(CYOutput &out, CYFlags flags) const {
190 out << '\r';
191 out.out_ << value_;
192 out.right_ = true;
193 out << '\r';
194 }
195
196 void CYCompound::Output(CYOutput &out, CYFlags flags) const {
197 if (CYExpression *expression = expressions_)
198 if (CYExpression *next = expression->next_) {
199 expression->Output(out, CYLeft(flags));
200 CYFlags center(CYCenter(flags));
201 while (next != NULL) {
202 expression = next;
203 out << ',' << ' ';
204 next = expression->next_;
205 CYFlags right(next != NULL ? center : CYRight(flags));
206 expression->Output(out, right);
207 }
208 } else
209 expression->Output(out, flags);
210 }
211
212 void CYCondition::Output(CYOutput &out, CYFlags flags) const {
213 test_->Output(out, Precedence() - 1, CYLeft(flags));
214 out << ' ' << '?' << ' ';
215 if (true_ != NULL)
216 true_->Output(out, CYAssign::Precedence_, CYNoFlags);
217 out << ' ' << ':' << ' ';
218 false_->Output(out, CYAssign::Precedence_, CYRight(flags));
219 }
220
221 void CYContinue::Output(CYOutput &out, CYFlags flags) const {
222 out << "continue";
223 if (label_ != NULL)
224 out << ' ' << *label_;
225 out << ';';
226 }
227
228 void CYClause::Output(CYOutput &out) const {
229 if (case_ != NULL)
230 out << "case" << ' ' << *case_;
231 else
232 out << "default";
233 out << ':' << '\n';
234 if (statements_ != NULL)
235 statements_->Multiple(out);
236 out << next_;
237 }
238
239 void CYDebugger::Output(CYOutput &out, CYFlags flags) const {
240 out << "debugger" << ';';
241 }
242
243 void CYDeclaration::ForIn(CYOutput &out, CYFlags flags) const {
244 out << "var";
245 Output(out, CYRight(flags));
246 }
247
248 void CYDeclaration::Output(CYOutput &out, CYFlags flags) const {
249 out << *identifier_;
250 //out.out_ << ':' << identifier_->usage_ << '#' << identifier_->offset_;
251 if (initialiser_ != NULL) {
252 out << ' ' << '=' << ' ';
253 initialiser_->Output(out, CYAssign::Precedence_, CYRight(flags));
254 }
255 }
256
257 void CYForDeclarations::Output(CYOutput &out, CYFlags flags) const {
258 out << "var";
259 Output(out, CYRight(flags));
260 }
261
262 void CYDeclarations::Output(CYOutput &out) const {
263 Output(out, CYNoFlags);
264 }
265
266 void CYDeclarations::Output(CYOutput &out, CYFlags flags) const {
267 const CYDeclarations *declaration(this);
268 bool first(true);
269
270 for (;;) {
271 CYDeclarations *next(declaration->next_);
272
273 CYFlags jacks(first ? CYLeft(flags) : next == NULL ? CYRight(flags) : CYCenter(flags));
274 first = false;
275 declaration->declaration_->Output(out, jacks);
276
277 if (next == NULL)
278 break;
279
280 out << ',' << ' ';
281 declaration = next;
282 }
283 }
284
285 void CYDirectMember::Output(CYOutput &out, CYFlags flags) const {
286 object_->Output(out, Precedence(), CYLeft(flags) | CYNoInteger);
287 if (const char *word = property_->Word())
288 out << '.' << word;
289 else
290 out << '[' << *property_ << ']';
291 }
292
293 void CYDoWhile::Output(CYOutput &out, CYFlags flags) const {
294 out << "do";
295 code_->Single(out, CYCenter(flags));
296 out << "while" << ' ' << '(' << *test_ << ')';
297 }
298
299 void CYElement::Output(CYOutput &out) const {
300 if (value_ != NULL)
301 value_->Output(out, CYAssign::Precedence_, CYNoFlags);
302 if (next_ != NULL || value_ == NULL) {
303 out << ',';
304 if (next_ != NULL && next_->value_ != NULL)
305 out << ' ';
306 }
307 if (next_ != NULL)
308 next_->Output(out);
309 }
310
311 void CYEmpty::Output(CYOutput &out, CYFlags flags) const {
312 out.Terminate();
313 }
314
315 void CYExpress::Output(CYOutput &out, CYFlags flags) const {
316 expression_->Output(out, flags | CYNoBF);
317 out << ';';
318 }
319
320 void CYExpression::ClassName(CYOutput &out, bool object) const {
321 Output(out, CYAssign::Precedence_, CYNoFlags);
322 }
323
324 void CYExpression::ForIn(CYOutput &out, CYFlags flags) const {
325 Output(out, flags | CYNoRightHand);
326 }
327
328 void CYExpression::Output(CYOutput &out) const {
329 Output(out, CYNoFlags);
330 }
331
332 void CYExpression::Output(CYOutput &out, unsigned precedence, CYFlags flags) const {
333 if (precedence < Precedence() || (flags & CYNoRightHand) != 0 && RightHand())
334 out << '(' << *this << ')';
335 else
336 Output(out, flags);
337 }
338
339 void CYFatArrow::Output(CYOutput &out, CYFlags flags) const {
340 out << '(' << parameters_ << ')' << ' ' << "=>" << ' ' << code_;
341 }
342
343 void CYFinally::Output(CYOutput &out) const {
344 out << ' ' << "finally" << ' ' << code_;
345 }
346
347 void CYFor::Output(CYOutput &out, CYFlags flags) const {
348 out << "for" << ' ' << '(';
349 if (initialiser_ != NULL)
350 initialiser_->Output(out, CYNoIn);
351 out.Terminate();
352 if (test_ != NULL)
353 out << ' ';
354 out << test_;
355 out.Terminate();
356 if (increment_ != NULL)
357 out << ' ';
358 out << increment_;
359 out << ')';
360 code_->Single(out, CYRight(flags));
361 }
362
363 void CYForOf::Output(CYOutput &out, CYFlags flags) const {
364 out << "for" << ' ' << "each" << ' ' << '(';
365 initialiser_->ForIn(out, CYNoIn);
366 out << "in" << *set_ << ')';
367 code_->Single(out, CYRight(flags));
368 }
369
370 void CYForOfComprehension::Output(CYOutput &out) const {
371 out << "for" << ' ' << "each" << ' ' << '(' << *name_ << ' ' << "in" << ' ' << *set_ << ')' << next_;
372 }
373
374 void CYForIn::Output(CYOutput &out, CYFlags flags) const {
375 out << "for" << ' ' << '(';
376 if (initialiser_ != NULL)
377 initialiser_->ForIn(out, CYNoIn);
378 out << "in" << *set_ << ')';
379 code_->Single(out, CYRight(flags));
380 }
381
382 void CYForInComprehension::Output(CYOutput &out) const {
383 out << "for" << ' ' << '(' << *name_ << ' ' << "in" << ' ' << *set_ << ')';
384 }
385
386 void CYFunction::Output(CYOutput &out, CYFlags flags) const {
387 // XXX: one could imagine using + here to save a byte
388 bool protect((flags & CYNoFunction) != 0);
389 if (protect)
390 out << '(';
391 out << "function";
392 if (name_ != NULL)
393 out << ' ' << *name_;
394 out << '(' << parameters_ << ')';
395 out << ' ' << code_;
396 if (protect)
397 out << ')';
398 }
399
400 void CYFunctionExpression::Output(CYOutput &out, CYFlags flags) const {
401 CYFunction::Output(out, flags);
402 }
403
404 void CYFunctionStatement::Output(CYOutput &out, CYFlags flags) const {
405 CYFunction::Output(out, flags);
406 }
407
408 void CYFunctionParameter::Output(CYOutput &out) const {
409 initialiser_->Output(out, CYNoFlags);
410 if (next_ != NULL)
411 out << ',' << ' ' << *next_;
412 }
413
414 const char *CYIdentifier::Word() const {
415 return replace_ == NULL || replace_ == this ? CYWord::Word() : replace_->Word();
416 }
417
418 void CYIf::Output(CYOutput &out, CYFlags flags) const {
419 bool protect(false);
420 if (false_ == NULL && (flags & CYNoDangle) != 0) {
421 protect = true;
422 out << '{';
423 }
424
425 out << "if" << ' ' << '(' << *test_ << ')';
426
427 CYFlags right(protect ? CYNoFlags : CYRight(flags));
428
429 CYFlags jacks(CYNoDangle);
430 if (false_ == NULL)
431 jacks |= right;
432 else
433 jacks |= protect ? CYNoFlags : CYCenter(flags);
434
435 true_->Single(out, jacks);
436
437 if (false_ != NULL) {
438 out << '\t' << "else";
439 false_->Single(out, right);
440 }
441
442 if (protect)
443 out << '}';
444 }
445
446 void CYIfComprehension::Output(CYOutput &out) const {
447 out << "if" << ' ' << '(' << *test_ << ')' << next_;
448 }
449
450 void CYIndirectMember::Output(CYOutput &out, CYFlags flags) const {
451 object_->Output(out, Precedence(), CYLeft(flags));
452 if (const char *word = property_->Word())
453 out << "->" << word;
454 else
455 out << "->" << '[' << *property_ << ']';
456 }
457
458 void CYInfix::Output(CYOutput &out, CYFlags flags) const {
459 const char *name(Operator());
460 bool protect((flags & CYNoIn) != 0 && strcmp(name, "in") == 0);
461 if (protect)
462 out << '(';
463 CYFlags left(protect ? CYNoFlags : CYLeft(flags));
464 lhs_->Output(out, Precedence(), left);
465 out << ' ' << name << ' ';
466 CYFlags right(protect ? CYNoFlags : CYRight(flags));
467 rhs_->Output(out, Precedence() - 1, right);
468 if (protect)
469 out << ')';
470 }
471
472 void CYLabel::Output(CYOutput &out, CYFlags flags) const {
473 out << *name_ << ':' << ' ';
474 statement_->Single(out, CYRight(flags));
475 }
476
477 void CYTypedIdentifier::Output(CYOutput &out) const {
478 // XXX: this is clearly wrong
479 out << "XXX";
480 }
481
482 void CYLambda::Output(CYOutput &out, CYFlags flags) const {
483 // XXX: this is seriously wrong
484 out << "[](";
485 out << ")->";
486 out << "{";
487 out << "}";
488 }
489
490 void CYTypeDefinition::Output(CYOutput &out, CYFlags flags) const {
491 out << "typedef" << *typed_;
492 }
493
494 void CYLetStatement::Output(CYOutput &out, CYFlags flags) const {
495 out << "let" << ' ' << '(' << *declarations_ << ')';
496 code_->Single(out, CYRight(flags));
497 }
498
499 namespace cy {
500 namespace Syntax {
501
502 void New::Output(CYOutput &out, CYFlags flags) const {
503 out << "new" << ' ';
504 CYFlags jacks(CYNoCall | CYCenter(flags));
505 constructor_->Output(out, Precedence(), jacks);
506 if (arguments_ != NULL)
507 out << '(' << *arguments_ << ')';
508 }
509
510 } }
511
512 void CYNull::Output(CYOutput &out, CYFlags flags) const {
513 CYWord::Output(out);
514 }
515
516 void CYNumber::Output(CYOutput &out, CYFlags flags) const {
517 std::ostringstream str;
518 CYNumerify(str, Value());
519 std::string value(str.str());
520 out << value.c_str();
521 // XXX: this should probably also handle hex conversions and exponents
522 if ((flags & CYNoInteger) != 0 && value.find('.') == std::string::npos)
523 out << '.';
524 }
525
526 void CYNumber::PropertyName(CYOutput &out) const {
527 Output(out, CYNoFlags);
528 }
529
530 void CYObject::Output(CYOutput &out, CYFlags flags) const {
531 bool protect((flags & CYNoBrace) != 0);
532 if (protect)
533 out << '(';
534 out << '{' << '\n';
535 ++out.indent_;
536 out << properties_;
537 --out.indent_;
538 out << '\t' << '}';
539 if (protect)
540 out << ')';
541 }
542
543 void CYPostfix::Output(CYOutput &out, CYFlags flags) const {
544 lhs_->Output(out, Precedence(), CYLeft(flags));
545 out << Operator();
546 }
547
548 void CYPrefix::Output(CYOutput &out, CYFlags flags) const {
549 const char *name(Operator());
550 out << name;
551 if (Alphabetic())
552 out << ' ';
553 rhs_->Output(out, Precedence(), CYRight(flags));
554 }
555
556 void CYProgram::Output(CYOutput &out) const {
557 if (statements_ != NULL)
558 statements_->Multiple(out);
559 }
560
561 void CYProperty::Output(CYOutput &out) const {
562 out << '\t';
563 name_->PropertyName(out);
564 out << ':' << ' ';
565 value_->Output(out, CYAssign::Precedence_, CYNoFlags);
566 if (next_ != NULL)
567 out << ',' << '\n' << *next_;
568 else
569 out << '\n';
570 }
571
572 void CYRegEx::Output(CYOutput &out, CYFlags flags) const {
573 out << Value();
574 }
575
576 void CYReturn::Output(CYOutput &out, CYFlags flags) const {
577 out << "return";
578 if (value_ != NULL)
579 out << ' ' << *value_;
580 out << ';';
581 }
582
583 void CYRubyBlock::Output(CYOutput &out, CYFlags flags) const {
584 call_->Output(out, CYLeft(flags));
585 out << ' ';
586 proc_->Output(out, CYRight(flags));
587 }
588
589 void CYRubyProc::Output(CYOutput &out, CYFlags flags) const {
590 // XXX: this is not outputting the parameters
591 out << code_;
592 }
593
594 void CYStatement::Multiple(CYOutput &out, CYFlags flags) const {
595 bool first(true);
596 CYForEach (next, this) {
597 bool last(next->next_ == NULL);
598 CYFlags jacks(first ? last ? flags : CYLeft(flags) : last ? CYRight(flags) : CYCenter(flags));
599 first = false;
600 out << '\t';
601 next->Output(out, jacks);
602 out << '\n';
603 }
604 }
605
606 void CYStatement::Single(CYOutput &out, CYFlags flags) const {
607 if (this == NULL)
608 return out.Terminate();
609
610 _assert(next_ == NULL);
611 out << '\n';
612 ++out.indent_;
613 out << '\t';
614 Output(out, flags);
615 out << '\n';
616 --out.indent_;
617 }
618
619 void CYString::Output(CYOutput &out, CYFlags flags) const {
620 std::ostringstream str;
621 CYStringify(str, value_, size_);
622 out << str.str().c_str();
623 }
624
625 void CYString::PropertyName(CYOutput &out) const {
626 if (const char *word = Word())
627 out << word;
628 else
629 out << *this;
630 }
631
632 static const char *Reserved_[] = {
633 "false", "null", "true",
634
635 "break", "case", "catch", "continue", "default",
636 "delete", "do", "else", "finally", "for", "function",
637 "if", "in", "instanceof", "new", "return", "switch",
638 "this", "throw", "try", "typeof", "var", "void",
639 "while", "with",
640
641 "debugger", "const",
642
643 "class", "enum", "export", "extends", "import", "super",
644
645 "abstract", "boolean", "byte", "char", "double", "final",
646 "float", "goto", "int", "long", "native", "short",
647 "synchronized", "throws", "transient", "volatile",
648
649 "let", "yield",
650
651 NULL
652 };
653
654 const char *CYString::Word() const {
655 if (size_ == 0 || !WordStartRange_[value_[0]])
656 return NULL;
657 for (size_t i(1); i != size_; ++i)
658 if (!WordEndRange_[value_[i]])
659 return NULL;
660 const char *value(Value());
661 for (const char **reserved(Reserved_); *reserved != NULL; ++reserved)
662 if (strcmp(*reserved, value) == 0)
663 return NULL;
664 return value;
665 }
666
667 void CYSwitch::Output(CYOutput &out, CYFlags flags) const {
668 out << "switch" << ' ' << '(' << *value_ << ')' << ' ' << '{';
669 out << clauses_;
670 out << '}';
671 }
672
673 void CYThis::Output(CYOutput &out, CYFlags flags) const {
674 CYWord::Output(out);
675 }
676
677 namespace cy {
678 namespace Syntax {
679
680 void Throw::Output(CYOutput &out, CYFlags flags) const {
681 out << "throw";
682 if (value_ != NULL)
683 out << ' ' << *value_;
684 out << ';';
685 }
686
687 void Try::Output(CYOutput &out, CYFlags flags) const {
688 out << "try" << ' ' << code_ << catch_ << finally_;
689 }
690
691 } }
692
693 void CYVar::Output(CYOutput &out, CYFlags flags) const {
694 out << "var";
695 declarations_->Output(out, flags);
696 out << ';';
697 }
698
699 void CYVariable::Output(CYOutput &out, CYFlags flags) const {
700 out << *name_;
701 }
702
703 void CYWhile::Output(CYOutput &out, CYFlags flags) const {
704 out << "while" << '(' << *test_ << ')';
705 code_->Single(out, CYRight(flags));
706 }
707
708 void CYWith::Output(CYOutput &out, CYFlags flags) const {
709 out << "with" << '(' << *scope_ << ')';
710 code_->Single(out, CYRight(flags));
711 }
712
713 void CYWord::ClassName(CYOutput &out, bool object) const {
714 if (object)
715 out << "objc_getClass(";
716 out << '"' << Word() << '"';
717 if (object)
718 out << ')';
719 }
720
721 void CYWord::Output(CYOutput &out) const {
722 out << Word();
723 if (out.options_.verbose_)
724 out.out_ << '@' << this;
725 }
726
727 void CYWord::PropertyName(CYOutput &out) const {
728 Output(out);
729 }
730
731 const char *CYWord::Word() const {
732 return word_;
733 }