casc_storage.hpp

Go to the documentation of this file.

#pragma once

 
#include <cstdint>
#include <filesystem>
#include <mutex>
#include <optional>
#include <span>
#include <string>
#include <vector>
 
#include <wowlib/core/buffer.hpp>
#include <wowlib/core/client_version.hpp>
#include <wowlib/core/error.hpp>
#include <wowlib/core/file_key.hpp>
 
namespace wowlib::fs {
  class CascStorage {
  public:
    struct Options {
      std::filesystem::path clientRoot; 
      std::string product = "wow"; 
      Locale locale = Locale::enUS; 
      std::optional<std::uint32_t> build; 
    };

    static Result<CascStorage> open(Options options);
 
    ~CascStorage() { _close(); }
 
    CascStorage(const CascStorage&) = delete;
    CascStorage& operator=(const CascStorage&) = delete;
 
    CascStorage(CascStorage&& other) noexcept
      : _options(std::move(other._options)), _storage(other._storage) {
      other._storage = nullptr;
    }
 
    CascStorage& operator=(CascStorage&& other) noexcept {
      if (this != &other) {
        _close();
        _options = std::move(other._options);
        _storage = other._storage;
        other._storage = nullptr;
      }
      return *this;
    }

    Result<FileBuffer> readFile(const FileKey& key);

    Result<void> addEncryptionKey(std::uint64_t keyName, std::span<const std::byte, 16> key);

    Result<void> importKeys(std::string_view keyList);

    bool exists(const FileKey& key);

    Result<std::vector<FileDataID>> enumerateFdids();

    static constexpr StorageKind kind() { return StorageKind::Casc; }
 
  private:
    explicit CascStorage(Options options)
      : _options(std::move(options)) {}

    Result<void> _openStorage();

    void _close() noexcept;

    bool _isOpen() const { return _storage != nullptr; }
 
    Options _options;
    void* _storage = nullptr; // CascLib HANDLE
    mutable std::mutex _mtx;
  };
}