furlang/furc/include/furc/front/token.hpp

#ifndef FURC_FRONT_TOKEN_HPP
#define FURC_FRONT_TOKEN_HPP

#include "furc/handle.hpp"

#include <cassert>
#include <ostream>
#include <string_view>

namespace furc {
namespace front {

enum class token_t {
    None,
    Identifier,
    String,
    Keyword,
    Integer,
    Lparen,
    Rparen,
    Lbrace,
    Rbrace,
    Lbracket,
    Rbracket,
    Semicolon,
    Colon,
    Comma,
    Dot,

    Plus,
    Minus,
    Star,
    Slash,
    Percent,
    DPlus,
    DMinus,

    Eq,
    PlusEq,
    MinusEq,
    StarEq,
    SlashEq,
    PercentEq,

    DEq,
    NotEq,
    LessThan,
    GreaterThan,
    LessEq,
    GreaterEq,
};

static inline std::ostream& operator<<(std::ostream& os, token_t type) {
    switch (type) {
    case token_t::None: return os << "none";
    case token_t::Identifier: return os << "identifier";
    case token_t::String: return os << "string";
    case token_t::Keyword: return os << "keyword";
    case token_t::Integer: return os << "integer";
    case token_t::Lparen: return os << "'('";
    case token_t::Rparen: return os << "')'";
    case token_t::Lbrace: return os << "'{'";
    case token_t::Rbrace: return os << "'}'";
    case token_t::Lbracket: return os << "'['";
    case token_t::Rbracket: return os << "']'";
    case token_t::Semicolon: return os << "';'";
    case token_t::Colon: return os << "':'";
    case token_t::Comma: return os << "','";
    case token_t::Dot: return os << "'.'";
    case token_t::Plus: return os << "'+'";
    case token_t::Minus: return os << "'-'";
    case token_t::Star: return os << "'*'";
    case token_t::Slash: return os << "'/'";
    case token_t::Percent: return os << "'%'";
    case token_t::DPlus: return os << "++";
    case token_t::DMinus: return os << "--";
    case token_t::Eq: return os << "=";
    case token_t::PlusEq: return os << "+=";
    case token_t::MinusEq: return os << "-=";
    case token_t::StarEq: return os << "*=";
    case token_t::SlashEq: return os << "/=";
    case token_t::PercentEq: return os << "%=";
    case token_t::DEq: return os << "==";
    case token_t::NotEq: return os << "!=";
    case token_t::LessThan: return os << "<";
    case token_t::GreaterThan: return os << ">";
    case token_t::LessEq: return os << "<=";
    case token_t::GreaterEq: return os << ">=";
    }
    return os;
}

static inline std::string operator+(const std::string& str, token_t type) {
    switch (type) {
    case token_t::None: return str + "none";
    case token_t::Identifier: return str + "identifier";
    case token_t::String: return str + "string";
    case token_t::Keyword: return str + "keyword";
    case token_t::Integer: return str + "integer";
    case token_t::Lparen: return str + "'('";
    case token_t::Rparen: return str + "')'";
    case token_t::Lbrace: return str + "'{'";
    case token_t::Rbrace: return str + "'}'";
    case token_t::Lbracket: return str + "'['";
    case token_t::Rbracket: return str + "']'";
    case token_t::Semicolon: return str + "';'";
    case token_t::Colon: return str + "':'";
    case token_t::Comma: return str + "','";
    case token_t::Dot: return str + "'.'";
    case token_t::Plus: return str + "'+'";
    case token_t::Minus: return str + "'-'";
    case token_t::Star: return str + "'*'";
    case token_t::Slash: return str + "'/'";
    case token_t::Percent: return str + "'%'";
    case token_t::DPlus: return str + "++";
    case token_t::DMinus: return str + "--";
    case token_t::Eq: return str + "=";
    case token_t::PlusEq: return str + "+=";
    case token_t::MinusEq: return str + "-=";
    case token_t::StarEq: return str + "*=";
    case token_t::SlashEq: return str + "/=";
    case token_t::PercentEq: return str + "%=";
    case token_t::DEq: return str + "==";
    case token_t::NotEq: return str + "!=";
    case token_t::LessThan: return str + "<";
    case token_t::GreaterThan: return str + ">";
    case token_t::LessEq: return str + "<=";
    case token_t::GreaterEq: return str + ">=";
    }
    return str;
}

enum class keyword_token {
    None,
    Func,
    Return,
};

static inline std::ostream& operator<<(std::ostream& os, keyword_token keyword) {
    switch (keyword) {
    case keyword_token::None: return os << "none";
    case keyword_token::Func: return os << "func";
    case keyword_token::Return: return os << "return";
    }
}

static inline std::string operator+(const std::string& str, keyword_token keyword) {
    switch (keyword) {
    case keyword_token::None: return str + "none";
    case keyword_token::Func: return str + "func";
    case keyword_token::Return: return str + "return";
    }
}

using integer_token = std::uint64_t;

struct token {
    token_t type = token_t::None;
    union value {
        std::nullptr_t   null;
        std::string_view string;
        keyword_token    keyword;
        integer_token    integer;

        value(std::nullptr_t value = nullptr)
          : null(value) {}

        value(std::string_view value)
          : string(value) {}

        value(keyword_token value)
          : keyword(value) {}

        value(integer_token value)
          : integer(value) {}
    } value;

    token() = default;

    token(token_t type, std::string_view value = {})
      : type(type), value(value) {}

    token(keyword_token keyword, std::string_view value = {})
      : type(token_t::Keyword), value(keyword) {}

    token(integer_token integer)
      : type(token_t::Integer), value(integer) {}

    union value*       operator->() { return &value; }
    const union value* operator->() const { return &value; }

    bool operator==(const token& rhs) const {
        if (type != rhs.type) return false;
        switch (type) {
        case token_t::Identifier:
        case token_t::String: return value.string == rhs.value.string;
        case token_t::Keyword: return value.keyword == rhs.value.keyword;
        case token_t::Integer: return value.integer == rhs.value.integer;
        default: return true;
        }
    }
};

static inline std::ostream& operator<<(std::ostream& os, const token& token) {
    switch (token.type) {
    case token_t::Identifier:
    case token_t::String: return os << token.value.string;
    case token_t::Keyword: return os << token.value.keyword;
    case token_t::Integer: return os << token.value.integer;
    default: return os << token.type;
    }
}

template <typename Error = std::string>
using token_handle = handle<token, Error>;

} // namespace front
} // namespace furc

#endif // FURC_FRONT_TOKEN_HPP