d7228db3f3ebf6eace7f8b906b794e22bb5adabf
[WebKit-https.git] / WebKit / win / WebScriptScope.cpp
1 /*
2  * Copyright (C) 2007 Apple Inc.  All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  *
8  * 1.  Redistributions of source code must retain the above copyright
9  *     notice, this list of conditions and the following disclaimer. 
10  * 2.  Redistributions in binary form must reproduce the above copyright
11  *     notice, this list of conditions and the following disclaimer in the
12  *     documentation and/or other materials provided with the distribution. 
13  * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of
14  *     its contributors may be used to endorse or promote products derived
15  *     from this software without specific prior written permission. 
16  *
17  * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
18  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
19  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
20  * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
21  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
22  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
23  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
24  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27  */
28
29 #include "config.h"
30 #include "WebKitDLL.h"
31 #include "WebScriptScope.h"
32
33 #include "COMEnumVariant.h"
34 #include "WebScriptCallFrame.h"
35
36 #include <memory>
37 #include <JavaScriptCore/PropertyNameArray.h>
38 #include <wtf/Assertions.h>
39 #include <wtf/OwnPtr.h>
40
41 #pragma warning(push, 0)
42 #include <WebCore/BString.h>
43 #include <WebCore/PlatformString.h>
44 #pragma warning(pop)
45
46 using namespace KJS;
47 using namespace std;
48
49 // WebScriptScope ------------------------------------------------------------
50
51 WebScriptScope::WebScriptScope(JSObject* scope)
52     : m_refCount(0)
53     , m_scope(scope)
54 {
55     gClassCount++;
56 }
57
58 WebScriptScope::~WebScriptScope()
59 {
60     gClassCount--;
61 }
62
63 WebScriptScope* WebScriptScope::createInstance(JSObject* scope)
64 {
65     WebScriptScope* instance = new WebScriptScope(scope);
66     instance->AddRef();
67     return instance;
68 }
69
70 // IUnknown -------------------------------------------------------------------
71
72 HRESULT STDMETHODCALLTYPE WebScriptScope::QueryInterface(REFIID riid, void** ppvObject)
73 {
74     *ppvObject = 0;
75     if (IsEqualGUID(riid, IID_IUnknown))
76         *ppvObject = static_cast<IWebScriptScope*>(this);
77     else if (IsEqualGUID(riid, IID_IWebScriptScope))
78         *ppvObject = static_cast<IWebScriptScope*>(this);
79     else
80         return E_NOINTERFACE;
81
82     AddRef();
83     return S_OK;
84 }
85
86 ULONG STDMETHODCALLTYPE WebScriptScope::AddRef(void)
87 {
88     return ++m_refCount;
89 }
90
91 ULONG STDMETHODCALLTYPE WebScriptScope::Release(void)
92 {
93     ULONG newRef = --m_refCount;
94     if (!newRef)
95         delete(this);
96
97     return newRef;
98 }
99
100 template<> struct COMVariantSetter<Identifier>
101 {
102     static void setVariant(VARIANT* variant, const Identifier& value)
103     {
104         ASSERT(V_VT(variant) == VT_EMPTY);
105
106         V_VT(variant) = VT_BSTR;
107         V_BSTR(variant) = WebCore::BString(reinterpret_cast<const wchar_t*>(value.data()), value.size()).release();
108     }
109 };
110
111 // WebScriptScope ------------------------------------------------------------
112
113 HRESULT STDMETHODCALLTYPE WebScriptScope::variableNames(
114     /* [in] */ IWebScriptCallFrame* frame,
115     /* [out, retval] */ IEnumVARIANT** variableNames)
116 {
117     if (!variableNames)
118         return E_POINTER;
119
120     *variableNames = 0;
121
122     if (!frame)
123         return E_FAIL;
124
125     PropertyNameArray propertyNames;
126     m_scope->getPropertyNames(frame->impl()->state(), propertyNames);
127
128     *variableNames = COMEnumVariant<PropertyNameArray>::adopt(propertyNames);
129
130     return S_OK;
131 }
132
133 HRESULT STDMETHODCALLTYPE WebScriptScope::valueForVariable(
134     /* [in] */ IWebScriptCallFrame* frame,
135     /* [in] */ BSTR key,
136     /* [out, retval] */ BSTR* value)
137 {
138     if (!frame || !key)
139         return E_FAIL;
140
141     if (!value)
142         return E_POINTER;
143
144     *value = 0;
145
146     Identifier identKey(reinterpret_cast<KJS::UChar*>(key), SysStringLen(key));
147
148     JSValue* jsvalue = m_scope->get(frame->impl()->state(), identKey);
149
150     UString string;
151     const JSType type = jsvalue->type();
152     switch (type) {
153         case NullType:
154         case UndefinedType:
155         case UnspecifiedType:
156         case GetterSetterType:
157             break;
158         case StringType:
159             string = jsvalue->getString();
160             break;
161         case NumberType:
162             string = UString::from(jsvalue->getNumber());
163             break;
164         case BooleanType:
165             string = jsvalue->getBoolean() ? "True" : "False";
166             break;
167         case ObjectType: {
168             JSObject* jsobject = jsvalue->getObject();
169             jsvalue = jsobject->defaultValue(frame->impl()->state(), StringType);
170             string = jsvalue->getString();
171             break;
172         }
173     }
174
175     *value = WebCore::BString(string).release();
176     return S_OK;
177 }