2009-03-26 Darin Adler <darin@apple.com>
[WebKit-https.git] / WebCore / bindings / js / JSMessagePortCustom.cpp
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 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 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 #include "config.h"
27 #include "JSMessagePort.h"
28
29 #include "AtomicString.h"
30 #include "Event.h"
31 #include "Frame.h"
32 #include "JSDOMGlobalObject.h"
33 #include "JSEvent.h"
34 #include "JSEventListener.h"
35 #include "MessagePort.h"
36
37 using namespace JSC;
38
39 namespace WebCore {
40
41 void JSMessagePort::mark()
42 {
43     DOMObject::mark();
44
45     markIfNotNull(m_impl->onmessage());
46     markIfNotNull(m_impl->onclose());
47
48     if (MessagePort* entangledPort = m_impl->entangledPort()) {
49         DOMObject* wrapper = getCachedDOMObjectWrapper(*Heap::heap(this)->globalData(), entangledPort);
50         if (wrapper && !wrapper->marked())
51             wrapper->mark();
52     }
53
54     typedef MessagePort::EventListenersMap EventListenersMap;
55     typedef MessagePort::ListenerVector ListenerVector;
56     EventListenersMap& eventListeners = m_impl->eventListeners();
57     for (EventListenersMap::iterator mapIter = eventListeners.begin(); mapIter != eventListeners.end(); ++mapIter) {
58         for (ListenerVector::iterator vecIter = mapIter->second.begin(); vecIter != mapIter->second.end(); ++vecIter) 
59             (*vecIter)->mark();
60     }
61 }
62
63 JSValuePtr JSMessagePort::startConversation(ExecState* exec, const ArgList& args)
64 {
65     JSDOMGlobalObject* globalObject = static_cast<JSDOMGlobalObject*>(exec->lexicalGlobalObject());
66     const UString& message = args.at(exec, 0).toString(exec);
67
68     return toJS(exec, impl()->startConversation(globalObject->scriptExecutionContext(), message).get());
69 }
70
71 JSValuePtr JSMessagePort::addEventListener(ExecState* exec, const ArgList& args)
72 {
73     JSDOMGlobalObject* globalObject = toJSDOMGlobalObject(impl()->scriptExecutionContext());
74     if (!globalObject)
75         return jsUndefined();
76     RefPtr<JSEventListener> listener = globalObject->findOrCreateJSEventListener(exec, args.at(exec, 1));
77     if (!listener)
78         return jsUndefined();
79     impl()->addEventListener(args.at(exec, 0).toString(exec), listener.release(), args.at(exec, 2).toBoolean(exec));
80     return jsUndefined();
81 }
82
83 JSValuePtr JSMessagePort::removeEventListener(ExecState* exec, const ArgList& args)
84 {
85     JSDOMGlobalObject* globalObject = toJSDOMGlobalObject(impl()->scriptExecutionContext());
86     if (!globalObject)
87         return jsUndefined();
88     JSEventListener* listener = globalObject->findJSEventListener(exec, args.at(exec, 1));
89     if (!listener)
90         return jsUndefined();
91     impl()->removeEventListener(args.at(exec, 0).toString(exec), listener, args.at(exec, 2).toBoolean(exec));
92     return jsUndefined();
93 }
94
95 } // namespace WebCore