Output.cpp

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