a29a8e4ebee89dfe0792d77fe998a33ec7ea3d23
[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, WorkerThread* thread)
51     : m_url(url)
52     , m_userAgent(userAgent)
53     , m_location(WorkerLocation::create(url))
54     , m_script(new WorkerScriptController(this))
55     , m_thread(thread)
56 {
57     setSecurityOrigin(SecurityOrigin::create(url));
58 }
59
60 WorkerContext::~WorkerContext()
61 {
62     ASSERT(currentThread() == m_thread->threadID());
63
64     m_thread->workerObjectProxy()->workerContextDestroyed();
65 }
66
67 ScriptExecutionContext* WorkerContext::scriptExecutionContext() const
68 {
69     return const_cast<WorkerContext*>(this);
70 }
71
72 const KURL& WorkerContext::virtualURL() const
73 {
74     return m_url;
75 }
76
77 KURL WorkerContext::virtualCompleteURL(const String& url) const
78 {
79     return completeURL(url);
80 }
81
82 KURL WorkerContext::completeURL(const String& url) const
83 {
84     // Always return a null URL when passed a null string.
85     // FIXME: Should we change the KURL constructor to have this behavior?
86     if (url.isNull())
87         return KURL();
88     // FIXME: does this need to provide a charset, like Document::completeURL does?
89     return KURL(m_location->url(), url);
90 }
91
92 String WorkerContext::userAgent(const KURL&) const
93 {
94     return m_userAgent;
95 }
96
97 WorkerNavigator* WorkerContext::navigator() const
98 {
99     if (!m_navigator)
100         m_navigator = WorkerNavigator::create(m_userAgent);
101     return m_navigator.get();
102 }
103
104 bool WorkerContext::hasPendingActivity() const
105 {
106     ActiveDOMObjectsMap& activeObjects = activeDOMObjects();
107     ActiveDOMObjectsMap::const_iterator activeObjectsEnd = activeObjects.end();
108     for (ActiveDOMObjectsMap::const_iterator iter = activeObjects.begin(); iter != activeObjectsEnd; ++iter) {
109         if (iter->first->hasPendingActivity())
110             return true;
111     }
112     return false;
113 }
114
115 void WorkerContext::reportException(const String& errorMessage, int lineNumber, const String& sourceURL)
116 {
117     m_thread->workerObjectProxy()->postExceptionToWorkerObject(errorMessage, lineNumber, sourceURL);
118 }
119
120 void WorkerContext::addMessage(MessageDestination destination, MessageSource source, MessageLevel level, const String& message, unsigned lineNumber, const String& sourceURL)
121 {
122     m_thread->workerObjectProxy()->postConsoleMessageToWorkerObject(destination, source, level, message, lineNumber, sourceURL);
123 }
124
125 void WorkerContext::resourceRetrievedByXMLHttpRequest(unsigned long, const ScriptString&)
126 {
127     // FIXME: The implementation is pending the fixes in https://bugs.webkit.org/show_bug.cgi?id=23175
128     notImplemented();
129 }
130
131 void WorkerContext::postMessage(const String& message)
132 {
133     m_thread->workerObjectProxy()->postMessageToWorkerObject(message);
134 }
135
136 void WorkerContext::addEventListener(const AtomicString& eventType, PassRefPtr<EventListener> eventListener, bool)
137 {
138     EventListenersMap::iterator iter = m_eventListeners.find(eventType);
139     if (iter == m_eventListeners.end()) {
140         ListenerVector listeners;
141         listeners.append(eventListener);
142         m_eventListeners.add(eventType, listeners);
143     } else {
144         ListenerVector& listeners = iter->second;
145         for (ListenerVector::iterator listenerIter = listeners.begin(); listenerIter != listeners.end(); ++listenerIter) {
146             if (*listenerIter == eventListener)
147                 return;
148         }
149         
150         listeners.append(eventListener);
151         m_eventListeners.add(eventType, listeners);
152     }    
153 }
154
155 void WorkerContext::removeEventListener(const AtomicString& eventType, EventListener* eventListener, bool)
156 {
157     EventListenersMap::iterator iter = m_eventListeners.find(eventType);
158     if (iter == m_eventListeners.end())
159         return;
160     
161     ListenerVector& listeners = iter->second;
162     for (ListenerVector::const_iterator listenerIter = listeners.begin(); listenerIter != listeners.end(); ++listenerIter) {
163         if (*listenerIter == eventListener) {
164             listeners.remove(listenerIter - listeners.begin());
165             return;
166         }
167     }
168 }
169
170 bool WorkerContext::dispatchEvent(PassRefPtr<Event> event, ExceptionCode& ec)
171 {
172     if (!event || event->type().isEmpty()) {
173         ec = EventException::UNSPECIFIED_EVENT_TYPE_ERR;
174         return true;
175     }
176     
177     ListenerVector listenersCopy = m_eventListeners.get(event->type());
178     for (ListenerVector::const_iterator listenerIter = listenersCopy.begin(); listenerIter != listenersCopy.end(); ++listenerIter) {
179         event->setTarget(this);
180         event->setCurrentTarget(this);
181         listenerIter->get()->handleEvent(event.get(), false);
182     }
183     
184     return !event->defaultPrevented();
185 }
186
187 void WorkerContext::postTask(PassRefPtr<Task> task)
188 {
189     thread()->runLoop().postTask(task);
190 }
191
192 int WorkerContext::installTimeout(ScheduledAction* action, int timeout, bool singleShot)
193 {
194     return DOMTimer::install(scriptExecutionContext(), action, timeout, singleShot);
195 }
196
197 void WorkerContext::removeTimeout(int timeoutId)
198 {
199     DOMTimer::removeById(scriptExecutionContext(), timeoutId);
200 }
201
202 void WorkerContext::dispatchMessage(const String& message)
203 {
204     RefPtr<Event> evt = MessageEvent::create(message, "", "", 0, 0);
205
206     if (m_onmessageListener.get()) {
207         evt->setTarget(this);
208         evt->setCurrentTarget(this);
209         m_onmessageListener->handleEvent(evt.get(), false);
210     }
211
212     ExceptionCode ec = 0;
213     dispatchEvent(evt.release(), ec);
214     ASSERT(!ec);
215 }
216
217 } // namespace WebCore
218
219 #endif // ENABLE(WORKERS)