WebCore/bindings/js/JSHTMLDocumentCustom.cpp
changeset 0 4f2f89ce4247
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/WebCore/bindings/js/JSHTMLDocumentCustom.cpp	Fri Sep 17 09:02:29 2010 +0300
@@ -0,0 +1,171 @@
+/*
+ * Copyright (C) 2007, 2008, 2009 Apple Inc. All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
+ * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+ * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR
+ * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
+ * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+ * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
+ * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
+ * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 
+ */
+
+#include "config.h"
+#include "JSHTMLDocument.h"
+
+#include "CharacterNames.h"
+#include "Frame.h"
+#include "HTMLAllCollection.h"
+#include "HTMLBodyElement.h"
+#include "HTMLCollection.h"
+#include "HTMLDocument.h"
+#include "HTMLElement.h"
+#include "HTMLIFrameElement.h"
+#include "HTMLNames.h"
+#include "JSDOMWindow.h"
+#include "JSDOMWindowCustom.h"
+#include "JSDOMWindowShell.h"
+#include "JSHTMLCollection.h"
+#include "SegmentedString.h"
+#include "DocumentParser.h"
+#include <runtime/Error.h>
+#include <runtime/JSCell.h>
+
+using namespace JSC;
+
+namespace WebCore {
+
+using namespace HTMLNames;
+
+bool JSHTMLDocument::canGetItemsForName(ExecState*, HTMLDocument* document, const Identifier& propertyName)
+{
+    AtomicStringImpl* atomicPropertyName = findAtomicString(propertyName);
+    return atomicPropertyName && (document->hasNamedItem(atomicPropertyName) || document->hasExtraNamedItem(atomicPropertyName));
+}
+
+JSValue JSHTMLDocument::nameGetter(ExecState* exec, JSValue slotBase, const Identifier& propertyName)
+{
+    JSHTMLDocument* thisObj = static_cast<JSHTMLDocument*>(asObject(slotBase));
+    HTMLDocument* document = static_cast<HTMLDocument*>(thisObj->impl());
+
+    String name = identifierToString(propertyName);
+    RefPtr<HTMLCollection> collection = document->documentNamedItems(name);
+
+    unsigned length = collection->length();
+    if (!length)
+        return jsUndefined();
+
+    if (length == 1) {
+        Node* node = collection->firstItem();
+
+        Frame* frame;
+        if (node->hasTagName(iframeTag) && (frame = static_cast<HTMLIFrameElement*>(node)->contentFrame()))
+            return toJS(exec, frame);
+
+        return toJS(exec, node);
+    } 
+
+    return toJS(exec, collection.get());
+}
+
+// Custom attributes
+
+JSValue JSHTMLDocument::all(ExecState* exec) const
+{
+    // If "all" has been overwritten, return the overwritten value
+    JSValue v = getDirect(Identifier(exec, "all"));
+    if (v)
+        return v;
+
+    return toJS(exec, static_cast<HTMLDocument*>(impl())->all().get());
+}
+
+void JSHTMLDocument::setAll(ExecState* exec, JSValue value)
+{
+    // Add "all" to the property map.
+    putDirect(Identifier(exec, "all"), value);
+}
+
+// Custom functions
+
+JSValue JSHTMLDocument::open(ExecState* exec)
+{
+    // For compatibility with other browsers, pass open calls with more than 2 parameters to the window.
+    if (exec->argumentCount() > 2) {
+        Frame* frame = static_cast<HTMLDocument*>(impl())->frame();
+        if (frame) {
+            JSDOMWindowShell* wrapper = toJSDOMWindowShell(frame, currentWorld(exec));
+            if (wrapper) {
+                JSValue function = wrapper->get(exec, Identifier(exec, "open"));
+                CallData callData;
+                CallType callType = ::getCallData(function, callData);
+                if (callType == CallTypeNone)
+                    return throwTypeError(exec);
+                return JSC::call(exec, function, callType, callData, wrapper, ArgList(exec));
+            }
+        }
+        return jsUndefined();
+    }
+
+    // document.open clobbers the security context of the document and
+    // aliases it with the active security context.
+    Document* activeDocument = asJSDOMWindow(exec->lexicalGlobalObject())->impl()->document();
+
+    // In the case of two parameters or fewer, do a normal document open.
+    static_cast<HTMLDocument*>(impl())->open(activeDocument);
+    return this;
+}
+
+enum NewlineRequirement { DoNotAddNewline, DoAddNewline };
+
+static inline void documentWrite(ExecState* exec, HTMLDocument* document, NewlineRequirement addNewline)
+{
+    // DOM only specifies single string argument, but browsers allow multiple or no arguments.
+
+    size_t size = exec->argumentCount();
+
+    UString firstString = exec->argument(0).toString(exec);
+    SegmentedString segmentedString = ustringToString(firstString);
+    if (size != 1) {
+        if (!size)
+            segmentedString.clear();
+        else {
+            for (size_t i = 1; i < size; ++i) {
+                UString subsequentString = exec->argument(i).toString(exec);
+                segmentedString.append(SegmentedString(ustringToString(subsequentString)));
+            }
+        }
+    }
+    if (addNewline)
+        segmentedString.append(SegmentedString(&newlineCharacter, 1));
+
+    Document* activeDocument = asJSDOMWindow(exec->lexicalGlobalObject())->impl()->document();
+    document->write(segmentedString, activeDocument);
+}
+
+JSValue JSHTMLDocument::write(ExecState* exec)
+{
+    documentWrite(exec, static_cast<HTMLDocument*>(impl()), DoNotAddNewline);
+    return jsUndefined();
+}
+
+JSValue JSHTMLDocument::writeln(ExecState* exec)
+{
+    documentWrite(exec, static_cast<HTMLDocument*>(impl()), DoAddNewline);
+    return jsUndefined();
+}
+
+} // namespace WebCore