runtime/observatory/lib/src/repositories/target.dart

// Copyright (c) 2016, 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.

part of repositories;

typedef bool IsConnectedVMTargetDelegate(M.Target target);

class TargetChangeEvent implements M.TargetChangeEvent {
  final TargetRepository repository;
  final bool disconnected;
  TargetChangeEvent(this.repository, [this.disconnected = false]);
}

class TargetRepository implements M.TargetRepository {
  static const _historyKey = 'history';

  final StreamController<TargetChangeEvent> _onChange;
  final Stream<TargetChangeEvent> onChange;
  final SettingsRepository _settings = new SettingsRepository('targetManager');

  final List<SC.WebSocketVMTarget> _list = <SC.WebSocketVMTarget>[];
  SC.WebSocketVMTarget? current;
  final IsConnectedVMTargetDelegate _isConnectedVMTarget;

  factory TargetRepository(IsConnectedVMTargetDelegate isConnectedVMTarget) {
    var controller = new StreamController<TargetChangeEvent>();
    var stream = controller.stream.asBroadcastStream();
    return new TargetRepository._(isConnectedVMTarget, controller, stream);
  }

  TargetRepository._(this._isConnectedVMTarget, this._onChange, this.onChange) {
    _restore();
    final defaultAddress = _networkAddressOfDefaultTarget();
    var defaultTarget = find(defaultAddress);
    // Add the default address if it doesn't already exist.
    if (defaultTarget == null) {
      defaultTarget = new SC.WebSocketVMTarget(defaultAddress);
      _list.insert(0, defaultTarget);
    }
    // Set the current target to the default target.
    current = defaultTarget;
  }

  void add(String address) {
    if (find(address) != null) {
      return;
    }
    _list.insert(0, new SC.WebSocketVMTarget(address));
    _onChange.add(new TargetChangeEvent(this));
    _store();
  }

  Iterable<M.Target> list() => _list.toList();

  void setCurrent(M.Target t) {
    SC.WebSocketVMTarget target = t as SC.WebSocketVMTarget;
    if (!_list.contains(target)) {
      return;
    }
    current = target;
    current!.lastConnectionTime = new DateTime.now().millisecondsSinceEpoch;
    _onChange.add(new TargetChangeEvent(this));
    _store();
  }

  void emitDisconnectEvent() {
    _onChange.add(new TargetChangeEvent(this, true));
  }

  void delete(o) {
    if (_list.remove(o)) {
      if (o == current) {
        current = null;
      }
      _onChange.add(new TargetChangeEvent(this));
      _store();
    }
  }

  /// Read settings from data store.
  void _restore() {
    _list.clear();
    var loaded = _settings.get(_historyKey);
    if (loaded == null) {
      return;
    }
    for (var i in loaded) {
      _list.add(new SC.WebSocketVMTarget.fromMap(i));
    }
    _list.sort((SC.WebSocketVMTarget a, SC.WebSocketVMTarget b) {
      return b.lastConnectionTime.compareTo(a.lastConnectionTime);
    });
  }

  /// After making a change, update settings.
  void _store() {
    _settings.set(_historyKey, _list);
  }

  /// Find by networkAddress.
  SC.WebSocketVMTarget? find(String networkAddress) {
    for (SC.WebSocketVMTarget item in _list) {
      if (item.networkAddress == networkAddress) {
        return item;
      }
    }
    return null;
  }

  static String _networkAddressOfDefaultTarget() {
    // It is possible to override the default port and host by adding extra
    // query parameters:
    // http://localhost:8080?override-port=8181
    // http://localhost:8080?override-port=8181&override-host=10.0.0.2
    final Uri serverAddress = Uri.parse(window.location.toString());
    final String? port = serverAddress.queryParameters['override-port'];
    final String? host = serverAddress.queryParameters['override-host'];
    final Uri wsAddress = new Uri(
      scheme: 'ws',
      host: host ?? serverAddress.host,
      port: int.tryParse(port ?? '') ?? serverAddress.port,
      path: serverAddress.path.isEmpty ? '/ws' : serverAddress.path + 'ws',
    );
    return wsAddress.toString();
  }

  bool isConnectedVMTarget(M.Target target) => _isConnectedVMTarget(target);
}