Polish the XML error message so that it indicates when a document is the result...
[WebKit-https.git] / WebCore / khtml / xsl / xslt_processorimpl.cpp
1 /**
2  * This file is part of the XSL implementation.
3  *
4  * Copyright (C) 2004 Apple Computer, Inc.
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Library General Public
8  * License as published by the Free Software Foundation; either
9  * version 2 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Library General Public License for more details.
15  *
16  * You should have received a copy of the GNU Library General Public License
17  * along with this library; see the file COPYING.LIB.  If not, write to
18  * the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
19  * Boston, MA 02111-1307, USA.
20  */
21
22 #ifdef KHTML_XSLT
23
24 #include "xslt_processorimpl.h"
25 #include "xsl_stylesheetimpl.h"
26 #include "html_documentimpl.h"
27 #include "loader.h"
28 #include "khtmlview.h"
29 #include "khtml_part.h"
30
31 #include <libxslt/xsltutils.h>
32 #include <libxslt/documents.h>
33 #include <libxslt/imports.h>
34
35 using namespace khtml;
36 using namespace DOM;
37
38 namespace DOM {
39     
40 XSLTProcessorImpl::XSLTProcessorImpl(XSLStyleSheetImpl* sheet, DocumentImpl* source)
41 :m_stylesheet(sheet), m_sourceDocument(source)
42 {
43     if (m_stylesheet)
44         m_stylesheet->ref();
45     if (m_sourceDocument)
46         m_sourceDocument->ref();
47 }
48
49 XSLTProcessorImpl::~XSLTProcessorImpl()
50 {
51     if (m_stylesheet)
52         m_stylesheet->deref();
53     if (m_sourceDocument)
54         m_sourceDocument->deref();
55 }
56
57 #ifdef NOT_YET_READY
58 static XSLStyleSheetImpl* globalSheet = 0;
59 static xmlDocPtr stylesheetLoadFunc(const xmlChar* uri,
60                                     xmlDictPtr dict,
61                                     int options,
62                                     void* ctxt,
63                                     xsltLoadType type)
64 {
65     if (type != XSLT_LOAD_STYLESHEET)
66         return NULL; // FIXME: Add support for XSLT_LOAD_DOCUMENT for the document() function.
67     
68     if (!globalSheet)
69         return NULL;
70     return globalSheet->locateStylesheetSubResource(((xsltStylesheetPtr)ctxt)->doc, uri);
71 }
72 #endif
73
74 DocumentImpl* XSLTProcessorImpl::transformDocument(DocumentImpl* doc)
75 {
76     // FIXME: Right now we assume |doc| is unparsed, but if it has been parsed we will need to serialize it
77     // and then feed that resulting source to libxslt.
78     m_resultOutput = "";
79
80     if (!m_stylesheet || !m_stylesheet->document()) return 0;
81         
82 #ifdef NOT_YET_READY
83     globalSheet = m_stylesheet;
84     xsltSetLoaderFunc(stylesheetLoadFunc);
85 #endif
86
87     xsltStylesheetPtr sheet = m_stylesheet->compileStyleSheet();
88
89 #ifdef NOT_YET_READY
90     globalSheet = 0;
91     xsltSetLoaderFunc(0);
92 #endif
93
94     if (!sheet) return 0;
95     m_stylesheet->clearDocuments();
96   
97     // Get the parsed source document.
98     xmlDocPtr sourceDoc = (xmlDocPtr)doc->transformSource();
99     xmlDocPtr resultDoc = xsltApplyStylesheet(sheet, sourceDoc, NULL);
100     DocumentImpl* result = documentFromXMLDocPtr(resultDoc, sheet);
101     xsltFreeStylesheet(sheet);
102     return result;
103 }
104
105 static int bufferWrite(void* context, const char* buffer, int len)
106 {
107     static_cast<XSLTProcessorImpl*>(context)->addToResult(buffer, len);
108     return len;
109 }
110
111 void XSLTProcessorImpl::addToResult(const char* buffer, int len)
112 {
113     m_resultOutput += QString(buffer, len);
114 }
115
116 DocumentImpl* XSLTProcessorImpl::documentFromXMLDocPtr(xmlDocPtr resultDoc, xsltStylesheetPtr sheet)
117 {
118     // FIXME: For now we serialize and then reparse.  It might be more optimal to write a DOM
119     // converter.
120     if (!resultDoc || !sheet) return 0;
121     DocumentImpl* result = 0;
122     xmlOutputBufferPtr outputBuf = xmlAllocOutputBuffer(NULL);
123     if (outputBuf) {
124         outputBuf->context = this;
125         outputBuf->writecallback = bufferWrite;
126         
127         if (xsltSaveResultTo(outputBuf, resultDoc, sheet) < 0)
128             return 0;
129         
130         // There are three types of output we need to be able to deal with:
131         // HTML (create an HTML document), XML (create an XML document), and text (wrap in a <pre> and
132         // make an XML document).
133         KHTMLView* view = m_sourceDocument->view();
134         const xmlChar* method;
135         XSLT_GET_IMPORT_PTR(method, sheet, method);
136         if (method == NULL && resultDoc->type == XML_HTML_DOCUMENT_NODE)
137             method = (const xmlChar*)"html";
138         if (xmlStrEqual(method, (const xmlChar*)"html"))
139             result = m_sourceDocument->implementation()->createHTMLDocument(view);
140         else
141             result = m_sourceDocument->implementation()->createDocument(view);
142         result->attach();
143         result->setURL(m_sourceDocument->URL());
144         result->setBaseURL(m_sourceDocument->baseURL());
145         result->setDecoder(m_sourceDocument->decoder()); // FIXME: Should just be UTF-16.
146         result->docLoader()->setShowAnimations(m_sourceDocument->docLoader()->showAnimations());
147         result->setTransformSourceDocument(m_sourceDocument);
148
149         if (xmlStrEqual(method, (const xmlChar*)"text")) {
150             // Modify the output so that it is a well-formed XHTML document with a <pre> tag enclosing
151             // the text.
152             QString beforeString("<!DOCTYPE html PUBLIC \"-//W3C//DTD XHTML 1.0 Strict//EN\" \"http://www.w3.org/TR/xhtml1/DTD/strict.dtd\">\n<html xmlns=\"http://www.w3.org/1999/xhtml\">\n<body>\n<pre>\n<![CDATA[");
153             QString afterString("]]>\n</pre>\n</body>\n</html>\n");
154             m_resultOutput = beforeString + m_resultOutput + afterString;
155         }
156         
157         // Before parsing, we need to detach the old document completely and get the new document
158         // in place.  We have to do this only if we're rendering the result document.
159         if (view) {
160             view->clear();
161             view->part()->replaceDocImpl(result);
162         }
163         
164         result->open();
165         result->determineParseMode(m_resultOutput); // Make sure we parse in the correct mode.
166         result->write(m_resultOutput);
167         result->finishParsing();
168         if (view)
169             view->part()->checkCompleted();
170         else
171             result->close(); // FIXME: Even viewless docs can load subresources. onload will fire too early.
172                              // This is probably a bug in XMLHttpRequestObjects as well.
173     }
174     return result;
175 }
176
177 }
178 #endif