73 lines
1.9 KiB
Rust
73 lines
1.9 KiB
Rust
use crate::types::{Epoch, Rank};
|
|
|
|
use serde::{Deserialize, Serialize};
|
|
use std::path::PathBuf;
|
|
|
|
#[derive(Clone, Debug, Default, Deserialize, Serialize)]
|
|
pub struct Dir {
|
|
pub path: PathBuf,
|
|
pub rank: Rank,
|
|
pub last_accessed: Epoch,
|
|
}
|
|
|
|
impl Dir {
|
|
pub fn is_dir(&self) -> bool {
|
|
self.path.is_dir()
|
|
}
|
|
|
|
#[cfg(unix)]
|
|
pub fn is_match(&self, query: &[String]) -> bool {
|
|
use bstr::ByteSlice;
|
|
use std::os::unix::ffi::OsStrExt;
|
|
|
|
let path_bytes = self.path.as_os_str().as_bytes().to_lowercase();
|
|
let mut subpath = path_bytes.as_slice();
|
|
|
|
for subquery in query.iter() {
|
|
let subquery_bytes = subquery.as_bytes();
|
|
match subpath.find(subquery_bytes) {
|
|
Some(idx) => subpath = &subpath[idx + subquery_bytes.len()..],
|
|
None => return false,
|
|
}
|
|
}
|
|
|
|
true
|
|
}
|
|
|
|
#[cfg(not(unix))]
|
|
pub fn is_match(&self, query: &[String]) -> bool {
|
|
let path_str = match self.path.to_str() {
|
|
Some(path_str) => path_str.to_lowercase(),
|
|
None => return false, // silently ignore invalid UTF-8
|
|
};
|
|
|
|
let mut subpath = path_str.as_str();
|
|
|
|
for subquery in query.iter() {
|
|
match subpath.find(subquery) {
|
|
Some(idx) => subpath = &subpath[idx + subquery.len()..],
|
|
None => return false,
|
|
}
|
|
}
|
|
|
|
true
|
|
}
|
|
|
|
pub fn get_frecency(&self, now: Epoch) -> Rank {
|
|
const HOUR: Epoch = 60 * 60;
|
|
const DAY: Epoch = 24 * HOUR;
|
|
const WEEK: Epoch = 7 * DAY;
|
|
|
|
let duration = now - self.last_accessed;
|
|
if duration < HOUR {
|
|
self.rank * 4.0
|
|
} else if duration < DAY {
|
|
self.rank * 2.0
|
|
} else if duration < WEEK {
|
|
self.rank / 2.0
|
|
} else {
|
|
self.rank / 4.0
|
|
}
|
|
}
|
|
}
|