#pragma once
#ifndef BLOBSTORE_IMPLEMENTATIONS_ONBLOCKS_DATANODESTORE_DATANODESTORE_H_
#define BLOBSTORE_IMPLEMENTATIONS_ONBLOCKS_DATANODESTORE_DATANODESTORE_H_

#include <memory>
#include "messmer/cpp-utils/macros.h"
#include "DataNodeView.h"
#include <messmer/blockstore/utils/Key.h>

namespace blockstore{
class Block;
class BlockStore;
}

namespace blobstore {
namespace onblocks {
namespace datanodestore {
class DataNode;
class DataLeafNode;
class DataInnerNode;

class DataNodeStore {
public:
  DataNodeStore(std::unique_ptr<blockstore::BlockStore> blockstore, uint32_t blocksizeBytes);
  virtual ~DataNodeStore();

  static constexpr uint8_t MAX_DEPTH = 10;

  DataNodeLayout layout() const;

  std::unique_ptr<DataNode> load(const blockstore::Key &key);

  std::unique_ptr<DataLeafNode> createNewLeafNode();
  std::unique_ptr<DataInnerNode> createNewInnerNode(const DataNode &first_child);

  std::unique_ptr<DataNode> createNewNodeAsCopyFrom(const DataNode &source);

  std::unique_ptr<DataNode> overwriteNodeWith(std::unique_ptr<DataNode> target, const DataNode &source);

  void remove(std::unique_ptr<DataNode> node);

  void removeSubtree(std::unique_ptr<DataNode> node);

  uint64_t numNodes() const;
  //TODO Test overwriteNodeWith(), createNodeAsCopyFrom(), removeSubtree()

private:
  std::unique_ptr<DataNode> load(std::unique_ptr<blockstore::Block> block);

  std::unique_ptr<blockstore::BlockStore> _blockstore;
  const DataNodeLayout _layout;

  DISALLOW_COPY_AND_ASSIGN(DataNodeStore);
};

}
}
}

#endif