2007-10-20 Rodney Dawes <dobey@wayofthemonkey.com>
[WebKit-https.git] / WebCore / bindings / js / JSDocumentCustom.cpp
1 /*
2  * Copyright (C) 2007 Apple, Inc. All rights reserved.
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Library General Public
6  * License as published by the Free Software Foundation; either
7  * version 2 of the License, or (at your option) any later version.
8  *
9  * This library is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Library General Public License for more details.
13  *
14  * You should have received a copy of the GNU Library General Public License
15  * along with this library; see the file COPYING.LIB.  If not, write to
16  * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
17  * Boston, MA 02110-1301, USA.
18  */
19
20 #include "config.h"
21 #include "JSDocument.h"
22
23 #include "Document.h"
24 #include "Frame.h"
25 #include "FrameLoader.h"
26 #include "HTMLDocument.h"
27 #include "JSHTMLDocument.h"
28 #include "kjs_binding.h"
29 #include "kjs_window.h"
30
31 #if ENABLE(SVG)
32 #include "JSSVGDocument.h"
33 #include "SVGDocument.h"
34 #endif
35
36 namespace WebCore {
37
38 using namespace KJS;
39
40 void JSDocument::mark()
41 {
42     DOMObject::mark();
43     ScriptInterpreter::markDOMNodesForDocument(static_cast<Document*>(impl()));
44 }
45
46 JSValue* JSDocument::location(ExecState* exec) const
47 {
48     Frame* frame = static_cast<Document*>(impl())->frame();
49     if (!frame)
50         return jsNull();
51
52     KJS::Window* win = KJS::Window::retrieveWindow(frame);
53     ASSERT(win);
54     return win->location();
55 }
56
57 void JSDocument::setLocation(ExecState* exec, JSValue* value)
58 {
59     Frame* frame = static_cast<Document*>(impl())->frame();
60     if (!frame)
61         return;
62
63     String str = value->toString(exec);
64
65     // IE and Mozilla both resolve the URL relative to the source frame,
66     // not the target frame.
67     Frame* activeFrame = static_cast<ScriptInterpreter*>(exec->dynamicInterpreter())->frame();
68     if (activeFrame)
69         str = activeFrame->document()->completeURL(str);
70
71     bool userGesture = static_cast<ScriptInterpreter*>(exec->dynamicInterpreter())->wasRunByUserGesture();
72     frame->loader()->scheduleLocationChange(str, activeFrame->loader()->outgoingReferrer(), false, userGesture);
73 }
74
75 JSValue* toJS(ExecState* exec, Document* doc)
76 {
77     if (!doc)
78         return jsNull();
79
80     ScriptInterpreter* interp = static_cast<ScriptInterpreter*>(exec->dynamicInterpreter());
81     JSDocument* ret =  static_cast<JSDocument*>(interp->getDOMObject(doc));
82     if (ret)
83         return ret;
84
85     if (doc->isHTMLDocument())
86         ret = new JSHTMLDocument(exec, static_cast<HTMLDocument*>(doc));
87 #if ENABLE(SVG)
88     else if (doc->isSVGDocument())
89         ret = new JSSVGDocument(exec, static_cast<SVGDocument*>(doc));
90 #endif
91     else
92         ret = new JSDocument(exec, doc);
93
94     // Make sure the document is kept around by the window object, and works right with the
95     // back/forward cache.
96     if (doc->frame())
97         KJS::Window::retrieveWindow(doc->frame())->putDirect("document", ret, DontDelete|ReadOnly);
98     else {
99         size_t nodeCount = 0;
100         for (Node* n = doc; n; n = n->traverseNextNode())
101             nodeCount++;
102         
103         Collector::reportExtraMemoryCost(nodeCount * sizeof(Node));
104     }
105
106     interp->putDOMObject(doc, ret);
107
108     return ret;
109 }
110
111 } // namespace WebCore