2009-05-01 Sam Weinig <sam@webkit.org>
[WebKit-https.git] / WebCore / workers / WorkerContext.h
1 /*
2  * Copyright (C) 2008, 2009 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  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR
17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 
24  *
25  */
26
27 #ifndef WorkerContext_h
28 #define WorkerContext_h
29
30 #if ENABLE(WORKERS)
31
32 #include "AtomicStringHash.h"
33 #include "EventListener.h"
34 #include "EventTarget.h"
35 #include "ScriptExecutionContext.h"
36 #include "WorkerScriptController.h"
37 #include <wtf/OwnPtr.h>
38 #include <wtf/PassRefPtr.h>
39 #include <wtf/RefCounted.h>
40 #include <wtf/RefPtr.h>
41
42 namespace WebCore {
43
44     class ScheduledAction;
45     class WorkerLocation;
46     class WorkerNavigator;
47     class WorkerThread;
48
49     class WorkerContext : public RefCounted<WorkerContext>, public ScriptExecutionContext, public EventTarget {
50     public:
51         static PassRefPtr<WorkerContext> create(const KURL& url, const String& userAgent, WorkerThread* thread)
52         {
53             return adoptRef(new WorkerContext(url, userAgent, thread));
54         }
55
56         virtual ~WorkerContext();
57
58         virtual bool isWorkerContext() const { return true; }
59
60         virtual WorkerContext* toWorkerContext() { return this; }
61
62         virtual ScriptExecutionContext* scriptExecutionContext() const;
63
64         const KURL& url() const { return m_url; }
65         KURL completeURL(const String&) const;
66
67         virtual String userAgent(const KURL&) const;
68
69         WorkerScriptController* script() { return m_script.get(); }
70         void clearScript() { return m_script.clear(); }
71
72         WorkerThread* thread() { return m_thread; }
73
74         bool hasPendingActivity() const;
75
76         virtual void reportException(const String& errorMessage, int lineNumber, const String& sourceURL);
77         virtual void addMessage(MessageDestination, MessageSource, MessageLevel, const String& message, unsigned lineNumber, const String& sourceURL);
78         virtual void resourceRetrievedByXMLHttpRequest(unsigned long identifier, const ScriptString& sourceString);
79         virtual void scriptImported(unsigned long identifier, const String& sourceString);
80
81         virtual void postTask(PassRefPtr<Task>); // Executes the task on context's thread asynchronously.
82
83
84         // WorkerGlobalScope
85         WorkerContext* self() { return this; }
86         WorkerLocation* location() const;
87
88         // WorkerUtils
89         void importScripts(const Vector<String>& urls, const String& callerURL, int callerLine, ExceptionCode&);
90         WorkerNavigator* navigator() const;
91
92
93         // DedicatedWorkerGlobalScope
94         void postMessage(const String& message);
95         void setOnmessage(PassRefPtr<EventListener> eventListener) { m_onmessageListener = eventListener; }
96         EventListener* onmessage() const { return m_onmessageListener.get(); }
97
98         // Timers
99         int setTimeout(ScheduledAction*, int timeout);
100         void clearTimeout(int timeoutId);
101         int setInterval(ScheduledAction*, int timeout);
102         void clearInterval(int timeoutId);
103
104         // EventTarget
105         virtual void addEventListener(const AtomicString& eventType, PassRefPtr<EventListener>, bool useCapture);
106         virtual void removeEventListener(const AtomicString& eventType, EventListener*, bool useCapture);
107         virtual bool dispatchEvent(PassRefPtr<Event>, ExceptionCode&);
108
109         typedef Vector<RefPtr<EventListener> > ListenerVector;
110         typedef HashMap<AtomicString, ListenerVector> EventListenersMap;
111         EventListenersMap& eventListeners() { return m_eventListeners; }
112
113         void dispatchMessage(const String&);
114
115         // These methods are used for GC marking. See JSWorkerContext::mark() in
116         // JSWorkerContextCustom.cpp.
117         WorkerNavigator* optionalNavigator() const { return m_navigator.get(); }
118         WorkerLocation* optionalLocation() const { return m_location.get(); }
119
120         using RefCounted<WorkerContext>::ref;
121         using RefCounted<WorkerContext>::deref;
122
123     private:
124         WorkerContext(const KURL&, const String&, WorkerThread*);
125
126         virtual void refScriptExecutionContext() { ref(); }
127         virtual void derefScriptExecutionContext() { deref(); }
128         virtual void refEventTarget() { ref(); }
129         virtual void derefEventTarget() { deref(); }
130
131         virtual const KURL& virtualURL() const;
132         virtual KURL virtualCompleteURL(const String&) const;
133
134         KURL m_url;
135         String m_userAgent;
136
137         mutable RefPtr<WorkerLocation> m_location;
138         mutable RefPtr<WorkerNavigator> m_navigator;
139
140         OwnPtr<WorkerScriptController> m_script;
141         WorkerThread* m_thread;
142
143         RefPtr<EventListener> m_onmessageListener;
144         EventListenersMap m_eventListeners;
145     };
146
147 } // namespace WebCore
148
149 #endif // ENABLE(WORKERS)
150
151 #endif // WorkerContext_h