pkgs/dart_mcp/lib/src/server/elicitation_request_support.dart

part of 'server.dart';

/// A mixin that adds support for making `elicitation/create` requests to a
/// [MCPServer].
base mixin ElicitationRequestSupport on LoggingSupport {
  /// Whether or not the connected client supports elicitation.
  ///
  /// Only safe to call after calling [initialize] on `super` since this is
  /// based on the client capabilities.
  bool get supportsElicitation => clientCapabilities.elicitation != null;

  @override
  FutureOr<InitializeResult> initialize(InitializeRequest request) {
    initialized.then((_) {
      if (!supportsElicitation) {
        log(
          LoggingLevel.warning,
          'Client does not support the elicitation capability, some '
          'functionality may be disabled.',
        );
      }
    });
    return super.initialize(request);
  }

  /// Sends an `elicitation/create` request to the client.
  ///
  /// This method will only succeed if the client has advertised the
  /// `elicitation` capability.
  Future<ElicitResult> elicit(ElicitRequest request) async {
    if (!supportsElicitation) {
      throw StateError('Client does not support elicitation');
    }
    return sendRequest(ElicitRequest.methodName, request);
  }
}