Reworked the QWebSettings API.
[WebKit-https.git] / WebKitTools / DumpRenderTree / qt / DumpRenderTree.cpp
1 /*
2  * Copyright (C) 2005, 2006 Apple Computer, Inc.  All rights reserved.
3  * Copyright (C) 2006 Nikolas Zimmermann <zimmermann@kde.org>
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  *
9  * 1.  Redistributions of source code must retain the above copyright
10  *     notice, this list of conditions and the following disclaimer.
11  * 2.  Redistributions in binary form must reproduce the above copyright
12  *     notice, this list of conditions and the following disclaimer in the
13  *     documentation and/or other materials provided with the distribution.
14  * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of
15  *     its contributors may be used to endorse or promote products derived
16  *     from this software without specific prior written permission.
17  *
18  * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
19  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
20  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
21  * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
22  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
23  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
24  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
25  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
27  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28  */
29
30 #include "DumpRenderTree.h"
31 #include "jsobjects.h"
32
33 #include <QDir>
34 #include <QFile>
35 #include <QTimer>
36 #include <QBoxLayout>
37 #include <QScrollArea>
38 #include <QApplication>
39 #include <QUrl>
40 #include <QFocusEvent>
41
42 #include <qwebpage.h>
43 #include <qwebframe.h>
44 #include <qwebsettings.h>
45
46 #include <unistd.h>
47 #include <qdebug.h>
48 extern void qt_drt_run(bool b);
49 extern void qt_dump_set_accepts_editing(bool b);
50
51
52 namespace WebCore {
53
54 // Choose some default values.
55 const unsigned int maxViewWidth = 800;
56 const unsigned int maxViewHeight = 600;
57
58 class WebPage : public QWebPage {
59 public:
60     WebPage(QWidget *parent, DumpRenderTree *drt);
61
62     QWebPage *createWindow();
63
64     void javaScriptAlert(QWebFrame *frame, const QString& message);
65     void javaScriptConsoleMessage(const QString& message, unsigned int lineNumber, const QString& sourceID);
66     bool javaScriptConfirm(QWebFrame *frame, const QString& msg);
67     bool javaScriptPrompt(QWebFrame *frame, const QString& msg, const QString& defaultValue, QString* result);
68
69 private:
70     DumpRenderTree *m_drt;
71 };
72
73 WebPage::WebPage(QWidget *parent, DumpRenderTree *drt)
74     : QWebPage(parent), m_drt(drt)
75 {
76     settings()->setAttribute(QWebSettings::JavascriptCanOpenWindows, true);
77 }
78
79 QWebPage *WebPage::createWindow()
80 {
81     return m_drt->createWindow();
82 }
83
84 void WebPage::javaScriptAlert(QWebFrame *frame, const QString& message)
85 {
86     fprintf(stdout, "ALERT: %s\n", message.toUtf8().constData());
87 }
88
89 void WebPage::javaScriptConsoleMessage(const QString& message, unsigned int lineNumber, const QString&)
90 {
91     fprintf (stdout, "CONSOLE MESSAGE: line %d: %s\n", lineNumber, message.toUtf8().constData());
92 }
93
94 bool WebPage::javaScriptConfirm(QWebFrame *frame, const QString& msg)
95 {
96     fprintf(stdout, "CONFIRM: %s\n", msg.toUtf8().constData());
97     return true;
98 }
99
100 bool WebPage::javaScriptPrompt(QWebFrame *frame, const QString& msg, const QString& defaultValue, QString* result)
101 {
102     fprintf(stdout, "PROMPT: %s, default text: %s\n", msg.toUtf8().constData(), defaultValue.toUtf8().constData());
103     *result = defaultValue;
104     return true;
105 }
106
107 DumpRenderTree::DumpRenderTree()
108     : m_stdin(0)
109     , m_notifier(0)
110 {
111     m_controller = new LayoutTestController(this);
112     connect(m_controller, SIGNAL(done()), this, SLOT(dump()), Qt::QueuedConnection);
113
114     m_page = new WebPage(0, this);
115     connect(m_page, SIGNAL(frameCreated(QWebFrame *)), this, SLOT(connectFrame(QWebFrame *)));
116     m_page->resize(maxViewWidth, maxViewHeight);
117     m_page->mainFrame()->setHorizontalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
118     m_page->mainFrame()->setVerticalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
119     connect(m_page, SIGNAL(titleChanged(const QString&)),
120             SLOT(titleChanged(const QString&)));
121
122     m_eventSender = new EventSender(m_page);
123     m_textInputController = new TextInputController(m_page);
124
125     QObject::connect(this, SIGNAL(quit()), qApp, SLOT(quit()), Qt::QueuedConnection);
126     qt_drt_run(true);
127     QFocusEvent event(QEvent::FocusIn, Qt::ActiveWindowFocusReason);
128     QApplication::sendEvent(m_page, &event);
129 }
130
131 DumpRenderTree::~DumpRenderTree()
132 {
133     delete m_page;
134
135     delete m_stdin;
136     delete m_notifier;
137 }
138
139 void DumpRenderTree::open()
140 {
141     if (!m_stdin) {
142         m_stdin = new QFile;
143         m_stdin->open(stdin, QFile::ReadOnly);
144     }
145
146     if (!m_notifier) {
147         m_notifier = new QSocketNotifier(STDIN_FILENO, QSocketNotifier::Read);
148         connect(m_notifier, SIGNAL(activated(int)), this, SLOT(readStdin(int)));
149     }
150 }
151
152 void DumpRenderTree::open(const QUrl& url)
153 {
154     resetJSObjects();
155     m_page->open(url);
156 }
157
158 void DumpRenderTree::readStdin(int /* socket */)
159 {
160     // Read incoming data from stdin...
161     QByteArray line = m_stdin->readLine();
162     if (line.endsWith('\n'))
163         line.truncate(line.size()-1);
164     //fprintf(stderr, "\n    opening %s\n", line.constData());
165     if (line.isEmpty())
166         quit();
167     QFileInfo fi(line);
168     open(QUrl::fromLocalFile(fi.absoluteFilePath()));
169     fflush(stdout);
170 }
171
172 void DumpRenderTree::resetJSObjects()
173 {
174     m_controller->reset();
175     foreach(QWidget *widget, windows)
176         delete widget;
177     windows.clear();
178 }
179
180 void DumpRenderTree::initJSObjects()
181 {
182     QWebFrame *frame = qobject_cast<QWebFrame*>(sender());
183     Q_ASSERT(frame);
184     frame->addToJSWindowObject("layoutTestController", m_controller);
185     frame->addToJSWindowObject("eventSender", m_eventSender);
186     frame->addToJSWindowObject("textInputController", m_textInputController);
187 }
188
189
190 QString DumpRenderTree::dumpFramesAsText(QWebFrame* frame)
191 {
192     if (!frame)
193         return QString();
194
195     QString result;
196     QWebFrame *parent = qobject_cast<QWebFrame *>(frame->parent());
197     if (parent) {
198         result.append(QLatin1String("\n--------\nFrame: '"));
199         result.append(frame->name());
200         result.append(QLatin1String("'\n--------\n"));
201     }
202
203     result.append(frame->innerText());
204     result.append(QLatin1String("\n"));
205
206     if (m_controller->shouldDumpChildrenAsText()) {
207         QList<QWebFrame *> children = frame->childFrames();
208         for (int i = 0; i < children.size(); ++i)
209             result += dumpFramesAsText(children.at(i));
210     }
211
212     return result;
213 }
214
215 void DumpRenderTree::dump()
216 {
217     QWebFrame *frame = m_page->mainFrame();
218
219     //fprintf(stderr, "    Dumping\n");
220     if (!m_notifier) {
221         // Dump markup in single file mode...
222         QString markup = frame->markup();
223         fprintf(stdout, "Source:\n\n%s\n", markup.toUtf8().constData());
224     }
225
226     // Dump render text...
227     QString renderDump;
228     if (m_controller->shouldDumpAsText()) {
229         renderDump = dumpFramesAsText(frame);
230     } else {
231         renderDump = frame->renderTreeDump();
232     }
233     if (renderDump.isEmpty()) {
234         printf("ERROR: nil result from %s", m_controller->shouldDumpAsText() ? "[documentElement innerText]" : "[frame renderTreeAsExternalRepresentation]");
235     } else {
236         fprintf(stdout, "%s", renderDump.toUtf8().constData());
237     }
238
239     fprintf(stdout, "#EOF\n");
240
241     fflush(stdout);
242
243     if (!m_notifier) {
244         // Exit now in single file mode...
245         quit();
246     }
247 }
248
249 void DumpRenderTree::titleChanged(const QString &s)
250 {
251     if (m_controller->shouldDumpTitleChanges())
252         printf("TITLE CHANGED: %s\n", s.toUtf8().data());
253 }
254
255 void DumpRenderTree::connectFrame(QWebFrame *frame)
256 {
257     connect(frame, SIGNAL(cleared()), this, SLOT(initJSObjects()));
258     connect(frame, SIGNAL(provisionalLoad()),
259             layoutTestController(), SLOT(provisionalLoad()));
260
261     if (frame == m_page->mainFrame()) {
262         connect(frame, SIGNAL(loadDone(bool)),
263                 layoutTestController(), SLOT(maybeDump(bool)));
264     }
265 }
266
267 QWebPage *DumpRenderTree::createWindow()
268 {
269     if (!m_controller->canOpenWindows())
270         return 0;
271     QWidget *container = new QWidget(0);
272     container->resize(0, 0);
273     container->move(-1, -1);
274     container->hide();
275     QWebPage *page = new WebPage(container, this);
276     connect(m_page, SIGNAL(frameCreated(QWebFrame *)), this, SLOT(connectFrame(QWebFrame *)));
277     windows.append(container);
278     return page;
279 }
280
281 int DumpRenderTree::windowCount() const
282 {
283     int count = 0;
284     foreach(QWidget *w, windows) {
285         if (w->children().count())
286             ++count;
287     }
288     return count + 1;
289 }
290
291 }