X-Git-Url: https://git.saurik.com/cycript.git/blobdiff_plain/cac61857bcc00ddc1e354756e11b24e93b8edbf3..46f4f308790f8c412abb5f81f278822aef198e20:/Output.cpp diff --git a/Output.cpp b/Output.cpp index 7580288..1feff42 100644 --- a/Output.cpp +++ b/Output.cpp @@ -1,184 +1,206 @@ +/* Cycript - Optimizing JavaScript Compiler/Runtime + * Copyright (C) 2009-2012 Jay Freeman (saurik) +*/ + +/* GNU Lesser General Public License, Version 3 {{{ */ +/* + * Cycript is free software: you can redistribute it and/or modify it under + * the terms of the GNU Lesser 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 MERCHANTABILITY or + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public + * License for more details. + * + * You should have received a copy of the GNU Lesser General Public License + * along with Cycript. If not, see . +**/ +/* }}} */ + +#include "cycript.hpp" #include "Parser.hpp" -#include -#include - -#include #include -_finline CYFlags operator ~(CYFlags rhs) { - return static_cast(~static_cast(rhs)); -} - -_finline CYFlags operator &(CYFlags lhs, CYFlags rhs) { - return static_cast(static_cast(lhs) & static_cast(rhs)); -} - -_finline CYFlags operator |(CYFlags lhs, CYFlags rhs) { - return static_cast(static_cast(lhs) | static_cast(rhs)); -} - -_finline CYFlags &operator |=(CYFlags &lhs, CYFlags rhs) { - return lhs = lhs | rhs; -} - -_finline CYFlags CYLeft(CYFlags flags) { - return flags & ~CYNoTrailer; -} - -_finline CYFlags CYCenter(CYFlags flags) { - return flags & CYNoIn; -} - -_finline CYFlags CYRight(CYFlags flags) { - return flags & (CYNoIn | CYNoTrailer); -} - -bool CYFalse::Value() const { - return false; -} - -bool CYTrue::Value() const { - return true; -} +void CYOutput::Terminate() { + out_ << ';'; + mode_ = NoMode; +} + +CYOutput &CYOutput::operator <<(char rhs) { + if (rhs == ' ' || rhs == '\n') + if (pretty_) + out_ << rhs; + else goto done; + else if (rhs == '\t') + if (pretty_) + for (unsigned i(0); i != indent_; ++i) + out_ << " "; + else goto done; + else if (rhs == '\r') { + if (right_) { + out_ << '\n'; + right_ = false; + } goto done; + } else goto work; + + right_ = true; + mode_ = NoMode; + goto done; + + work: + if (mode_ == Terminated && rhs != '}') { + right_ = true; + out_ << ';'; + } -#define CYPA 16 + if (rhs == ';') { + if (pretty_) + goto none; + else { + mode_ = Terminated; + goto done; + } + } else if (rhs == '+') { + if (mode_ == NoPlus) + out_ << ' '; + mode_ = NoPlus; + } else if (rhs == '-') { + if (mode_ == NoHyphen) + out_ << ' '; + mode_ = NoHyphen; + } else if (WordEndRange_[rhs]) { + if (mode_ == NoLetter) + out_ << ' '; + mode_ = NoLetter; + } else none: + mode_ = NoMode; + + right_ = true; + out_ << rhs; + done: + return *this; +} + +CYOutput &CYOutput::operator <<(const char *rhs) { + size_t size(strlen(rhs)); + + if (size == 1) + return *this << *rhs; + + if (mode_ == Terminated) + out_ << ';'; + else if ( + mode_ == NoPlus && *rhs == '+' || + mode_ == NoHyphen && *rhs == '-' || + mode_ == NoLetter && WordEndRange_[*rhs] + ) + out_ << ' '; + + if (WordEndRange_[rhs[size - 1]]) + mode_ = NoLetter; + else + mode_ = NoMode; -void CYAddressOf::Output(std::ostream &out, CYFlags flags) const { - rhs_->Output(out, 1, CYLeft(flags)); - out << ".$cya()"; + right_ = true; + out_ << rhs; + return *this; } -void CYArgument::Output(std::ostream &out) const { +void CYArgument::Output(CYOutput &out) const { if (name_ != NULL) { out << *name_; if (value_ != NULL) - out << ":"; + out << ':' << ' '; } if (value_ != NULL) - value_->Output(out, CYPA, CYNoFlags); + value_->Output(out, CYAssign::Precedence_, CYNoFlags); if (next_ != NULL) { if (next_->name_ == NULL) out << ','; - else - out << ' '; - next_->Output(out); + out << ' ' << *next_; } } -void CYArray::Output(std::ostream &out, CYFlags flags) const { - out << '['; - if (elements_ != NULL) - elements_->Output(out); - out << ']'; -} - -void CYArrayComprehension::Output(std::ostream &out, CYFlags flags) const { - // XXX: I don't necc. need the ()s - out << "(function($cyv"; - for (CYComprehension *comprehension(comprehensions_); comprehension != NULL; comprehension = comprehension->next_) - if (const char *name = comprehension->Name()) - out << ',' << name; - out << "){"; - out << "$cyv=[];"; - comprehensions_->Output(out); - out << "$cyv.push("; - expression_->Output(out, CYPA, CYNoFlags); - out << ");"; - for (CYComprehension *comprehension(comprehensions_); comprehension != NULL; comprehension = comprehension->next_) - comprehension->End_(out); - out << "return $cyv;"; - out << "}())"; -} - -void CYAssignment::Output(std::ostream &out, CYFlags flags) const { - lhs_->Output(out, Precedence() - 1, CYLeft(flags)); - out << Operator(); +void CYArray::Output(CYOutput &out, CYFlags flags) const { + out << '[' << elements_ << ']'; +} + +void CYArrayComprehension::Output(CYOutput &out, CYFlags flags) const { + out << '[' << *expression_ << ' ' << *comprehensions_ << ']'; +} + +void CYAssignment::Output(CYOutput &out, CYFlags flags) const { + lhs_->Output(out, Precedence() - 1, CYLeft(flags) | CYNoRightHand); + out << ' ' << Operator() << ' '; rhs_->Output(out, Precedence(), CYRight(flags)); } -void CYBlock::Output(std::ostream &out) const { - for (CYSource *statement(statements_); statement != NULL; statement = statement->next_) - statement->Output(out); +void CYBlock::Output(CYOutput &out) const { + out << '{' << '\n'; + ++out.indent_; + if (statements_ != NULL) + statements_->Multiple(out); + --out.indent_; + out << '\t' << '}'; } -void CYBoolean::Output(std::ostream &out, CYFlags flags) const { - if ((flags & CYNoLeader) != 0) - out << ' '; +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"); - if ((flags & CYNoTrailer) != 0) - out << ' '; } -void CYBreak::Output(std::ostream &out) const { +void CYBreak::Output(CYOutput &out, CYFlags flags) const { out << "break"; if (label_ != NULL) out << ' ' << *label_; out << ';'; } -void CYCall::Output(std::ostream &out, CYFlags flags) const { - function_->Output(out, Precedence(), CYLeft(flags)); - out << '('; - if (arguments_ != NULL) - arguments_->Output(out); - out << ')'; +void CYCall::Output(CYOutput &out, CYFlags flags) const { + bool protect((flags & CYNoCall) != 0); + if (protect) + out << '('; + function_->Output(out, Precedence(), protect ? CYNoFlags : flags); + out << '(' << arguments_ << ')'; + if (protect) + out << ')'; } -void CYCatch::Output(std::ostream &out) const { - out << "catch(" << *name_ << ')'; - code_->Output(out, true); -} +namespace cy { +namespace Syntax { -void CYCategory::Output(std::ostream &out) const { - out << "(function($cys,$cyp,$cyc,$cyn,$cyt){"; - out << "$cyp=object_getClass($cys);"; - out << "$cyc=$cys;"; - if (messages_ != NULL) - messages_->Output(out, true); - out << "})("; - name_->ClassName(out, true); - out << ");"; +void Catch::Output(CYOutput &out) const { + out << ' ' << "catch" << ' ' << '(' << *name_ << ')' << ' ' << code_; } -void CYClass::Output(std::ostream &out) const { - Output(out, CYNoBF); - out << ";"; -} +} } -void CYClass::Output(std::ostream &out, CYFlags flags) const { - // XXX: I don't necc. need the ()s - out << "(function($cys,$cyp,$cyc,$cyn,$cyt,$cym){"; - out << "$cyp=object_getClass($cys);"; - out << "$cyc=objc_allocateClassPair($cys,"; - if (name_ != NULL) - name_->ClassName(out, false); - else - out << "$cyq(\"CY$\")"; - out << ",0);"; - out << "$cym=object_getClass($cyc);"; - if (fields_ != NULL) - fields_->Output(out); - if (messages_ != NULL) - messages_->Output(out, false); - out << "objc_registerClassPair($cyc);"; - out << "return $cyc;"; - out << "}("; - if (super_ != NULL) - super_->Output(out, CYPA, CYNoFlags); - else - out << "null"; - out << "))"; +void CYComment::Output(CYOutput &out, CYFlags flags) const { + out << '\r'; + out.out_ << value_; + out.right_ = true; + out << '\r'; } -void CYCompound::Output(std::ostream &out, CYFlags flags) const { +void CYCompound::Output(CYOutput &out, CYFlags flags) const { if (CYExpression *expression = expressions_) if (CYExpression *next = expression->next_) { expression->Output(out, CYLeft(flags)); CYFlags center(CYCenter(flags)); while (next != NULL) { expression = next; - out << ','; + out << ',' << ' '; next = expression->next_; CYFlags right(next != NULL ? center : CYRight(flags)); expression->Output(out, right); @@ -187,620 +209,508 @@ void CYCompound::Output(std::ostream &out, CYFlags flags) const { expression->Output(out, flags); } -void CYComprehension::Output(std::ostream &out) const { - Begin_(out); - if (next_ != NULL) - next_->Output(out); -} - -void CYCondition::Output(std::ostream &out, CYFlags flags) const { +void CYCondition::Output(CYOutput &out, CYFlags flags) const { test_->Output(out, Precedence() - 1, CYLeft(flags)); - out << '?'; + out << ' ' << '?' << ' '; if (true_ != NULL) - true_->Output(out, CYPA, CYNoFlags); - out << ':'; - false_->Output(out, CYPA, CYRight(flags)); + true_->Output(out, CYAssign::Precedence_, CYNoFlags); + out << ' ' << ':' << ' '; + false_->Output(out, CYAssign::Precedence_, CYRight(flags)); } -void CYContinue::Output(std::ostream &out) const { +void CYContinue::Output(CYOutput &out, CYFlags flags) const { out << "continue"; if (label_ != NULL) out << ' ' << *label_; out << ';'; } -void CYClause::Output(std::ostream &out) const { - if (case_ != NULL) { - out << "case"; - case_->Output(out, CYNoLeader); - } else +void CYClause::Output(CYOutput &out) const { + if (case_ != NULL) + out << "case" << ' ' << *case_; + else out << "default"; - out << ':'; - if (code_ != NULL) - code_->Output(out, false); - if (next_ != NULL) - out << *next_; + out << ':' << '\n'; + if (statements_ != NULL) + statements_->Multiple(out); + out << next_; } -const char *CYDeclaration::ForEachIn() const { - return identifier_->Value(); +void CYDebugger::Output(CYOutput &out, CYFlags flags) const { + out << "debugger" << ';'; } -void CYDeclaration::ForIn(std::ostream &out, CYFlags flags) const { - if ((flags & CYNoLeader) != 0) - out << ' '; - out << "var "; +void CYDeclaration::ForIn(CYOutput &out, CYFlags flags) const { + out << "var"; Output(out, CYRight(flags)); } -void CYDeclaration::ForEachIn(std::ostream &out) const { +void CYDeclaration::Output(CYOutput &out, CYFlags flags) const { out << *identifier_; + //out.out_ << ':' << identifier_->usage_ << '#' << identifier_->offset_; + if (initialiser_ != NULL) { + out << ' ' << '=' << ' '; + initialiser_->Output(out, CYAssign::Precedence_, CYRight(flags)); + } } -void CYDeclaration::Output(std::ostream &out, CYFlags flags) const { - out << *identifier_; - if (initialiser_ != NULL) { - out << '='; - initialiser_->Output(out, CYPA, CYRight(flags)); - } else if ((flags & CYNoTrailer) != 0) - out << ' '; +void CYForDeclarations::Output(CYOutput &out, CYFlags flags) const { + out << "var"; + Output(out, CYRight(flags)); } -void CYDeclarations::For(std::ostream &out) const { - out << "var "; - Output(out, CYNoIn); +void CYDeclarations::Output(CYOutput &out) const { + Output(out, CYNoFlags); } -void CYDeclarations::Output(std::ostream &out, CYFlags flags) const { +void CYDeclarations::Output(CYOutput &out, CYFlags flags) const { const CYDeclarations *declaration(this); - output: - CYDeclarations *next(declaration->next_); - CYFlags right(next == NULL ? CYRight(flags) : CYCenter(flags)); - declaration->declaration_->Output(out, right); + bool first(true); - if (next != NULL) { - out << ','; + for (;;) { + CYDeclarations *next(declaration->next_); + + CYFlags jacks(first ? CYLeft(flags) : next == NULL ? CYRight(flags) : CYCenter(flags)); + first = false; + declaration->declaration_->Output(out, jacks); + + if (next == NULL) + break; + + out << ',' << ' '; declaration = next; - goto output; } } -void CYDirectMember::Output(std::ostream &out, CYFlags flags) const { - object_->Output(out, Precedence(), CYLeft(flags)); +void CYDirectMember::Output(CYOutput &out, CYFlags flags) const { + object_->Output(out, Precedence(), CYLeft(flags) | CYNoInteger); if (const char *word = property_->Word()) out << '.' << word; - else { - out << '['; - property_->Output(out, CYNoFlags); - out << ']'; - } + else + out << '[' << *property_ << ']'; } -void CYDoWhile::Output(std::ostream &out) const { - // XXX: extra space character! - out << "do "; - code_->Output(out, false); - out << "while("; - test_->Output(out, CYNoFlags); - out << ')'; +void CYDoWhile::Output(CYOutput &out, CYFlags flags) const { + out << "do"; + code_->Single(out, CYCenter(flags)); + out << "while" << ' ' << '(' << *test_ << ')'; } -void CYElement::Output(std::ostream &out) const { +void CYElement::Output(CYOutput &out) const { if (value_ != NULL) - value_->Output(out, CYPA, CYNoFlags); - if (next_ != NULL || value_ == NULL) + value_->Output(out, CYAssign::Precedence_, CYNoFlags); + if (next_ != NULL || value_ == NULL) { out << ','; + if (next_ != NULL && next_->value_ != NULL) + out << ' '; + } if (next_ != NULL) next_->Output(out); } -void CYEmpty::Output(std::ostream &out) const { - out << ';'; +void CYEmpty::Output(CYOutput &out, CYFlags flags) const { + out.Terminate(); } -void CYEmpty::Output(std::ostream &out, bool block) const { - if (next_ != NULL) - CYSource::Output(out, block); - else - out << "{}"; -} - -void CYExpress::Output(std::ostream &out) const { - expression_->Output(out, CYNoBF); +void CYExpress::Output(CYOutput &out, CYFlags flags) const { + expression_->Output(out, flags | CYNoBF); out << ';'; } -void CYExpression::ClassName(std::ostream &out, bool object) const { - Output(out, CYPA, CYNoFlags); -} - -const char *CYExpression::ForEachIn() const { - return NULL; +void CYExpression::ClassName(CYOutput &out, bool object) const { + Output(out, CYAssign::Precedence_, CYNoFlags); } -void CYExpression::For(std::ostream &out) const { - Output(out, CYNoIn); +void CYExpression::ForIn(CYOutput &out, CYFlags flags) const { + Output(out, flags | CYNoRightHand); } -void CYExpression::ForEachIn(std::ostream &out) const { - // XXX: this should handle LeftHandSideExpression - Output(out, CYPA, CYNoFlags); +void CYExpression::Output(CYOutput &out) const { + Output(out, CYNoFlags); } -void CYExpression::ForIn(std::ostream &out, CYFlags flags) const { - // XXX: this should handle LeftHandSideExpression - Output(out, flags); +void CYExpression::Output(CYOutput &out, unsigned precedence, CYFlags flags) const { + if (precedence < Precedence() || (flags & CYNoRightHand) != 0 && RightHand()) + out << '(' << *this << ')'; + else + Output(out, flags); } -void CYExpression::Output(std::ostream &out, unsigned precedence, CYFlags flags) const { - if (precedence < Precedence()) { - out << '('; - Output(out, CYNoFlags); - out << ')'; - } else - Output(out, flags); +void CYFatArrow::Output(CYOutput &out, CYFlags flags) const { + out << '(' << parameters_ << ')' << ' ' << "=>" << ' ' << code_; } -void CYField::Output(std::ostream &out) const { - // XXX: implement! +void CYFinally::Output(CYOutput &out) const { + out << ' ' << "finally" << ' ' << code_; } -void CYFor::Output(std::ostream &out) const { - out << "for("; +void CYFor::Output(CYOutput &out, CYFlags flags) const { + out << "for" << ' ' << '('; if (initialiser_ != NULL) - initialiser_->For(out); - out << ';'; + initialiser_->Output(out, CYNoIn); + out.Terminate(); if (test_ != NULL) - test_->Output(out, CYNoFlags); - out << ';'; + out << ' '; + out << test_; + out.Terminate(); if (increment_ != NULL) - increment_->Output(out, CYNoFlags); + out << ' '; + out << increment_; out << ')'; - code_->Output(out, false); + code_->Single(out, CYRight(flags)); } -void CYForEachIn::Output(std::ostream &out) const { - const char *name(initialiser_->ForEachIn()); - - if (name != NULL) { - out << '{'; - out << "var " << name << ';'; - } - - out << "(function($cys,$cyt){"; - out << "$cys="; - set_->Output(out, CYPA, CYNoFlags); - out << ";"; - - out << "for($cyt in $cys){"; - - initialiser_->ForEachIn(out); - out << "=$cys[$cyt];"; - - code_->Show(out); +void CYForOf::Output(CYOutput &out, CYFlags flags) const { + out << "for" << ' ' << "each" << ' ' << '('; + initialiser_->ForIn(out, CYNoIn); + out << "in" << *set_ << ')'; + code_->Single(out, CYRight(flags)); +} - out << "}}());"; +void CYForOfComprehension::Output(CYOutput &out) const { + out << "for" << ' ' << "each" << ' ' << '(' << *name_ << ' ' << "in" << ' ' << *set_ << ')' << next_; +} - if (name != NULL) - out << '}'; +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 CYForEachInComprehension::Begin_(std::ostream &out) const { - out << "(function($cys){"; - out << "$cys="; - set_->Output(out, CYPA, CYNoFlags); - out << ";"; +void CYForInComprehension::Output(CYOutput &out) const { + out << "for" << ' ' << '(' << *name_ << ' ' << "in" << ' ' << *set_ << ')'; +} - out << "for(" << *name_ << " in $cys){"; - out << *name_ << "=$cys[" << *name_ << "];"; +void CYFunction::Output(CYOutput &out, CYFlags flags) const { + // XXX: one could imagine using + here to save a byte + bool protect((flags & CYNoFunction) != 0); + if (protect) + out << '('; + out << "function"; + if (name_ != NULL) + out << ' ' << *name_; + out << '(' << parameters_ << ')'; + out << ' ' << code_; + if (protect) + out << ')'; } -void CYForEachInComprehension::End_(std::ostream &out) const { - out << "}}());"; +void CYFunctionExpression::Output(CYOutput &out, CYFlags flags) const { + CYFunction::Output(out, flags); } -void CYForIn::Output(std::ostream &out) const { - out << "for("; - initialiser_->ForIn(out, CYNoIn | CYNoTrailer); - out << "in"; - set_->Output(out, CYNoLeader); - out << ')'; - code_->Output(out, false); +void CYFunctionStatement::Output(CYOutput &out, CYFlags flags) const { + CYFunction::Output(out, flags); } -void CYForInComprehension::Begin_(std::ostream &out) const { - out << "for(" << *name_ << " in"; - set_->Output(out, CYNoLeader); - out << ')'; +void CYFunctionParameter::Output(CYOutput &out) const { + initialiser_->Output(out, CYNoFlags); + if (next_ != NULL) + out << ',' << ' ' << *next_; } -void CYFunction::Output(std::ostream &out) const { - CYLambda::Output(out, CYNoFlags); +const char *CYIdentifier::Word() const { + return replace_ == NULL || replace_ == this ? CYWord::Word() : replace_->Word(); } -void CYFunctionParameter::Output(std::ostream &out) const { - out << *name_; - if (next_ != NULL) { - out << ','; - out << *next_; +void CYIf::Output(CYOutput &out, CYFlags flags) const { + bool protect(false); + if (false_ == NULL && (flags & CYNoDangle) != 0) { + protect = true; + out << '{'; } -} -void CYIf::Output(std::ostream &out) const { - out << "if("; - test_->Output(out, CYNoFlags); - out << ')'; - true_->Output(out, true); + out << "if" << ' ' << '(' << *test_ << ')'; + + CYFlags right(protect ? CYNoFlags : CYRight(flags)); + + CYFlags jacks(CYNoDangle); + if (false_ == NULL) + jacks |= right; + else + jacks |= protect ? CYNoFlags : CYCenter(flags); + + true_->Single(out, jacks); + if (false_ != NULL) { - out << "else "; - false_->Output(out, false); + out << '\t' << "else"; + false_->Single(out, right); } -} -void CYIfComprehension::Begin_(std::ostream &out) const { - out << "if("; - test_->Output(out, CYNoFlags); - out << ')'; + if (protect) + out << '}'; } -void CYIndirect::Output(std::ostream &out, CYFlags flags) const { - rhs_->Output(out, 1, CYLeft(flags)); - out << ".$cyi"; +void CYIfComprehension::Output(CYOutput &out) const { + out << "if" << ' ' << '(' << *test_ << ')' << next_; } -void CYIndirectMember::Output(std::ostream &out, CYFlags flags) const { +void CYIndirectMember::Output(CYOutput &out, CYFlags flags) const { object_->Output(out, Precedence(), CYLeft(flags)); - out << ".$cyi"; if (const char *word = property_->Word()) - out << '.' << word; - else { - out << '['; - property_->Output(out, CYNoFlags); - out << ']'; - } + out << "->" << word; + else + out << "->" << '[' << *property_ << ']'; } -void CYInfix::Output(std::ostream &out, CYFlags flags) const { +void CYInfix::Output(CYOutput &out, CYFlags flags) const { const char *name(Operator()); - bool protect((flags & CYNoIn) != 0 && strcmp(name, "in")); + bool protect((flags & CYNoIn) != 0 && strcmp(name, "in") == 0); if (protect) out << '('; - bool alphabetic(Alphabetic()); CYFlags left(protect ? CYNoFlags : CYLeft(flags)); - if (alphabetic) - left |= CYNoTrailer; lhs_->Output(out, Precedence(), left); - out << name; + out << ' ' << name << ' '; CYFlags right(protect ? CYNoFlags : CYRight(flags)); - if (alphabetic) - right |= CYNoLeader; - if (strcmp(name, "-") == 0) - right |= CYNoHyphen; rhs_->Output(out, Precedence() - 1, right); if (protect) out << ')'; } -void CYLambda::Output(std::ostream &out, CYFlags flags) const { - bool protect((flags & CYNoFunction) != 0); - if (protect) - out << '('; - else if ((flags & CYNoLeader) != 0) - out << ' '; - out << "function"; - if (name_ != NULL) - out << ' ' << *name_; - out << '('; - if (parameters_ != NULL) - out << *parameters_; - out << "){"; - if (body_ != NULL) - body_->Show(out); - out << '}'; - if (protect) - out << ')'; +void CYLabel::Output(CYOutput &out, CYFlags flags) const { + out << *name_ << ':' << ' '; + statement_->Single(out, CYRight(flags)); } -void CYLet::Output(std::ostream &out) const { - out << "let("; - declarations_->Output(out, CYNoFlags); - out << "){"; - if (statements_ != NULL) - statements_->Show(out); - out << "}"; +void CYLetStatement::Output(CYOutput &out, CYFlags flags) const { + out << "let" << ' ' << '(' << *declarations_ << ')'; + code_->Single(out, CYRight(flags)); } -void CYMessage::Output(std::ostream &out, bool replace) const { - if (next_ != NULL) - next_->Output(out, replace); - out << "$cyn=new Selector(\""; - for (CYMessageParameter *parameter(parameter_); parameter != NULL; parameter = parameter->next_) - if (parameter->tag_ != NULL) { - out << *parameter->tag_; - if (parameter->name_ != NULL) - out << ':'; - } - out << "\");"; - out << "$cyt=$cyn.type($cy" << (instance_ ? 's' : 'p') << ");"; - out << "class_" << (replace ? "replace" : "add") << "Method($cy" << (instance_ ? 'c' : 'm') << ",$cyn,"; - out << "new Functor(function(self,_cmd"; - for (CYMessageParameter *parameter(parameter_); parameter != NULL; parameter = parameter->next_) - if (parameter->name_ != NULL) - out << ',' << *parameter->name_; - out << "){return function(){"; - if (body_ != NULL) - body_->Show(out); - out << "}.call(self);},$cyt),$cyt);"; -} - -void CYNew::Output(std::ostream &out, CYFlags flags) const { - if ((flags & CYNoLeader) != 0) - out << ' '; - out << "new"; - constructor_->Output(out, Precedence(), CYCenter(flags) | CYNoLeader); - out << '('; +namespace cy { +namespace Syntax { + +void New::Output(CYOutput &out, CYFlags flags) const { + out << "new" << ' '; + CYFlags jacks(CYNoCall | CYCenter(flags)); + constructor_->Output(out, Precedence(), jacks); if (arguments_ != NULL) - arguments_->Output(out); - out << ')'; + out << '(' << *arguments_ << ')'; } -void CYNull::Output(std::ostream &out, CYFlags flags) const { - if ((flags & CYNoLeader) != 0) - out << ' '; +} } + +void CYNull::Output(CYOutput &out, CYFlags flags) const { CYWord::Output(out); - if ((flags & CYNoTrailer) != 0) - out << ' '; } -void CYNumber::Output(std::ostream &out, CYFlags flags) const { - double value(Value()); - if ((flags & CYNoLeader) != 0 || value < 0 && (flags & CYNoHyphen) != 0) - out << ' '; - // XXX: decide on correct precision - out << std::setprecision(9) << value; - if ((flags & CYNoTrailer) != 0) - out << ' '; +void CYNumber::Output(CYOutput &out, CYFlags flags) const { + std::ostringstream str; + CYNumerify(str, Value()); + std::string value(str.str()); + out << value.c_str(); + // XXX: this should probably also handle hex conversions and exponents + if ((flags & CYNoInteger) != 0 && value.find('.') == std::string::npos) + out << '.'; } -void CYNumber::PropertyName(std::ostream &out) const { - Output(out); +void CYNumber::PropertyName(CYOutput &out) const { + Output(out, CYNoFlags); } -void CYObject::Output(std::ostream &out, CYFlags flags) const { +void CYObject::Output(CYOutput &out, CYFlags flags) const { bool protect((flags & CYNoBrace) != 0); if (protect) out << '('; - out << '{'; - if (property_ != NULL) - property_->Output(out); - out << '}'; + out << '{' << '\n'; + ++out.indent_; + out << properties_; + --out.indent_; + out << '\t' << '}'; if (protect) out << ')'; } -void CYPostfix::Output(std::ostream &out, CYFlags flags) const { +void CYPostfix::Output(CYOutput &out, CYFlags flags) const { lhs_->Output(out, Precedence(), CYLeft(flags)); out << Operator(); } -void CYPrefix::Output(std::ostream &out, CYFlags flags) const { +void CYPrefix::Output(CYOutput &out, CYFlags flags) const { const char *name(Operator()); - bool alphabetic(Alphabetic()); - if (alphabetic && (flags & CYNoLeader) != 0 || name[0] == '-' && (flags & CYNoHyphen) != 0) - out << ' '; out << name; - CYFlags right(CYRight(flags)); - if (alphabetic) - right |= CYNoLeader; - rhs_->Output(out, Precedence(), right); + if (Alphabetic()) + out << ' '; + rhs_->Output(out, Precedence(), CYRight(flags)); +} + +void CYProgram::Output(CYOutput &out) const { + if (statements_ != NULL) + statements_->Multiple(out); } -void CYProperty::Output(std::ostream &out) const { +void CYProperty::Output(CYOutput &out) const { + out << '\t'; name_->PropertyName(out); - out << ':'; - value_->Output(out, CYPA, CYNoFlags); - if (next_ != NULL) { - out << ','; - next_->Output(out); - } + out << ':' << ' '; + value_->Output(out, CYAssign::Precedence_, CYNoFlags); + if (next_ != NULL) + out << ',' << '\n' << *next_; + else + out << '\n'; +} + +void CYRegEx::Output(CYOutput &out, CYFlags flags) const { + out << Value(); } -void CYReturn::Output(std::ostream &out) const { +void CYReturn::Output(CYOutput &out, CYFlags flags) const { out << "return"; if (value_ != NULL) - value_->Output(out, CYNoLeader); + out << ' ' << *value_; out << ';'; } -void CYSelector::Output(std::ostream &out, CYFlags flags) const { - if ((flags & CYNoLeader) != 0) - out << ' '; - out << "new Selector(\""; - if (name_ != NULL) - name_->Output(out); - out << "\")"; +void CYRubyBlock::Output(CYOutput &out, CYFlags flags) const { + call_->Output(out, CYLeft(flags)); + out << ' '; + proc_->Output(out, CYRight(flags)); } -void CYSelectorPart::Output(std::ostream &out) const { - if (name_ != NULL) - out << *name_; - if (value_) - out << ':'; - if (next_ != NULL) - next_->Output(out); +void CYRubyProc::Output(CYOutput &out, CYFlags flags) const { + // XXX: this is not outputting the parameters + out << code_; } -void CYSend::Output(std::ostream &out, CYFlags flags) const { - if ((flags & CYNoLeader) != 0) - out << ' '; - out << "objc_msgSend("; - self_->Output(out, CYPA, CYNoFlags); - out << ","; - std::ostringstream name; - for (CYArgument *argument(arguments_); argument != NULL; argument = argument->next_) - if (argument->name_ != NULL) { - name << *argument->name_; - if (argument->value_ != NULL) - name << ':'; - } - out << reinterpret_cast(sel_registerName(name.str().c_str())); - for (CYArgument *argument(arguments_); argument != NULL; argument = argument->next_) - if (argument->value_ != NULL) { - out << ","; - argument->value_->Output(out, CYPA, CYNoFlags); - } - out << ')'; +void CYStatement::Multiple(CYOutput &out, CYFlags flags) const { + bool first(true); + CYForEach (next, this) { + bool last(next->next_ == NULL); + CYFlags jacks(first ? last ? flags : CYLeft(flags) : last ? CYRight(flags) : CYCenter(flags)); + first = false; + out << '\t'; + next->Output(out, jacks); + out << '\n'; + } } -void CYSource::Show(std::ostream &out) const { - for (const CYSource *next(this); next != NULL; next = next->next_) - next->Output_(out); -} +void CYStatement::Single(CYOutput &out, CYFlags flags) const { + if (this == NULL) + return out.Terminate(); -void CYSource::Output(std::ostream &out, bool block) const { - if (!block && !IsBlock()) - Output(out); - else { - out << '{'; - Show(out); - out << '}'; - } + _assert(next_ == NULL); + out << '\n'; + ++out.indent_; + out << '\t'; + Output(out, flags); + out << '\n'; + --out.indent_; } -void CYSource::Output_(std::ostream &out) const { - Output(out); +void CYString::Output(CYOutput &out, CYFlags flags) const { + std::ostringstream str; + CYStringify(str, value_, size_); + out << str.str().c_str(); } -void CYStatement::Output_(std::ostream &out) const { - for (CYLabel *label(labels_); label != NULL; label = label->next_) - out << *label->name_ << ':'; - Output(out); +void CYString::PropertyName(CYOutput &out) const { + if (const char *word = Word()) + out << word; + else + out << *this; } -void CYString::Output(std::ostream &out, CYFlags flags) const { - unsigned quot(0), apos(0); - for (const char *value(value_), *end(value_ + size_); value != end; ++value) - if (*value == '"') - ++quot; - else if (*value == '\'') - ++apos; - - bool single(quot > apos); - - out << (single ? '\'' : '"'); - for (const char *value(value_), *end(value_ + size_); value != end; ++value) - switch (*value) { - case '\\': out << "\\\\"; break; - case '\b': out << "\\b"; break; - case '\f': out << "\\f"; break; - case '\n': out << "\\n"; break; - case '\r': out << "\\r"; break; - case '\t': out << "\\t"; break; - case '\v': out << "\\v"; break; - - case '"': - if (!single) - out << "\\\""; - else goto simple; - break; +static const char *Reserved_[] = { + "false", "null", "true", - case '\'': - if (single) - out << "\\'"; - else goto simple; - break; + "break", "case", "catch", "continue", "default", + "delete", "do", "else", "finally", "for", "function", + "if", "in", "instanceof", "new", "return", "switch", + "this", "throw", "try", "typeof", "var", "void", + "while", "with", - default: - if (*value < 0x20 || *value >= 0x7f) - out << "\\x" << std::setbase(16) << std::setw(2) << std::setfill('0') << unsigned(*value); - else simple: - out << *value; - } - out << (single ? '\'' : '"'); -} + "debugger", "const", -void CYString::PropertyName(std::ostream &out) const { - if (const char *word = Word()) - out << word; - else - Output(out); + "class", "enum", "export", "extends", "import", "super", + + "abstract", "boolean", "byte", "char", "double", "final", + "float", "goto", "int", "long", "native", "short", + "synchronized", "throws", "transient", "volatile", + + "let", "yield", + + NULL +}; + +const char *CYString::Word() const { + if (size_ == 0 || !WordStartRange_[value_[0]]) + return NULL; + for (size_t i(1); i != size_; ++i) + if (!WordEndRange_[value_[i]]) + return NULL; + const char *value(Value()); + for (const char **reserved(Reserved_); *reserved != NULL; ++reserved) + if (strcmp(*reserved, value) == 0) + return NULL; + return value; } -void CYSwitch::Output(std::ostream &out) const { - out << "switch("; - value_->Output(out, CYNoFlags); - out << "){"; - if (clauses_ != NULL) - out << *clauses_; +void CYSwitch::Output(CYOutput &out, CYFlags flags) const { + out << "switch" << ' ' << '(' << *value_ << ')' << ' ' << '{'; + out << clauses_; out << '}'; } -void CYThis::Output(std::ostream &out, CYFlags flags) const { - if ((flags & CYNoLeader) != 0) - out << ' '; +void CYThis::Output(CYOutput &out, CYFlags flags) const { CYWord::Output(out); - if ((flags & CYNoTrailer) != 0) - out << ' '; } -void CYThrow::Output(std::ostream &out) const { +namespace cy { +namespace Syntax { + +void Throw::Output(CYOutput &out, CYFlags flags) const { out << "throw"; if (value_ != NULL) - value_->Output(out, CYNoLeader); + out << ' ' << *value_; out << ';'; } -void CYTry::Output(std::ostream &out) const { - out << "try"; - try_->Output(out, true); - if (catch_ != NULL) - catch_->Output(out); - if (finally_ != NULL) { - out << "finally"; - finally_->Output(out, true); - } +void Try::Output(CYOutput &out, CYFlags flags) const { + out << "try" << ' ' << code_ << catch_ << finally_; } -void CYVar::Output(std::ostream &out) const { - out << "var "; - declarations_->Output(out, CYNoFlags); +} } + +void CYVar::Output(CYOutput &out, CYFlags flags) const { + out << "var"; + declarations_->Output(out, flags); out << ';'; } -void CYVariable::Output(std::ostream &out, CYFlags flags) const { - if ((flags & CYNoLeader) != 0) - out << ' '; +void CYVariable::Output(CYOutput &out, CYFlags flags) const { out << *name_; - if ((flags & CYNoTrailer) != 0) - out << ' '; } -void CYWhile::Output(std::ostream &out) const { - out << "while("; - test_->Output(out, CYNoFlags); - out << ')'; - code_->Output(out, false); +void CYWhile::Output(CYOutput &out, CYFlags flags) const { + out << "while" << '(' << *test_ << ')'; + code_->Single(out, CYRight(flags)); } -void CYWith::Output(std::ostream &out) const { - out << "with("; - scope_->Output(out, CYNoFlags); - out << ')'; - code_->Output(out, false); +void CYWith::Output(CYOutput &out, CYFlags flags) const { + out << "with" << '(' << *scope_ << ')'; + code_->Single(out, CYRight(flags)); } -void CYWord::ClassName(std::ostream &out, bool object) const { +void CYWord::ClassName(CYOutput &out, bool object) const { if (object) out << "objc_getClass("; - out << '"' << Value() << '"'; + out << '"' << Word() << '"'; if (object) out << ')'; } -void CYWord::Output(std::ostream &out) const { - out << Value(); +void CYWord::Output(CYOutput &out) const { + out << Word(); + if (out.options_.verbose_) + out.out_ << '@' << this; } -void CYWord::PropertyName(std::ostream &out) const { +void CYWord::PropertyName(CYOutput &out) const { Output(out); } + +const char *CYWord::Word() const { + return word_; +}