]>
Commit | Line | Data |
---|---|---|
b3378a02 | 1 | /* Cycript - Optimizing JavaScript Compiler/Runtime |
c15969fd | 2 | * Copyright (C) 2009-2013 Jay Freeman (saurik) |
b53b30c1 JF |
3 | */ |
4 | ||
c15969fd | 5 | /* GNU General Public License, Version 3 {{{ */ |
b53b30c1 | 6 | /* |
c15969fd JF |
7 | * Cycript is free software: you can redistribute it and/or modify |
8 | * it under the terms of the GNU General Public License as published | |
9 | * by the Free Software Foundation, either version 3 of the License, | |
10 | * or (at your option) any later version. | |
b53b30c1 | 11 | * |
c15969fd JF |
12 | * Cycript is distributed in the hope that it will be useful, but |
13 | * WITHOUT ANY WARRANTY; without even the implied warranty of | |
14 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
15 | * GNU General Public License for more details. | |
b53b30c1 | 16 | * |
c15969fd | 17 | * You should have received a copy of the GNU General Public License |
b3378a02 JF |
18 | * along with Cycript. If not, see <http://www.gnu.org/licenses/>. |
19 | **/ | |
b53b30c1 JF |
20 | /* }}} */ |
21 | ||
4de0686f | 22 | #include "Replace.hpp" |
3c1c3635 | 23 | #include "ObjectiveC/Syntax.hpp" |
4de0686f | 24 | |
4de0686f JF |
25 | #include <sstream> |
26 | ||
9a7c375c JF |
27 | static CYExpression *MessageType(CYContext &context, CYExpression *type, CYMessageParameter *next, CYExpression *extra = NULL) { |
28 | if (type == NULL) | |
29 | return NULL; | |
30 | ||
31 | CYExpression *left($C0($M(type, $S("toString")))); | |
32 | if (extra != NULL) | |
33 | left = $ CYAdd(left, extra); | |
34 | ||
cfd73c6d | 35 | if (next == NULL || next->name_ == NULL) |
9a7c375c JF |
36 | return left; |
37 | ||
38 | CYExpression *right(next->TypeSignature(context)); | |
39 | if (right == NULL) | |
40 | return NULL; | |
41 | ||
42 | return $ CYAdd(left, right); | |
43 | } | |
44 | ||
4de0686f JF |
45 | CYStatement *CYCategory::Replace(CYContext &context) { |
46 | CYVariable *cyc($V("$cyc")), *cys($V("$cys")); | |
47 | ||
b639f46d | 48 | return $E($C1($F(NULL, $P6($L("$cys"), $L("$cyp"), $L("$cyc"), $L("$cyn"), $L("$cyt"), $L("$cym")), $$->* |
4de0686f JF |
49 | $E($ CYAssign($V("$cyp"), $C1($V("object_getClass"), cys)))->* |
50 | $E($ CYAssign(cyc, cys))->* | |
725075ae | 51 | $E($ CYAssign($V("$cym"), $C1($V("object_getClass"), cyc)))->* |
4de0686f JF |
52 | messages_->Replace(context, true) |
53 | ), name_->ClassName(context, true))); | |
54 | } | |
55 | ||
56 | CYExpression *CYClass::Replace_(CYContext &context) { | |
57 | CYVariable *cyc($V("$cyc")), *cys($V("$cys")); | |
58 | ||
59 | CYExpression *name(name_ != NULL ? name_->ClassName(context, false) : $C1($V("$cyq"), $S("CY$"))); | |
60 | ||
c8a0500b | 61 | return $C1($F(NULL, $P6($L("$cys"), $L("$cyp"), $L("$cyc"), $L("$cyn"), $L("$cyt"), $L("$cym")), $$->* |
4de0686f JF |
62 | $E($ CYAssign($V("$cyp"), $C1($V("object_getClass"), cys)))->* |
63 | $E($ CYAssign(cyc, $C3($V("objc_allocateClassPair"), cys, name, $D(0))))->* | |
64 | $E($ CYAssign($V("$cym"), $C1($V("object_getClass"), cyc)))->* | |
64b8d29f | 65 | protocols_->Replace(context)->* |
4de0686f JF |
66 | fields_->Replace(context)->* |
67 | messages_->Replace(context, false)->* | |
68 | $E($C1($V("objc_registerClassPair"), cyc))->* | |
69 | $ CYReturn(cyc) | |
70 | ), super_ == NULL ? $ CYNull() : super_); | |
71 | } | |
72 | ||
73 | CYExpression *CYClassExpression::Replace(CYContext &context) { | |
74 | return Replace_(context); | |
75 | } | |
76 | ||
77 | CYStatement *CYClassStatement::Replace(CYContext &context) { | |
78 | return $E(Replace_(context)); | |
79 | } | |
80 | ||
cfd73c6d JF |
81 | CYStatement *CYField::Replace(CYContext &context) const { $T(NULL) |
82 | CYVariable *cyn($V("$cyn")); | |
83 | CYVariable *cyt($V("$cyt")); | |
84 | ||
85 | CYExpression *type($C0($M(type_, $S("toString")))); | |
86 | ||
87 | return $ CYBlock($$->* | |
88 | next_->Replace(context)->* | |
89 | $E($ CYAssign(cyt, type))->* | |
90 | $E($ CYAssign(cyn, $N1($V("Type"), cyt)))->* | |
91 | $E($C5($V("class_addIvar"), $V("$cyc"), $S(name_->Word()), $M(cyn, $S("size")), $M(cyn, $S("alignment")), cyt)) | |
92 | ); | |
4de0686f JF |
93 | } |
94 | ||
1ba6903e JF |
95 | CYStatement *CYImport::Replace(CYContext &context) { |
96 | return this; | |
97 | } | |
98 | ||
4de0686f JF |
99 | CYStatement *CYMessage::Replace(CYContext &context, bool replace) const { $T(NULL) |
100 | CYVariable *cyn($V("$cyn")); | |
101 | CYVariable *cyt($V("$cyt")); | |
cacd1a88 JF |
102 | CYVariable *self($V("self")); |
103 | CYVariable *_class($V(instance_ ? "$cys" : "$cyp")); | |
4de0686f | 104 | |
9a7c375c JF |
105 | CYExpression *type(TypeSignature(context) ?: $C1($M(cyn, $S("type")), _class)); |
106 | ||
4de0686f JF |
107 | return $ CYBlock($$->* |
108 | next_->Replace(context, replace)->* | |
109 | $E($ CYAssign(cyn, parameters_->Selector(context)))->* | |
9a7c375c | 110 | $E($ CYAssign(cyt, type))->* |
4de0686f JF |
111 | $E($C4($V(replace ? "class_replaceMethod" : "class_addMethod"), |
112 | $V(instance_ ? "$cyc" : "$cym"), | |
113 | cyn, | |
c8a0500b JF |
114 | $N2($V("Functor"), $F(NULL, $P2($L("self"), $L("_cmd"), parameters_->Parameters(context)), $$->* |
115 | $ CYVar($L1($L("$cyr", $N2($V("Super"), self, _class))))->* | |
cacd1a88 | 116 | $ CYReturn($C1($M($F(NULL, NULL, code_), $S("call")), self)) |
4de0686f JF |
117 | ), cyt), |
118 | cyt | |
119 | )) | |
120 | ); | |
121 | } | |
122 | ||
9a7c375c JF |
123 | CYExpression *CYMessage::TypeSignature(CYContext &context) const { |
124 | return MessageType(context, type_, parameters_, $S("@:")); | |
125 | } | |
126 | ||
4de0686f JF |
127 | CYFunctionParameter *CYMessageParameter::Parameters(CYContext &context) const { $T(NULL) |
128 | CYFunctionParameter *next(next_->Parameters(context)); | |
c8a0500b | 129 | return name_ == NULL ? next : $ CYFunctionParameter($ CYDeclaration(name_), next); |
4de0686f JF |
130 | } |
131 | ||
132 | CYSelector *CYMessageParameter::Selector(CYContext &context) const { | |
133 | return $ CYSelector(SelectorPart(context)); | |
134 | } | |
135 | ||
136 | CYSelectorPart *CYMessageParameter::SelectorPart(CYContext &context) const { $T(NULL) | |
137 | CYSelectorPart *next(next_->SelectorPart(context)); | |
138 | return tag_ == NULL ? next : $ CYSelectorPart(tag_, name_ != NULL, next); | |
139 | } | |
140 | ||
9a7c375c JF |
141 | CYExpression *CYMessageParameter::TypeSignature(CYContext &context) const { |
142 | return MessageType(context, type_, next_); | |
143 | } | |
144 | ||
f2f0d1d1 JF |
145 | CYExpression *CYBox::Replace(CYContext &context) { |
146 | return $C1($M($V("Instance"), $S("box")), value_); | |
147 | } | |
148 | ||
56e02e5b | 149 | CYExpression *CYObjCBlock::Replace(CYContext &context) { |
9a39f705 | 150 | return $N2($V("Functor"), $ CYFunctionExpression(NULL, $ CYFunctionParameter($ CYDeclaration($ CYIdentifier("$cyt")), parameters_->Parameters(context)), statements_), parameters_->TypeSignature(context, $ CYAdd(typed_->Replace(context), $ CYString("@")))); |
56e02e5b JF |
151 | } |
152 | ||
64b8d29f JF |
153 | CYStatement *CYProtocol::Replace(CYContext &context) const { $T(NULL) |
154 | return $ CYBlock($$->* | |
155 | next_->Replace(context)->* | |
156 | $E($C2($V("class_addProtocol"), | |
157 | $V("$cyc"), name_ | |
158 | )) | |
159 | ); | |
160 | } | |
161 | ||
4de0686f | 162 | CYExpression *CYSelector::Replace(CYContext &context) { |
575f1f1b | 163 | return $C1($V("sel_registerName"), name_->Replace(context)); |
4de0686f JF |
164 | } |
165 | ||
166 | CYString *CYSelectorPart::Replace(CYContext &context) { | |
167 | std::ostringstream str; | |
c2c9f509 | 168 | CYForEach (part, this) { |
4de0686f | 169 | if (part->name_ != NULL) |
69688ca1 | 170 | str << part->name_->Word(); |
4de0686f JF |
171 | if (part->value_) |
172 | str << ':'; | |
173 | } | |
b799113b | 174 | return $S($pool.strdup(str.str().c_str())); |
4de0686f JF |
175 | } |
176 | ||
cacd1a88 | 177 | CYExpression *CYSendDirect::Replace(CYContext &context) { |
4de0686f JF |
178 | std::ostringstream name; |
179 | CYArgument **argument(&arguments_); | |
2385c806 | 180 | CYSelectorPart *selector(NULL), *current(NULL); |
4de0686f JF |
181 | |
182 | while (*argument != NULL) { | |
183 | if ((*argument)->name_ != NULL) { | |
2385c806 JF |
184 | CYSelectorPart *part($ CYSelectorPart((*argument)->name_, (*argument)->value_ != NULL)); |
185 | if (selector == NULL) | |
186 | selector = part; | |
187 | if (current != NULL) | |
188 | current->SetNext(part); | |
189 | current = part; | |
4de0686f | 190 | (*argument)->name_ = NULL; |
4de0686f JF |
191 | } |
192 | ||
193 | if ((*argument)->value_ == NULL) | |
194 | *argument = (*argument)->next_; | |
195 | else | |
196 | argument = &(*argument)->next_; | |
197 | } | |
198 | ||
2385c806 | 199 | return $C2($V("objc_msgSend"), self_, ($ CYSelector(selector))->Replace(context), arguments_); |
4de0686f | 200 | } |
cacd1a88 JF |
201 | |
202 | CYExpression *CYSendSuper::Replace(CYContext &context) { | |
57a65431 | 203 | return $ CYSendDirect($V("$cyr"), arguments_); |
cacd1a88 | 204 | } |