stdexec::stoppable_token & ::stop_token_for_t: Support std::stop_token (#1726)

Both the stdexec::stoppable_token concept and the stdexec::
stop_token_for_t template type alias depend on the stop token type
having a unary member template named callback_type. While P2300 added
the aforementioned member template to std::stop_token that means that
before C++26 the aforementioned utilities don't work therewith.

Special cased:

- stdexec::stoppable_token to allow std::stop_token without checking for
  the above-mentioned member template, and
- stdexec::stop_token_t to resolve to std::stop_callback<Callback> when
  its first template argument is std::stop_token

---------

Co-authored-by: Eric Niebler <[email protected]>

Author: RobertLeahy

include/stdexec/__detail/__stop_token.hpp

@@ -20,26 +20,53 @@
 
 #include "__concepts.hpp"
 
+STDEXEC_NAMESPACE_STD_BEGIN
+class stop_token;
+
+template <class _Callback>
+class stop_callback;
+STDEXEC_NAMESPACE_STD_END
+
 namespace stdexec {
   namespace __stok {
     template <template <class> class>
     struct __check_type_alias_exists;
   } // namespace __stok
 
   template <class _Token, class _Callback>
-  using stop_callback_for_t = _Token::template callback_type<_Callback>;
+  struct __stop_callback_for {
+    using __t = _Token::template callback_type<_Callback>;
+  };
+  template <class _Callback>
+  struct __stop_callback_for<std::stop_token, _Callback> {
+    using __t = std::stop_callback<_Callback>;
+  };
+
+  template <class _Token, class _Callback>
+  using stop_callback_for_t = __stop_callback_for<_Token, _Callback>::__t;
 
   template <class _Token>
-  concept stoppable_token =
+  concept __stoppable_token =
     __nothrow_copy_constructible<_Token> && __nothrow_move_constructible<_Token>
     && equality_comparable<_Token> && requires(const _Token& __token) {
          { __token.stop_requested() } noexcept -> __boolean_testable_;
          { __token.stop_possible() } noexcept -> __boolean_testable_;
-    // workaround ICE in appleclang 13.1
+       }
+  // workaround ICE in appleclang 13.1
 #if !defined(__clang__)
+       && requires {
          typename __stok::__check_type_alias_exists<_Token::template callback_type>;
+       }
 #endif
-       };
+      ;
+
+  template <class _Token>
+  concept __stoppable_token_or = __same_as<_Token, std::stop_token> || __stoppable_token<_Token>;
+
+  // The cast to bool below is to make __stoppable_token_or<_Token> an atomic constraint,
+  // hiding the disjunction within it for the sake of better compile-time performance.
+  template <class _Token>
+  concept stoppable_token = bool(__stoppable_token_or<_Token>);
 
   template <class _Token, typename _Callback, typename _Initializer = _Callback>
   concept stoppable_token_for =

test/CMakeLists.txt

@@ -29,6 +29,7 @@ set(stdexec_test_sources
     stdexec/concepts/test_concepts_receiver.cpp
     stdexec/concepts/test_concept_operation_state.cpp
     stdexec/concepts/test_concepts_sender.cpp
+    stdexec/concepts/test_concepts_stop_tokens.cpp
     stdexec/concepts/test_awaitables.cpp
     stdexec/algos/factories/test_just.cpp
     stdexec/algos/factories/test_transfer_just.cpp

test/stdexec/concepts/test_concepts_stop_tokens.cpp

@@ -0,0 +1,42 @@
+/*
+ * SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+ *                         Copyright (c) 2025 Robert Leahy. All rights reserved.
+ * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+ *
+ * Licensed under the Apache License, Version 2.0 with LLVM Exceptions (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * https://llvm.org/LICENSE.txt
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include <stdexec/execution.hpp>
+
+#include <type_traits>
+
+namespace {
+
+  struct on_stop_request {
+    void operator()() && noexcept {
+    }
+  };
+
+  static_assert(::stdexec::stoppable_token<::stdexec::never_stop_token>);
+  static_assert(::stdexec::unstoppable_token<::stdexec::never_stop_token>);
+  static_assert(::stdexec::stoppable_token<::stdexec::inplace_stop_token>);
+  static_assert(!::stdexec::unstoppable_token<::stdexec::inplace_stop_token>);
+  static_assert(std::is_same_v<
+                ::stdexec::stop_callback_for_t<::stdexec::never_stop_token, on_stop_request>,
+                ::stdexec::never_stop_token::callback_type<on_stop_request>>);
+  static_assert(::stdexec::stoppable_token<std::stop_token>);
+  static_assert(std::is_same_v<
+                ::stdexec::stop_callback_for_t<std::stop_token, on_stop_request>,
+                std::stop_callback<on_stop_request>>);
+
+} // namespace