70 lines
1.9 KiB
Rust
70 lines
1.9 KiB
Rust
//! Module containing the definition of [`OpenedFilesModel`],
|
|
//! which tracks files that have been opened, organized by repository.
|
|
|
|
use std::{collections::HashMap, path::PathBuf};
|
|
|
|
use instant::Instant;
|
|
use warpui::{Entity, ModelContext, SingletonEntity};
|
|
|
|
#[derive(Default, Clone)]
|
|
pub struct OpenedFilesInRepo(HashMap<PathBuf, Instant>);
|
|
|
|
impl OpenedFilesInRepo {
|
|
pub fn get(&self, file_path: &PathBuf) -> Option<&Instant> {
|
|
self.0.get(file_path)
|
|
}
|
|
|
|
pub fn iter(&self) -> impl Iterator<Item = (&PathBuf, &Instant)> {
|
|
self.0.iter()
|
|
}
|
|
}
|
|
|
|
/// Model that tracks files that have been opened, organized by repository.
|
|
/// Maps repository paths to files and when they were last opened.
|
|
#[derive(Default)]
|
|
pub struct OpenedFilesModel {
|
|
opened_files: HashMap<PathBuf, OpenedFilesInRepo>,
|
|
}
|
|
|
|
impl Entity for OpenedFilesModel {
|
|
type Event = ();
|
|
}
|
|
|
|
impl SingletonEntity for OpenedFilesModel {}
|
|
|
|
impl OpenedFilesModel {
|
|
pub fn new() -> Self {
|
|
Self::default()
|
|
}
|
|
|
|
/// Get all opened files for a specific repository.
|
|
pub fn opened_files_for_repo(&self, repo_path: &PathBuf) -> Option<&OpenedFilesInRepo> {
|
|
self.opened_files.get(repo_path)
|
|
}
|
|
|
|
/// Record that a file has been opened in a repository. If the `file_path` is not within the `repo_path`,
|
|
/// then the file is not recorded.
|
|
#[cfg_attr(not(feature = "local_fs"), allow(dead_code))]
|
|
pub fn file_opened(
|
|
&mut self,
|
|
repo_path: PathBuf,
|
|
file_path: PathBuf,
|
|
ctx: &mut ModelContext<Self>,
|
|
) {
|
|
let opened_at = Instant::now();
|
|
|
|
// Convert absolute file path to relative path from repo root
|
|
let Ok(relative_file_path) = file_path.strip_prefix(&repo_path) else {
|
|
return;
|
|
};
|
|
|
|
self.opened_files
|
|
.entry(repo_path.clone())
|
|
.or_default()
|
|
.0
|
|
.insert(relative_file_path.into(), opened_at);
|
|
|
|
ctx.notify();
|
|
}
|
|
}
|