X-Git-Url: https://git.saurik.com/cycript.git/blobdiff_plain/7b750785c5d53fef6b885cc1fc4bd02ba6daeb1b..b23692f3038296c94d25c84c55ed4c1bb49619cf:/Output.cpp diff --git a/Output.cpp b/Output.cpp index 778acb5..22e2149 100644 --- a/Output.cpp +++ b/Output.cpp @@ -1,47 +1,48 @@ /* Cycript - Optimizing JavaScript Compiler/Runtime - * Copyright (C) 2009-2013 Jay Freeman (saurik) + * Copyright (C) 2009-2015 Jay Freeman (saurik) */ -/* GNU General Public License, Version 3 {{{ */ +/* GNU Affero General Public License, Version 3 {{{ */ /* - * Cycript is free software: you can redistribute it and/or modify - * it under the terms of the GNU General Public License as published - * by the Free Software Foundation, either version 3 of the License, - * or (at your option) any later version. - * - * Cycript is distributed in the hope that it will be useful, but - * WITHOUT ANY WARRANTY; without even the implied warranty of + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU Affero General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - * GNU General Public License for more details. - * - * You should have received a copy of the GNU General Public License - * along with Cycript. If not, see . + * GNU Affero General Public License for more details. + + * You should have received a copy of the GNU Affero General Public License + * along with this program. If not, see . **/ /* }}} */ #include "cycript.hpp" -#include "Parser.hpp" #include +#include "Syntax.hpp" + void CYOutput::Terminate() { - out_ << ';'; + operator ()(';'); mode_ = NoMode; } CYOutput &CYOutput::operator <<(char rhs) { if (rhs == ' ' || rhs == '\n') if (pretty_) - out_ << rhs; + operator ()(rhs); else goto done; else if (rhs == '\t') if (pretty_) for (unsigned i(0); i != indent_; ++i) - out_ << " "; + operator ()(" ", 4); else goto done; else if (rhs == '\r') { if (right_) { - out_ << '\n'; + operator ()('\n'); right_ = false; } goto done; } else goto work; @@ -53,7 +54,7 @@ CYOutput &CYOutput::operator <<(char rhs) { work: if (mode_ == Terminated && rhs != '}') { right_ = true; - out_ << ';'; + operator ()(';'); } if (rhs == ';') { @@ -65,21 +66,21 @@ CYOutput &CYOutput::operator <<(char rhs) { } } else if (rhs == '+') { if (mode_ == NoPlus) - out_ << ' '; + operator ()(' '); mode_ = NoPlus; } else if (rhs == '-') { if (mode_ == NoHyphen) - out_ << ' '; + operator ()(' '); mode_ = NoHyphen; } else if (WordEndRange_[rhs]) { if (mode_ == NoLetter) - out_ << ' '; + operator ()(' '); mode_ = NoLetter; } else none: mode_ = NoMode; right_ = true; - out_ << rhs; + operator ()(rhs); done: return *this; } @@ -91,21 +92,22 @@ CYOutput &CYOutput::operator <<(const char *rhs) { return *this << *rhs; if (mode_ == Terminated) - out_ << ';'; + operator ()(';'); else if ( mode_ == NoPlus && *rhs == '+' || mode_ == NoHyphen && *rhs == '-' || mode_ == NoLetter && WordEndRange_[*rhs] ) - out_ << ' '; + operator ()(' '); - if (WordEndRange_[rhs[size - 1]]) + char last(rhs[size - 1]); + if (WordEndRange_[last] || last == '/') mode_ = NoLetter; else mode_ = NoMode; right_ = true; - out_ << rhs; + operator ()(rhs, size); return *this; } @@ -137,26 +139,18 @@ void CYAssignment::Output(CYOutput &out, CYFlags flags) const { rhs_->Output(out, Precedence(), CYRight(flags)); } -void CYBlock::Output(CYOutput &out) const { +void CYBlock::Output(CYOutput &out, CYFlags flags) const { out << '{' << '\n'; ++out.indent_; - if (statements_ != NULL) - statements_->Multiple(out); + out << code_; --out.indent_; out << '\t' << '}'; } -void CYBlock::Output(CYOutput &out, CYFlags flags) const { - if (statements_ == NULL) - out.Terminate(); - else if (statements_->next_ == NULL) - statements_->Single(out, flags); - else - Output(out); -} - void CYBoolean::Output(CYOutput &out, CYFlags flags) const { - out << (Value() ? "true" : "false"); + out << '!' << (Value() ? "0" : "1"); + if ((flags & CYNoInteger) != 0) + out << '.'; } void CYBreak::Output(CYOutput &out, CYFlags flags) const { @@ -180,16 +174,49 @@ namespace cy { namespace Syntax { void Catch::Output(CYOutput &out) const { - out << ' ' << "catch" << ' ' << '(' << *name_ << ')' << ' ' << code_; + out << ' ' << "catch" << ' ' << '(' << *name_ << ')' << ' '; + out << '{' << '\n'; + ++out.indent_; + out << code_; + --out.indent_; + out << '\t' << '}'; } } } -void CYComment::Output(CYOutput &out, CYFlags flags) const { - out << '\r'; - out.out_ << value_; - out.right_ = true; - out << '\r'; +void CYClassExpression::Output(CYOutput &out, CYFlags flags) const { + bool protect((flags & CYNoClass) != 0); + if (protect) + out << '('; + out << "class"; + if (name_ != NULL) + out << ' ' << *name_; + out << *tail_;; + if (protect) + out << ')'; +} + +void CYClassStatement::Output(CYOutput &out, CYFlags flags) const { + out << "class" << ' ' << *name_ << *tail_; +} + +void CYClassTail::Output(CYOutput &out) const { + if (extends_ == NULL) + out << ' '; + else { + out << '\n'; + ++out.indent_; + out << "extends" << ' '; + extends_->Output(out, CYAssign::Precedence_ - 1, CYNoFlags); + out << '\n'; + --out.indent_; + } + + out << '{' << '\n'; + ++out.indent_; + + --out.indent_; + out << '}'; } void CYCompound::Output(CYOutput &out, CYFlags flags) const { @@ -202,6 +229,12 @@ void CYCompound::Output(CYOutput &out, CYFlags flags) const { } } +void CYComputed::PropertyName(CYOutput &out) const { + out << '['; + expression_->Output(out, CYAssign::Precedence_, CYNoFlags); + out << ']'; +} + void CYCondition::Output(CYOutput &out, CYFlags flags) const { test_->Output(out, Precedence() - 1, CYLeft(flags)); out << ' ' << '?' << ' '; @@ -219,13 +252,15 @@ void CYContinue::Output(CYOutput &out, CYFlags flags) const { } void CYClause::Output(CYOutput &out) const { + out << '\t'; if (case_ != NULL) out << "case" << ' ' << *case_; else out << "default"; out << ':' << '\n'; - if (statements_ != NULL) - statements_->Multiple(out); + ++out.indent_; + out << code_; + --out.indent_; out << next_; } @@ -233,11 +268,6 @@ void CYDebugger::Output(CYOutput &out, CYFlags flags) const { out << "debugger" << ';'; } -void CYDeclaration::ForIn(CYOutput &out, CYFlags flags) const { - out << "var"; - Output(out, CYRight(flags)); -} - void CYDeclaration::Output(CYOutput &out, CYFlags flags) const { out << *identifier_; //out.out_ << ':' << identifier_->usage_ << '#' << identifier_->offset_; @@ -247,11 +277,6 @@ void CYDeclaration::Output(CYOutput &out, CYFlags flags) const { } } -void CYForDeclarations::Output(CYOutput &out, CYFlags flags) const { - out << "var"; - Output(out, CYRight(flags)); -} - void CYDeclarations::Output(CYOutput &out) const { Output(out, CYNoFlags); } @@ -285,16 +310,29 @@ void CYDirectMember::Output(CYOutput &out, CYFlags flags) const { void CYDoWhile::Output(CYOutput &out, CYFlags flags) const { out << "do"; - code_->Single(out, CYCenter(flags)); + + unsigned line(out.position_.line); + unsigned indent(out.indent_); + code_->Single(out, CYCenter(flags), CYCompactLong); + + if (out.position_.line != line && out.recent_ == indent) + out << ' '; + else + out << '\n' << '\t'; + out << "while" << ' ' << '(' << *test_ << ')'; } -void CYElement::Output(CYOutput &out) const { +void CYElementSpread::Output(CYOutput &out) const { + out << "..." << value_; +} + +void CYElementValue::Output(CYOutput &out) const { if (value_ != NULL) value_->Output(out, CYAssign::Precedence_, CYNoFlags); if (next_ != NULL || value_ == NULL) { out << ','; - if (next_ != NULL && next_->value_ != NULL) + if (next_ != NULL && !next_->Elision()) out << ' '; } if (next_ != NULL) @@ -305,17 +343,13 @@ void CYEmpty::Output(CYOutput &out, CYFlags flags) const { out.Terminate(); } -void CYExpress::Output(CYOutput &out, CYFlags flags) const { - expression_->Output(out, flags | CYNoBF); - out << ';'; -} - -void CYExpression::ClassName(CYOutput &out, bool object) const { - Output(out, CYAssign::Precedence_, CYNoFlags); +void CYEval::Output(CYOutput &out, CYFlags flags) const { + _assert(false); } -void CYExpression::ForIn(CYOutput &out, CYFlags flags) const { - Output(out, flags | CYNoRightHand); +void CYExpress::Output(CYOutput &out, CYFlags flags) const { + expression_->Output(out, flags | CYNoBFC); + out << ';'; } void CYExpression::Output(CYOutput &out) const { @@ -329,12 +363,21 @@ void CYExpression::Output(CYOutput &out, int precedence, CYFlags flags) const { Output(out, flags); } +void CYExternal::Output(CYOutput &out, CYFlags flags) const { + out << "extern" << abi_ << typed_ << ';'; +} + void CYFatArrow::Output(CYOutput &out, CYFlags flags) const { - out << '(' << parameters_ << ')' << ' ' << "=>" << ' ' << code_; + out << '(' << parameters_ << ')' << ' ' << "=>" << ' ' << '{' << code_ << '}'; } void CYFinally::Output(CYOutput &out) const { - out << ' ' << "finally" << ' ' << code_; + out << ' ' << "finally" << ' '; + out << '{' << '\n'; + ++out.indent_; + out << code_; + --out.indent_; + out << '\t' << '}'; } void CYFor::Output(CYOutput &out, CYFlags flags) const { @@ -350,33 +393,62 @@ void CYFor::Output(CYOutput &out, CYFlags flags) const { out << ' '; out << increment_; out << ')'; - code_->Single(out, CYRight(flags)); + code_->Single(out, CYRight(flags), CYCompactShort); +} + +void CYForLexical::Output(CYOutput &out, CYFlags flags) const { + out << (constant_ ? "const" : "let") << ' '; + declaration_->Output(out, CYRight(flags)); +} + +void CYForIn::Output(CYOutput &out, CYFlags flags) const { + out << "for" << ' ' << '('; + initialiser_->Output(out, CYNoIn | CYNoRightHand); + out << ' ' << "in" << ' ' << *set_ << ')'; + code_->Single(out, CYRight(flags), CYCompactShort); +} + +void CYForInitialized::Output(CYOutput &out, CYFlags flags) const { + out << "for" << ' ' << '(' << "var" << ' '; + declaration_->Output(out, CYNoIn | CYNoRightHand); + out << ' ' << "in" << ' ' << *set_ << ')'; + code_->Single(out, CYRight(flags), CYCompactShort); +} + +void CYForInComprehension::Output(CYOutput &out) const { + out << "for" << ' ' << '('; + declaration_->Output(out, CYNoIn | CYNoRightHand); + out << ' ' << "in" << ' ' << *set_ << ')'; } void CYForOf::Output(CYOutput &out, CYFlags flags) const { - out << "for" << ' ' << "each" << ' ' << '('; - initialiser_->ForIn(out, CYNoIn); - out << "in" << *set_ << ')'; - code_->Single(out, CYRight(flags)); + out << "for" << ' ' << '('; + initialiser_->Output(out, CYNoRightHand); + out << ' ' << "of" << ' ' << *set_ << ')'; + code_->Single(out, CYRight(flags), CYCompactShort); } void CYForOfComprehension::Output(CYOutput &out) const { - out << "for" << ' ' << "each" << ' ' << '(' << *name_ << ' ' << "in" << ' ' << *set_ << ')' << next_; + out << "for" << ' ' << '('; + declaration_->Output(out, CYNoRightHand); + out << ' ' << "of" << ' ' << *set_ << ')' << next_; } -void CYForIn::Output(CYOutput &out, CYFlags flags) const { - out << "for" << ' ' << '('; - if (initialiser_ != NULL) - initialiser_->ForIn(out, CYNoIn); - out << "in" << *set_ << ')'; - code_->Single(out, CYRight(flags)); +void CYForVariable::Output(CYOutput &out, CYFlags flags) const { + out << "var" << ' '; + declaration_->Output(out, CYRight(flags)); } -void CYForInComprehension::Output(CYOutput &out) const { - out << "for" << ' ' << '(' << *name_ << ' ' << "in" << ' ' << *set_ << ')'; +void CYFunction::Output(CYOutput &out) const { + out << '(' << parameters_ << ')' << ' '; + out << '{' << '\n'; + ++out.indent_; + out << code_; + --out.indent_; + out << '\t' << '}'; } -void CYFunction::Output(CYOutput &out, CYFlags flags) const { +void CYFunctionExpression::Output(CYOutput &out, CYFlags flags) const { // XXX: one could imagine using + here to save a byte bool protect((flags & CYNoFunction) != 0); if (protect) @@ -384,18 +456,14 @@ void CYFunction::Output(CYOutput &out, CYFlags flags) const { out << "function"; if (name_ != NULL) out << ' ' << *name_; - out << '(' << parameters_ << ')'; - out << ' ' << code_; + CYFunction::Output(out); if (protect) out << ')'; } -void CYFunctionExpression::Output(CYOutput &out, CYFlags flags) const { - CYFunction::Output(out, flags); -} - void CYFunctionStatement::Output(CYOutput &out, CYFlags flags) const { - CYFunction::Output(out, flags); + out << "function" << ' ' << *name_; + CYFunction::Output(out); } void CYFunctionParameter::Output(CYOutput &out) const { @@ -405,7 +473,7 @@ void CYFunctionParameter::Output(CYOutput &out) const { } const char *CYIdentifier::Word() const { - return replace_ == NULL || replace_ == this ? CYWord::Word() : replace_->Word(); + return next_ == NULL || next_ == this ? CYWord::Word() : next_->Word(); } void CYIf::Output(CYOutput &out, CYFlags flags) const { @@ -425,11 +493,18 @@ void CYIf::Output(CYOutput &out, CYFlags flags) const { else jacks |= protect ? CYNoFlags : CYCenter(flags); - true_->Single(out, jacks); + unsigned line(out.position_.line); + unsigned indent(out.indent_); + true_->Single(out, jacks, CYCompactShort); if (false_ != NULL) { - out << '\t' << "else"; - false_->Single(out, right); + if (out.position_.line != line && out.recent_ == indent) + out << ' '; + else + out << '\n' << '\t'; + + out << "else"; + false_->Single(out, right, CYCompactLong); } if (protect) @@ -444,6 +519,11 @@ void CYImport::Output(CYOutput &out, CYFlags flags) const { out << "@import"; } +void CYIndirect::Output(CYOutput &out, CYFlags flags) const { + out << "*"; + rhs_->Output(out, Precedence(), CYRight(flags)); +} + void CYIndirectMember::Output(CYOutput &out, CYFlags flags) const { object_->Output(out, Precedence(), CYLeft(flags)); if (const char *word = property_->Word()) @@ -467,8 +547,22 @@ void CYInfix::Output(CYOutput &out, CYFlags flags) const { } void CYLabel::Output(CYOutput &out, CYFlags flags) const { - out << *name_ << ':' << ' '; - statement_->Single(out, CYRight(flags)); + out << *name_ << ':'; + statement_->Single(out, CYRight(flags), CYCompactShort); +} + +void CYParenthetical::Output(CYOutput &out, CYFlags flags) const { + out << '('; + expression_->Output(out, CYCompound::Precedence_, CYNoFlags); + out << ')'; +} + +void CYStatement::Output(CYOutput &out) const { + Multiple(out); +} + +void CYTemplate::Output(CYOutput &out, CYFlags flags) const { + _assert(false); } void CYTypeArrayOf::Output(CYOutput &out, CYIdentifier *identifier) const { @@ -485,7 +579,7 @@ void CYTypeBlockWith::Output(CYOutput &out, CYIdentifier *identifier) const { } void CYTypeConstant::Output(CYOutput &out, CYIdentifier *identifier) const { - out << "const"; + out << "const" << ' '; next_->Output(out, Precedence(), identifier); } @@ -543,12 +637,13 @@ void CYLambda::Output(CYOutput &out, CYFlags flags) const { } void CYTypeDefinition::Output(CYOutput &out, CYFlags flags) const { - out << "typedef" << *typed_; + out << "typedef" << ' ' << *typed_; } -void CYLetStatement::Output(CYOutput &out, CYFlags flags) const { - out << "let" << ' ' << '(' << *declarations_ << ')'; - code_->Single(out, CYRight(flags)); +void CYLet::Output(CYOutput &out, CYFlags flags) const { + out << "let" << ' '; + declarations_->Output(out, flags); // XXX: flags + out << ';'; } void CYModule::Output(CYOutput &out) const { @@ -571,7 +666,7 @@ void New::Output(CYOutput &out, CYFlags flags) const { } } void CYNull::Output(CYOutput &out, CYFlags flags) const { - CYWord::Output(out); + out << "null"; } void CYNumber::Output(CYOutput &out, CYFlags flags) const { @@ -614,20 +709,42 @@ void CYPrefix::Output(CYOutput &out, CYFlags flags) const { rhs_->Output(out, Precedence(), CYRight(flags)); } -void CYProgram::Output(CYOutput &out) const { - if (statements_ != NULL) - statements_->Multiple(out); +void CYScript::Output(CYOutput &out) const { + out << code_; } void CYProperty::Output(CYOutput &out) const { + if (next_ != NULL || out.pretty_) + out << ','; + out << '\n' << next_; +} + +void CYPropertyGetter::Output(CYOutput &out) const { + out << "get" << ' '; + name_->PropertyName(out); + CYFunction::Output(out); + CYProperty::Output(out); +} + +void CYPropertyMethod::Output(CYOutput &out) const { + name_->PropertyName(out); + CYFunction::Output(out); + CYProperty::Output(out); +} + +void CYPropertySetter::Output(CYOutput &out) const { + out << "set" << ' '; + name_->PropertyName(out); + CYFunction::Output(out); + CYProperty::Output(out); +} + +void CYPropertyValue::Output(CYOutput &out) const { out << '\t'; name_->PropertyName(out); out << ':' << ' '; value_->Output(out, CYAssign::Precedence_, CYNoFlags); - if (next_ != NULL) - out << ',' << '\n' << *next_; - else - out << '\n'; + CYProperty::Output(out); } void CYRegEx::Output(CYOutput &out, CYFlags flags) const { @@ -648,8 +765,11 @@ void CYRubyBlock::Output(CYOutput &out, CYFlags flags) const { } void CYRubyProc::Output(CYOutput &out, CYFlags flags) const { - // XXX: this is not outputting the parameters + out << '{' << ' ' << '|' << parameters_ << '|' << '\n'; + ++out.indent_; out << code_; + --out.indent_; + out << '\t' << '}'; } void CYStatement::Multiple(CYOutput &out, CYFlags flags) const { @@ -664,17 +784,26 @@ void CYStatement::Multiple(CYOutput &out, CYFlags flags) const { } } -void CYStatement::Single(CYOutput &out, CYFlags flags) const { +void CYStatement::Single(CYOutput &out, CYFlags flags, CYCompactType request) const { if (this == NULL) return out.Terminate(); _assert(next_ == NULL); - out << '\n'; - ++out.indent_; - out << '\t'; + + CYCompactType compact(Compact()); + + if (compact >= request) + out << ' '; + else { + out << '\n'; + ++out.indent_; + out << '\t'; + } + Output(out, flags); - out << '\n'; - --out.indent_; + + if (compact < request) + --out.indent_; } void CYString::Output(CYOutput &out, CYFlags flags) const { @@ -725,14 +854,28 @@ const char *CYString::Word() const { return value; } +void CYSuperAccess::Output(CYOutput &out, CYFlags flags) const { + out << "super"; + if (const char *word = property_->Word()) + out << '.' << word; + else + out << '[' << *property_ << ']'; +} + +void CYSuperCall::Output(CYOutput &out, CYFlags flags) const { + out << "super" << '(' << arguments_ << ')'; +} + void CYSwitch::Output(CYOutput &out, CYFlags flags) const { - out << "switch" << ' ' << '(' << *value_ << ')' << ' ' << '{'; + out << "switch" << ' ' << '(' << *value_ << ')' << ' ' << '{' << '\n'; + ++out.indent_; out << clauses_; - out << '}'; + --out.indent_; + out << '\t' << '}'; } void CYThis::Output(CYOutput &out, CYFlags flags) const { - CYWord::Output(out); + out << "this"; } namespace cy { @@ -746,7 +889,13 @@ void Throw::Output(CYOutput &out, CYFlags flags) const { } void Try::Output(CYOutput &out, CYFlags flags) const { - out << "try" << ' ' << code_ << catch_ << finally_; + out << "try" << ' '; + out << '{' << '\n'; + ++out.indent_; + out << code_; + --out.indent_; + out << '\t' << '}'; + out << catch_ << finally_; } } } @@ -780,8 +929,8 @@ void CYTypeVoid::Output(CYOutput &out) const { } void CYVar::Output(CYOutput &out, CYFlags flags) const { - out << "var"; - declarations_->Output(out, flags); + out << "var" << ' '; + declarations_->Output(out, flags); // XXX: flags out << ';'; } @@ -790,27 +939,23 @@ void CYVariable::Output(CYOutput &out, CYFlags flags) const { } void CYWhile::Output(CYOutput &out, CYFlags flags) const { - out << "while" << '(' << *test_ << ')'; - code_->Single(out, CYRight(flags)); + out << "while" << ' ' << '(' << *test_ << ')'; + code_->Single(out, CYRight(flags), CYCompactShort); } void CYWith::Output(CYOutput &out, CYFlags flags) const { - out << "with" << '(' << *scope_ << ')'; - code_->Single(out, CYRight(flags)); -} - -void CYWord::ClassName(CYOutput &out, bool object) const { - if (object) - out << "objc_getClass("; - out << '"' << Word() << '"'; - if (object) - out << ')'; + out << "with" << ' ' << '(' << *scope_ << ')'; + code_->Single(out, CYRight(flags), CYCompactShort); } void CYWord::Output(CYOutput &out) const { out << Word(); - if (out.options_.verbose_) - out.out_ << '@' << this; + if (out.options_.verbose_) { + out('@'); + char number[32]; + sprintf(number, "%p", this); + out(number); + } } void CYWord::PropertyName(CYOutput &out) const {