// Copyright (c) 2020, 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 '../watcher.dart';

/// A factory to produce custom watchers for specific file paths.
class _CustomWatcherFactory {
  final String id;
  final DirectoryWatcher? Function(String path, {Duration? pollingDelay})
      createDirectoryWatcher;
  final FileWatcher? Function(String path, {Duration? pollingDelay})
      createFileWatcher;

  _CustomWatcherFactory(
      this.id, this.createDirectoryWatcher, this.createFileWatcher);
}

/// Registers a custom watcher.
///
/// Each custom watcher must have a unique [id] and the same watcher may not be
/// registered more than once.
/// [createDirectoryWatcher] and [createFileWatcher] should return watchers for
/// the file paths they are able to handle. If the custom watcher is not able to
/// handle the path it should return null.
/// The paths handled by each custom watch may not overlap, at most one custom
/// matcher may return a non-null watcher for a given path.
///
/// When a file or directory watcher is created the path is checked against each
/// registered custom watcher, and if exactly one custom watcher is available it
/// will be used instead of the default.
void registerCustomWatcher(
  String id,
  DirectoryWatcher? Function(String path, {Duration? pollingDelay})?
      createDirectoryWatcher,
  FileWatcher? Function(String path, {Duration? pollingDelay})?
      createFileWatcher,
) {
  if (_customWatcherFactories.containsKey(id)) {
    throw ArgumentError('A custom watcher with id `$id` '
        'has already been registered');
  }
  _customWatcherFactories[id] = _CustomWatcherFactory(
      id,
      createDirectoryWatcher ?? (_, {pollingDelay}) => null,
      createFileWatcher ?? (_, {pollingDelay}) => null);
}

/// Tries to create a custom [DirectoryWatcher] and returns it.
///
/// Returns `null` if no custom watcher was applicable and throws a [StateError]
/// if more than one was.
DirectoryWatcher? createCustomDirectoryWatcher(String path,
    {Duration? pollingDelay}) {
  DirectoryWatcher? customWatcher;
  String? customFactoryId;
  for (var watcherFactory in _customWatcherFactories.values) {
    if (customWatcher != null) {
      throw StateError('Two `CustomWatcherFactory`s applicable: '
          '`$customFactoryId` and `${watcherFactory.id}` for `$path`');
    }
    customWatcher =
        watcherFactory.createDirectoryWatcher(path, pollingDelay: pollingDelay);
    customFactoryId = watcherFactory.id;
  }
  return customWatcher;
}

/// Tries to create a custom [FileWatcher] and returns it.
///
/// Returns `null` if no custom watcher was applicable and throws a [StateError]
/// if more than one was.
FileWatcher? createCustomFileWatcher(String path, {Duration? pollingDelay}) {
  FileWatcher? customWatcher;
  String? customFactoryId;
  for (var watcherFactory in _customWatcherFactories.values) {
    if (customWatcher != null) {
      throw StateError('Two `CustomWatcherFactory`s applicable: '
          '`$customFactoryId` and `${watcherFactory.id}` for `$path`');
    }
    customWatcher =
        watcherFactory.createFileWatcher(path, pollingDelay: pollingDelay);
    customFactoryId = watcherFactory.id;
  }
  return customWatcher;
}

final _customWatcherFactories = <String, _CustomWatcherFactory>{};