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/PagedList.h>
#include <CesiumUtility/Result.h>
#include <CesiumUtility/Uri.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();
 
  Connection(
      const CesiumAsync::AsyncSystem& asyncSystem,
      const std::shared_ptr<CesiumAsync::IAssetAccessor>& pAssetAccessor,
      const AuthenticationToken& accessToken,
      const std::optional<std::string>& refreshToken,
      const CesiumClientCommon::OAuth2ClientOptions& clientOptions)
      : _asyncSystem(asyncSystem),
        _pAssetAccessor(pAssetAccessor),
        _accessToken(accessToken),
        _refreshToken(refreshToken),
        _clientOptions(clientOptions) {}
 
  const AuthenticationToken& getAccessToken() const { return _accessToken; }
  void setAccessToken(const AuthenticationToken& authToken) {
    _accessToken = authToken;
  }
 
  const std::optional<std::string>& getRefreshToken() const {
    return _refreshToken;
  }
  void setRefreshToken(const std::optional<std::string>& refreshToken) {
    _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<std::string_view>>
  ensureValidToken();
 
  CesiumAsync::AsyncSystem _asyncSystem;
  std::shared_ptr<CesiumAsync::IAssetAccessor> _pAssetAccessor;
  AuthenticationToken _accessToken;
  std::optional<std::string> _refreshToken;
  CesiumClientCommon::OAuth2ClientOptions _clientOptions;
};
} // namespace CesiumITwinClient