quarkdrive-webdav 1.3.6

use std::path::Path;
use std::time::Duration;
use moka::future::Cache as MokaCache;
use tracing::debug;
use crate::drive::{QuarkDrive};
use crate::drive::model::QuarkFile;

#[derive(Clone)]
pub struct Cache {
    inner: MokaCache<String, Vec<QuarkFile>>,
    drive: QuarkDrive,
}
const ONE_PAGE: u32 = 500;

impl Cache {
    pub fn new(max_capacity: u64, ttl: u64, drive: QuarkDrive) -> Self {
        let inner = MokaCache::builder()
            .max_capacity(max_capacity)
            .time_to_live(Duration::from_secs(ttl))
            .build();
        
        Self { inner , drive}
    }
    pub async fn get_or_insert(&self, key: &str) -> Option<Vec<QuarkFile>> {
        debug!(key = %key, "cache: get_or_insert");
        if let Some(files) = self.get(key).await {
            return Some(files);
        }
        if key == "/" {
            self.dfs(QuarkFile::new_root(), key, "/").await;
        }else {
            let mut path = Path::new(key);
            let mut dsf_root_file = None;
            while let Some(parent) = path.parent() {
                if let Some(c_files) = self.get(parent.to_str().unwrap()).await {
                    let file_name = path.file_name().and_then(|os_str| os_str.to_str());
                    let found = c_files.iter().find(|quark_file| {
                        Some(quark_file.file_name.as_str()) == file_name
                    }).cloned();
                    if found.is_none() {
                        debug!(key = %key, "cache: no file found for path: {}", path.to_str().unwrap());
                        path = parent;
                        continue;
                    }
                    dsf_root_file = found;
                    break;
                }

                path = parent;

                if path.to_str() == Some("/") {
                    break;
                }

            }
            if path.to_str() == Some("/") {
                self.dfs(QuarkFile::new_root(), key, "/").await;
            }else {
                match dsf_root_file { 
                    Some(dsf_root_fil) => {
                        debug!(key = %key, "cache: found root file: {}", dsf_root_fil.file_name);
                        self.dfs(dsf_root_fil, key, path.to_str().unwrap()).await;
                    },
                    None => {
                        debug!(key = %key, "cache: no root file found for path: {}", path.to_str().unwrap());
                        return None;
                    }
                }
            }

        }
        if let Some(files) = self.get(key).await {
            Some(files)
        }else {
            debug!(key = %key, "cache: no files found for key");
            None
        }
    }

    async fn dfs(&self, file: QuarkFile, target_path: &str, dfs_path: &str) {
        if file.dir {
            let mut current_files = Vec::<QuarkFile>::new();
            for page_no in 1..=20 {
                let (files, total) =
                    match self.drive.get_files_by_pdir_fid(&file.fid, page_no, ONE_PAGE).await{
                    Ok((k, v)) => (k, v),
                    Err(e) => {
                        debug!(error = %e, file_id = &file.fid, file_name = &file.file_name,
                                page_no = page_no,
                            "Failed to get files from drive");
                        return;
                    }
                };
                let mut files = files.unwrap();
                // add dfs_path to each file
                for f in files.list.iter_mut() {
                    f.parent_path = Some(dfs_path.to_string());
                }
                let size = files.list.len();
                current_files.extend(files.list);
                // guess: es limit is 10000
                if size < ONE_PAGE as usize || page_no >= total / ONE_PAGE + 1   {
                    break;
                }
            }

            self.insert(dfs_path.to_string(), current_files.clone()).await;
            debug!("{} in cache", &dfs_path);
            if dfs_path == target_path {
                return;
            }
            for curr_f in current_files {
                let file_path = if dfs_path == "/" {
                    format!("{}{}", dfs_path, curr_f.file_name)
                }else {
                    format!("{}/{}", dfs_path, curr_f.file_name)
                };
                if target_path.starts_with(&file_path) {
                    Box::pin(self.dfs(curr_f, target_path, &file_path)).await;
                }
            }

        }
    }

    async fn get(&self, key: &str) -> Option<Vec<QuarkFile>> {
        debug!(key = %key, "cache: get");
        self.inner.get(key).await
    }

    async fn insert(&self, key: String, value: Vec<QuarkFile>) {
        debug!(key = %key, "cache: insert");
        self.inner.insert(key, value).await;
    }

    pub async fn invalidate(&self, path: &Path) {
        let key = path.to_string_lossy().into_owned();
        debug!(path = %path.display(), key = %key, "cache: invalidate");
        self.inner.invalidate(&key).await;
    }

    pub async fn invalidate_parent(&self, path: &Path) {
        if let Some(parent) = path.parent() {
            self.invalidate(parent).await;
        }
    }

    pub fn invalidate_all(&self) {
        debug!("cache: invalidate all");
        self.inner.invalidate_all();
    }

}