cymbal 0.1.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
65
66
67

use std::{fmt::Display, thread::JoinHandle};

use anyhow::Context;
use crossbeam::channel::Sender;

use crate::{ext::ResultExt, symbol::Symbol};

pub enum Message {
  Symbol(String),
  Stop,
}

#[derive(Clone)]
pub struct Writer {
  delimiter: char,
  separator: char,

  pub sink: Sender<Message>,
}

impl Writer {
  pub fn spawn(delimiter: char, separator: char, capacity: usize) -> Result<(Self, JoinHandle<()>), anyhow::Error> {
    let (send, recv) = crossbeam::channel::bounded(capacity);

    let handle = std::thread::spawn(move || {
      while let Ok(Message::Symbol(symbol)) = recv.recv() {
        print!("{symbol}")
      }
    });

    (
      Self {
        delimiter,
        separator,
        sink: send,
      },
      handle,
    )
      .ok()
  }

  pub fn send<Path, Text>(&self, symbol: Symbol<Path, Text>) -> Result<(), anyhow::Error>
  where
    Path: Display,
    Text: Display,
  {
    self
      .sink
      .send(Message::Symbol(format!(
        "{kind}{dlm}{path}{dlm}{line}{dlm}{col}{dlm}{lead}{dlm}{text}{dlm}{tail}{end}",
        kind = symbol.kind.colored_abbreviation(),
        path = symbol.path,
        line = symbol.span.start.line,
        col = symbol.span.start.column,
        lead = symbol.lead,
        text = symbol.text,
        tail = symbol.tail,
        dlm = self.delimiter,
        end = self.separator,
      )))
      .context("failed to send message")
  }

  pub fn stop(&self) -> Result<(), anyhow::Error> {
    self.sink.send(Message::Stop).context("failed to send stop message")
  }
}