cymbal 0.3.1

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

mod args;
mod cache;
mod config;
mod ext;
mod parser;
mod symbol;
mod template;
mod text;
mod utils;
mod walker;
mod worker;
mod writer;

use std::sync::Arc;

use anyhow::Context;
use clap::Parser;
use parking_lot::RwLock;

use crate::{args::Args, ext::Leak, walker::Walker, worker::Worker, writer::Writer};

// TODO(enricozb):
// - add daemonization
// - investigate caching TSQuery: https://github.com/tree-sitter/tree-sitter/issues/1942
fn main() -> Result<(), anyhow::Error> {
  let args = Args::parse();

  let config = args.config()?.leak();
  let cache = Arc::new(RwLock::new(args.cache()?));
  let num_threads = args.num_threads();

  let walker = if let Some(file) = args.file() {
    Walker::single(file)?
  } else {
    Walker::spawn(config.extensions(), args.num_threads() * 8)?
  };

  let (writer, writer_handle) = Writer::spawn(args.delimiter, args.separator, args.num_threads() * 8)?;

  let mut worker_handles = Vec::new();
  for _ in 0..num_threads {
    let worker_handle = Worker::new(config, cache.clone(), walker.files.clone(), writer.clone()).spawn();

    worker_handles.push(worker_handle);
  }

  for worker_handle in worker_handles {
    worker_handle
      .join()
      .expect("join worker")
      .context("failed to join worker")?;
  }

  cache.read().save().context("failed to save cache")?;

  writer.stop()?;
  writer_handle.join().expect("join writer");

  Ok(())
}