2009-06-16 Jian Li <jianli@chromium.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         void close();
88
89         // WorkerUtils
90         void importScripts(const Vector<String>& urls, const String& callerURL, int callerLine, ExceptionCode&);
91         WorkerNavigator* navigator() const;
92
93
94         // DedicatedWorkerGlobalScope
95         void postMessage(const String& message);
96         void setOnmessage(PassRefPtr<EventListener> eventListener) { m_onmessageListener = eventListener; }
97         EventListener* onmessage() const { return m_onmessageListener.get(); }
98
99         // Timers
100         int setTimeout(ScheduledAction*, int timeout);
101         void clearTimeout(int timeoutId);
102         int setInterval(ScheduledAction*, int timeout);
103         void clearInterval(int timeoutId);
104
105         // EventTarget
106         virtual void addEventListener(const AtomicString& eventType, PassRefPtr<EventListener>, bool useCapture);
107         virtual void removeEventListener(const AtomicString& eventType, EventListener*, bool useCapture);
108         virtual bool dispatchEvent(PassRefPtr<Event>, ExceptionCode&);
109
110         typedef Vector<RefPtr<EventListener> > ListenerVector;
111         typedef HashMap<AtomicString, ListenerVector> EventListenersMap;
112         EventListenersMap& eventListeners() { return m_eventListeners; }
113
114         void dispatchMessage(const String&);
115
116         // These methods are used for GC marking. See JSWorkerContext::mark() in
117         // JSWorkerContextCustom.cpp.
118         WorkerNavigator* optionalNavigator() const { return m_navigator.get(); }
119         WorkerLocation* optionalLocation() const { return m_location.get(); }
120
121         using RefCounted<WorkerContext>::ref;
122         using RefCounted<WorkerContext>::deref;
123
124     private:
125         WorkerContext(const KURL&, const String&, WorkerThread*);
126
127         virtual void refScriptExecutionContext() { ref(); }
128         virtual void derefScriptExecutionContext() { deref(); }
129         virtual void refEventTarget() { ref(); }
130         virtual void derefEventTarget() { deref(); }
131
132         virtual const KURL& virtualURL() const;
133         virtual KURL virtualCompleteURL(const String&) const;
134
135         KURL m_url;
136         String m_userAgent;
137
138         mutable RefPtr<WorkerLocation> m_location;
139         mutable RefPtr<WorkerNavigator> m_navigator;
140
141         OwnPtr<WorkerScriptController> m_script;
142         WorkerThread* m_thread;
143
144         RefPtr<EventListener> m_onmessageListener;
145         EventListenersMap m_eventListeners;
146
147         bool m_closing;
148     };
149
150 } // namespace WebCore
151
152 #endif // ENABLE(WORKERS)
153
154 #endif // WorkerContext_h