#ifndef ELOHIMSCRIPT_LEXTOKEN_HPP
#define ELOHIMSCRIPT_LEXTOKEN_HPP

#include <memory>
#include <utility>
#include "LexTokenKind.hpp"

namespace ElohimScript::Parser {
    class LexToken {
        friend class Lexer;

        std::unique_ptr<const LexToken> _next;
        TextSpan _span;

    public:
        LexToken(TextSpan span) : _span(span) {}
        virtual ~LexToken() = default;
        [[nodiscard]] virtual LexTokenKind GetKind() const noexcept = 0;
        [[nodiscard]] const std::unique_ptr<const LexToken>& GetNext() const noexcept { return _next; }
        [[nodiscard]] const TextSpan& GetSpan() const noexcept { return _span; }
    };

    template <LexTokenKind kind> class LexTokenImpl : public LexToken {
    public:
        LexTokenImpl(TextSpan span) : LexToken(span){};
        [[nodiscard]] LexTokenKind GetKind() const noexcept override { return kind; }
    };

    class IntegerLiteral : public LexTokenImpl<LexTokenKind::IntegerLiteral> {
        uint64_t _value;

    public:
        IntegerLiteral(TextSpan span, uint64_t value)
            : LexTokenImpl<LexTokenKind::IntegerLiteral>(span), _value(value) {}
        [[nodiscard]] uint64_t GetValue() const noexcept { return _value; }
    };

    class FloatLiteral : public LexTokenImpl<LexTokenKind::FloatLiteral> {
        double _value;

    public:
        FloatLiteral(TextSpan span, double value) : LexTokenImpl<LexTokenKind::FloatLiteral>(span), _value(value) {}
        [[nodiscard]] double GetValue() const noexcept { return _value; }
    };

    class StringLiteral : public LexTokenImpl<LexTokenKind::StringLiteral> {
        std::u8string _value;

    public:
        StringLiteral(TextSpan span, std::u8string value)
            : LexTokenImpl<LexTokenKind::StringLiteral>(span), _value(std::move(value)) {}
        [[nodiscard]] const std::u8string& GetValue() const noexcept { return _value; }
    };

    class IdentifierToken : public LexTokenImpl<LexTokenKind::Identifier> {
        std::u8string _value;

    public:
        IdentifierToken(TextSpan span, std::u8string value)
            : LexTokenImpl<LexTokenKind::Identifier>(span), _value(std::move(value)) {}
        [[nodiscard]] const std::u8string& GetValue() const noexcept { return _value; }
    };
}

#endif // ELOHIMSCRIPT_LEXTOKEN_HPP