cymbal 0.9.0

search for symbols in a codebase
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

use std::collections::HashSet;
use std::path::PathBuf;

use anyhow::Result;
use ignore::Walk;
use tokio::task::JoinHandle;

use crate::cache::Cache;
use crate::channel::{FileTask, Sender};
use crate::config::Language;
use crate::ext::IntoExt;

pub struct Walker {
  path: PathBuf,
  sender: Sender,
  cache: Option<Cache>,
  should_clean_cache: bool,
}

impl Walker {
  pub fn new(path: PathBuf, sender: Sender, cache: Option<Cache>, should_clean_cache: bool) -> Self {
    Self {
      path,
      sender,
      cache,
      should_clean_cache,
    }
  }

  pub fn spawn(self) -> JoinHandle<Result<()>> {
    tokio::spawn(self.run())
  }

  async fn run(self) -> Result<()> {
    let walker = Walk::new(self.path).filter_map(Result::ok).filter_map(|dir_entry| {
      let metadata = dir_entry.metadata().ok()?;
      if !metadata.is_file() {
        return None;
      }
      let file_modified = metadata.modified().ok()?;
      let file_path = dir_entry.into_path();
      let language = Language::from_file_path(&file_path)?;

      (file_path, file_modified, language).some()
    });

    let mut file_paths = HashSet::new();

    for (file_path, file_modified, language) in walker {
      let file_task = FileTask::new(file_path.clone(), file_modified.into(), language);

      self.sender.send(file_task).await?;
      file_paths.insert(file_path.clone());
    }

    if let Some(cache) = &self.cache
      && self.should_clean_cache
    {
      cache.delete_stale_file_paths(&file_paths).await?;
    }

    ().ok()
  }
}