ladybird/Userland/Libraries/LibSQL/Index.h

/*
 * Copyright (c) 2021, Jan de Visser <jan@de-visser.net>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#pragma once

#include <AK/RefCounted.h>
#include <LibSQL/Forward.h>
#include <LibSQL/Meta.h>
#include <LibSQL/Serializer.h>

namespace SQL {

class IndexNode {
public:
    virtual ~IndexNode() = default;
    [[nodiscard]] Block::Index block_index() const { return m_block_index; }
    IndexNode* as_index_node() { return dynamic_cast<IndexNode*>(this); }

protected:
    explicit IndexNode(Block::Index block_index)
        : m_block_index(block_index)
    {
    }

    void set_block_index(Block::Index block_index) { m_block_index = block_index; }

private:
    Block::Index m_block_index;
};

class Index : public RefCounted<Index> {
public:
    virtual ~Index() = default;

    NonnullRefPtr<TupleDescriptor> descriptor() const { return m_descriptor; }
    [[nodiscard]] bool duplicates_allowed() const { return !m_unique; }
    [[nodiscard]] bool unique() const { return m_unique; }
    [[nodiscard]] Block::Index block_index() const { return m_block_index; }

protected:
    Index(Serializer&, NonnullRefPtr<TupleDescriptor> const&, bool unique, Block::Index block_index);
    Index(Serializer&, NonnullRefPtr<TupleDescriptor> const&, Block::Index block_index);

    [[nodiscard]] Serializer& serializer() { return m_serializer; }
    void set_block_index(Block::Index block_index) { m_block_index = block_index; }
    u32 request_new_block_index() { return m_serializer.request_new_block_index(); }

private:
    Serializer m_serializer;
    NonnullRefPtr<TupleDescriptor> m_descriptor;
    bool m_unique { false };
    Block::Index m_block_index { 0 };
};

}