jscript: Added propertyIsEnumerable implementation.
[wine] / dlls / jscript / object.c
1 /*
2  * Copyright 2008 Jacek Caban for CodeWeavers
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * This library is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with this library; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
17  */
18
19 #include "jscript.h"
20
21 #include "wine/debug.h"
22
23 WINE_DEFAULT_DEBUG_CHANNEL(jscript);
24
25 static const WCHAR toStringW[] = {'t','o','S','t','r','i','n','g',0};
26 static const WCHAR toLocaleStringW[] = {'t','o','L','o','c','a','l','e','S','t','r','i','n','g',0};
27 static const WCHAR valueOfW[] = {'v','a','l','u','e','O','f',0};
28 static const WCHAR hasOwnPropertyW[] = {'h','a','s','O','w','n','P','r','o','p','e','r','t','y',0};
29 static const WCHAR propertyIsEnumerableW[] =
30     {'p','r','o','p','e','r','t','y','I','s','E','n','u','m','e','r','a','b','l','e',0};
31 static const WCHAR isPrototypeOfW[] = {'i','s','P','r','o','t','o','t','y','p','e','O','f',0};
32
33 static const WCHAR default_valueW[] = {'[','o','b','j','e','c','t',' ','O','b','j','e','c','t',']',0};
34
35 static HRESULT Object_toString(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, unsigned argc, jsval_t *argv,
36         jsval_t *r)
37 {
38     jsdisp_t *jsdisp;
39     const WCHAR *str;
40
41     static const WCHAR formatW[] = {'[','o','b','j','e','c','t',' ','%','s',']',0};
42
43     static const WCHAR arrayW[] = {'A','r','r','a','y',0};
44     static const WCHAR booleanW[] = {'B','o','o','l','e','a','n',0};
45     static const WCHAR dateW[] = {'D','a','t','e',0};
46     static const WCHAR errorW[] = {'E','r','r','o','r',0};
47     static const WCHAR functionW[] = {'F','u','n','c','t','i','o','n',0};
48     static const WCHAR mathW[] = {'M','a','t','h',0};
49     static const WCHAR numberW[] = {'N','u','m','b','e','r',0};
50     static const WCHAR objectW[] = {'O','b','j','e','c','t',0};
51     static const WCHAR regexpW[] = {'R','e','g','E','x','p',0};
52     static const WCHAR stringW[] = {'S','t','r','i','n','g',0};
53     /* Keep in sync with jsclass_t enum */
54     static const WCHAR *names[] = {objectW, arrayW, booleanW, dateW, errorW,
55         functionW, NULL, mathW, numberW, objectW, regexpW, stringW, objectW, objectW};
56
57     TRACE("\n");
58
59     jsdisp = get_jsdisp(jsthis);
60     if(!jsdisp) {
61         str = objectW;
62     }else if(names[jsdisp->builtin_info->class]) {
63         str = names[jsdisp->builtin_info->class];
64     }else {
65         FIXME("jdisp->builtin_info->class = %d\n", jsdisp->builtin_info->class);
66         return E_FAIL;
67     }
68
69     if(r) {
70         jsstr_t *ret;
71
72         ret = jsstr_alloc_buf(9+strlenW(str));
73         if(!ret)
74             return E_OUTOFMEMORY;
75
76         sprintfW(ret->str, formatW, str);
77         *r = jsval_string(ret);
78     }
79
80     return S_OK;
81 }
82
83 static HRESULT Object_toLocaleString(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, unsigned argc, jsval_t *argv,
84         jsval_t *r)
85 {
86     TRACE("\n");
87
88     if(!is_jsdisp(jsthis)) {
89         FIXME("Host object this\n");
90         return E_FAIL;
91     }
92
93     return jsdisp_call_name(jsthis->u.jsdisp, toStringW, DISPATCH_METHOD, 0, NULL, r);
94 }
95
96 static HRESULT Object_valueOf(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, unsigned argc, jsval_t *argv,
97         jsval_t *r)
98 {
99     TRACE("\n");
100
101     if(r) {
102         IDispatch_AddRef(jsthis->u.disp);
103         *r = jsval_disp(jsthis->u.disp);
104     }
105     return S_OK;
106 }
107
108 static HRESULT Object_hasOwnProperty(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, unsigned argc, jsval_t *argv,
109         jsval_t *r)
110 {
111     jsstr_t *name;
112     DISPID id;
113     HRESULT hres;
114
115     TRACE("\n");
116
117     if(!argc) {
118         if(r)
119             *r = jsval_bool(FALSE);
120         return S_OK;
121     }
122
123     hres = to_string(ctx, argv[0], &name);
124     if(FAILED(hres))
125         return hres;
126
127     if(is_jsdisp(jsthis)) {
128         BOOL result;
129
130         hres = jsdisp_is_own_prop(jsthis->u.jsdisp, name->str, &result);
131         if(FAILED(hres))
132             return hres;
133
134         if(r)
135             *r = jsval_bool(result);
136         return S_OK;
137     }
138
139     if(is_dispex(jsthis)) {
140         BSTR bstr;
141
142         bstr = SysAllocStringLen(name->str, jsstr_length(name));
143         if(!bstr)
144             return E_OUTOFMEMORY;
145
146         hres = IDispatchEx_GetDispID(jsthis->u.dispex, bstr,
147                 make_grfdex(ctx, fdexNameCaseSensitive), &id);
148         SysFreeString(bstr);
149     } else {
150         OLECHAR *names = name->str;
151         hres = IDispatch_GetIDsOfNames(jsthis->u.disp, &IID_NULL,
152                 &names, 1, ctx->lcid, &id);
153     }
154
155     if(r)
156         *r = jsval_bool(SUCCEEDED(hres));
157     return S_OK;
158 }
159
160 static HRESULT Object_propertyIsEnumerable(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, unsigned argc, jsval_t *argv,
161         jsval_t *r)
162 {
163     jsstr_t *name;
164     BOOL ret;
165     HRESULT hres;
166
167     TRACE("\n");
168
169     if(argc != 1) {
170         FIXME("argc %d not supported\n", argc);
171         return E_NOTIMPL;
172     }
173
174     if(!is_jsdisp(jsthis)) {
175         FIXME("Host object this\n");
176         return E_FAIL;
177     }
178
179     hres = to_string(ctx, argv[0], &name);
180     if(FAILED(hres))
181         return hres;
182
183     hres = jsdisp_is_enumerable(jsthis->u.jsdisp, name->str, &ret);
184     jsstr_release(name);
185     if(FAILED(hres))
186         return hres;
187
188     if(r)
189         *r = jsval_bool(ret);
190     return S_OK;
191 }
192
193 static HRESULT Object_isPrototypeOf(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, unsigned argc, jsval_t *argv,
194         jsval_t *r)
195 {
196     FIXME("\n");
197     return E_NOTIMPL;
198 }
199
200 static HRESULT Object_value(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, unsigned argc, jsval_t *argv,
201         jsval_t *r)
202 {
203     TRACE("\n");
204
205     switch(flags) {
206     case INVOKE_FUNC:
207         return throw_type_error(ctx, JS_E_FUNCTION_EXPECTED, NULL);
208     case DISPATCH_PROPERTYGET: {
209         jsstr_t *ret = jsstr_alloc(default_valueW);
210         if(!ret)
211             return E_OUTOFMEMORY;
212         *r = jsval_string(ret);
213         break;
214     }
215     default:
216         FIXME("unimplemented flags %x\n", flags);
217         return E_NOTIMPL;
218     }
219
220     return S_OK;
221 }
222
223 static void Object_destructor(jsdisp_t *dispex)
224 {
225     heap_free(dispex);
226 }
227
228 static const builtin_prop_t Object_props[] = {
229     {hasOwnPropertyW,        Object_hasOwnProperty,        PROPF_METHOD|1},
230     {isPrototypeOfW,         Object_isPrototypeOf,         PROPF_METHOD|1},
231     {propertyIsEnumerableW,  Object_propertyIsEnumerable,  PROPF_METHOD|1},
232     {toLocaleStringW,        Object_toLocaleString,        PROPF_METHOD},
233     {toStringW,              Object_toString,              PROPF_METHOD},
234     {valueOfW,               Object_valueOf,               PROPF_METHOD}
235 };
236
237 static const builtin_info_t Object_info = {
238     JSCLASS_OBJECT,
239     {NULL, Object_value, 0},
240     sizeof(Object_props)/sizeof(*Object_props),
241     Object_props,
242     Object_destructor,
243     NULL
244 };
245
246 static const builtin_info_t ObjectInst_info = {
247     JSCLASS_OBJECT,
248     {NULL, Object_value, 0},
249     0, NULL,
250     Object_destructor,
251     NULL
252 };
253
254 static HRESULT ObjectConstr_value(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, unsigned argc, jsval_t *argv,
255         jsval_t *r)
256 {
257     HRESULT hres;
258
259     TRACE("\n");
260
261     switch(flags) {
262     case DISPATCH_METHOD:
263         if(argc) {
264             if(!is_undefined(argv[0]) && !is_null(argv[0]) && (!is_object_instance(argv[0]) || get_object(argv[0]))) {
265                 IDispatch *disp;
266
267                 hres = to_object(ctx, argv[0], &disp);
268                 if(FAILED(hres))
269                     return hres;
270
271                 if(r)
272                     *r = jsval_disp(disp);
273                 else
274                     IDispatch_Release(disp);
275                 return S_OK;
276             }
277         }
278         /* fall through */
279     case DISPATCH_CONSTRUCT: {
280         jsdisp_t *obj;
281
282         hres = create_object(ctx, NULL, &obj);
283         if(FAILED(hres))
284             return hres;
285
286         if(r)
287             *r = jsval_obj(obj);
288         else
289             jsdisp_release(obj);
290         break;
291     }
292
293     default:
294         FIXME("unimplemented flags: %x\n", flags);
295         return E_NOTIMPL;
296     }
297
298     return S_OK;
299 }
300
301 HRESULT create_object_constr(script_ctx_t *ctx, jsdisp_t *object_prototype, jsdisp_t **ret)
302 {
303     static const WCHAR ObjectW[] = {'O','b','j','e','c','t',0};
304
305     return create_builtin_constructor(ctx, ObjectConstr_value, ObjectW, NULL, PROPF_CONSTR,
306             object_prototype, ret);
307 }
308
309 HRESULT create_object_prototype(script_ctx_t *ctx, jsdisp_t **ret)
310 {
311     return create_dispex(ctx, &Object_info, NULL, ret);
312 }
313
314 HRESULT create_object(script_ctx_t *ctx, jsdisp_t *constr, jsdisp_t **ret)
315 {
316     jsdisp_t *object;
317     HRESULT hres;
318
319     object = heap_alloc_zero(sizeof(jsdisp_t));
320     if(!object)
321         return E_OUTOFMEMORY;
322
323     hres = init_dispex_from_constr(object, ctx, &ObjectInst_info, constr ? constr : ctx->object_constr);
324     if(FAILED(hres)) {
325         heap_free(object);
326         return hres;
327     }
328
329     *ret = object;
330     return S_OK;
331 }