ObjectiveC/Output.mm

   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 "Replace.hpp"
  23 #include "ObjectiveC/Syntax.hpp"
  24
  25 #include <Foundation/Foundation.h>
  26 #include <sstream>
  27
  28 void CYCategory::Output(CYOutput &out, CYFlags flags) const {
  29     out << "(function($cys,$cyp,$cyc,$cyn,$cyt){";
  30     out << "$cyp=object_getClass($cys);";
  31     out << "$cyc=$cys;";
  32     if (messages_ != NULL)
  33         messages_->Output(out, true);
  34     out << "})(";
  35     name_->ClassName(out, true);
  36     out << ')';
  37     out << ';';
  38 }
  39
  40 void CYClass::Output(CYOutput &out, CYFlags flags) const {
  41     // XXX: I don't necc. need the ()s
  42     out << "(function($cys,$cyp,$cyc,$cyn,$cyt,$cym){";
  43     out << "$cyp=object_getClass($cys);";
  44     out << "$cyc=objc_allocateClassPair($cys,";
  45     if (name_ != NULL)
  46         name_->ClassName(out, false);
  47     else
  48         out << "$cyq(\"CY$\")";
  49     out << ",0);";
  50     out << "$cym=object_getClass($cyc);";
  51     if (fields_ != NULL)
  52         fields_->Output(out);
  53     if (messages_ != NULL)
  54         messages_->Output(out, false);
  55     if (protocols_ != NULL) {
  56         out << '<';
  57         out << *protocols_;
  58         out << '>';
  59     }
  60     out << "objc_registerClassPair($cyc);";
  61     out << "return $cyc;";
  62     out << "}(";
  63     if (super_ != NULL)
  64         super_->Output(out, CYAssign::Precedence_, CYNoFlags);
  65     else
  66         out << "null";
  67     out << "))";
  68 }
  69
  70 void CYClassExpression::Output(CYOutput &out, CYFlags flags) const {
  71     CYClass::Output(out, flags);
  72 }
  73
  74 void CYClassStatement::Output(CYOutput &out, CYFlags flags) const {
  75     CYClass::Output(out, flags);
  76 }
  77
  78 void CYField::Output(CYOutput &out) const {
  79 }
  80
  81 void CYImport::Output(CYOutput &out, CYFlags flags) const {
  82     out << "@import";
  83 }
  84
  85 void CYMessage::Output(CYOutput &out, bool replace) const {
  86     out << (instance_ ? '-' : '+');
  87
  88     CYForEach (parameter, parameters_)
  89         if (parameter->tag_ != NULL) {
  90             out << ' ' << *parameter->tag_;
  91             if (parameter->name_ != NULL)
  92                 out << ':' << *parameter->name_;
  93         }
  94
  95     out << code_;
  96 }
  97
  98 void CYProtocol::Output(CYOutput &out) const {
  99     name_->Output(out, CYAssign::Precedence_, CYNoFlags);
 100     if (next_ != NULL)
 101         out << ',' << ' ' << *next_;
 102 }
 103
 104 void CYSelector::Output(CYOutput &out, CYFlags flags) const {
 105     out << "@selector" << '(' << name_ << ')';
 106 }
 107
 108 void CYSelectorPart::Output(CYOutput &out) const {
 109     out << name_;
 110     if (value_)
 111         out << ':';
 112     out << next_;
 113 }
 114
 115 void CYSend::Output(CYOutput &out, CYFlags flags) const {
 116     CYForEach (argument, arguments_)
 117         if (argument->name_ != NULL) {
 118             out << ' ' << *argument->name_;
 119             if (argument->value_ != NULL)
 120                 out << ':' << *argument->value_;
 121         }
 122 }
 123
 124 void CYSendDirect::Output(CYOutput &out, CYFlags flags) const {
 125     out << '[';
 126     self_->Output(out, CYAssign::Precedence_, CYNoFlags);
 127     CYSend::Output(out, flags);
 128     out << ']';
 129 }
 130
 131 void CYSendSuper::Output(CYOutput &out, CYFlags flags) const {
 132     out << '[' << "super";
 133     CYSend::Output(out, flags);
 134     out << ']';
 135 }