From fd1eaafff57fb40dd0e504c3d5fd5e8abe14b5f5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Bla=C5=BE=20Hrastnik?= <blaz@mxxn.io>
Date: Fri, 13 Aug 2021 13:15:36 +0900
Subject: [PATCH] Add :tree-sitter-scopes, useful when developing indents.toml

---
 helix-core/src/indent.rs   | 21 +++++++++++++++++++++
 helix-term/src/commands.rs | 21 +++++++++++++++++++++
 2 files changed, 42 insertions(+)

diff --git a/helix-core/src/indent.rs b/helix-core/src/indent.rs
index d272dd68..9489baa7 100644
--- a/helix-core/src/indent.rs
+++ b/helix-core/src/indent.rs
@@ -313,6 +313,27 @@ pub fn suggested_indent_for_pos(
     }
 }
 
+pub fn get_scopes(syntax: Option<&Syntax>, text: RopeSlice, pos: usize) -> Vec<&'static str> {
+    let mut scopes = Vec::new();
+    if let Some(syntax) = syntax {
+        let byte_start = text.char_to_byte(pos);
+        let node = match get_highest_syntax_node_at_bytepos(syntax, byte_start) {
+            Some(node) => node,
+            None => return scopes,
+        };
+
+        scopes.push(node.kind());
+
+        while let Some(parent) = node.parent() {
+            scopes.push(parent.kind())
+        }
+    }
+
+    scopes.reverse();
+
+    return scopes;
+}
+
 #[cfg(test)]
 mod test {
     use super::*;
diff --git a/helix-term/src/commands.rs b/helix-term/src/commands.rs
index 0d118dbc..b5671723 100644
--- a/helix-term/src/commands.rs
+++ b/helix-term/src/commands.rs
@@ -1883,6 +1883,20 @@ mod cmd {
         doc.reload(view.id)
     }
 
+    fn tree_sitter_scopes(
+        cx: &mut compositor::Context,
+        _args: &[&str],
+        _event: PromptEvent,
+    ) -> anyhow::Result<()> {
+        let (view, doc) = current!(cx.editor);
+        let text = doc.text().slice(..);
+
+        let pos = doc.selection(view.id).primary().cursor(text);
+        let scopes = indent::get_scopes(doc.syntax(), text, pos);
+        cx.editor.set_status(format!("scopes: {:?}", &scopes));
+        Ok(())
+    }
+
     pub const TYPABLE_COMMAND_LIST: &[TypableCommand] = &[
         TypableCommand {
             name: "quit",
@@ -2114,6 +2128,13 @@ mod cmd {
             doc: "Discard changes and reload from the source file.",
             fun: reload,
             completer: None,
+        },
+        TypableCommand {
+            name: "tree-sitter-scopes",
+            alias: None,
+            doc: "Display tree sitter scopes, primarily for theming and development.",
+            fun: tree_sitter_scopes,
+            completer: None,
         }
     ];