+/* 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);
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_);
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);
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()
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 {
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) {
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);
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);
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;
projects / cycript.git / blobdiff