syntax_highlighter.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <asmgrader/api/expression_inspection.hpp>
#include <asmgrader/logging.hpp>
 
#include <fmt/base.h>
#include <fmt/color.h>
#include <fmt/format.h>
#include <libassert/assert.hpp>
#include <range/v3/algorithm/contains.hpp>
#include <range/v3/algorithm/find.hpp>
#include <range/v3/algorithm/find_if.hpp>
#include <range/v3/algorithm/fold_left.hpp>
#include <range/v3/algorithm/transform.hpp>
#include <range/v3/iterator/concepts.hpp>
#include <range/v3/range/access.hpp>
#include <range/v3/range/concepts.hpp>
#include <range/v3/range/conversion.hpp>
#include <range/v3/range/primitives.hpp>
#include <range/v3/view/enumerate.hpp>
#include <range/v3/view/join.hpp>
#include <range/v3/view/split.hpp>
#include <range/v3/view/split_when.hpp>
#include <range/v3/view/transform.hpp>
 
#include <array>
#include <charconv>
#include <cstddef>
#include <functional>
#include <span>
#include <string>
#include <string_view>
#include <system_error>
#include <utility>
#include <vector>
 
namespace asmgrader::highlight {
 
using Option = std::pair<inspection::Token::Kind, fmt::text_style>;
 
using enum inspection::Token::Kind;
 
struct Options
{
    static constexpr auto num_token_kinds = static_cast<std::size_t>(EndDelimiter);
 
    struct Opt
    {
        fmt::text_style style;
        std::function<fmt::text_style(std::string_view)> style_fn;
        std::function<std::string(std::string_view)> transform_fn;
 
        std::string apply(std::string_view str) const;
    };
 
    static fmt::text_style style_basic_ident_types(fmt::text_style type_style, fmt::text_style default_style,
                                                   std::string_view ident);
    static fmt::text_style style_op_keywords(fmt::text_style keyword_style, fmt::text_style default_style,
                                             std::string_view op);
    static std::string basic_binary_op_spacing(std::string_view op);
    // Add some spacing around unary and ternary ops
    static std::string basic_op_spacing(std::string_view op);
 
    static Options get_default_options();
 
    std::array<Opt, num_token_kinds> token_opts{};
 
    constexpr Opt& operator[](std::size_t idx) { return token_opts.at(idx); }
 
    constexpr Opt& operator[](inspection::Token::Kind kind) { return token_opts.at(static_cast<std::size_t>(kind)); }
 
    constexpr const Opt& operator[](std::size_t idx) const { return token_opts.at(idx); }
 
    constexpr const Opt& operator[](inspection::Token::Kind kind) const {
        return token_opts.at(static_cast<std::size_t>(kind));
    }
};
 
std::string render_blocks(std::string_view str, bool skip_styling);
 
std::string highlight(std::string_view str, const Options& opts = Options::get_default_options());
 
std::string highlight(const inspection::Token& token, const Options& opts = Options::get_default_options());
 
std::string highlight(std::span<const inspection::Token> tokens, const Options& opts = Options::get_default_options());
 
} // namespace asmgrader::highlight