Replace static_casts with to* functions for document types.
[WebKit-https.git] / Source / WebCore / bindings / js / JSHTMLDocumentCustom.cpp
1 /*
2  * Copyright (C) 2007, 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 #include "config.h"
27 #include "JSHTMLDocument.h"
28
29 #include "Frame.h"
30 #include "HTMLAllCollection.h"
31 #include "HTMLBodyElement.h"
32 #include "HTMLCollection.h"
33 #include "HTMLDocument.h"
34 #include "HTMLElement.h"
35 #include "HTMLIFrameElement.h"
36 #include "HTMLNames.h"
37 #include "JSDOMWindow.h"
38 #include "JSDOMWindowCustom.h"
39 #include "JSDOMWindowShell.h"
40 #include "JSHTMLCollection.h"
41 #include "JSMainThreadExecState.h"
42 #include "SegmentedString.h"
43 #include "DocumentParser.h"
44 #include <runtime/Error.h>
45 #include <runtime/JSCell.h>
46 #include <wtf/unicode/CharacterNames.h>
47
48 using namespace JSC;
49
50 namespace WebCore {
51
52 using namespace HTMLNames;
53
54 bool JSHTMLDocument::canGetItemsForName(ExecState*, HTMLDocument* document, PropertyName propertyName)
55 {
56     AtomicStringImpl* atomicPropertyName = findAtomicString(propertyName);
57     return atomicPropertyName && (document->hasNamedItem(atomicPropertyName) || document->hasExtraNamedItem(atomicPropertyName));
58 }
59
60 JSValue JSHTMLDocument::nameGetter(ExecState* exec, JSValue slotBase, PropertyName propertyName)
61 {
62     JSHTMLDocument* thisObj = jsCast<JSHTMLDocument*>(asObject(slotBase));
63     HTMLDocument* document = toHTMLDocument(thisObj->impl());
64
65     RefPtr<HTMLCollection> collection = document->documentNamedItems(propertyNameToAtomicString(propertyName));
66
67     if (collection->isEmpty())
68         return jsUndefined();
69
70     if (collection->hasExactlyOneItem()) {
71         Node* node = collection->item(0);
72
73         Frame* frame;
74         if (node->hasTagName(iframeTag) && (frame = static_cast<HTMLIFrameElement*>(node)->contentFrame()))
75             return toJS(exec, frame);
76
77         return toJS(exec, thisObj->globalObject(), node);
78     } 
79
80     return toJS(exec, thisObj->globalObject(), WTF::getPtr(collection));
81 }
82
83 // Custom attributes
84
85 JSValue JSHTMLDocument::all(ExecState* exec) const
86 {
87     // If "all" has been overwritten, return the overwritten value
88     JSValue v = getDirect(exec->globalData(), Identifier(exec, "all"));
89     if (v)
90         return v;
91
92     return toJS(exec, globalObject(), toHTMLDocument(impl())->all());
93 }
94
95 void JSHTMLDocument::setAll(ExecState* exec, JSValue value)
96 {
97     // Add "all" to the property map.
98     putDirect(exec->globalData(), Identifier(exec, "all"), value);
99 }
100
101 // Custom functions
102
103 JSValue JSHTMLDocument::open(ExecState* exec)
104 {
105     // For compatibility with other browsers, pass open calls with more than 2 parameters to the window.
106     if (exec->argumentCount() > 2) {
107         Frame* frame = toHTMLDocument(impl())->frame();
108         if (frame) {
109             JSDOMWindowShell* wrapper = toJSDOMWindowShell(frame, currentWorld(exec));
110             if (wrapper) {
111                 JSValue function = wrapper->get(exec, Identifier(exec, "open"));
112                 CallData callData;
113                 CallType callType = ::getCallData(function, callData);
114                 if (callType == CallTypeNone)
115                     return throwTypeError(exec);
116                 return JSMainThreadExecState::call(exec, function, callType, callData, wrapper, ArgList(exec));
117             }
118         }
119         return jsUndefined();
120     }
121
122     // document.open clobbers the security context of the document and
123     // aliases it with the active security context.
124     Document* activeDocument = asJSDOMWindow(exec->lexicalGlobalObject())->impl()->document();
125
126     // In the case of two parameters or fewer, do a normal document open.
127     toHTMLDocument(impl())->open(activeDocument);
128     return this;
129 }
130
131 enum NewlineRequirement { DoNotAddNewline, DoAddNewline };
132
133 static inline void documentWrite(ExecState* exec, HTMLDocument* document, NewlineRequirement addNewline)
134 {
135     // DOM only specifies single string argument, but browsers allow multiple or no arguments.
136
137     size_t size = exec->argumentCount();
138
139     String firstString = exec->argument(0).toString(exec)->value(exec);
140     SegmentedString segmentedString = firstString;
141     if (size != 1) {
142         if (!size)
143             segmentedString.clear();
144         else {
145             for (size_t i = 1; i < size; ++i) {
146                 String subsequentString = exec->argument(i).toString(exec)->value(exec);
147                 segmentedString.append(SegmentedString(subsequentString));
148             }
149         }
150     }
151     if (addNewline)
152         segmentedString.append(SegmentedString(String(&newlineCharacter, 1)));
153
154     Document* activeDocument = asJSDOMWindow(exec->lexicalGlobalObject())->impl()->document();
155     document->write(segmentedString, activeDocument);
156 }
157
158 JSValue JSHTMLDocument::write(ExecState* exec)
159 {
160     documentWrite(exec, toHTMLDocument(impl()), DoNotAddNewline);
161     return jsUndefined();
162 }
163
164 JSValue JSHTMLDocument::writeln(ExecState* exec)
165 {
166     documentWrite(exec, toHTMLDocument(impl()), DoAddNewline);
167     return jsUndefined();
168 }
169
170 } // namespace WebCore