5074606b46701f82533b375447e061475c62764c
[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_dump_editing_callbacks(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 WebFrame : public QWebFrame {
59 public:
60     WebFrame(QWebPage *parent, QWebFrameData *frameData)
61         : QWebFrame(parent, frameData) {}
62     WebFrame(QWebFrame *parent, QWebFrameData *frameData)
63         : QWebFrame(parent, frameData) {}
64 };
65
66 class WebPage : public QWebPage {
67 public:
68     WebPage(QWidget *parent, DumpRenderTree *drt);
69
70     QWebFrame *createFrame(QWebFrame *parentFrame, QWebFrameData *frameData);
71     QWebPage *createWindow();
72
73     void javaScriptAlert(QWebFrame *frame, const QString& message);
74     void javaScriptConsoleMessage(const QString& message, unsigned int lineNumber, const QString& sourceID);
75
76 private:
77     DumpRenderTree *m_drt;
78 };
79
80 WebPage::WebPage(QWidget *parent, DumpRenderTree *drt)
81     : QWebPage(parent), m_drt(drt)
82 {
83     QWebSettings s = settings();
84     s.setAttribute(QWebSettings::JavascriptCanOpenWindows, true);
85     setSettings(s);
86 }
87
88 QWebFrame *WebPage::createFrame(QWebFrame *parentFrame, QWebFrameData *frameData)
89 {
90     if (parentFrame) {
91         WebFrame *f = new WebFrame(parentFrame, frameData);
92         connect(f, SIGNAL(cleared()), m_drt, SLOT(initJSObjects()));
93         connect(f, SIGNAL(provisionalLoad()),
94                 m_drt->layoutTestController(), SLOT(provisionalLoad()));
95         return f;
96     }
97     WebFrame *f = new WebFrame(this, frameData);
98
99     connect(f, SIGNAL(cleared()), m_drt, SLOT(initJSObjects()));
100     connect(f, SIGNAL(provisionalLoad()),
101             m_drt->layoutTestController(), SLOT(provisionalLoad()));
102     connect(f, SIGNAL(loadDone(bool)),
103             m_drt->layoutTestController(), SLOT(maybeDump(bool)));
104
105     return f;
106 }
107
108 QWebPage *WebPage::createWindow()
109 {
110     return m_drt->createWindow();
111 }
112
113 void WebPage::javaScriptAlert(QWebFrame *frame, const QString& message)
114 {
115     fprintf(stdout, "ALERT: %s\n", message.toUtf8().constData());
116 }
117
118 void WebPage::javaScriptConsoleMessage(const QString& message, unsigned int lineNumber, const QString&)
119 {
120     fprintf (stdout, "CONSOLE MESSAGE: line %d: %s\n", lineNumber, message.toUtf8().constData());
121 }
122
123 DumpRenderTree::DumpRenderTree()
124     : m_stdin(0)
125     , m_notifier(0)
126 {
127     m_controller = new LayoutTestController(this);
128     connect(m_controller, SIGNAL(done()), this, SLOT(dump()), Qt::QueuedConnection);
129
130     m_page = new WebPage(0, this);
131     m_page->resize(maxViewWidth, maxViewHeight);
132     m_page->mainFrame()->setHorizontalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
133     m_page->mainFrame()->setVerticalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
134     connect(m_page, SIGNAL(titleChanged(const QString&)),
135             SLOT(titleChanged(const QString&)));
136
137     m_eventSender = new EventSender(m_page);
138
139     QObject::connect(this, SIGNAL(quit()), qApp, SLOT(quit()), Qt::QueuedConnection);
140     QFocusEvent event(QEvent::FocusIn, Qt::ActiveWindowFocusReason);
141     QApplication::sendEvent(m_page, &event);
142 }
143
144 DumpRenderTree::~DumpRenderTree()
145 {
146     delete m_page;
147
148     delete m_stdin;
149     delete m_notifier;
150 }
151
152 void DumpRenderTree::open()
153 {
154     if (!m_stdin) {
155         m_stdin = new QFile;
156         m_stdin->open(stdin, QFile::ReadOnly);
157     }
158
159     if (!m_notifier) {
160         m_notifier = new QSocketNotifier(STDIN_FILENO, QSocketNotifier::Read);
161         connect(m_notifier, SIGNAL(activated(int)), this, SLOT(readStdin(int)));
162     }
163 }
164
165 void DumpRenderTree::open(const QUrl& url)
166 {
167     resetJSObjects();
168     m_page->open(url);
169 }
170
171 void DumpRenderTree::readStdin(int /* socket */)
172 {
173     // Read incoming data from stdin...
174     QByteArray line = m_stdin->readLine();
175     if (line.endsWith('\n'))
176         line.truncate(line.size()-1);
177     //fprintf(stderr, "\n    opening %s\n", line.constData());
178     if (line.isEmpty())
179         quit();
180     QFileInfo fi(line);
181     open(QUrl::fromLocalFile(fi.absoluteFilePath()));
182     fflush(stdout);
183 }
184
185 void DumpRenderTree::resetJSObjects()
186 {
187     m_controller->reset();
188     foreach(QWidget *widget, windows)
189         delete widget;
190     windows.clear();
191 }
192
193 void DumpRenderTree::initJSObjects()
194 {
195     QWebFrame *frame = qobject_cast<QWebFrame*>(sender());
196     Q_ASSERT(frame);
197     frame->addToJSWindowObject("layoutTestController", m_controller);
198     frame->addToJSWindowObject("eventSender", m_eventSender);
199 }
200
201
202 QString DumpRenderTree::dumpFramesAsText(QWebFrame* frame)
203 {
204     if (!frame)
205         return QString();
206
207     QString result;
208     QWebFrame *parent = qobject_cast<QWebFrame *>(frame->parent());
209     if (parent) {
210         result.append(QLatin1String("\n--------\nFrame: '"));
211         result.append(frame->name());
212         result.append(QLatin1String("'\n--------\n"));
213     }
214
215     result.append(frame->innerText());
216     result.append(QLatin1String("\n"));
217
218     if (m_controller->shouldDumpChildrenAsText()) {
219         QList<QWebFrame *> children = frame->childFrames();
220         for (int i = 0; i < children.size(); ++i)
221             result += dumpFramesAsText(children.at(i));
222     }
223
224     return result;
225 }
226
227 void DumpRenderTree::dump()
228 {
229     QWebFrame *frame = m_page->mainFrame();
230
231     //fprintf(stderr, "    Dumping\n");
232     if (!m_notifier) {
233         // Dump markup in single file mode...
234         QString markup = frame->markup();
235         fprintf(stdout, "Source:\n\n%s\n", markup.toUtf8().constData());
236     }
237
238     // Dump render text...
239     QString renderDump;
240     if (m_controller->shouldDumpAsText()) {
241         renderDump = dumpFramesAsText(frame);
242     } else {
243         renderDump = frame->renderTreeDump();
244     }
245     if (renderDump.isEmpty()) {
246         printf("ERROR: nil result from %s", m_controller->shouldDumpAsText() ? "[documentElement innerText]" : "[frame renderTreeAsExternalRepresentation]");
247     } else {
248         fprintf(stdout, "%s", renderDump.toUtf8().constData());
249     }
250
251     fprintf(stdout, "#EOF\n");
252
253     fflush(stdout);
254
255     if (!m_notifier) {
256         // Exit now in single file mode...
257         quit();
258     }
259 }
260
261 void DumpRenderTree::titleChanged(const QString &s)
262 {
263     if (m_controller->shouldDumpTitleChanges())
264         printf("TITLE CHANGED: %s\n", s.toUtf8().data());
265 }
266
267
268 QWebPage *DumpRenderTree::createWindow()
269 {
270     if (!m_controller->canOpenWindows())
271         return 0;
272     QWidget *container = new QWidget(0);
273     container->resize(0, 0);
274     container->move(-1, -1);
275     container->hide();
276     QWebPage *page = new WebPage(container, this);
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 }