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