X-Git-Url: https://git.saurik.com/cycript.git/blobdiff_plain/3b52fd1a20c639d7a716d3e0f3176a55f2fcba72..b64ab4dac5a76da209ff3cb12aa8bf83c6142607:/Replace.cpp

diff --git a/Replace.cpp b/Replace.cpp
index 7d861c9..b41aac1 100644
--- a/Replace.cpp
+++ b/Replace.cpp
@@ -1,95 +1,77 @@
+/* Cycript - Inlining/Optimizing JavaScript Compiler
+ * Copyright (C) 2009  Jay Freeman (saurik)
+*/
+
+/* Modified BSD License {{{ */
+/*
+ *        Redistribution and use in source and binary
+ * forms, with or without modification, are permitted
+ * provided that the following conditions are met:
+ *
+ * 1. Redistributions of source code must retain the
+ *    above copyright notice, this list of conditions
+ *    and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the
+ *    above copyright notice, this list of conditions
+ *    and the following disclaimer in the documentation
+ *    and/or other materials provided with the
+ *    distribution.
+ * 3. The name of the author may not be used to endorse
+ *    or promote products derived from this software
+ *    without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS''
+ * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,
+ * BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE
+ * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
+ * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
+ * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+ * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+ * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
+ * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
+ * TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
+ * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
+ * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+*/
+/* }}} */
+
 #include "Parser.hpp"
 
-#include <iostream>
 #include <iomanip>
 
-#include <objc/runtime.h>
-#include <sstream>
-
-#define $ \
-    new(context.pool_)
-
-#define $D(args...) \
-    ($ CYNumber(args))
-#define $E(args...) \
-    ($ CYExpress(args))
-#define $F(args...) \
-    ($ CYFunctionExpression(args))
-#define $I(args...) \
-    ($ CYIdentifier(args))
-#define $M(args...) \
-    ($ CYDirectMember(args))
-#define $P(args...) \
-    ($ CYFunctionParameter(args))
-#define $S(args...) \
-    ($ CYString(args))
-#define $V(name) \
-    ($ CYVariable($I(name)))
-
-#define $T(value) \
-    if (this == NULL) \
-        return value;
-#define $$ \
-    CYStatements()
-
-#define $P1(arg0, args...) \
-    $P($I(arg0), ##args)
-#define $P2(arg0, arg1, args...) \
-    $P($I(arg0), $P1(arg1, ##args))
-#define $P3(arg0, arg1, arg2, args...) \
-    $P($I(arg0), $P2(arg1, arg2, ##args))
-#define $P4(arg0, arg1, arg2, arg3, args...) \
-    $P($I(arg0), $P3(arg1, arg2, arg3, ##args))
-#define $P5(arg0, arg1, arg2, arg3, arg4, args...) \
-    $P($I(arg0), $P4(arg1, arg2, arg3, arg4, ##args))
-#define $P6(arg0, arg1, arg2, arg3, arg4, arg5, args...) \
-    $P($I(arg0), $P5(arg1, arg2, arg3, arg4, arg5, ##args))
-
-#define $C(args...) \
-    ($ CYCall(args))
-#define $C_(args...) \
-    ($ CYArgument(args))
-#define $N(args...) \
-    ($ CYNew(args))
-
-#define $C1_(arg0, args...) \
-    $C_(arg0, ##args)
-#define $C2_(arg0, arg1, args...) \
-    $C_(arg0, $C1_(arg1, ##args))
-#define $C3_(arg0, arg1, arg2, args...) \
-    $C_(arg0, $C2_(arg1, arg2, ##args))
-#define $C4_(arg0, arg1, arg2, arg3, args...) \
-    $C_(arg0, $C3_(arg1, arg2, arg3, ##args))
-#define $C5_(arg0, arg1, arg2, arg3, arg4, args...) \
-    $C_(arg0, $C4_(arg1, arg2, arg3, arg4, ##args))
-#define $C6_(arg0, arg1, arg2, arg3, arg4, arg5, args...) \
-    $C_(arg0, $C5_(arg1, arg2, arg3, arg4, arg5, ##args))
-
-#define $C0(func, args...) \
-    $C(func, ##args)
-#define $C1(func, args...) \
-    $C(func, $C1_(args))
-#define $C2(func, args...) \
-    $C(func, $C2_(args))
-#define $C3(func, args...) \
-    $C(func, $C3_(args))
-#define $C4(func, args...) \
-    $C(func, $C4_(args))
-#define $C5(func, args...) \
-    $C(func, $C5_(args))
-
-#define $N0(func, args...) \
-    $N(func, ##args)
-#define $N1(func, args...) \
-    $N(func, $C1_(args))
-#define $N2(func, args...) \
-    $N(func, $C2_(args))
-#define $N3(func, args...) \
-    $N(func, $C3_(args))
-#define $N4(func, args...) \
-    $N(func, $C4_(args))
-#define $N5(func, args...) \
-    $N(func, $C5_(args))
+#include "Replace.hpp"
+
+CYExpression *CYAdd::Replace(CYContext &context) {
+    CYInfix::Replace(context);
+
+    CYExpression *lhp(lhs_->Primitive(context));
+    CYExpression *rhp(rhs_->Primitive(context));
+
+    CYString *lhs(dynamic_cast<CYString *>(lhp));
+    CYString *rhs(dynamic_cast<CYString *>(rhp));
+
+    if (lhs != NULL || rhs != NULL) {
+        if (lhs == NULL) {
+            lhs = lhp->String(context);
+            if (lhs == NULL)
+                return NULL;
+        } else if (rhs == NULL) {
+            rhs = rhp->String(context);
+            if (rhs == NULL)
+                return NULL;
+        }
+
+        return lhs->Concat(context, rhs);
+    }
+
+    if (CYNumber *lhn = lhp->Number(context))
+        if (CYNumber *rhn = rhp->Number(context))
+            return $D(lhn->Value() + rhn->Value());
+
+    return NULL;
+}
 
 CYExpression *CYAddressOf::Replace(CYContext &context) {
     CYPrefix::Replace(context);
@@ -137,43 +119,14 @@ CYExpression *CYCall::Replace(CYContext &context) {
     return NULL;
 }
 
-void CYCatch::Replace(CYContext &context) { $T()
-    code_.Replace(context);
-}
-
-CYStatement *CYCategory::Replace(CYContext &context) {
-    CYVariable *cyc($V("$cyc")), *cys($V("$cys"));
-
-    return $E($C1($F(NULL, $P5("$cys", "$cyp", "$cyc", "$cyn", "$cyt"), $$->*
-        $E($ CYAssign($V("$cyp"), $C1($V("object_getClass"), cys)))->*
-        $E($ CYAssign(cyc, cys))->*
-        messages_->Replace(context, true)
-    ), name_->ClassName(context, true)));
-}
-
-CYExpression *CYClass::Replace_(CYContext &context) {
-    CYVariable *cyc($V("$cyc")), *cys($V("$cys"));
-
-    CYExpression *name(name_ != NULL ? name_->ClassName(context, false) : $C1($V("$cyq"), $S("CY$")));
+namespace cy {
+namespace Syntax {
 
-    return $C1($F(NULL, $P6("$cys", "$cyp", "$cyc", "$cyn", "$cyt", "$cym"), $$->*
-        $E($ CYAssign($V("$cyp"), $C1($V("object_getClass"), cys)))->*
-        $E($ CYAssign(cyc, $C3($V("objc_allocateClassPair"), cys, name, $D(0))))->*
-        $E($ CYAssign($V("$cym"), $C1($V("object_getClass"), cyc)))->*
-        fields_->Replace(context)->*
-        messages_->Replace(context, false)->*
-        $E($C1($V("objc_registerClassPair"), cyc))->*
-        $ CYReturn(cyc)
-    ), super_ == NULL ? $ CYNull() : super_);
+void Catch::Replace(CYContext &context) { $T()
+    code_.Replace(context);
 }
 
-CYExpression *CYClassExpression::Replace(CYContext &context) {
-    return Replace_(context);
-}
-
-CYStatement *CYClassStatement::Replace(CYContext &context) {
-    return $E(Replace_(context));
-}
+} }
 
 void CYClause::Replace(CYContext &context) { $T()
     context.Replace(case_);
@@ -218,6 +171,10 @@ void CYDeclaration::Replace(CYContext &context) {
     context.Replace(initialiser_);
 }
 
+CYProperty *CYDeclarations::Property(CYContext &context) { $T(NULL)
+    return $ CYProperty(declaration_->identifier_, declaration_->initialiser_ ?: $U, next_->Property(context));
+}
+
 void CYDeclarations::Replace(CYContext &context) { $T()
     declaration_->Replace(context);
     next_->Replace(context);
@@ -268,8 +225,12 @@ CYExpression *CYExpression::ReplaceAll(CYContext &context) { $T(NULL)
     return replace;
 }
 
-CYStatement *CYField::Replace(CYContext &context) const {
-    return NULL;
+CYNumber *CYFalse::Number(CYContext &context) {
+    return $D(0);
+}
+
+CYString *CYFalse::String(CYContext &context) {
+    return $S("false");
 }
 
 void CYFinally::Replace(CYContext &context) { $T()
@@ -302,13 +263,12 @@ CYStatement *CYForInComprehension::Replace(CYContext &context, CYStatement *stat
 CYStatement *CYForEachIn::Replace(CYContext &context) {
     CYVariable *cys($V("$cys")), *cyt($V("$cyt"));
 
-    return $ CYWith($ CYObject($ CYProperty($S("$cys"), $D(0), $ CYProperty($S("$cyt"), $D(0)))), $ CYBlock($$->*
-        $E($ CYAssign(cys, set_))->*
+    return $ CYLet($L2($L($I("$cys"), set_), $L($I("$cyt"))), $$->*
         $ CYForIn(cyt, cys, $ CYBlock($$->*
             $E($ CYAssign(initialiser_->ForEachIn(context), $M(cys, cyt)))->*
             code_
         ))
-    ));
+    );
 }
 
 CYFunctionParameter *CYForEachInComprehension::Parameter(CYContext &context) const {
@@ -377,48 +337,36 @@ CYStatement *CYLabel::Replace(CYContext &context) {
     return NULL;
 }
 
+CYStatement *CYLet::Replace(CYContext &context) {
+    return $ CYWith($ CYObject(declarations_->Property(context)), &code_);
+}
+
 void CYMember::Replace_(CYContext &context) {
     context.Replace(object_);
     context.Replace(property_);
 }
 
-CYStatement *CYMessage::Replace(CYContext &context, bool replace) const { $T(NULL)
-    CYVariable *cyn($V("$cyn"));
-    CYVariable *cyt($V("$cyt"));
-
-    return $ CYBlock($$->*
-        next_->Replace(context, replace)->*
-        $E($ CYAssign(cyn, parameters_->Selector(context)))->*
-        $E($ CYAssign(cyt, $C1($M(cyn, $S("type")), $V(instance_ ? "$cys" : "$cyp"))))->*
-        $E($C4($V(replace ? "class_replaceMethod" : "class_addMethod"),
-            $V(instance_ ? "$cyc" : "$cym"),
-            cyn,
-            $N2($V("Functor"), $F(NULL, $P2("self", "_cmd", parameters_->Parameters(context)), $$->*
-                $ CYReturn($C1($M($F(NULL, NULL, statements_), $S("call")), $V("self")))
-            ), cyt),
-            cyt
-        ))
-    );
+CYExpression *CYNew::Replace(CYContext &context) {
+    context.Replace(constructor_);
+    arguments_->Replace(context);
+    return NULL;
 }
 
-CYFunctionParameter *CYMessageParameter::Parameters(CYContext &context) const { $T(NULL)
-    CYFunctionParameter *next(next_->Parameters(context));
-    return name_ == NULL ? next : $ CYFunctionParameter(name_, next);
+CYNumber *CYNull::Number(CYContext &context) {
+    return $D(0);
 }
 
-CYSelector *CYMessageParameter::Selector(CYContext &context) const {
-    return $ CYSelector(SelectorPart(context));
+CYString *CYNull::String(CYContext &context) {
+    return $S("null");
 }
 
-CYSelectorPart *CYMessageParameter::SelectorPart(CYContext &context) const { $T(NULL)
-    CYSelectorPart *next(next_->SelectorPart(context));
-    return tag_ == NULL ? next : $ CYSelectorPart(tag_, name_ != NULL, next);
+CYNumber *CYNumber::Number(CYContext &context) {
+    return this;
 }
 
-CYExpression *CYNew::Replace(CYContext &context) {
-    context.Replace(constructor_);
-    arguments_->Replace(context);
-    return NULL;
+CYString *CYNumber::String(CYContext &context) {
+    // XXX: there is a precise algorithm for this
+    return $S(apr_psprintf(context.pool_, "%.17g", Value()));
 }
 
 CYExpression *CYObject::Replace(CYContext &context) {
@@ -450,45 +398,6 @@ CYStatement *CYReturn::Replace(CYContext &context) {
     return NULL;
 }
 
-CYExpression *CYSelector::Replace(CYContext &context) {
-    return $N1($V("Selector"), name_->Replace(context));
-}
-
-CYExpression *CYSend::Replace(CYContext &context) {
-    std::ostringstream name;
-    CYArgument **argument(&arguments_);
-
-    while (*argument != NULL) {
-        if ((*argument)->name_ != NULL) {
-            name << *(*argument)->name_;
-            (*argument)->name_ = NULL;
-            if ((*argument)->value_ != NULL)
-                name << ':';
-        }
-
-        if ((*argument)->value_ == NULL)
-            *argument = (*argument)->next_;
-        else
-            argument = &(*argument)->next_;
-    }
-
-    SEL sel(sel_registerName(name.str().c_str()));
-    double address(static_cast<double>(reinterpret_cast<uintptr_t>(sel)));
-
-    return $C2($V("objc_msgSend"), self_, $D(address), arguments_);
-}
-
-CYString *CYSelectorPart::Replace(CYContext &context) {
-    std::ostringstream str;
-    for (const CYSelectorPart *part(this); part != NULL; part = part->next_) {
-        if (part->name_ != NULL)
-            str << part->name_->Value();
-        if (part->value_)
-            str << ':';
-    }
-    return $S(apr_pstrdup(context.pool_, str.str().c_str()));
-}
-
 CYStatement *CYStatement::ReplaceAll(CYContext &context) { $T(NULL)
     CYStatement *replace(this);
     context.Replace(replace);
@@ -501,6 +410,24 @@ CYStatement *CYStatement::ReplaceAll(CYContext &context) { $T(NULL)
     return replace;
 }
 
+CYString *CYString::Concat(CYContext &context, CYString *rhs) const {
+    size_t size(size_ + rhs->size_);
+    char *value(new(context.pool_) char[size + 1]);
+    memcpy(value, value_, size_);
+    memcpy(value + size_, rhs->value_, rhs->size_);
+    value[size] = '\0';
+    return $S(value, size);
+}
+
+CYNumber *CYString::Number(CYContext &context) {
+    // XXX: there is a precise algorithm for this
+    return NULL;
+}
+
+CYString *CYString::String(CYContext &context) {
+    return this;
+}
+
 CYStatement *CYSwitch::Replace(CYContext &context) {
     context.Replace(value_);
     clauses_->Replace(context);
@@ -511,22 +438,40 @@ CYExpression *CYThis::Replace(CYContext &context) {
     return NULL;
 }
 
-CYStatement *CYThrow::Replace(CYContext &context) {
+namespace cy {
+namespace Syntax {
+
+CYStatement *Throw::Replace(CYContext &context) {
     context.Replace(value_);
     return NULL;
 }
 
+} }
+
 CYExpression *CYTrivial::Replace(CYContext &context) {
     return NULL;
 }
 
-CYStatement *CYTry::Replace(CYContext &context) {
+CYNumber *CYTrue::Number(CYContext &context) {
+    return $D(1);
+}
+
+CYString *CYTrue::String(CYContext &context) {
+    return $S("true");
+}
+
+namespace cy {
+namespace Syntax {
+
+CYStatement *Try::Replace(CYContext &context) {
     code_.Replace(context);
     catch_->Replace(context);
     finally_->Replace(context);
     return NULL;
 }
 
+} }
+
 CYStatement *CYVar::Replace(CYContext &context) {
     declarations_->Replace(context);
     return NULL;