cymbal 0.8.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

mod args;
mod cache;
mod channel;
mod color;
mod config;
mod ext;
mod parser;
mod symbol;
mod template;
mod utils;
mod walker;
mod worker;

use anyhow::Result;
use clap::Parser;
use tokio::task::JoinSet;

use crate::args::Args;
use crate::ext::{IntoExt, IteratorExt, Leak};
use crate::walker::Walker;
use crate::worker::Worker;

#[tokio::main]
async fn main() -> Result<()> {
  let args = Args::parse();
  let available_concurrency = args.concurrency()?.convert::<usize>();
  let cache = args.cache().await?;
  let config = args.config().await?.leak();
  let (sender, receiver) = args.channel();
  let walker = Walker::new(args.search_path().to_path_buf(), sender, cache.clone()).spawn();
  let delimiter = args.delimiter();
  let separator = args.separator();

  // TODO:
  // - tree sitter languages and queries must be shared

  let mut workers = JoinSet::new();
  for _ in 0..available_concurrency {
    workers.spawn(Worker::new(cache.clone(), config, receiver.clone(), delimiter, separator).run());
  }
  workers.join_all().await.ok_all()?;

  walker.await?
}