ObjectiveC/Replace.mm

   1 /* Cycript - Inlining/Optimizing JavaScript Compiler
   2  * Copyright (C) 2009  Jay Freeman (saurik)
   3 */
   4
   5 /* Modified BSD License {{{ */
   6 /*
   7  *        Redistribution and use in source and binary
   8  * forms, with or without modification, are permitted
   9  * provided that the following conditions are met:
  10  *
  11  * 1. Redistributions of source code must retain the
  12  *    above copyright notice, this list of conditions
  13  *    and the following disclaimer.
  14  * 2. Redistributions in binary form must reproduce the
  15  *    above copyright notice, this list of conditions
  16  *    and the following disclaimer in the documentation
  17  *    and/or other materials provided with the
  18  *    distribution.
  19  * 3. The name of the author may not be used to endorse
  20  *    or promote products derived from this software
  21  *    without specific prior written permission.
  22  *
  23  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS''
  24  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,
  25  * BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
  26  * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  27  * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE
  28  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  29  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  30  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
  31  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  32  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  33  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
  34  * TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
  35  * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
  36  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  37 */
  38 /* }}} */
  39
  40 #include "Replace.hpp"
  41 #include "ObjectiveC/Syntax.hpp"
  42
  43 #include <Foundation/Foundation.h>
  44 #include <sstream>
  45
  46 CYStatement *CYCategory::Replace(CYContext &context) {
  47     CYVariable *cyc($V("$cyc")), *cys($V("$cys"));
  48
  49     return $E($C1($F(NULL, $P5("$cys", "$cyp", "$cyc", "$cyn", "$cyt"), $$->*
  50         $E($ CYAssign($V("$cyp"), $C1($V("object_getClass"), cys)))->*
  51         $E($ CYAssign(cyc, cys))->*
  52         $E($ CYAssign($V("$cym"), $C1($V("object_getClass"), cyc)))->*
  53         messages_->Replace(context, true)
  54     ), name_->ClassName(context, true)));
  55 }
  56
  57 CYExpression *CYClass::Replace_(CYContext &context) {
  58     CYVariable *cyc($V("$cyc")), *cys($V("$cys"));
  59
  60     CYExpression *name(name_ != NULL ? name_->ClassName(context, false) : $C1($V("$cyq"), $S("CY$")));
  61
  62     return $C1($F(NULL, $P6("$cys", "$cyp", "$cyc", "$cyn", "$cyt", "$cym"), $$->*
  63         $E($ CYAssign($V("$cyp"), $C1($V("object_getClass"), cys)))->*
  64         $E($ CYAssign(cyc, $C3($V("objc_allocateClassPair"), cys, name, $D(0))))->*
  65         $E($ CYAssign($V("$cym"), $C1($V("object_getClass"), cyc)))->*
  66         protocols_->Replace(context)->*
  67         fields_->Replace(context)->*
  68         messages_->Replace(context, false)->*
  69         $E($C1($V("objc_registerClassPair"), cyc))->*
  70         $ CYReturn(cyc)
  71     ), super_ == NULL ? $ CYNull() : super_);
  72 }
  73
  74 CYExpression *CYClassExpression::Replace(CYContext &context) {
  75     return Replace_(context);
  76 }
  77
  78 CYStatement *CYClassStatement::Replace(CYContext &context) {
  79     return $E(Replace_(context));
  80 }
  81
  82 CYStatement *CYField::Replace(CYContext &context) const {
  83     return NULL;
  84 }
  85
  86 CYStatement *CYImport::Replace(CYContext &context) {
  87     return this;
  88 }
  89
  90 CYStatement *CYMessage::Replace(CYContext &context, bool replace) const { $T(NULL)
  91     CYVariable *cyn($V("$cyn"));
  92     CYVariable *cyt($V("$cyt"));
  93     CYVariable *self($V("self"));
  94     CYVariable *_class($V(instance_ ? "$cys" : "$cyp"));
  95
  96     return $ CYBlock($$->*
  97         next_->Replace(context, replace)->*
  98         $E($ CYAssign(cyn, parameters_->Selector(context)))->*
  99         $E($ CYAssign(cyt, $C1($M(cyn, $S("type")), _class)))->*
 100         $E($C4($V(replace ? "class_replaceMethod" : "class_addMethod"),
 101             $V(instance_ ? "$cyc" : "$cym"),
 102             cyn,
 103             $N2($V("Functor"), $F(NULL, $P2("self", "_cmd", parameters_->Parameters(context)), $$->*
 104                 $ CYVar($L1($L($I("$cyr"), $N2($V("Super"), self, _class))))->*
 105                 $ CYReturn($C1($M($F(NULL, NULL, code_), $S("call")), self))
 106             ), cyt),
 107             cyt
 108         ))
 109     );
 110 }
 111
 112 CYFunctionParameter *CYMessageParameter::Parameters(CYContext &context) const { $T(NULL)
 113     CYFunctionParameter *next(next_->Parameters(context));
 114     return name_ == NULL ? next : $ CYFunctionParameter(name_, next);
 115 }
 116
 117 CYSelector *CYMessageParameter::Selector(CYContext &context) const {
 118     return $ CYSelector(SelectorPart(context));
 119 }
 120
 121 CYSelectorPart *CYMessageParameter::SelectorPart(CYContext &context) const { $T(NULL)
 122     CYSelectorPart *next(next_->SelectorPart(context));
 123     return tag_ == NULL ? next : $ CYSelectorPart(tag_, name_ != NULL, next);
 124 }
 125
 126 CYStatement *CYProtocol::Replace(CYContext &context) const { $T(NULL)
 127     return $ CYBlock($$->*
 128         next_->Replace(context)->*
 129         $E($C2($V("class_addProtocol"),
 130             $V("$cyc"), name_
 131         ))
 132     );
 133 }
 134
 135 CYExpression *CYSelector::Replace(CYContext &context) {
 136     return $N1($V("Selector"), name_->Replace(context));
 137 }
 138
 139 CYString *CYSelectorPart::Replace(CYContext &context) {
 140     std::ostringstream str;
 141     for (const CYSelectorPart *part(this); part != NULL; part = part->next_) {
 142         if (part->name_ != NULL)
 143             str << part->name_->Word();
 144         if (part->value_)
 145             str << ':';
 146     }
 147     return $S(apr_pstrdup(context.pool_, str.str().c_str()));
 148 }
 149
 150 CYExpression *CYSendDirect::Replace(CYContext &context) {
 151     std::ostringstream name;
 152     CYArgument **argument(&arguments_);
 153     CYSelectorPart *selector(NULL), *current(NULL);
 154
 155     while (*argument != NULL) {
 156         if ((*argument)->name_ != NULL) {
 157             CYSelectorPart *part($ CYSelectorPart((*argument)->name_, (*argument)->value_ != NULL));
 158             if (selector == NULL)
 159                 selector = part;
 160             if (current != NULL)
 161                 current->SetNext(part);
 162             current = part;
 163             (*argument)->name_ = NULL;
 164         }
 165
 166         if ((*argument)->value_ == NULL)
 167             *argument = (*argument)->next_;
 168         else
 169             argument = &(*argument)->next_;
 170     }
 171
 172     return $C2($V("objc_msgSend"), self_, ($ CYSelector(selector))->Replace(context), arguments_);
 173 }
 174
 175 CYExpression *CYSendSuper::Replace(CYContext &context) {
 176     return $ CYSendDirect($V("$cyr"), arguments_);
 177 }