Connection.h

#pragma once
 
#include "AuthenticationToken.h"
#include "CesiumCuratedContent.h"
#include "IModel.h"
#include "IModelMeshExport.h"
#include "ITwin.h"
#include "ITwinRealityData.h"
#include "Library.h"
#include "Profile.h"
 
#include <CesiumAsync/AsyncSystem.h>
#include <CesiumAsync/Future.h>
#include <CesiumAsync/IAssetAccessor.h>
#include <CesiumClientCommon/OAuth2PKCE.h>
#include <CesiumITwinClient/GeospatialFeatureCollection.h>
#include <CesiumITwinClient/PagedList.h>
#include <CesiumUtility/Result.h>
#include <CesiumUtility/Uri.h>
#include <CesiumVectorData/GeoJsonObjectTypes.h>
 
#include <memory>
#include <optional>
#include <string>
#include <vector>
 
namespace CesiumITwinClient {

struct QueryParameters {
public:
  std::optional<std::string> search;
  std::optional<std::string> orderBy;

  std::optional<uint32_t> top;

  std::optional<uint32_t> skip;

  void addToQuery(CesiumUtility::UriQuery& uriQuery) const;
  void addToUri(CesiumUtility::Uri& uri) const;
};

class CESIUMITWINCLIENT_API Connection {
public:
  static CesiumAsync::Future<CesiumUtility::Result<Connection>> authorize(
      const CesiumAsync::AsyncSystem& asyncSystem,
      const std::shared_ptr<CesiumAsync::IAssetAccessor>& pAssetAccessor,
      const std::string& friendlyApplicationName,
      const std::string& clientID,
      const std::string& redirectPath,
      const std::optional<int>& redirectPort,
      const std::vector<std::string>& scopes,
      std::function<void(const std::string&)>&& openUrlCallback);

  CesiumAsync::Future<CesiumUtility::Result<UserProfile>> me();

  CesiumAsync::Future<CesiumUtility::Result<PagedList<ITwin>>>
  itwins(const QueryParameters& params);

  CesiumAsync::Future<CesiumUtility::Result<PagedList<IModel>>>
  imodels(const std::string& iTwinId, const QueryParameters& params);

  CesiumAsync::Future<CesiumUtility::Result<PagedList<IModelMeshExport>>>
  meshExports(const std::string& iModelId, const QueryParameters& params);

  CesiumAsync::Future<CesiumUtility::Result<PagedList<ITwinRealityData>>>
  realityData(const std::string& iTwinId, const QueryParameters& params);

  CesiumAsync::Future<
      CesiumUtility::Result<std::vector<CesiumCuratedContentAsset>>>
  cesiumCuratedContent();

  CesiumAsync::Future<
      CesiumUtility::Result<std::vector<GeospatialFeatureCollection>>>
  geospatialFeatureCollections(const std::string& iTwinId);

  CesiumAsync::Future<
      CesiumUtility::Result<PagedList<CesiumVectorData::GeoJsonFeature>>>
  geospatialFeatures(
      const std::string& iTwinId,
      const std::string& collectionId,
      uint32_t limit = 10000);

  Connection(
      const CesiumAsync::AsyncSystem& asyncSystem,
      const std::shared_ptr<CesiumAsync::IAssetAccessor>& pAssetAccessor,
      const AuthenticationToken& authenticationToken,
      const std::optional<std::string>& refreshToken,
      const CesiumClientCommon::OAuth2ClientOptions& clientOptions)
      : _asyncSystem(asyncSystem),
        _pAssetAccessor(pAssetAccessor),
        _authenticationToken(authenticationToken),
        _refreshToken(refreshToken),
        _clientOptions(clientOptions) {}

  const AuthenticationToken& getAuthenticationToken() const {
    return this->_authenticationToken;
  }

  void setAuthenticationToken(const AuthenticationToken& authToken) {
    this->_authenticationToken = authToken;
  }

  const std::optional<std::string>& getRefreshToken() const {
    return this->_refreshToken;
  }

  void setRefreshToken(const std::optional<std::string>& refreshToken) {
    this->_refreshToken = refreshToken;
  }
 
private:
  CesiumAsync::Future<CesiumUtility::Result<PagedList<ITwin>>>
  listITwins(const std::string& url);
  CesiumAsync::Future<CesiumUtility::Result<PagedList<IModel>>>
  listIModels(const std::string& url);
  CesiumAsync::Future<CesiumUtility::Result<PagedList<IModelMeshExport>>>
  listIModelMeshExports(const std::string& url);
  CesiumAsync::Future<CesiumUtility::Result<PagedList<ITwinRealityData>>>
  listITwinRealityData(const std::string& url);
  CesiumAsync::Future<
      CesiumUtility::Result<PagedList<CesiumVectorData::GeoJsonFeature>>>
  listGeospatialFeatures(const std::string& url);
 
  CesiumAsync::Future<CesiumUtility::Result<std::string>> ensureValidToken();
 
  CesiumAsync::AsyncSystem _asyncSystem;
  std::shared_ptr<CesiumAsync::IAssetAccessor> _pAssetAccessor;
  AuthenticationToken _authenticationToken;
  std::optional<std::string> _refreshToken;
  CesiumClientCommon::OAuth2ClientOptions _clientOptions;
};
} // namespace CesiumITwinClient