pkg/dtd_impl/lib/src/dtd_stream_manager.dart

// Copyright (c) 2023, the Dart project authors.  Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.

import 'package:dart_service_protocol_shared/dart_service_protocol_shared.dart';
import 'package:dtd/dtd.dart' show DtdParameters;
import 'package:vm_service/vm_service.dart' show RPCError;

import 'dart_tooling_daemon.dart';
import 'dtd_client.dart' show DTDClient;

/// Manages state related to stream subscriptions made by [DTDClient]s.
class DTDStreamManager extends StreamManager {
  DTDStreamManager(this.dtd);

  final DartToolingDaemon dtd;

  /// Send an event to the [stream].
  ///
  /// [stream] must be a registered custom stream (i.e., not a stream specified
  /// as part of the VM service protocol).
  ///
  /// If [stream] is not a registered custom stream, an [RPCError] with code
  /// `kCustomStreamDoesNotExist` will be thrown.
  ///
  /// If [stream] is a core stream, an [RPCError] with code
  /// `kCoreStreamNotAllowed` will be thrown.
  void postEventHelper(
    String stream,
    String eventKind,
    Map<String, Object?> eventData,
  ) {
    super.postEvent(stream, <String, Object?>{
      DtdParameters.streamId: stream,
      DtdParameters.eventKind: eventKind,
      DtdParameters.eventData: eventData,
      DtdParameters.timestamp: DateTime.now().millisecondsSinceEpoch,
    });
  }

  /// Send `streamNotify` notifications to clients subscribed to `streamId`.
  void streamNotify(String streamId, Map<String, Object?> data) {
    super.postEvent(streamId, data);
  }
}