/* * Copyright (c) 2023, Tim Flynn * * SPDX-License-Identifier: BSD-2-Clause */ #include #include #include #include namespace WebView { static String generate_style() { StringBuilder builder; builder.append(HTML_HIGHLIGHTER_STYLE); builder.append(R"~~~( .html { counter-reset: line; } .line { counter-increment: line; white-space: nowrap; } .line::before { content: counter(line) " "; display: inline-block; width: 2.5em; padding-right: 0.5em; text-align: right; } @media (prefers-color-scheme: dark) { .line::before { color: darkgrey; } } @media (prefers-color-scheme: light) { .line::before { color: dimgray; } } )~~~"sv); return MUST(builder.to_string()); } String highlight_source(URL::URL const& url, StringView source) { Web::HTML::HTMLTokenizer tokenizer { source, "utf-8"sv }; StringBuilder builder; builder.append(R"~~~( )~~~"sv); builder.appendff("View Source - {}", url); builder.appendff("", generate_style()); builder.append(R"~~~( 
)~~~"sv);

    size_t previous_position = 0;

    auto append_source = [&](auto end_position, Optional const& class_name = {}) {
        if (end_position <= previous_position)
            return;

        auto segment = source.substring_view(previous_position, end_position - previous_position);

        auto append_class_start = [&]() {
            if (class_name.has_value())
                builder.appendff(""sv, *class_name);
        };
        auto append_class_end = [&]() {
            if (class_name.has_value())
                builder.append(""sv);
        };

        append_class_start();

        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 if (code_point == '\n') {
                append_class_end();
                builder.append("\n"sv);
                append_class_start();
            } else {
                builder.append_code_point(code_point);
            }
        }

        append_class_end();
        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);

            if (token->is_end_of_file())
                break;
        }
    }

    builder.append(R"~~~(
)~~~"sv); return MUST(builder.to_string()); } }