WebCore:
[WebKit-https.git] / WebCore / workers / WorkerContext.cpp
1 /*
2  * Copyright (C) 2008 Apple Inc. All Rights Reserved.
3  * Copyright (C) 2009 Google Inc. All Rights Reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
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  *
14  * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
15  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
17  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR
18  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
19  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
20  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
21  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
22  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
24  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 
25  *
26  */
27
28 #include "config.h"
29
30 #if ENABLE(WORKERS)
31
32 #include "WorkerContext.h"
33
34 #include "ActiveDOMObject.h"
35 #include "DOMTimer.h"
36 #include "DOMWindow.h"
37 #include "Event.h"
38 #include "EventException.h"
39 #include "MessageEvent.h"
40 #include "NotImplemented.h"
41 #include "SecurityOrigin.h"
42 #include "WorkerLocation.h"
43 #include "WorkerNavigator.h"
44 #include "WorkerObjectProxy.h"
45 #include "WorkerThread.h"
46 #include <wtf/RefPtr.h>
47
48 namespace WebCore {
49
50 WorkerContext::WorkerContext(const KURL& url, const String& userAgent, const String& encoding, WorkerThread* thread)
51     : m_url(url)
52     , m_userAgent(userAgent)
53     , m_encoding(encoding)
54     , m_location(WorkerLocation::create(url))
55     , m_script(new WorkerScriptController(this))
56     , m_thread(thread)
57 {
58     setSecurityOrigin(SecurityOrigin::create(url));
59 }
60
61 WorkerContext::~WorkerContext()
62 {
63     ASSERT(currentThread() == m_thread->threadID());
64
65     m_thread->workerObjectProxy()->workerContextDestroyed();
66 }
67
68 ScriptExecutionContext* WorkerContext::scriptExecutionContext() const
69 {
70     return const_cast<WorkerContext*>(this);
71 }
72
73 const KURL& WorkerContext::virtualURL() const
74 {
75     return m_url;
76 }
77
78 KURL WorkerContext::virtualCompleteURL(const String& url) const
79 {
80     return completeURL(url);
81 }
82
83 KURL WorkerContext::completeURL(const String& url) const
84 {
85     // Always return a null URL when passed a null string.
86     // FIXME: Should we change the KURL constructor to have this behavior?
87     if (url.isNull())
88         return KURL();
89     // FIXME: Document::completeURL provides a charset here (to encode query portion of url when
90     // submitting charset-encoded forms data). FF3.1b2 uses original document's encoding for nested
91     // Workers and importScripts but UTF-8 for XHR. If it's not a bug, we probably should match FF.
92     return KURL(m_location->url(), url);
93 }
94
95 String WorkerContext::userAgent(const KURL&) const
96 {
97     return m_userAgent;
98 }
99
100 String WorkerContext::encoding() const
101 {
102     return m_encoding;
103 }
104
105 WorkerNavigator* WorkerContext::navigator() const
106 {
107     if (!m_navigator)
108         m_navigator = WorkerNavigator::create(m_userAgent);
109     return m_navigator.get();
110 }
111
112 bool WorkerContext::hasPendingActivity() const
113 {
114     ActiveDOMObjectsMap& activeObjects = activeDOMObjects();
115     ActiveDOMObjectsMap::const_iterator activeObjectsEnd = activeObjects.end();
116     for (ActiveDOMObjectsMap::const_iterator iter = activeObjects.begin(); iter != activeObjectsEnd; ++iter) {
117         if (iter->first->hasPendingActivity())
118             return true;
119     }
120     return false;
121 }
122
123 void WorkerContext::reportException(const String& errorMessage, int lineNumber, const String& sourceURL)
124 {
125     m_thread->workerObjectProxy()->postExceptionToWorkerObject(errorMessage, lineNumber, sourceURL);
126 }
127
128 void WorkerContext::addMessage(MessageDestination destination, MessageSource source, MessageLevel level, const String& message, unsigned lineNumber, const String& sourceURL)
129 {
130     m_thread->workerObjectProxy()->postConsoleMessageToWorkerObject(destination, source, level, message, lineNumber, sourceURL);
131 }
132
133 void WorkerContext::resourceRetrievedByXMLHttpRequest(unsigned long, const ScriptString&)
134 {
135     // FIXME: The implementation is pending the fixes in https://bugs.webkit.org/show_bug.cgi?id=23175
136     notImplemented();
137 }
138
139 void WorkerContext::postMessage(const String& message)
140 {
141     m_thread->workerObjectProxy()->postMessageToWorkerObject(message);
142 }
143
144 void WorkerContext::addEventListener(const AtomicString& eventType, PassRefPtr<EventListener> eventListener, bool)
145 {
146     EventListenersMap::iterator iter = m_eventListeners.find(eventType);
147     if (iter == m_eventListeners.end()) {
148         ListenerVector listeners;
149         listeners.append(eventListener);
150         m_eventListeners.add(eventType, listeners);
151     } else {
152         ListenerVector& listeners = iter->second;
153         for (ListenerVector::iterator listenerIter = listeners.begin(); listenerIter != listeners.end(); ++listenerIter) {
154             if (*listenerIter == eventListener)
155                 return;
156         }
157         
158         listeners.append(eventListener);
159         m_eventListeners.add(eventType, listeners);
160     }    
161 }
162
163 void WorkerContext::removeEventListener(const AtomicString& eventType, EventListener* eventListener, bool)
164 {
165     EventListenersMap::iterator iter = m_eventListeners.find(eventType);
166     if (iter == m_eventListeners.end())
167         return;
168     
169     ListenerVector& listeners = iter->second;
170     for (ListenerVector::const_iterator listenerIter = listeners.begin(); listenerIter != listeners.end(); ++listenerIter) {
171         if (*listenerIter == eventListener) {
172             listeners.remove(listenerIter - listeners.begin());
173             return;
174         }
175     }
176 }
177
178 bool WorkerContext::dispatchEvent(PassRefPtr<Event> event, ExceptionCode& ec)
179 {
180     if (!event || event->type().isEmpty()) {
181         ec = EventException::UNSPECIFIED_EVENT_TYPE_ERR;
182         return true;
183     }
184     
185     ListenerVector listenersCopy = m_eventListeners.get(event->type());
186     for (ListenerVector::const_iterator listenerIter = listenersCopy.begin(); listenerIter != listenersCopy.end(); ++listenerIter) {
187         event->setTarget(this);
188         event->setCurrentTarget(this);
189         listenerIter->get()->handleEvent(event.get(), false);
190     }
191     
192     return !event->defaultPrevented();
193 }
194
195 void WorkerContext::postTask(PassRefPtr<Task> task)
196 {
197     thread()->runLoop().postTask(task);
198 }
199
200 int WorkerContext::installTimeout(ScheduledAction* action, int timeout, bool singleShot)
201 {
202     return DOMTimer::install(scriptExecutionContext(), action, timeout, singleShot);
203 }
204
205 void WorkerContext::removeTimeout(int timeoutId)
206 {
207     DOMTimer::removeById(scriptExecutionContext(), timeoutId);
208 }
209
210 void WorkerContext::dispatchMessage(const String& message)
211 {
212     RefPtr<Event> evt = MessageEvent::create(message, "", "", 0, 0);
213
214     if (m_onmessageListener.get()) {
215         evt->setTarget(this);
216         evt->setCurrentTarget(this);
217         m_onmessageListener->handleEvent(evt.get(), false);
218     }
219
220     ExceptionCode ec = 0;
221     dispatchEvent(evt.release(), ec);
222     ASSERT(!ec);
223 }
224
225 } // namespace WebCore
226
227 #endif // ENABLE(WORKERS)