kak-tree-sitter 0.4.4

Server between Kakoune and tree-sitter
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98

use std::fs;

use colored::Colorize;
use kak_tree_sitter_config::Config;

use crate::{
  error::OhNo,
  highlighting::{BufferId, Highlighters},
  languages::Languages,
  request::Request,
  response::Response,
  server::SessionFifo,
  session::KakSession,
};

/// Type responsible for handling requests.
///
/// This type is stateful, as requests might have side-effect (i.e. tree-sitter parsing generates trees/highlighters
/// that can be reused, for instance).
pub struct Handler {
  /// Map a highlighter to a [`BufferId`].
  highlighters: Highlighters,

  /// Known languages.
  langs: Languages,
}

impl Handler {
  pub fn new(config: &Config) -> Result<Self, OhNo> {
    let highlighters = Highlighters::new();
    let langs = Languages::load_from_dir(config)?;

    Ok(Self {
      highlighters,
      langs,
    })
  }

  /// Handle the request and return an optional response to send back to Kakoune.
  pub fn handle_request(
    &mut self,
    session: &KakSession,
    session_fifo: &mut SessionFifo,
    req: &Request,
  ) -> Result<Response, OhNo> {
    match req {
      Request::TryEnableHighlight { lang, .. } => {
        log::info!("try enable highlight for language {lang}, session {session:?}");

        let supported = self.langs.get(lang).is_some();

        if !supported {
          log::warn!("{}", format!("language {lang} is not supported").red());
        }

        Ok(Response::FiletypeSupported { supported })
      }

      Request::Highlight {
        buffer,
        lang,
        timestamp,
        ..
      } => {
        log::debug!(
          "highlight for session {session:?}, buffer {buffer}, lang {lang}, timestamp {timestamp}"
        );

        let buffer_id = BufferId::new(&session.session_name, buffer);

        // read the buffer content from the buffer FIFO; this is the law
        log::debug!("waiting for buffer FIFO…");
        let payload = fs::read_to_string(session_fifo.buffer_fifo_path())?;
        log::debug!("buffer FIFO read");

        Ok(self.handle_highlight_req(buffer_id, lang, *timestamp, &payload)?)
      }
    }
  }

  fn handle_highlight_req(
    &mut self,
    buffer_id: BufferId,
    lang_name: &str,
    timestamp: u64,
    source: &str,
  ) -> Result<Response, OhNo> {
    if let Some(lang) = self.langs.get(lang_name) {
      self
        .highlighters
        .highlight(lang, &self.langs, buffer_id, timestamp, source)
    } else {
      Ok(Response::status(format!(
        "unsupported language: {lang_name}"
      )))
    }
  }
}