NameLexer.hpp

Go to the documentation of this file.

//          Copyright Dominic (DNKpp) Koepke 2024 - 2025.
// Distributed under the Boost Software License, Version 1.0.
//    (See accompanying file LICENSE_1_0.txt or copy at
//          https://www.boost.org/LICENSE_1_0.txt)
 
#ifndef MIMICPP_PRINTING_TYPE_NAME_LEXER_HPP
#define MIMICPP_PRINTING_TYPE_NAME_LEXER_HPP
 
#pragma once
 
#include "mimic++/Fwd.hpp"
#include "mimic++/config/Config.hpp"
#include "mimic++/utilities/Algorithm.hpp"
 
#ifndef MIMICPP_DETAIL_IS_MODULE
    #include <algorithm>
    #include <array>
    #include <cctype>
    #include <functional>
    #include <span>
    #include <tuple>
    #include <variant>
#endif
 
namespace mimicpp::printing::type::lexing
{
    // see: https://en.cppreference.com/w/cpp/string/byte/isspace
    inline auto constexpr is_space = [](char const c) noexcept {
        return static_cast<bool>(
            std::isspace(static_cast<unsigned char>(c)));
    };
 
    // see: https://en.cppreference.com/w/cpp/string/byte/isdigit
    inline auto constexpr is_digit = [](char const c) noexcept {
        return static_cast<bool>(
            std::isdigit(static_cast<unsigned char>(c)));
    };
 
    namespace texts
    {
        // just list the noteworthy ones here
        inline std::array constexpr visibilityKeywords = std::to_array<StringViewT>({"public", "protected", "private"});
        inline std::array constexpr specKeywords = std::to_array<StringViewT>({"const", "constexpr", "volatile", "noexcept", "static"});
        inline std::array constexpr contextKeywords = std::to_array<StringViewT>({"operator", "struct", "class", "enum"});
        inline std::array constexpr typeKeywords = std::to_array<StringViewT>(
            // The `__int64` keyword is used by msvc as an alias for `long long`.
            {"auto", "void", "bool", "char", "char8_t", "char16_t", "char32_t", "wchar_t", "double", "float", "int", "long", "__int64", "short", "signed", "unsigned"});
        inline std::array constexpr otherKeywords = std::to_array<StringViewT>({"new", "delete", "co_await"});
        inline std::array constexpr digraphs = std::to_array<StringViewT>(
            {"and", "or", "xor", "not", "bitand", "bitor", "compl", "and_eq", "or_eq", "xor_eq", "not_eq"});
 
        inline std::array constexpr braceLikes = std::to_array<StringViewT>({"{", "}", "[", "]", "(", ")", "`", "'"});
        inline std::array constexpr comparison = std::to_array<StringViewT>({"==", "!=", "<", "<=", ">", ">=", "<=>"});
        inline std::array constexpr assignment = std::to_array<StringViewT>({"=", "+=", "-=", "*=", "/=", "%=", "&=", "|=", "^=", "<<=", ">>="});
        inline std::array constexpr incOrDec = std::to_array<StringViewT>({"++", "--"});
        inline std::array constexpr arithmetic = std::to_array<StringViewT>({"+", "-", "*", "/", "%"});
        inline std::array constexpr bitArithmetic = std::to_array<StringViewT>({"~", "&", "|", "^", "<<", ">>"});
        inline std::array constexpr logical = std::to_array<StringViewT>({"!", "&&", "||"});
        inline std::array constexpr access = std::to_array<StringViewT>({".", ".*", "->", "->*"});
        inline std::array constexpr specialAngles = std::to_array<StringViewT>({"<:", ":>", "<%", "%>"});
        inline std::array constexpr rest = std::to_array<StringViewT>({"::", ";", ",", ":", "...", "?"});
    }
 
    [[nodiscard]]
    consteval auto make_keyword_collection() noexcept
    {
        std::array collection = util::concat_arrays(
            texts::visibilityKeywords,
            texts::specKeywords,
            texts::contextKeywords,
            texts::otherKeywords,
            texts::typeKeywords,
            texts::digraphs);
 
        std::ranges::sort(collection);
        MIMICPP_ASSERT(collection.cend() == std::ranges::unique(collection).begin(), "Fix your input!");
 
        return collection;
    }
 
    [[nodiscard]]
    consteval auto make_operator_or_punctuator_collection() noexcept
    {
        // see: https://eel.is/c++draft/lex.operators#nt:operator-or-punctuator
        std::array collection = util::concat_arrays(
            texts::braceLikes,
            texts::comparison,
            texts::assignment,
            texts::incOrDec,
            texts::arithmetic,
            texts::bitArithmetic,
            texts::logical,
            texts::access,
            texts::specialAngles,
            texts::rest);
        std::ranges::sort(collection);
        MIMICPP_ASSERT(collection.cend() == std::ranges::unique(collection).begin(), "Fix your input!");
 
        return collection;
    }
 
    struct space
    {
        [[nodiscard]]
        bool operator==(space const&) const = default;
    };
 
    struct keyword
    {
    public:
        static constexpr std::array textCollection = make_keyword_collection();
 
        [[nodiscard]]
        explicit constexpr keyword(StringViewT const& text) noexcept
            : keyword{
                  std::ranges::distance(
                      textCollection.cbegin(),
                      util::binary_find(textCollection, text))}
        {
        }
 
        [[nodiscard]]
        explicit constexpr keyword(std::ptrdiff_t const keywordIndex) noexcept
            : m_KeywordIndex{keywordIndex}
        {
            MIMICPP_ASSERT(0 <= m_KeywordIndex && m_KeywordIndex < std::ranges::ssize(textCollection), "Invalid keyword.");
        }
 
        [[nodiscard]]
        constexpr StringViewT text() const noexcept
        {
            return textCollection[m_KeywordIndex];
        }
 
        [[nodiscard]]
        bool operator==(keyword const&) const = default;
 
    private:
        std::ptrdiff_t m_KeywordIndex;
    };
 
    struct operator_or_punctuator
    {
    public:
        static constexpr std::array textCollection = make_operator_or_punctuator_collection();
 
        [[nodiscard]]
        explicit constexpr operator_or_punctuator(StringViewT const& text) noexcept
            : operator_or_punctuator{
                  std::ranges::distance(
                      textCollection.cbegin(),
                      util::binary_find(textCollection, text))}
        {
        }
 
        [[nodiscard]]
        explicit constexpr operator_or_punctuator(std::ptrdiff_t const textIndex) noexcept
            : m_TextIndex{textIndex}
        {
            MIMICPP_ASSERT(0 <= m_TextIndex && m_TextIndex < std::ranges::ssize(textCollection), "Invalid operator or punctuator.");
        }
 
        [[nodiscard]]
        constexpr StringViewT text() const noexcept
        {
            return textCollection[m_TextIndex];
        }
 
        [[nodiscard]]
        bool operator==(operator_or_punctuator const&) const = default;
 
    private:
        std::ptrdiff_t m_TextIndex;
    };
 
    struct identifier
    {
        StringViewT content;
 
        [[nodiscard]]
        bool operator==(identifier const&) const = default;
    };
 
    struct end
    {
        [[nodiscard]]
        bool operator==(end const&) const = default;
    };
 
    using token_class = std::variant<
        end,
        space,
        keyword,
        operator_or_punctuator,
        identifier>;
 
    struct token
    {
        StringViewT content;
        token_class classification;
    };
 
    class NameLexer
    {
    public:
        [[nodiscard]]
        explicit constexpr NameLexer(StringViewT text) noexcept
            : m_Text{std::move(text)},
              m_Next{find_next()}
        {
        }
 
        [[nodiscard]]
        constexpr token next() noexcept
        {
            return std::exchange(m_Next, find_next());
        }
 
        [[nodiscard]]
        constexpr token const& peek() const noexcept
        {
            return m_Next;
        }
 
    private:
        StringViewT m_Text;
        token m_Next;
 
        [[nodiscard]]
        constexpr token find_next() noexcept
        {
            if (m_Text.empty())
            {
                return token{
                    .content = {m_Text.cend(), m_Text.cend()},
                    .classification = end{}
                };
            }
 
            if (is_space(m_Text.front()))
            {
                // Multiple consecutive spaces or any whitespace character other than a single space
                // carry no meaningful semantic value beyond delimitation.
                // Although single spaces may sometimes influence the result and sometimes not,
                // complicating the overall process, we filter out all non-single whitespace characters here.
                if (StringViewT const content = next_as_space();
                    " " == content)
                {
                    return token{
                        .content = content,
                        .classification = space{}};
                }
 
                return find_next();
            }
 
            if (auto const options = util::prefix_range(
                    operator_or_punctuator::textCollection,
                    m_Text.substr(0u, 1u)))
            {
                return next_as_op_or_punctuator(options);
            }
 
            StringViewT const content = next_as_identifier();
            // As we do not perform any prefix-checks, we need to check now whether the token actually denotes a keyword.
            if (auto const iter = util::binary_find(keyword::textCollection, content);
                iter != keyword::textCollection.cend())
            {
                return token{
                    .content = content,
                    .classification = keyword{std::ranges::distance(keyword::textCollection.begin(), iter)}};
            }
 
            return token{
                .content = content,
                .classification = identifier{.content = content}};
        }
 
        [[nodiscard]]
        constexpr StringViewT next_as_space() noexcept
        {
            auto const end = std::ranges::find_if_not(m_Text.cbegin() + 1, m_Text.cend(), is_space);
            StringViewT const content{m_Text.cbegin(), end};
            m_Text = StringViewT{end, m_Text.cend()};
 
            return content;
        }

        [[nodiscard]]
        constexpr token next_as_op_or_punctuator(std::span<StringViewT const> options) noexcept
        {
            MIMICPP_ASSERT(m_Text.substr(0u, 1u) == options.front(), "Assumption does not hold.");
 
            auto const try_advance = [&, this](std::size_t const n) {
                if (n <= m_Text.size())
                {
                    return util::prefix_range(
                        options,
                        StringViewT{m_Text.cbegin(), m_Text.cbegin() + n});
                }
 
                return std::ranges::subrange{options.end(), options.end()};
            };
 
            std::size_t length{1u};
            StringViewT const* lastMatch = &options.front();
            while (auto const nextOptions = try_advance(length + 1))
            {
                ++length;
                options = {nextOptions.begin(), nextOptions.end()};
 
                // If the first string is exactly the size of the prefix, it's a match.
                if (auto const& front = options.front();
                    length == front.size())
                {
                    lastMatch = &front;
                }
            }
 
            MIMICPP_ASSERT(!options.empty(), "Invalid state.");
            MIMICPP_ASSERT(lastMatch, "Invalid state.");
 
            auto const index = std::ranges::distance(operator_or_punctuator::textCollection.data(), lastMatch);
            StringViewT const content{m_Text.substr(0u, lastMatch->size())};
            m_Text.remove_prefix(lastMatch->size());
 
            return token{
                .content = content,
                .classification = operator_or_punctuator{index}};
        }

        [[nodiscard]]
        constexpr StringViewT next_as_identifier() noexcept
        {
            auto const last = std::ranges::find_if_not(
                m_Text.cbegin() + 1,
                m_Text.cend(),
                [](auto const c) {
                    return !is_space(c)
                        && !std::ranges::binary_search(operator_or_punctuator::textCollection, StringViewT{&c, 1u});
                });
 
            StringViewT const content{m_Text.cbegin(), last};
            m_Text = {last, m_Text.cend()};
 
            return content;
        }
    };
}
 
#endif