e800605ad3
This URL library ends up being a relatively fundamental base library of the system, as LibCore depends on LibURL. This change has two main benefits: * Moving AK back more towards being an agnostic library that can be used between the kernel and userspace. URL has never really fit that description - and is not used in the kernel. * URL _should_ depend on LibUnicode, as it needs punnycode support. However, it's not really possible to do this inside of AK as it can't depend on any external library. This change brings us a little closer to being able to do that, but unfortunately we aren't there quite yet, as the code generators depend on LibCore.
98 lines
3 KiB
C++
98 lines
3 KiB
C++
/*
|
|
* Copyright (c) 2023, Tim Flynn <trflynn89@serenityos.org>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include <AK/StringBuilder.h>
|
|
#include <LibURL/URL.h>
|
|
#include <LibWeb/HTML/Parser/HTMLTokenizer.h>
|
|
#include <LibWebView/SourceHighlighter.h>
|
|
|
|
namespace WebView {
|
|
|
|
String highlight_source(URL::URL const& url, StringView source)
|
|
{
|
|
Web::HTML::HTMLTokenizer tokenizer { source, "utf-8"sv };
|
|
StringBuilder builder;
|
|
|
|
builder.append(R"~~~(
|
|
<!DOCTYPE html>
|
|
<html>
|
|
<head>
|
|
<meta name="color-scheme" content="dark light">)~~~"sv);
|
|
|
|
builder.appendff("<title>View Source - {}</title>", url);
|
|
builder.appendff("<style type=\"text/css\">{}</style>", HTML_HIGHLIGHTER_STYLE);
|
|
builder.append(R"~~~(
|
|
</head>
|
|
<body>
|
|
<pre class="html">
|
|
)~~~"sv);
|
|
|
|
size_t previous_position = 0;
|
|
|
|
auto append_source = [&](auto end_position, Optional<StringView> const& class_name = {}) {
|
|
if (end_position <= previous_position)
|
|
return;
|
|
|
|
auto segment = source.substring_view(previous_position, end_position - previous_position);
|
|
|
|
if (class_name.has_value())
|
|
builder.appendff("<span class=\"{}\">"sv, *class_name);
|
|
|
|
for (auto code_point : Utf8View { segment }) {
|
|
if (code_point == '&')
|
|
builder.append("&"sv);
|
|
else if (code_point == 0xA0)
|
|
builder.append(" "sv);
|
|
else if (code_point == '<')
|
|
builder.append("<"sv);
|
|
else if (code_point == '>')
|
|
builder.append(">"sv);
|
|
else
|
|
builder.append_code_point(code_point);
|
|
}
|
|
|
|
if (class_name.has_value())
|
|
builder.append("</span>"sv);
|
|
|
|
previous_position = end_position;
|
|
};
|
|
|
|
for (auto token = tokenizer.next_token(); token.has_value(); token = tokenizer.next_token()) {
|
|
if (token->is_comment()) {
|
|
append_source(token->start_position().byte_offset);
|
|
append_source(token->end_position().byte_offset, "comment"sv);
|
|
} else if (token->is_start_tag() || token->is_end_tag()) {
|
|
auto tag_name_start = token->start_position().byte_offset;
|
|
|
|
append_source(tag_name_start);
|
|
append_source(tag_name_start + token->tag_name().bytes().size(), "tag"sv);
|
|
|
|
token->for_each_attribute([&](auto const& attribute) {
|
|
append_source(attribute.name_start_position.byte_offset);
|
|
append_source(attribute.name_end_position.byte_offset, "attribute-name"sv);
|
|
|
|
append_source(attribute.value_start_position.byte_offset);
|
|
append_source(attribute.value_end_position.byte_offset, "attribute-value"sv);
|
|
|
|
return IterationDecision::Continue;
|
|
});
|
|
|
|
append_source(token->end_position().byte_offset);
|
|
} else {
|
|
append_source(token->end_position().byte_offset);
|
|
}
|
|
}
|
|
|
|
builder.append(R"~~~(
|
|
</pre>
|
|
</body>
|
|
</html>
|
|
)~~~"sv);
|
|
|
|
return MUST(builder.to_string());
|
|
}
|
|
|
|
}
|