//! Code to watch configuration files for any changes.

//!

// TODO: perhaps this shouldn't live in tor-config? But it doesn't seem substantial enough to have

// its own crate, and it can't live in e.g. tor-basic-utils, because it depends on tor-rtcompat.

use std::collections::hash_map::Entry;

use std::collections::{HashMap, HashSet};

use std::io;

use std::path::{Path, PathBuf};

use std::pin::Pin;

use std::sync::Arc;

use std::task::{Context, Poll};

use tor_rtcompat::Runtime;

use amplify::Getters;

use futures::lock::Mutex;

use notify::{EventKind, Watcher};

use postage::watch;

use futures::{SinkExt as _, Stream, StreamExt as _};

/// `Result` whose `Err` is [`FileWatcherBuildError`].

pub type Result<T> = std::result::Result<T, FileWatcherBuildError>;

cfg_if::cfg_if! {

    if #[cfg(any(target_os = "linux", target_os = "android", target_os = "windows"))] {

        /// The concrete type of the underlying watcher.

        type NotifyWatcher = notify::RecommendedWatcher;

    } else {

        /// The concrete type of the underlying watcher.

        type NotifyWatcher = notify::PollWatcher;

    }

}

/// A wrapper around a `notify::Watcher` to watch a set of parent

/// directories in order to learn about changes in some specific files that they

/// contain.

///

/// The `Watcher` implementation in `notify` has a weakness: it gives sensible

/// results when you're watching directories, but if you start watching

/// non-directory files, it won't notice when those files get replaced.  That's

/// a problem for users who want to change their configuration atomically by

/// making new files and then moving them into place over the old ones.

///

/// For more background on the issues with `notify`, see

/// <https://github.com/notify-rs/notify/issues/165> and

/// <https://github.com/notify-rs/notify/pull/166>.

///

/// ## Limitations

///

/// On backends using kqueue, this uses a polling watcher

/// to work around a bug in the `notify` crate[^1].

/// This introduces a perceivable delay,

/// and can be very expensive for large file trees.

///

/// [^1]: See <https://github.com/notify-rs/notify/issues/644>

#[derive(Getters)]

pub struct FileWatcher {

    /// An underlying `notify` watcher that tells us about directory changes.

    // this field is kept only so the watcher is not dropped

    #[getter(skip)]

    _watcher: NotifyWatcher,

    /// The list of directories that we're currently watching.

    watching_dirs: HashSet<PathBuf>,

}

impl FileWatcher {

    /// Create a `FileWatcherBuilder`

}

/// Event possibly triggering a configuration reload

#[derive(Debug, Clone, PartialEq)]

#[non_exhaustive]

pub enum Event {

    /// Some files may have been modified.

    FileChanged,

    /// Some filesystem events may have been missed.

    Rescan,

}

/// Builder used to configure a [`FileWatcher`] before it starts watching for changes.

pub struct FileWatcherBuilder<R: Runtime> {

    /// The runtime.

    runtime: R,

    /// The list of directories that we're currently watching.

    ///

    /// Each directory has a set of filters that indicates whether a given notify::Event

    /// is relevant or not.

    watching_dirs: HashMap<PathBuf, HashSet<DirEventFilter>>,

}

/// A filter for deciding what to do with a notify::Event pertaining

/// to files that are relative to one of the directories we are watching.

///

// Private, as this is an implementation detail.

// If/when we decide to make this public, this might need revisiting.

#[derive(Clone, Debug, Hash, PartialEq, Eq)]

enum DirEventFilter {

    /// Notify the caller about the event, if the file has the specified extension.

    MatchesExtension(String),

    /// Notify the caller about the event, if the file has the specified path.

    MatchesPath(PathBuf),

}

impl DirEventFilter {

    /// Check whether this filter accepts `path`.

        }

}

impl<R: Runtime> FileWatcherBuilder<R> {

    /// Create a `FileWatcherBuilder`

    /// Add a single path to the list of things to watch.

    ///

    /// The event receiver will be notified if the path is created, modified, renamed, or removed.

    ///

    /// If the path is a directory, its contents will **not** be watched.

    /// To watch the contents of a directory, use [`watch_dir`](FileWatcherBuilder::watch_dir).

    ///

    /// Idempotent: does nothing if we're already watching that path.

    /// Add a directory (but not any subdirs) to the list of things to watch.

    ///

    /// The event receiver will be notified whenever a file with the specified `extension`

    /// is created within this directory, or if an existing file with this extension

    /// is modified, renamed, or removed.

    /// Changes to files that have a different extension are ignored.

    ///

    /// Idempotent.

    /// Add the parents of `path` to the list of things to watch.

    ///

    /// Returns the absolute path of `path`.

    ///

    /// Idempotent.

        // Make the path absolute (without necessarily making it canonical).

        //

        // We do this because `notify` reports all of its events in terms of

        // absolute paths, so if we were to tell it to watch a directory by its

        // relative path, we'd get reports about the absolute paths of the files

        // in that directory.

        // See what directory we should watch in order to watch this file.

            // The file has a parent, so watch that.

            // The file has no parent.  Given that it's absolute, that means

            // that we're looking at the root directory.  There's nowhere to go

            // "up" from there.

        };

        // Note this file as one that we're watching, so that we can see changes

        // to it later on.

    /// Add just this (absolute) directory to the list of things to watch.

    ///

    /// Does not watch any of the parents.

    ///

    /// Idempotent.

        }

    /// Build a `FileWatcher` and start sending events to `tx`.

    ///

    /// On startup, the watcher sends a [`Rescan`](Event::Rescan) event.

    /// This helps mitigate the event loss that occurs if the watched files are modified between

    /// the time they are initially loaded and the time when the watcher is set up.

        cfg_if::cfg_if! {

            if #[cfg(any(target_os = "linux", target_os = "android", target_os = "windows"))] {

            } else {

                /// The polling frequency, for use with the `PollWatcher`.

                #[cfg(not(any(test, feature = "testing")))]

                const WATCHER_POLL_INTERVAL: std::time::Duration = std::time::Duration::from_secs(5);

                #[cfg(any(test, feature = "testing"))]

                const WATCHER_POLL_INTERVAL: std::time::Duration = std::time::Duration::from_millis(10);

                let config = notify::Config::default()

                    .with_poll_interval(WATCHER_POLL_INTERVAL);

                // When testing, compare the contents of the files too, not just their mtime

                // Otherwise, because the polling backend detects changes based on mtime,

                // if the test creates/writes files too fast,

                // it will fail to notice changes (this can happen, for example, on a tmpfs).

                #[cfg(any(test, feature = "testing"))]

                let config = config.with_compare_contents(true);

            }

        }

        }

}

/// Map a `notify` event to the [`Event`] type returned by [`FileWatcher`].

        {

            }

        }

    // filter events we don't want and map to event code

            } else {

            }

        }

            } else {

            }

        }

    }

/// Check whether this is a kind of [`notify::Event`] that we want to ignore.

///

/// Returns `true` for

///   * events that trigger on non-mutating file accesses

///   * catch-all events (used by `notify` for unsupported/unknown events)

///   * "other" meta-events

    use EventKind::*;

/// The sender half of a watch channel used by a [`FileWatcher`] for sending [`Event`]s.

///

/// For use with [`FileWatcherBuilder::start_watching`].

///

/// **Important**: to avoid contention, avoid sharing clones of the same `FileEventSender`

/// with multiple [`FileWatcherBuilder`]s. This type is [`Clone`] to support creating new

/// [`FileWatcher`]s from an existing [`channel`], which enables existing receivers to receive

/// events from new `FileWatcher`s (any old `FileWatcher`s are supposed to be discarded).

#[derive(Clone)]

pub struct FileEventSender(Arc<Mutex<watch::Sender<Event>>>);

/// The receiver half of a watch channel used for receiving [`Event`]s sent by a [`FileWatcher`].

#[derive(Clone)]

pub struct FileEventReceiver(watch::Receiver<Event>);

impl Stream for FileEventReceiver {

    type Item = Event;

}

impl FileEventReceiver {

    /// Try to read a message from the stream, without blocking.

    ///

    /// Returns `Some` if a message is ready.

    /// Returns `None` if the stream is open, but no messages are available,

    /// or if the stream is closed.

        use postage::prelude::Stream;

}

/// Create a new channel for use with a [`FileWatcher`].

//

// Note: the [`FileEventSender`] and [`FileEventReceiver`]  wrappers exist

// so we don't expose the channel's underlying type

// in our public API.

/// An error coming from a [`FileWatcherBuilder`].

#[derive(Debug, Clone, thiserror::Error)]

#[non_exhaustive]

pub enum FileWatcherBuildError {

    /// Invalid current working directory.

    ///

    /// This error can happen if the current directory does not exist,

    /// or if we don't have the necessary permissions to access it.

    #[error("Invalid current working directory")]

    CurrentDirectory(#[source] Arc<io::Error>),

    /// Encountered a problem while creating a `Watcher`.

    #[error("Problem creating Watcher")]

    Notify(#[from] Arc<notify::Error>),

}

#[cfg(test)]

mod test {

    // @@ begin test lint list maintained by maint/add_warning @@

    #![allow(clippy::bool_assert_comparison)]

    #![allow(clippy::clone_on_copy)]

    #![allow(clippy::dbg_macro)]

    #![allow(clippy::mixed_attributes_style)]

    #![allow(clippy::print_stderr)]

    #![allow(clippy::print_stdout)]

    #![allow(clippy::single_char_pattern)]

    #![allow(clippy::unwrap_used)]

    #![allow(clippy::unchecked_duration_subtraction)]

    #![allow(clippy::useless_vec)]

    #![allow(clippy::needless_pass_by_value)]

    //! <!-- @@ end test lint list maintained by maint/add_warning @@ -->

    use super::*;

    use notify::event::{AccessKind, ModifyKind};

    use test_temp_dir::{test_temp_dir, TestTempDir};

    /// Write `data` to file `name` within `dir`.

    fn write_file(dir: &TestTempDir, name: &str, data: &[u8]) -> PathBuf {

        let path = dir.as_path_untracked().join(name);

        std::fs::write(&path, data).unwrap();

        path

    }

    /// Return an event that has the Rescan flag set

    fn rescan_event() -> notify::Event {

        let event = notify::Event::new(notify::EventKind::Any);

        event.set_flag(notify::event::Flag::Rescan)

    }

    /// Assert that at least one FileChanged event is received.

    async fn assert_file_changed(rx: &mut FileEventReceiver) {

        assert_eq!(rx.next().await, Some(Event::FileChanged));

        // The write might trigger more than one event

        while let Some(ev) = rx.try_recv() {

            assert_eq!(ev, Event::FileChanged);

        }

    }

    /// Set the `EventKind` of `event` to an uninteresting `EventKind`

    /// and assert that it is ignored by `handle_event`.

    fn assert_ignored(event: &notify::Event, watching: &HashMap<PathBuf, HashSet<DirEventFilter>>) {

        for kind in [EventKind::Access(AccessKind::Any), EventKind::Other] {

            let ignored_event = event.clone().set_kind(kind);

            assert_eq!(handle_event(Ok(ignored_event.clone()), watching), None);

            // ...but if the rescan flag is set, the event is *not* ignored

            let event = ignored_event.set_flag(notify::event::Flag::Rescan);

            assert_eq!(handle_event(Ok(event), watching), Some(Event::Rescan));

        }

    }

    #[test]

    fn notify_event_handler() {

        let mut event = notify::Event::new(notify::EventKind::Modify(ModifyKind::Any));

        let mut watching_dirs = Default::default();

        assert_eq!(handle_event(Ok(event.clone()), &watching_dirs), None);

        assert_eq!(

            handle_event(Ok(rescan_event()), &watching_dirs),

            Some(Event::Rescan)

        );

        // Watch some directories

        watching_dirs.insert(

            "/foo/baz".into(),

            HashSet::from([DirEventFilter::MatchesExtension("auth".into())]),

        );

        assert_eq!(handle_event(Ok(event.clone()), &watching_dirs), None);

        assert_eq!(

            handle_event(Ok(rescan_event()), &watching_dirs),

            Some(Event::Rescan)

        );

        event = event.add_path("/foo/bar/alice.authh".into());

        assert_eq!(handle_event(Ok(event.clone()), &watching_dirs), None);

        event = event.add_path("/foo/bar/alice.auth".into());

        assert_eq!(handle_event(Ok(event.clone()), &watching_dirs), None);

        event = event.add_path("/foo/baz/bob.auth".into());

        assert_eq!(

            handle_event(Ok(event.clone()), &watching_dirs),

            Some(Event::FileChanged)

        );

        // The same event, but with an irrelevant kind, gets ignored:

        assert_ignored(&event, &watching_dirs);

        // Watch some files within /foo/bar

        watching_dirs.insert(

            "/foo/bar".into(),

            HashSet::from([DirEventFilter::MatchesPath("/foo/bar/abc".into())]),

        );

        assert_eq!(

            handle_event(Ok(event.clone()), &watching_dirs),

            Some(Event::FileChanged)

        );

        assert_eq!(

            handle_event(Ok(rescan_event()), &watching_dirs),

            Some(Event::Rescan)

        );

        // The same event, but with an irrelevant kind, gets ignored:

        assert_ignored(&event, &watching_dirs);

        // Watch some other files

        let event = notify::Event::new(notify::EventKind::Modify(ModifyKind::Any))

            .add_path("/a/b/c/d".into());

        let watching_dirs = [(

            "/a/b/c/".into(),

            HashSet::from([DirEventFilter::MatchesPath("/a/b/c/d".into())]),

        )]

        .into_iter()

        .collect();

        assert_eq!(

            handle_event(Ok(event), &watching_dirs),

            Some(Event::FileChanged)

        );

        assert_eq!(

            handle_event(Ok(rescan_event()), &watching_dirs),

            Some(Event::Rescan)

        );

        // Errors can also trigger an event

        let err = notify::Error::path_not_found();

        assert_eq!(handle_event(Err(err), &watching_dirs), None);

        let mut err = notify::Error::path_not_found();

        err = err.add_path("/a/b/c/d".into());

        assert_eq!(

            handle_event(Err(err), &watching_dirs),

            Some(Event::FileChanged)

        );

    }

    #[test]

    fn watch_dirs() {

        tor_rtcompat::test_with_one_runtime!(|rt| async move {

            let temp_dir = test_temp_dir!();

            let (tx, mut rx) = channel();

            // Watch for changes in .foo files from temp_dir

            let mut builder = FileWatcher::builder(rt.clone());

            builder

                .watch_dir(temp_dir.as_path_untracked(), "foo")

                .unwrap();

            let watcher = builder.start_watching(tx).unwrap();

            // On startup, the watcher sends a Event::Rescan event.

            // This is because the watcher is often set up after loading

            // the files or directories it is watching.

            assert_eq!(rx.try_recv(), Some(Event::Rescan));

            assert_eq!(rx.try_recv(), None);

            // Write a file with extension "foo".

            write_file(&temp_dir, "bar.foo", b"hello");

            assert_eq!(rx.next().await, Some(Event::FileChanged));

            drop(watcher);

            // The write might trigger more than one event

            while let Some(ev) = rx.next().await {

                assert_eq!(ev.clone(), Event::FileChanged);

            }

        });

    }

    #[test]

    fn watch_file_path() {

        tor_rtcompat::test_with_one_runtime!(|rt| async move {

            let temp_dir = test_temp_dir!();

            let (tx, mut rx) = channel();

            // Watch for changes to hello.txt

            let path = write_file(&temp_dir, "hello.txt", b"hello");

            let mut builder = FileWatcher::builder(rt.clone());

            builder.watch_path(&path).unwrap();

            let _watcher = builder.start_watching(tx).unwrap();

            // On startup, the watcher sends a Event::Rescan event.

            assert_eq!(rx.try_recv(), Some(Event::Rescan));

            assert_eq!(rx.try_recv(), None);

            // Write to hello.txt

            let _: PathBuf = write_file(&temp_dir, "hello.txt", b"good-bye");

            assert_file_changed(&mut rx).await;

            // Remove hello.txt

            std::fs::remove_file(&path).unwrap();

            assert_file_changed(&mut rx).await;

            // Create a new file

            let tmp_hello = write_file(&temp_dir, "hello.tmp", b"new hello");

            // Copy it over to the watched hello.txt location

            std::fs::rename(&tmp_hello, &path).unwrap();

            assert_file_changed(&mut rx).await;

        });

    }

    #[test]

    fn watch_dir_path() {

        tor_rtcompat::test_with_one_runtime!(|rt| async move {

            let temp_dir1 = tempfile::TempDir::new().unwrap();

            let (tx, mut rx) = channel();

            // Watch temp_dir for changes

            let mut builder = FileWatcher::builder(rt.clone());

            builder.watch_path(temp_dir1.path()).unwrap();

            let _watcher = builder.start_watching(tx).unwrap();

            // On startup, the watcher sends a Event::Rescan event.

            assert_eq!(rx.try_recv(), Some(Event::Rescan));

            assert_eq!(rx.try_recv(), None);

            // Writing a file to this directory shouldn't trigger an event

            std::fs::write(temp_dir1.path().join("hello.txt"), b"hello").unwrap();

            assert_eq!(rx.try_recv(), None);

            // Move temp_dir1 to temp_dir2

            let temp_dir2 = tempfile::TempDir::new().unwrap();

            std::fs::rename(&temp_dir1, &temp_dir2).unwrap();

            // Moving the directory triggers an event...

            assert_file_changed(&mut rx).await;

            // ...and so does moving it back to its original location

            std::fs::rename(&temp_dir2, &temp_dir1).unwrap();

            assert_file_changed(&mut rx).await;

        });

    }

}