f10c0ef8b1a54116a332c0fdf5d818925c665724
[WebKit-https.git] / Source / WebCore / xml / XSLTProcessor.cpp
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 "ContentSecurityPolicy.h"
32 #include "DocumentFragment.h"
33 #include "Frame.h"
34 #include "FrameLoader.h"
35 #include "FrameView.h"
36 #include "HTMLBodyElement.h"
37 #include "HTMLDocument.h"
38 #include "Page.h"
39 #include "SecurityOrigin.h"
40 #include "Text.h"
41 #include "TextResourceDecoder.h"
42 #include "markup.h"
43
44 #include <wtf/Assertions.h>
45 #include <wtf/Vector.h>
46
47 namespace WebCore {
48
49 static inline void transformTextStringToXHTMLDocumentString(String& text)
50 {
51     // Modify the output so that it is a well-formed XHTML document with a <pre> tag enclosing the text.
52     text.replaceWithLiteral('&', "&amp;");
53     text.replaceWithLiteral('<', "&lt;");
54     text = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n"
55         "<!DOCTYPE html PUBLIC \"-//W3C//DTD XHTML 1.0 Strict//EN\" \"http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd\">\n"
56         "<html xmlns=\"http://www.w3.org/1999/xhtml\">\n"
57         "<head><title/></head>\n"
58         "<body>\n"
59         "<pre>" + text + "</pre>\n"
60         "</body>\n"
61         "</html>\n";
62 }
63
64 XSLTProcessor::~XSLTProcessor()
65 {
66     // Stylesheet shouldn't outlive its root node.
67     ASSERT(!m_stylesheetRootNode || !m_stylesheet || m_stylesheet->hasOneRef());
68 }
69
70 PassRefPtr<Document> XSLTProcessor::createDocumentFromSource(const String& sourceString,
71     const String& sourceEncoding, const String& sourceMIMEType, Node* sourceNode, Frame* frame)
72 {
73     Ref<Document> ownerDocument(sourceNode->document());
74     bool sourceIsDocument = (sourceNode == &ownerDocument.get());
75     String documentSource = sourceString;
76
77     RefPtr<Document> result;
78     if (sourceMIMEType == "text/plain") {
79         result = Document::createXHTML(frame, sourceIsDocument ? ownerDocument->url() : URL());
80         transformTextStringToXHTMLDocumentString(documentSource);
81     } else
82         result = DOMImplementation::createDocument(sourceMIMEType, frame, sourceIsDocument ? ownerDocument->url() : URL());
83
84     // Before parsing, we need to save & detach the old document and get the new document
85     // in place. We have to do this only if we're rendering the result document.
86     if (frame) {
87         if (FrameView* view = frame->view())
88             view->clear();
89
90         if (Document* oldDocument = frame->document()) {
91             result->setTransformSourceDocument(oldDocument);
92             result->takeDOMWindowFrom(oldDocument);
93             result->setSecurityOrigin(oldDocument->securityOrigin());
94             result->setCookieURL(oldDocument->cookieURL());
95             result->setFirstPartyForCookies(oldDocument->firstPartyForCookies());
96             result->contentSecurityPolicy()->copyStateFrom(oldDocument->contentSecurityPolicy());
97         }
98
99         frame->setDocument(result);
100     }
101
102     RefPtr<TextResourceDecoder> decoder = TextResourceDecoder::create(sourceMIMEType);
103     decoder->setEncoding(sourceEncoding.isEmpty() ? UTF8Encoding() : TextEncoding(sourceEncoding), TextResourceDecoder::EncodingFromXMLHeader);
104     result->setDecoder(decoder.release());
105
106     result->setContent(documentSource);
107
108     return result.release();
109 }
110
111 PassRefPtr<Document> XSLTProcessor::transformToDocument(Node* sourceNode)
112 {
113     if (!sourceNode)
114         return 0;
115
116     String resultMIMEType;
117     String resultString;
118     String resultEncoding;
119     if (!transformToString(*sourceNode, resultMIMEType, resultString, resultEncoding))
120         return 0;
121     return createDocumentFromSource(resultString, resultEncoding, resultMIMEType, sourceNode, 0);
122 }
123
124 PassRefPtr<DocumentFragment> XSLTProcessor::transformToFragment(Node* sourceNode, Document* outputDoc)
125 {
126     if (!sourceNode || !outputDoc)
127         return 0;
128
129     String resultMIMEType;
130     String resultString;
131     String resultEncoding;
132
133     // If the output document is HTML, default to HTML method.
134     if (outputDoc->isHTMLDocument())
135         resultMIMEType = "text/html";
136
137     if (!transformToString(*sourceNode, resultMIMEType, resultString, resultEncoding))
138         return 0;
139     return createFragmentForTransformToFragment(resultString, resultMIMEType, outputDoc);
140 }
141
142 void XSLTProcessor::setParameter(const String& /*namespaceURI*/, const String& localName, const String& value)
143 {
144     // FIXME: namespace support?
145     // should make a QualifiedName here but we'd have to expose the impl
146     m_parameters.set(localName, value);
147 }
148
149 String XSLTProcessor::getParameter(const String& /*namespaceURI*/, const String& localName) const
150 {
151     // FIXME: namespace support?
152     // should make a QualifiedName here but we'd have to expose the impl
153     return m_parameters.get(localName);
154 }
155
156 void XSLTProcessor::removeParameter(const String& /*namespaceURI*/, const String& localName)
157 {
158     // FIXME: namespace support?
159     m_parameters.remove(localName);
160 }
161
162 void XSLTProcessor::reset()
163 {
164     m_stylesheet.clear();
165     m_stylesheetRootNode.clear();
166     m_parameters.clear();
167 }
168
169 } // namespace WebCore
170
171 #endif // ENABLE(XSLT)