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 CYLambda::Output(CYOutput &out, CYFlags flags) const {
 478     // XXX: this is seriously wrong
 479     out << "[](";
 480     out << ")->";
 481     out << "{";
 482     out << "}";
 483 }
 484
 485 void CYLetStatement::Output(CYOutput &out, CYFlags flags) const {
 486     out << "let" << ' ' << '(' << *declarations_ << ')';
 487     code_->Single(out, CYRight(flags));
 488 }
 489
 490 namespace cy {
 491 namespace Syntax {
 492
 493 void New::Output(CYOutput &out, CYFlags flags) const {
 494     out << "new" << ' ';
 495     CYFlags jacks(CYNoCall | CYCenter(flags));
 496     constructor_->Output(out, Precedence(), jacks);
 497     if (arguments_ != NULL)
 498         out << '(' << *arguments_ << ')';
 499 }
 500
 501 } }
 502
 503 void CYNull::Output(CYOutput &out, CYFlags flags) const {
 504     CYWord::Output(out);
 505 }
 506
 507 void CYNumber::Output(CYOutput &out, CYFlags flags) const {
 508     std::ostringstream str;
 509     CYNumerify(str, Value());
 510     std::string value(str.str());
 511     out << value.c_str();
 512     // XXX: this should probably also handle hex conversions and exponents
 513     if ((flags & CYNoInteger) != 0 && value.find('.') == std::string::npos)
 514         out << '.';
 515 }
 516
 517 void CYNumber::PropertyName(CYOutput &out) const {
 518     Output(out, CYNoFlags);
 519 }
 520
 521 void CYObject::Output(CYOutput &out, CYFlags flags) const {
 522     bool protect((flags & CYNoBrace) != 0);
 523     if (protect)
 524         out << '(';
 525     out << '{' << '\n';
 526     ++out.indent_;
 527     out << properties_;
 528     --out.indent_;
 529     out << '\t' << '}';
 530     if (protect)
 531         out << ')';
 532 }
 533
 534 void CYPostfix::Output(CYOutput &out, CYFlags flags) const {
 535     lhs_->Output(out, Precedence(), CYLeft(flags));
 536     out << Operator();
 537 }
 538
 539 void CYPrefix::Output(CYOutput &out, CYFlags flags) const {
 540     const char *name(Operator());
 541     out << name;
 542     if (Alphabetic())
 543         out << ' ';
 544     rhs_->Output(out, Precedence(), CYRight(flags));
 545 }
 546
 547 void CYProgram::Output(CYOutput &out) const {
 548     if (statements_ != NULL)
 549         statements_->Multiple(out);
 550 }
 551
 552 void CYProperty::Output(CYOutput &out) const {
 553     out << '\t';
 554     name_->PropertyName(out);
 555     out << ':' << ' ';
 556     value_->Output(out, CYAssign::Precedence_, CYNoFlags);
 557     if (next_ != NULL)
 558         out << ',' << '\n' << *next_;
 559     else
 560         out << '\n';
 561 }
 562
 563 void CYRegEx::Output(CYOutput &out, CYFlags flags) const {
 564     out << Value();
 565 }
 566
 567 void CYReturn::Output(CYOutput &out, CYFlags flags) const {
 568     out << "return";
 569     if (value_ != NULL)
 570         out << ' ' << *value_;
 571     out << ';';
 572 }
 573
 574 void CYRubyBlock::Output(CYOutput &out, CYFlags flags) const {
 575     call_->Output(out, CYLeft(flags));
 576     out << ' ';
 577     proc_->Output(out, CYRight(flags));
 578 }
 579
 580 void CYRubyProc::Output(CYOutput &out, CYFlags flags) const {
 581     // XXX: this is not outputting the parameters
 582     out << code_;
 583 }
 584
 585 void CYStatement::Multiple(CYOutput &out, CYFlags flags) const {
 586     bool first(true);
 587     CYForEach (next, this) {
 588         bool last(next->next_ == NULL);
 589         CYFlags jacks(first ? last ? flags : CYLeft(flags) : last ? CYRight(flags) : CYCenter(flags));
 590         first = false;
 591         out << '\t';
 592         next->Output(out, jacks);
 593         out << '\n';
 594     }
 595 }
 596
 597 void CYStatement::Single(CYOutput &out, CYFlags flags) const {
 598     if (this == NULL)
 599         return out.Terminate();
 600
 601     _assert(next_ == NULL);
 602     out << '\n';
 603     ++out.indent_;
 604     out << '\t';
 605     Output(out, flags);
 606     out << '\n';
 607     --out.indent_;
 608 }
 609
 610 void CYString::Output(CYOutput &out, CYFlags flags) const {
 611     std::ostringstream str;
 612     CYStringify(str, value_, size_);
 613     out << str.str().c_str();
 614 }
 615
 616 void CYString::PropertyName(CYOutput &out) const {
 617     if (const char *word = Word())
 618         out << word;
 619     else
 620         out << *this;
 621 }
 622
 623 static const char *Reserved_[] = {
 624     "false", "null", "true",
 625
 626     "break", "case", "catch", "continue", "default",
 627     "delete", "do", "else", "finally", "for", "function",
 628     "if", "in", "instanceof", "new", "return", "switch",
 629     "this", "throw", "try", "typeof", "var", "void",
 630     "while", "with",
 631
 632     "debugger", "const",
 633
 634     "class", "enum", "export", "extends", "import", "super",
 635
 636     "abstract", "boolean", "byte", "char", "double", "final",
 637     "float", "goto", "int", "long", "native", "short",
 638     "synchronized", "throws", "transient", "volatile",
 639
 640     "let", "yield",
 641
 642     NULL
 643 };
 644
 645 const char *CYString::Word() const {
 646     if (size_ == 0 || !WordStartRange_[value_[0]])
 647         return NULL;
 648     for (size_t i(1); i != size_; ++i)
 649         if (!WordEndRange_[value_[i]])
 650             return NULL;
 651     const char *value(Value());
 652     for (const char **reserved(Reserved_); *reserved != NULL; ++reserved)
 653         if (strcmp(*reserved, value) == 0)
 654             return NULL;
 655     return value;
 656 }
 657
 658 void CYSwitch::Output(CYOutput &out, CYFlags flags) const {
 659     out << "switch" << ' ' << '(' << *value_ << ')' << ' ' << '{';
 660     out << clauses_;
 661     out << '}';
 662 }
 663
 664 void CYThis::Output(CYOutput &out, CYFlags flags) const {
 665     CYWord::Output(out);
 666 }
 667
 668 namespace cy {
 669 namespace Syntax {
 670
 671 void Throw::Output(CYOutput &out, CYFlags flags) const {
 672     out << "throw";
 673     if (value_ != NULL)
 674         out << ' ' << *value_;
 675     out << ';';
 676 }
 677
 678 void Try::Output(CYOutput &out, CYFlags flags) const {
 679     out << "try" << ' ' << code_ << catch_ << finally_;
 680 }
 681
 682 } }
 683
 684 void CYVar::Output(CYOutput &out, CYFlags flags) const {
 685     out << "var";
 686     declarations_->Output(out, flags);
 687     out << ';';
 688 }
 689
 690 void CYVariable::Output(CYOutput &out, CYFlags flags) const {
 691     out << *name_;
 692 }
 693
 694 void CYWhile::Output(CYOutput &out, CYFlags flags) const {
 695     out << "while" << '(' << *test_ << ')';
 696     code_->Single(out, CYRight(flags));
 697 }
 698
 699 void CYWith::Output(CYOutput &out, CYFlags flags) const {
 700     out << "with" << '(' << *scope_ << ')';
 701     code_->Single(out, CYRight(flags));
 702 }
 703
 704 void CYWord::ClassName(CYOutput &out, bool object) const {
 705     if (object)
 706         out << "objc_getClass(";
 707     out << '"' << Word() << '"';
 708     if (object)
 709         out << ')';
 710 }
 711
 712 void CYWord::Output(CYOutput &out) const {
 713     out << Word();
 714     if (out.options_.verbose_)
 715         out.out_ << '@' << this;
 716 }
 717
 718 void CYWord::PropertyName(CYOutput &out) const {
 719     Output(out);
 720 }
 721
 722 const char *CYWord::Word() const {
 723     return word_;
 724 }