1 /*
2 * This file is part of the XSL implementation.
3 *
4 * Copyright (C) 2004, 2005, 2006, 2007, 2008 Apple, Inc. All rights reserved.
5 * Copyright (C) 2005, 2006 Alexey Proskuryakov <ap@webkit.org>
6 *
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Library General Public
9 * License as published by the Free Software Foundation; either
10 * version 2 of the License, or (at your option) any later version.
11 *
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Library General Public License for more details.
16 *
17 * You should have received a copy of the GNU Library General Public License
18 * along with this library; see the file COPYING.LIB. If not, write to
19 * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
20 * Boston, MA 02110-1301, USA.
21 */
22
23 #include "config.h"
24
25 #if ENABLE(XSLT)
26
27 #include "XSLTProcessor.h"
28
29 #include "DOMImplementation.h"
30 #include "CachedResourceLoader.h"
31 #include "DocumentFragment.h"
32 #include "Frame.h"
33 #include "FrameLoader.h"
34 #include "FrameView.h"
35 #include "HTMLBodyElement.h"
36 #include "HTMLDocument.h"
37 #include "Page.h"
38 #include "Text.h"
39 #include "TextResourceDecoder.h"
40 #include "markup.h"
41
42 #include <wtf/Assertions.h>
43 #include <wtf/Vector.h>
44
45 namespace WebCore {
46
transformTextStringToXHTMLDocumentString(String & text)47 static inline void transformTextStringToXHTMLDocumentString(String& text)
48 {
49 // Modify the output so that it is a well-formed XHTML document with a <pre> tag enclosing the text.
50 text.replace('&', "&");
51 text.replace('<', "<");
52 text = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n"
53 "<!DOCTYPE html PUBLIC \"-//W3C//DTD XHTML 1.0 Strict//EN\" \"http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd\">\n"
54 "<html xmlns=\"http://www.w3.org/1999/xhtml\">\n"
55 "<head><title/></head>\n"
56 "<body>\n"
57 "<pre>" + text + "</pre>\n"
58 "</body>\n"
59 "</html>\n";
60 }
61
~XSLTProcessor()62 XSLTProcessor::~XSLTProcessor()
63 {
64 // Stylesheet shouldn't outlive its root node.
65 ASSERT(!m_stylesheetRootNode || !m_stylesheet || m_stylesheet->hasOneRef());
66 }
67
createDocumentFromSource(const String & sourceString,const String & sourceEncoding,const String & sourceMIMEType,Node * sourceNode,Frame * frame)68 PassRefPtr<Document> XSLTProcessor::createDocumentFromSource(const String& sourceString,
69 const String& sourceEncoding, const String& sourceMIMEType, Node* sourceNode, Frame* frame)
70 {
71 RefPtr<Document> ownerDocument = sourceNode->document();
72 bool sourceIsDocument = (sourceNode == ownerDocument.get());
73 String documentSource = sourceString;
74
75 RefPtr<Document> result;
76 if (sourceMIMEType == "text/plain") {
77 result = Document::create(frame, sourceIsDocument ? ownerDocument->url() : KURL());
78 transformTextStringToXHTMLDocumentString(documentSource);
79 } else
80 result = DOMImplementation::createDocument(sourceMIMEType, frame, sourceIsDocument ? ownerDocument->url() : KURL(), false);
81
82 // Before parsing, we need to save & detach the old document and get the new document
83 // in place. We have to do this only if we're rendering the result document.
84 if (frame) {
85 if (FrameView* view = frame->view())
86 view->clear();
87 result->setTransformSourceDocument(frame->document());
88 frame->setDocument(result);
89 }
90
91 RefPtr<TextResourceDecoder> decoder = TextResourceDecoder::create(sourceMIMEType);
92 decoder->setEncoding(sourceEncoding.isEmpty() ? UTF8Encoding() : TextEncoding(sourceEncoding), TextResourceDecoder::EncodingFromXMLHeader);
93 result->setDecoder(decoder.release());
94
95 result->setContent(documentSource);
96
97 return result.release();
98 }
99
createFragmentFromSource(const String & sourceString,const String & sourceMIMEType,Document * outputDoc)100 static inline RefPtr<DocumentFragment> createFragmentFromSource(const String& sourceString, const String& sourceMIMEType, Document* outputDoc)
101 {
102 RefPtr<DocumentFragment> fragment = outputDoc->createDocumentFragment();
103
104 if (sourceMIMEType == "text/html") {
105 // As far as I can tell, there isn't a spec for how transformToFragment
106 // is supposed to work. Based on the documentation I can find, it looks
107 // like we want to start parsing the fragment in the InBody insertion
108 // mode. Unfortunately, that's an implementation detail of the parser.
109 // We achieve that effect here by passing in a fake body element as
110 // context for the fragment.
111 RefPtr<HTMLBodyElement> fakeBody = HTMLBodyElement::create(outputDoc);
112 fragment->parseHTML(sourceString, fakeBody.get());
113 } else if (sourceMIMEType == "text/plain")
114 fragment->parserAddChild(Text::create(outputDoc, sourceString));
115 else {
116 bool successfulParse = fragment->parseXML(sourceString, 0);
117 if (!successfulParse)
118 return 0;
119 }
120
121 // FIXME: Do we need to mess with URLs here?
122
123 return fragment;
124 }
125
transformToDocument(Node * sourceNode)126 PassRefPtr<Document> XSLTProcessor::transformToDocument(Node* sourceNode)
127 {
128 String resultMIMEType;
129 String resultString;
130 String resultEncoding;
131 if (!transformToString(sourceNode, resultMIMEType, resultString, resultEncoding))
132 return 0;
133 return createDocumentFromSource(resultString, resultEncoding, resultMIMEType, sourceNode, 0);
134 }
135
transformToFragment(Node * sourceNode,Document * outputDoc)136 PassRefPtr<DocumentFragment> XSLTProcessor::transformToFragment(Node* sourceNode, Document* outputDoc)
137 {
138 String resultMIMEType;
139 String resultString;
140 String resultEncoding;
141
142 // If the output document is HTML, default to HTML method.
143 if (outputDoc->isHTMLDocument())
144 resultMIMEType = "text/html";
145
146 if (!transformToString(sourceNode, resultMIMEType, resultString, resultEncoding))
147 return 0;
148 return createFragmentFromSource(resultString, resultMIMEType, outputDoc);
149 }
150
setParameter(const String &,const String & localName,const String & value)151 void XSLTProcessor::setParameter(const String& /*namespaceURI*/, const String& localName, const String& value)
152 {
153 // FIXME: namespace support?
154 // should make a QualifiedName here but we'd have to expose the impl
155 m_parameters.set(localName, value);
156 }
157
getParameter(const String &,const String & localName) const158 String XSLTProcessor::getParameter(const String& /*namespaceURI*/, const String& localName) const
159 {
160 // FIXME: namespace support?
161 // should make a QualifiedName here but we'd have to expose the impl
162 return m_parameters.get(localName);
163 }
164
removeParameter(const String &,const String & localName)165 void XSLTProcessor::removeParameter(const String& /*namespaceURI*/, const String& localName)
166 {
167 // FIXME: namespace support?
168 m_parameters.remove(localName);
169 }
170
reset()171 void XSLTProcessor::reset()
172 {
173 m_stylesheet.clear();
174 m_stylesheetRootNode.clear();
175 m_parameters.clear();
176 }
177
178 } // namespace WebCore
179
180 #endif // ENABLE(XSLT)
181