/*

 * Copyright (c) 2024, Andreas Kling <kling@serenityos.org>

 *

 * SPDX-License-Identifier: BSD-2-Clause

 */

#pragma once

#include <AK/RefCounted.h>

#include <LibJS/Bytecode/Operand.h>

namespace JS::Bytecode {

class ScopedOperandImpl : public RefCounted<ScopedOperandImpl> {

public:

    ScopedOperandImpl(Generator& generator, Operand operand)

        : m_generator(generator)

        , m_operand(operand)

    {

    }

    ~ScopedOperandImpl();

    [[nodiscard]] Operand const& operand() const { return m_operand; }

    [[nodiscard]] Operand& operand() { return m_operand; }

private:

    Generator& m_generator;

    Operand m_operand;

};

class ScopedOperand {

public:

    explicit ScopedOperand(Generator& generator, Operand operand)

        : m_impl(adopt_ref(*new ScopedOperandImpl(generator, operand)))

    {

    }

    [[nodiscard]] Operand const& operand() const { return m_impl->operand(); }

    [[nodiscard]] Operand& operand() { return m_impl->operand(); }

    operator Operand() const { return operand(); }

    [[nodiscard]] bool operator==(ScopedOperand const& other) const { return operand() == other.operand(); }

    [[nodiscard]] size_t ref_count() const { return m_impl->ref_count(); }

private:

    NonnullRefPtr<ScopedOperandImpl> m_impl;

};

}