JsonReaderOptions.h

#pragma once
 
#include "IExtensionJsonHandler.h"
#include "Library.h"
 
#include <functional>
#include <map>
#include <memory>
#include <optional>
#include <string>
#include <vector>
 
namespace CesiumJsonReader {
 
enum class ExtensionState {
  Enabled,
 
  JsonOnly,
 
  Disabled
};
 
class CESIUMJSONREADER_API JsonReaderOptions {
public:
  bool getCaptureUnknownProperties() const {
    return this->_captureUnknownProperties;
  }
 
  void setCaptureUnknownProperties(bool value) {
    this->_captureUnknownProperties = value;
  }
 
  template <typename TExtended, typename TExtensionHandler>
  void registerExtension(const std::string& extensionName) {
    auto it =
        this->_extensions.emplace(extensionName, ObjectTypeToHandler()).first;
    it->second.insert_or_assign(
        TExtended::TypeName,
        ExtensionReaderFactory([](const JsonReaderOptions& context) {
          return std::make_unique<TExtensionHandler>(context);
        }));
  }
 
  template <typename TExtended, typename TExtensionHandler>
  void registerExtension() {
    auto it =
        this->_extensions
            .emplace(TExtensionHandler::ExtensionName, ObjectTypeToHandler())
            .first;
    it->second.insert_or_assign(
        TExtended::TypeName,
        ExtensionHandlerFactory([](const JsonReaderOptions& context) {
          return std::make_unique<TExtensionHandler>(context);
        }));
  }
 
  ExtensionState getExtensionState(const std::string& extensionName) const;
 
  void
  setExtensionState(const std::string& extensionName, ExtensionState newState);
 
  std::unique_ptr<IExtensionJsonHandler> createExtensionHandler(
      const std::string_view& extensionName,
      const std::string& extendedObjectType) const;
 
private:
  using ExtensionHandlerFactory =
      std::function<std::unique_ptr<IExtensionJsonHandler>(
          const JsonReaderOptions&)>;
  using ObjectTypeToHandler = std::map<std::string, ExtensionHandlerFactory>;
  using ExtensionNameMap = std::map<std::string, ObjectTypeToHandler>;
 
  ExtensionNameMap _extensions;
  std::unordered_map<std::string, ExtensionState> _extensionStates;
  bool _captureUnknownProperties = true;
};
 
} // namespace CesiumJsonReader