feat(prog): use treesitter-context-fold for code folding

modules/me-prog.el

@@ -417,6 +417,32 @@ Fall back to the default `citre--project-root'."
   (setq treesitter-context-background-color "#EEEEEE"
         treesitter-context-border-color "#000000"))
 
+(use-package treesitter-context-fold
+  :hook (prog-mode . treesitter-context-fold-mode)
+  :when (+emacs-features-p 'tree-sitter)
+  :config
+  (require 'treesitter-context)
+  (with-eval-after-load 'evil
+    (add-to-list 'evil-fold-list `((treesitter-context-fold-mode)
+                                   :delete     nil
+                                   :open-all   nil
+                                   :close-all  nil
+                                   :toggle     treesitter-context-fold-toggle
+                                   :open       treesitter-context-fold-show
+                                   :open-rec   nil
+                                   :close      treesitter-context-fold-hide)))
+
+  ;; Redefine the major minor mode
+  (define-minor-mode treesitter-context-fold-mode
+    "Fold code according to treesit syntax tree."
+    :global nil
+    :init-value nil
+    :keymap treesitter-context-fold-mode-map
+    (if treesitter-context-fold-mode
+        (unless (and (treesit-available-p) (member major-mode treesitter-context--supported-mode))
+          (message "Treesitter context fold mode cannot be enabled.")
+          (setq treesitter-context-fold-mode nil)))))
+
 (use-package breadcrumb
   :straight t)