From 78970967861f6e60c0631eb9b6731989e5ecfefb Mon Sep 17 00:00:00 2001
From: mattiasdrp <5543639+mattiasdrp@users.noreply.github.com>
Date: Fri, 26 Aug 2022 12:04:02 +0200
Subject: [PATCH] Add a keymap in the close overlay

This map allows to bind a keybind to the closed overlay to open it with
a key when inside it without having to bind a key to `ts-fold-open`
that would need to be global
---
 ts-fold.el | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/ts-fold.el b/ts-fold.el
index e1655dc..223af79 100644
--- a/ts-fold.el
+++ b/ts-fold.el
@@ -167,10 +167,22 @@ the fold in a cons cell.  See `ts-fold-range-python' for an example."
   (let ((tree-sitter-mode t))
     (ts-fold-open-all)))
 
+(defvar ts-fold-close-keymap-default
+  (let ((map (make-sparse-keymap)))
+    (define-key map (kbd "TAB") 'ts-fold-open)
+    map))
+
+(defvar ts-fold-close-keymap 'ts-fold-close-keymap-default
+  "Keymap used within ts-fold-close overlay.
+It should be set before the overlay is created.")
+
+(make-local-variable 'ts-fold-close-keymap)
+
 ;;;###autoload
 (define-minor-mode ts-fold-mode
   "Folding code using tree sitter."
   :init-value nil
+  :keymap ts-fold-close-keymap
   :lighter "TS-Fold"
   (if ts-fold-mode (ts-fold--enable) (ts-fold--disable)))
 
@@ -218,6 +230,7 @@ This function is borrowed from `tree-sitter-node-at-point'."
     (let* ((beg (car range)) (end (cdr range)) (ov (make-overlay beg end)))
       (overlay-put ov 'creator 'ts-fold)
       (overlay-put ov 'invisible 'ts-fold)
+      (overlay-put ov 'keymap ts-fold-close-keymap-default)
       (overlay-put ov 'display (or (and ts-fold-summary-show
                                         (ts-fold-summary--get (buffer-substring beg end)))
                                    ts-fold-replacement))