csv_listfile.hpp

Go to the documentation of this file.

#pragma once

 
#include <filesystem>
#include <optional>
#include <shared_mutex>
#include <string>
#include <string_view>
#include <unordered_map>
 
#include <wowlib/core/error.hpp>
#include <wowlib/core/file_key.hpp>
#include <wowlib/core/shared_mutex.hpp>
#include <wowlib/fs/fdid_allocator.hpp>
#include <wowlib/fs/listfile.hpp>
 
namespace wowlib::fs {
  struct CsvListfileOptions {
    FileDataID customFdidStart{1'000'000'000};
  };
 
  // (namespace-scope rather than nested: gcc late-parses nested-class NSDMIs, so a
  // `= {}` default argument cannot aggregate-initialize a nested struct)

  class CsvListfile {
  public:
    using Options = CsvListfileOptions;
 
    CsvListfile() = default;

    static Result<CsvListfile> load(const std::filesystem::path& csv, Options options = {});

    std::optional<FileDataID> pathToFdid(std::string_view path) const;

    std::optional<std::string> fdidToPath(FileDataID fdid) const;

    Result<FileDataID> registerPath(std::string_view path);

    bool contains(std::string_view path) const;

    Result<void> save() const;

    std::size_t size() const;

    const std::filesystem::path& source() const { return _source; }
 
    CsvListfile(const CsvListfile&) = delete;
    CsvListfile& operator=(const CsvListfile&) = delete;
    CsvListfile(CsvListfile&& other) noexcept;
    CsvListfile& operator=(CsvListfile&& other) noexcept;
 
  private:
    std::filesystem::path _source;
    mutable SharedMutex _mtx;
    std::unordered_map<std::string, FileDataID> _pathToId;
    std::unordered_map<std::uint32_t, std::string> _idToPath;
    detail::FdidAllocator _allocator;
  };
 
  static_assert(ListfileProvider<CsvListfile>);
}