Output.cpp

   1 /* Cycript - Optimizing JavaScript Compiler/Runtime
   2  * Copyright (C) 2009-2010  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 CYDeclaration::ForIn(CYOutput &out, CYFlags flags) const {
 240     out << "var";
 241     Output(out, CYRight(flags));
 242 }
 243
 244 void CYDeclaration::Output(CYOutput &out, CYFlags flags) const {
 245     out << *identifier_;
 246     //out.out_ << ':' << identifier_->usage_ << '#' << identifier_->offset_;
 247     if (initialiser_ != NULL) {
 248         out << ' ' << '=' << ' ';
 249         initialiser_->Output(out, CYAssign::Precedence_, CYRight(flags));
 250     }
 251 }
 252
 253 void CYDeclarations::For(CYOutput &out) const {
 254     out << "var";
 255     Output(out, CYNoIn);
 256 }
 257
 258 void CYDeclarations::Output(CYOutput &out) const {
 259     Output(out, CYNoFlags);
 260 }
 261
 262 void CYDeclarations::Output(CYOutput &out, CYFlags flags) const {
 263     const CYDeclarations *declaration(this);
 264     bool first(true);
 265   output:
 266     CYDeclarations *next(declaration->next_);
 267     CYFlags jacks(first ? CYLeft(flags) : next == NULL ? CYRight(flags) : CYCenter(flags));
 268     first = false;
 269     declaration->declaration_->Output(out, jacks);
 270
 271     if (next != NULL) {
 272         out << ',' << ' ';
 273         declaration = next;
 274         goto output;
 275     }
 276 }
 277
 278 void CYDirectMember::Output(CYOutput &out, CYFlags flags) const {
 279     object_->Output(out, Precedence(), CYLeft(flags) | CYNoInteger);
 280     if (const char *word = property_->Word())
 281         out << '.' << word;
 282     else
 283         out << '[' << *property_ << ']';
 284 }
 285
 286 void CYDoWhile::Output(CYOutput &out, CYFlags flags) const {
 287     out << "do";
 288     code_->Single(out, CYCenter(flags));
 289     out << "while" << ' ' << '(' << *test_ << ')';
 290 }
 291
 292 void CYElement::Output(CYOutput &out) const {
 293     if (value_ != NULL)
 294         value_->Output(out, CYAssign::Precedence_, CYNoFlags);
 295     if (next_ != NULL || value_ == NULL) {
 296         out << ',';
 297         if (next_ != NULL && next_->value_ != NULL)
 298             out << ' ';
 299     }
 300     if (next_ != NULL)
 301         next_->Output(out);
 302 }
 303
 304 void CYEmpty::Output(CYOutput &out, CYFlags flags) const {
 305     out.Terminate();
 306 }
 307
 308 void CYExpress::Output(CYOutput &out, CYFlags flags) const {
 309     expression_->Output(out, flags | CYNoBF);
 310     out << ';';
 311 }
 312
 313 void CYExpression::ClassName(CYOutput &out, bool object) const {
 314     Output(out, CYAssign::Precedence_, CYNoFlags);
 315 }
 316
 317 void CYExpression::For(CYOutput &out) const {
 318     Output(out, CYNoIn);
 319 }
 320
 321 void CYExpression::ForIn(CYOutput &out, CYFlags flags) const {
 322     Output(out, flags | CYNoRightHand);
 323 }
 324
 325 void CYExpression::Output(CYOutput &out) const {
 326     Output(out, CYNoFlags);
 327 }
 328
 329 void CYExpression::Output(CYOutput &out, unsigned precedence, CYFlags flags) const {
 330     if (precedence < Precedence() || (flags & CYNoRightHand) != 0 && RightHand())
 331         out << '(' << *this << ')';
 332     else
 333         Output(out, flags);
 334 }
 335
 336 void CYFinally::Output(CYOutput &out) const {
 337     out << ' ' << "finally" << ' ' << code_;
 338 }
 339
 340 void CYFor::Output(CYOutput &out, CYFlags flags) const {
 341     out << "for" << ' ' << '(';
 342     if (initialiser_ != NULL)
 343         initialiser_->For(out);
 344     out.Terminate();
 345     if (test_ != NULL)
 346         out << ' ';
 347     out << test_;
 348     out.Terminate();
 349     if (increment_ != NULL)
 350         out << ' ';
 351     out << increment_;
 352     out << ')';
 353     code_->Single(out, CYRight(flags));
 354 }
 355
 356 void CYForEachIn::Output(CYOutput &out, CYFlags flags) const {
 357     out << "for" << ' ' << "each" << ' ' << '(';
 358     initialiser_->ForIn(out, CYNoIn);
 359     out << "in" << *set_ << ')';
 360     code_->Single(out, CYRight(flags));
 361 }
 362
 363 void CYForEachInComprehension::Output(CYOutput &out) const {
 364     out << "for" << ' ' << "each" << ' ' << '(' << *name_ << ' ' << "in" << ' ' << *set_ << ')' << next_;
 365 }
 366
 367 void CYForIn::Output(CYOutput &out, CYFlags flags) const {
 368     out << "for" << ' ' << '(';
 369     if (initialiser_ != NULL)
 370         initialiser_->ForIn(out, CYNoIn);
 371     out << "in" << *set_ << ')';
 372     code_->Single(out, CYRight(flags));
 373 }
 374
 375 void CYForInComprehension::Output(CYOutput &out) const {
 376     out << "for" << ' ' << '(' << *name_ << ' ' << "in" << ' ' << *set_ << ')';
 377 }
 378
 379 void CYFunction::Output(CYOutput &out, CYFlags flags) const {
 380     // XXX: one could imagine using + here to save a byte
 381     bool protect((flags & CYNoFunction) != 0);
 382     if (protect)
 383         out << '(';
 384     out << "function";
 385     if (name_ != NULL)
 386         out << ' ' << *name_;
 387     out << '(' << parameters_ << ')';
 388     out << ' ' << code_;
 389     if (protect)
 390         out << ')';
 391 }
 392
 393 void CYFunctionExpression::Output(CYOutput &out, CYFlags flags) const {
 394     CYFunction::Output(out, flags);
 395 }
 396
 397 void CYFunctionStatement::Output(CYOutput &out, CYFlags flags) const {
 398     CYFunction::Output(out, flags);
 399 }
 400
 401 void CYFunctionParameter::Output(CYOutput &out) const {
 402     out << *name_;
 403     if (next_ != NULL)
 404         out << ',' << ' ' << *next_;
 405 }
 406
 407 const char *CYIdentifier::Word() const {
 408     return replace_ == NULL || replace_ == this ? CYWord::Word() : replace_->Word();
 409 }
 410
 411 void CYIf::Output(CYOutput &out, CYFlags flags) const {
 412     bool protect(false);
 413     if (false_ == NULL && (flags & CYNoDangle) != 0) {
 414         protect = true;
 415         out << '{';
 416     }
 417
 418     out << "if" << ' ' << '(' << *test_ << ')';
 419
 420     CYFlags right(protect ? CYNoFlags : CYRight(flags));
 421
 422     CYFlags jacks(CYNoDangle);
 423     if (false_ == NULL)
 424         jacks |= right;
 425     else
 426         jacks |= protect ? CYNoFlags : CYCenter(flags);
 427
 428     true_->Single(out, jacks);
 429
 430     if (false_ != NULL) {
 431         out << '\t' << "else";
 432         false_->Single(out, right);
 433     }
 434
 435     if (protect)
 436         out << '}';
 437 }
 438
 439 void CYIfComprehension::Output(CYOutput &out) const {
 440     out << "if" << ' ' << '(' << *test_ << ')' << next_;
 441 }
 442
 443 void CYIndirectMember::Output(CYOutput &out, CYFlags flags) const {
 444     object_->Output(out, Precedence(), CYLeft(flags));
 445     if (const char *word = property_->Word())
 446         out << "->" << word;
 447     else
 448         out << "->" << '[' << *property_ << ']';
 449 }
 450
 451 void CYInfix::Output(CYOutput &out, CYFlags flags) const {
 452     const char *name(Operator());
 453     bool protect((flags & CYNoIn) != 0 && strcmp(name, "in") == 0);
 454     if (protect)
 455         out << '(';
 456     CYFlags left(protect ? CYNoFlags : CYLeft(flags));
 457     lhs_->Output(out, Precedence(), left);
 458     out << ' ' << name << ' ';
 459     CYFlags right(protect ? CYNoFlags : CYRight(flags));
 460     rhs_->Output(out, Precedence() - 1, right);
 461     if (protect)
 462         out << ')';
 463 }
 464
 465 void CYLabel::Output(CYOutput &out, CYFlags flags) const {
 466     out << *name_ << ':' << ' ';
 467     statement_->Single(out, CYRight(flags));
 468 }
 469
 470 void CYLet::Output(CYOutput &out, CYFlags flags) const {
 471     out << "let" << ' ' << '(' << *declarations_ << ')' << ' ' << code_;
 472 }
 473
 474 namespace cy {
 475 namespace Syntax {
 476
 477 void New::Output(CYOutput &out, CYFlags flags) const {
 478     out << "new" << ' ';
 479     CYFlags jacks(CYNoCall | CYCenter(flags));
 480     constructor_->Output(out, Precedence(), jacks);
 481     if (arguments_ != NULL)
 482         out << '(' << *arguments_ << ')';
 483 }
 484
 485 } }
 486
 487 void CYNull::Output(CYOutput &out, CYFlags flags) const {
 488     CYWord::Output(out);
 489 }
 490
 491 void CYNumber::Output(CYOutput &out, CYFlags flags) const {
 492     std::ostringstream str;
 493     CYNumerify(str, Value());
 494     std::string value(str.str());
 495     out << value.c_str();
 496     // XXX: this should probably also handle hex conversions and exponents
 497     if ((flags & CYNoInteger) != 0 && value.find('.') == std::string::npos)
 498         out << '.';
 499 }
 500
 501 void CYNumber::PropertyName(CYOutput &out) const {
 502     Output(out, CYNoFlags);
 503 }
 504
 505 void CYObject::Output(CYOutput &out, CYFlags flags) const {
 506     bool protect((flags & CYNoBrace) != 0);
 507     if (protect)
 508         out << '(';
 509     out << '{' << '\n';
 510     ++out.indent_;
 511     out << properties_;
 512     --out.indent_;
 513     out << '\t' << '}';
 514     if (protect)
 515         out << ')';
 516 }
 517
 518 void CYOptionalFunctionParameter::Output(CYOutput &out) const {
 519     out << *name_ << '=';
 520     initializer_->Output(out, CYAssign::Precedence_, CYNoFlags);
 521     if (next_ != NULL)
 522         out << ',' << ' ' << *next_;
 523 }
 524
 525 void CYPostfix::Output(CYOutput &out, CYFlags flags) const {
 526     lhs_->Output(out, Precedence(), CYLeft(flags));
 527     out << Operator();
 528 }
 529
 530 void CYPrefix::Output(CYOutput &out, CYFlags flags) const {
 531     const char *name(Operator());
 532     out << name;
 533     if (Alphabetic())
 534         out << ' ';
 535     rhs_->Output(out, Precedence(), CYRight(flags));
 536 }
 537
 538 void CYProgram::Output(CYOutput &out) const {
 539     if (statements_ != NULL)
 540         statements_->Multiple(out);
 541 }
 542
 543 void CYProperty::Output(CYOutput &out) const {
 544     out << '\t';
 545     name_->PropertyName(out);
 546     out << ':' << ' ';
 547     value_->Output(out, CYAssign::Precedence_, CYNoFlags);
 548     if (next_ != NULL)
 549         out << ',' << '\n' << *next_;
 550     else
 551         out << '\n';
 552 }
 553
 554 void CYRegEx::Output(CYOutput &out, CYFlags flags) const {
 555     out << Value();
 556 }
 557
 558 void CYReturn::Output(CYOutput &out, CYFlags flags) const {
 559     out << "return";
 560     if (value_ != NULL)
 561         out << ' ' << *value_;
 562     out << ';';
 563 }
 564
 565 void CYRubyBlock::Output(CYOutput &out, CYFlags flags) const {
 566     call_->Output(out, CYLeft(flags));
 567     out << ' ';
 568     proc_->Output(out, CYRight(flags));
 569 }
 570
 571 void CYRubyProc::Output(CYOutput &out, CYFlags flags) const {
 572     // XXX: this is not outputting the parameters
 573     out << code_;
 574 }
 575
 576 void CYStatement::Multiple(CYOutput &out, CYFlags flags) const {
 577     bool first(true);
 578     CYForEach (next, this) {
 579         bool last(next->next_ == NULL);
 580         CYFlags jacks(first ? last ? flags : CYLeft(flags) : last ? CYRight(flags) : CYCenter(flags));
 581         first = false;
 582         out << '\t';
 583         next->Output(out, jacks);
 584         out << '\n';
 585     }
 586 }
 587
 588 void CYStatement::Single(CYOutput &out, CYFlags flags) const {
 589     _assert(next_ == NULL);
 590     out << '\n';
 591     ++out.indent_;
 592     out << '\t';
 593     Output(out, flags);
 594     out << '\n';
 595     --out.indent_;
 596 }
 597
 598 void CYString::Output(CYOutput &out, CYFlags flags) const {
 599     std::ostringstream str;
 600     CYStringify(str, value_, size_);
 601     out << str.str().c_str();
 602 }
 603
 604 void CYString::PropertyName(CYOutput &out) const {
 605     if (const char *word = Word())
 606         out << word;
 607     else
 608         out << *this;
 609 }
 610
 611 static const char *Reserved_[] = {
 612     "false", "null", "true",
 613
 614     "break", "case", "catch", "continue", "default",
 615     "delete", "do", "else", "finally", "for", "function",
 616     "if", "in", "instanceof", "new", "return", "switch",
 617     "this", "throw", "try", "typeof", "var", "void",
 618     "while", "with",
 619
 620     "debugger", "const",
 621
 622     "class", "enum", "export", "extends", "import", "super",
 623
 624     "abstract", "boolean", "byte", "char", "double", "final",
 625     "float", "goto", "int", "long", "native", "short",
 626     "synchronized", "throws", "transient", "volatile",
 627
 628     "let", "yield",
 629
 630     NULL
 631 };
 632
 633 const char *CYString::Word() const {
 634     if (size_ == 0 || !WordStartRange_[value_[0]])
 635         return NULL;
 636     for (size_t i(1); i != size_; ++i)
 637         if (!WordEndRange_[value_[i]])
 638             return NULL;
 639     const char *value(Value());
 640     for (const char **reserved(Reserved_); *reserved != NULL; ++reserved)
 641         if (strcmp(*reserved, value) == 0)
 642             return NULL;
 643     return value;
 644 }
 645
 646 void CYSwitch::Output(CYOutput &out, CYFlags flags) const {
 647     out << "switch" << ' ' << '(' << *value_ << ')' << ' ' << '{';
 648     out << clauses_;
 649     out << '}';
 650 }
 651
 652 void CYThis::Output(CYOutput &out, CYFlags flags) const {
 653     CYWord::Output(out);
 654 }
 655
 656 namespace cy {
 657 namespace Syntax {
 658
 659 void Throw::Output(CYOutput &out, CYFlags flags) const {
 660     out << "throw";
 661     if (value_ != NULL)
 662         out << ' ' << *value_;
 663     out << ';';
 664 }
 665
 666 void Try::Output(CYOutput &out, CYFlags flags) const {
 667     out << "try" << ' ' << code_ << catch_ << finally_;
 668 }
 669
 670 } }
 671
 672 void CYVar::Output(CYOutput &out, CYFlags flags) const {
 673     out << "var";
 674     declarations_->Output(out, flags);
 675     out << ';';
 676 }
 677
 678 void CYVariable::Output(CYOutput &out, CYFlags flags) const {
 679     out << *name_;
 680 }
 681
 682 void CYWhile::Output(CYOutput &out, CYFlags flags) const {
 683     out << "while" << '(' << *test_ << ')';
 684     code_->Single(out, CYRight(flags));
 685 }
 686
 687 void CYWith::Output(CYOutput &out, CYFlags flags) const {
 688     out << "with" << '(' << *scope_ << ')';
 689     code_->Single(out, CYRight(flags));
 690 }
 691
 692 void CYWord::ClassName(CYOutput &out, bool object) const {
 693     if (object)
 694         out << "objc_getClass(";
 695     out << '"' << Word() << '"';
 696     if (object)
 697         out << ')';
 698 }
 699
 700 void CYWord::Output(CYOutput &out) const {
 701     out << Word();
 702     if (out.options_.verbose_)
 703         out.out_ << '@' << this;
 704 }
 705
 706 void CYWord::PropertyName(CYOutput &out) const {
 707     Output(out);
 708 }
 709
 710 const char *CYWord::Word() const {
 711     return word_;
 712 }