/*
 * Copyright (c) 2020-2021, the SerenityOS developers.
 * Copyright (c) 2021-2022, Sam Atkins <atkinssj@serenityos.org>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#pragma once

#include <AK/NonnullRefPtr.h>
#include <AK/RefPtr.h>
#include <LibWeb/CSS/Parser/Function.h>
#include <LibWeb/CSS/Parser/Token.h>
#include <LibWeb/Forward.h>

namespace Web::CSS::Parser {

// https://www.w3.org/TR/css-syntax-3/#component-value
class ComponentValue {
public:
    ComponentValue(Token);
    explicit ComponentValue(NonnullRefPtr<Function>);
    explicit ComponentValue(NonnullRefPtr<Block>);
    ~ComponentValue();

    bool is_block() const { return m_value.has<NonnullRefPtr<Block>>(); }
    Block const& block() const { return m_value.get<NonnullRefPtr<Block>>(); }

    bool is_function() const { return m_value.has<NonnullRefPtr<Function>>(); }
    Function const& function() const { return m_value.get<NonnullRefPtr<Function>>(); }

    bool is_token() const { return m_value.has<Token>(); }
    bool is(Token::Type type) const { return is_token() && token().is(type); }
    Token const& token() const { return m_value.get<Token>(); }
    operator Token() const { return m_value.get<Token>(); }

    String to_string() const;
    String to_debug_string() const;

private:
    Variant<Token, NonnullRefPtr<Function>, NonnullRefPtr<Block>> m_value;
};
}

template<>
struct AK::Formatter<Web::CSS::Parser::ComponentValue> : Formatter<StringView> {
    ErrorOr<void> format(FormatBuilder& builder, Web::CSS::Parser::ComponentValue const& component_value)
    {
        return Formatter<StringView>::format(builder, component_value.to_string());
    }
};