// Copyright 2021 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef SRC_STORAGE_BLOBFS_ALLOCATOR_HOST_ALLOCATOR_H_
#define SRC_STORAGE_BLOBFS_ALLOCATOR_HOST_ALLOCATOR_H_

#include <lib/stdcompat/span.h>
#include <lib/zx/status.h>
#include <zircon/errors.h>
#include <zircon/types.h>

#include <cstdint>
#include <memory>

#include "src/storage/blobfs/allocator/base_allocator.h"
#include "src/storage/blobfs/common.h"
#include "src/storage/blobfs/format.h"
#include "src/storage/blobfs/node_finder.h"

namespace blobfs {

// A simple allocator for manipulating node and block allocations in blobfs images on a host device.
class HostAllocator : public BaseAllocator {
 public:
  // Does not take ownership of |block_bitmap|.
  static zx::status<std::unique_ptr<HostAllocator>> Create(RawBitmap block_bitmap,
                                                           cpp20::span<Inode> node_map);

  // blobfs::NodeFinder interface.
  zx::status<InodePtr> GetNode(uint32_t node_index) final;

  void* GetBlockBitmapData();

 protected:
  // blobfs::BaseAllocator interface.
  zx::status<> AddBlocks(uint64_t block_count) final { return zx::error(ZX_ERR_NOT_SUPPORTED); }
  zx::status<> AddNodes() final { return zx::error(ZX_ERR_NOT_SUPPORTED); }

 private:
  HostAllocator(RawBitmap block_bitmap, cpp20::span<Inode> node_map,
                std::unique_ptr<id_allocator::IdAllocator> node_bitmap);

  cpp20::span<Inode> node_map_;
};

}  // namespace blobfs

#endif  // SRC_STORAGE_BLOBFS_ALLOCATOR_HOST_ALLOCATOR_H_