/*
* Copyright (C) 2006, 2008, 2009 Apple Inc. All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
* EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE COMPUTER, INC. OR
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
* OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
#include "config.h"
#include "HTMLViewSourceDocument.h"
#include "DOMImplementation.h"
#include "HTMLAnchorElement.h"
#include "HTMLBodyElement.h"
#include "HTMLDivElement.h"
#include "HTMLHtmlElement.h"
#include "HTMLNames.h"
#include "HTMLTableCellElement.h"
#include "HTMLTableElement.h"
#include "HTMLTableRowElement.h"
#include "HTMLTableSectionElement.h"
#include "HTMLTokenizer.h"
#include "MappedAttribute.h"
#include "Text.h"
#include "TextDocument.h"
namespace WebCore {
using namespace HTMLNames;
HTMLViewSourceDocument::HTMLViewSourceDocument(Frame* frame, const String& mimeType)
: HTMLDocument(frame)
, m_type(mimeType)
{
setUsesBeforeAfterRules(true);
}
Tokenizer* HTMLViewSourceDocument::createTokenizer()
{
// Use HTMLTokenizer if applicable, otherwise use TextTokenizer.
if (m_type == "text/html" || m_type == "application/xhtml+xml" || m_type == "image/svg+xml" || implementation()->isXMLMIMEType(m_type)
#if ENABLE(XHTMLMP)
|| m_type == "application/vnd.wap.xhtml+xml"
#endif
) {
return new HTMLTokenizer(this);
}
return createTextTokenizer(this);
}
void HTMLViewSourceDocument::createContainingTable()
{
RefPtr<HTMLHtmlElement> html = new HTMLHtmlElement(htmlTag, this);
addChild(html);
html->attach();
RefPtr<HTMLBodyElement> body = new HTMLBodyElement(bodyTag, this);
html->addChild(body);
body->attach();
// Create a line gutter div that can be used to make sure the gutter extends down the height of the whole
// document.
RefPtr<HTMLDivElement> div = new HTMLDivElement(divTag, this);
RefPtr<NamedMappedAttrMap> attrs = NamedMappedAttrMap::create();
attrs->addAttribute(MappedAttribute::create(classAttr, "webkit-line-gutter-backdrop"));
div->setAttributeMap(attrs.release());
body->addChild(div);
div->attach();
RefPtr<HTMLTableElement> table = new HTMLTableElement(tableTag, this);
body->addChild(table);
table->attach();
m_tbody = new HTMLTableSectionElement(tbodyTag, this);
table->addChild(m_tbody);
m_tbody->attach();
m_current = m_tbody;
}
void HTMLViewSourceDocument::addViewSourceText(const String& text)
{
if (!m_current)
createContainingTable();
addText(text, "");
}
void HTMLViewSourceDocument::addViewSourceToken(Token* token)
{
if (!m_current)
createContainingTable();
if (token->tagName == textAtom)
addText(token->text.get(), "");
else if (token->tagName == commentAtom) {
if (token->beginTag) {
m_current = addSpanWithClassName("webkit-html-comment");
addText(String("<!--") + token->text.get() + "-->", "webkit-html-comment");
}
} else {
// Handle the tag.
String classNameStr = "webkit-html-tag";
m_current = addSpanWithClassName(classNameStr);
String text = "<";
if (!token->beginTag)
text += "/";
text += token->tagName;
Vector<UChar>* guide = token->m_sourceInfo.get();
if (!guide || !guide->size())
text += ">";
addText(text, classNameStr);
// Walk our guide string that tells us where attribute names/values should go.
if (guide && guide->size()) {
unsigned size = guide->size();
unsigned begin = 0;
unsigned currAttr = 0;
RefPtr<Attribute> attr = 0;
for (unsigned i = 0; i < size; i++) {
if (guide->at(i) == 'a' || guide->at(i) == 'x' || guide->at(i) == 'v') {
// Add in the string.
addText(String(static_cast<UChar*>(guide->data()) + begin, i - begin), classNameStr);
begin = i + 1;
if (guide->at(i) == 'a') {
if (token->attrs && currAttr < token->attrs->length())
attr = token->attrs->attributeItem(currAttr++);
else
attr = 0;
}
if (attr) {
if (guide->at(i) == 'a') {
String name = attr->name().toString();
m_current = addSpanWithClassName("webkit-html-attribute-name");
addText(name, "webkit-html-attribute-name");
if (m_current != m_tbody)
m_current = static_cast<Element*>(m_current->parent());
} else {
const String& value = attr->value().string();
// Compare ignoring case since HTMLTokenizer doesn't
// lower names when passing in tokens to
// HTMLViewSourceDocument.
if (equalIgnoringCase(token->tagName, "base") && equalIgnoringCase(attr->name().localName(), "href")) {
// Catch the href attribute in the base element.
// It will be used for rendering anchors created
// by addLink() below.
setBaseElementURL(KURL(url(), value));
}
// FIXME: XML could use namespace prefixes and confuse us.
if (equalIgnoringCase(attr->name().localName(), "src") || equalIgnoringCase(attr->name().localName(), "href"))
m_current = addLink(value, equalIgnoringCase(token->tagName, "a"));
else
m_current = addSpanWithClassName("webkit-html-attribute-value");
addText(value, "webkit-html-attribute-value");
if (m_current != m_tbody)
m_current = static_cast<Element*>(m_current->parent());
}
}
}
}
// Add in any string that might be left.
if (begin < size)
addText(String(static_cast<UChar*>(guide->data()) + begin, size - begin), classNameStr);
// Add in the end tag.
addText(">", classNameStr);
}
m_current = m_td;
}
}
void HTMLViewSourceDocument::addViewSourceDoctypeToken(DoctypeToken* doctypeToken)
{
if (!m_current)
createContainingTable();
m_current = addSpanWithClassName("webkit-html-doctype");
String text = "<";
text += String::adopt(doctypeToken->m_source);
text += ">";
addText(text, "webkit-html-doctype");
}
PassRefPtr<Element> HTMLViewSourceDocument::addSpanWithClassName(const String& className)
{
if (m_current == m_tbody) {
addLine(className);
return m_current;
}
RefPtr<HTMLElement> span = HTMLElement::create(spanTag, this);
RefPtr<NamedMappedAttrMap> attrs = NamedMappedAttrMap::create();
attrs->addAttribute(MappedAttribute::create(classAttr, className));
span->setAttributeMap(attrs.release());
m_current->addChild(span);
span->attach();
return span.release();
}
void HTMLViewSourceDocument::addLine(const String& className)
{
// Create a table row.
RefPtr<HTMLTableRowElement> trow = new HTMLTableRowElement(trTag, this);
m_tbody->addChild(trow);
trow->attach();
// Create a cell that will hold the line number (it is generated in the stylesheet using counters).
RefPtr<HTMLTableCellElement> td = new HTMLTableCellElement(tdTag, this);
RefPtr<NamedMappedAttrMap> attrs = NamedMappedAttrMap::create();
attrs->addAttribute(MappedAttribute::create(classAttr, "webkit-line-number"));
td->setAttributeMap(attrs.release());
trow->addChild(td);
td->attach();
// Create a second cell for the line contents
td = new HTMLTableCellElement(tdTag, this);
attrs = NamedMappedAttrMap::create();
attrs->addAttribute(MappedAttribute::create(classAttr, "webkit-line-content"));
td->setAttributeMap(attrs.release());
trow->addChild(td);
td->attach();
m_current = m_td = td;
#ifdef DEBUG_LINE_NUMBERS
RefPtr<Text> lineNumberText = Text::create(this, String::number(tokenizer()->lineNumber() + 1) + " ");
td->addChild(lineNumberText);
lineNumberText->attach();
#endif
// Open up the needed spans.
if (!className.isEmpty()) {
if (className == "webkit-html-attribute-name" || className == "webkit-html-attribute-value")
m_current = addSpanWithClassName("webkit-html-tag");
m_current = addSpanWithClassName(className);
}
}
void HTMLViewSourceDocument::addText(const String& text, const String& className)
{
if (text.isEmpty())
return;
// Add in the content, splitting on newlines.
Vector<String> lines;
text.split('\n', true, lines);
unsigned size = lines.size();
for (unsigned i = 0; i < size; i++) {
String substring = lines[i];
if (substring.isEmpty()) {
if (i == size - 1)
break;
substring = " ";
}
if (m_current == m_tbody)
addLine(className);
RefPtr<Text> t = Text::create(this, substring);
m_current->addChild(t);
t->attach();
if (i < size - 1)
m_current = m_tbody;
}
// Set current to m_tbody if the last character was a newline.
if (text[text.length() - 1] == '\n')
m_current = m_tbody;
}
PassRefPtr<Element> HTMLViewSourceDocument::addLink(const String& url, bool isAnchor)
{
if (m_current == m_tbody)
addLine("webkit-html-tag");
// Now create a link for the attribute value instead of a span.
RefPtr<HTMLAnchorElement> anchor = HTMLAnchorElement::create(this);
RefPtr<NamedMappedAttrMap> attrs = NamedMappedAttrMap::create();
const char* classValue;
if (isAnchor)
classValue = "webkit-html-attribute-value webkit-html-external-link";
else
classValue = "webkit-html-attribute-value webkit-html-resource-link";
attrs->addAttribute(MappedAttribute::create(classAttr, classValue));
attrs->addAttribute(MappedAttribute::create(targetAttr, "_blank"));
attrs->addAttribute(MappedAttribute::create(hrefAttr, url));
anchor->setAttributeMap(attrs.release());
m_current->addChild(anchor);
anchor->attach();
return anchor.release();
}
}