cancellation_token_source.hpp

#pragma once

#include "cancellation_token.hpp"

namespace Cvb
{

CVB_BEGIN_INLINE_NS

template <>
inline HandleGuard<CancellationTokenSource>::HandleGuard(void* handle) noexcept
  : HandleGuard<CancellationTokenSource>(handle, [](void* handle) { CVB_CALL_CAPI(ReleaseObject(handle)); })
{
}


class CancellationTokenSource
{
  public:


    CancellationTokenSource()
      : handle_(nullptr)
    {
      CExports::CVCANCELLATIONTOKENSOURCE sourceHandle = nullptr;
      CExports::CVCANCELLATIONTOKEN tokenHandle = nullptr;
      CVB_CALL_CAPI_CHECKED(CVCCreateCancellationToken(sourceHandle, tokenHandle));
      handle_ = HandleGuard<CancellationTokenSource>(sourceHandle);
      token_ = std::make_shared<CancellationToken>(HandleGuard<CancellationToken>(tokenHandle), CancellationToken::PrivateTag{});
    }


    void* Handle() const noexcept
    {
      return handle_.Handle();
    }


    CancellationTokenPtr Token() const noexcept
    {
      return token_;
    }


    void Cancel()
    {
      CVB_CALL_CAPI_CHECKED(CVCCancellationTokenSourceCancel(Handle()));
    }

  private:

    HandleGuard<CancellationTokenSource> handle_;
    CancellationTokenPtr token_;
};

CVB_END_INLINE_NS

}