clipmem 0.5.6

use crate::model::{CaptureStoreResult, ClipboardSnapshot};

/// Tracks the pasteboard change count accounted for by the watch loop.
///
/// Callers decide whether to skip or capture a change, then call
/// `mark_change_handled` once that change count has been accounted for.
#[derive(Debug, Clone)]
pub struct WatchState {
    last_handled_change_count: Option<i64>,
    first_loop: bool,
}

impl WatchState {
    #[must_use]
    pub fn new() -> Self {
        Self {
            last_handled_change_count: None,
            first_loop: true,
        }
    }
}

impl Default for WatchState {
    fn default() -> Self {
        Self::new()
    }
}

pub fn should_skip_initial_change(skip_initial: bool, state: &WatchState) -> bool {
    skip_initial && state.first_loop
}

pub fn should_capture_change(change_count: i64, state: &WatchState) -> bool {
    state.last_handled_change_count != Some(change_count)
}

pub fn mark_change_handled(change_count: i64, state: &mut WatchState) {
    state.last_handled_change_count = Some(change_count);
    state.first_loop = false;
}

#[must_use]
pub fn format_watch_capture_line(
    snapshot: &ClipboardSnapshot,
    result: &CaptureStoreResult,
) -> String {
    let source = snapshot
        .frontmost_app_name()
        .or(snapshot.frontmost_app_bundle_id())
        .unwrap_or("unknown app");

    format!(
        "event={} snapshot={} {} kind={} bytes={} source={} preview={}",
        result.event_id(),
        result.snapshot_id(),
        if result.inserted_new_snapshot() {
            "new"
        } else {
            "seen"
        },
        snapshot.snapshot_kind(),
        snapshot.total_bytes(),
        source,
        snapshot.preview_text()
    )
}

#[cfg(test)]
mod tests {
    use anyhow::Result;

    use super::{
        format_watch_capture_line, mark_change_handled, should_capture_change,
        should_skip_initial_change, WatchState,
    };
    use crate::db::{Database, RetrievalFilters};
    use crate::model::{
        build_item, build_representation, build_snapshot, CaptureContext, CaptureStoreResult,
        ClipboardSnapshot,
    };

    fn process_watch_snapshot(
        db: &mut Database,
        snapshot: &ClipboardSnapshot,
        state: &mut WatchState,
    ) -> Result<Option<CaptureStoreResult>> {
        if !should_capture_change(snapshot.change_count(), state) {
            return Ok(None);
        }

        let result = db.store_capture(snapshot)?;
        mark_change_handled(snapshot.change_count(), state);
        Ok(Some(result))
    }

    #[test]
    fn skip_initial_marks_first_change_as_seen_without_storing() -> Result<()> {
        let db = Database::open_in_memory()?;
        let mut state = WatchState::new();
        let snapshot = build_snapshot(
            CaptureContext::new(7).with_frontmost_app_name("Editor"),
            Vec::new(),
        );

        if should_skip_initial_change(true, &state) {
            mark_change_handled(snapshot.change_count(), &mut state);
        }

        assert!(!should_capture_change(snapshot.change_count(), &state));
        assert!(db
            .recent(10, &RetrievalFilters::default())?
            .hits()
            .is_empty());
        Ok(())
    }

    #[test]
    fn repeated_change_counts_are_ignored_after_the_first_store() -> Result<()> {
        let mut db = Database::open_in_memory()?;
        let mut state = WatchState::new();
        let snapshot = build_snapshot(
            CaptureContext::new(7).with_frontmost_app_name("Editor"),
            Vec::new(),
        );

        let first = process_watch_snapshot(&mut db, &snapshot, &mut state)?;
        let second = process_watch_snapshot(&mut db, &snapshot, &mut state)?;

        assert!(first.is_some());
        assert!(second.is_none());
        assert_eq!(db.recent(10, &RetrievalFilters::default())?.hits().len(), 1);
        Ok(())
    }

    #[test]
    fn changed_change_counts_store_new_events_for_existing_content() -> Result<()> {
        let mut db = Database::open_in_memory()?;
        let mut state = WatchState::new();
        let first = build_snapshot(
            CaptureContext::new(7).with_frontmost_app_name("Editor"),
            Vec::new(),
        );
        let second = build_snapshot(
            CaptureContext::new(8).with_frontmost_app_name("Editor"),
            Vec::new(),
        );

        let first_store = anyhow::Context::context(
            process_watch_snapshot(&mut db, &first, &mut state)?,
            "expected first watch snapshot to store",
        )?;
        let second_store = anyhow::Context::context(
            process_watch_snapshot(&mut db, &second, &mut state)?,
            "expected second watch snapshot to store",
        )?;
        let details = db
            .find_snapshot(first_store.snapshot_id(), 10)?
            .ok_or_else(|| anyhow::anyhow!("expected stored snapshot details"))?;

        assert_eq!(first_store.snapshot_id(), second_store.snapshot_id());
        assert!(!second_store.inserted_new_snapshot());
        assert_eq!(details.capture_count(), 2);
        Ok(())
    }

    #[test]
    fn watch_log_line_includes_capture_context() {
        let snapshot = build_snapshot(
            CaptureContext::new(3)
                .with_frontmost_app_name("Editor")
                .with_frontmost_app_bundle_id("com.example.Editor"),
            vec![build_item(
                0,
                vec![build_representation(
                    "public.utf8-plain-text".to_string(),
                    None,
                    b"hello world".to_vec(),
                )],
            )],
        );
        let result = CaptureStoreResult::new(42, 7, true);

        let line = format_watch_capture_line(&snapshot, &result);

        assert!(line.contains("event=7"));
        assert!(line.contains("snapshot=42"));
        assert!(line.contains("new"));
        assert!(line.contains("source=Editor"));
        assert!(line.contains("preview=hello world"));
    }
}