Add support for various treesitter based modes

This commit adds support for c-ts-mode, python-ts-mode and
rust-ts-mode.

These modes are builtin modes included in GNU Emacs since 29.1

Author: pksadiq

electric-operator.el

@@ -588,6 +588,7 @@ Any better ideas would be welcomed."
 
                                       )
 
+(apply #'electric-operator-add-rules-for-mode 'c-ts-mode (electric-operator-get-rules-for-mode 'c-mode))
 
 ;; Use the same rules for c++
 (apply #'electric-operator-add-rules-for-mode 'c++-mode (electric-operator-get-rules-for-mode 'c-mode))
@@ -650,6 +651,20 @@ could be added here.")
    ;; Check for any user-defined types
    (electric-operator-looking-back-locally electric-operator-c-user-types-regex)))
 
+(defun electric-operator-c-ts-is-function-parameter? ()
+  "Try to guess if we are in function definition/declaration
+
+Using `c-ts-mode''s treesitter analysis."
+  (require 'treesit nil t)
+  (let ((node (treesit-node-at (point)))
+        (value nil))
+    (when (and (treesit-node-top-level node "^parameter_list$")
+               (treesit-node-top-level node "^function_declarator$"))
+      (setq node (treesit-node-top-level node "^parameter_list$"))
+      (setq value (list (treesit-node-start node) (treesit-node-end node)))
+      (and (> (point) (nth 0 value))
+           (<= (point) (nth 1 value))))))
+
 (defvar electric-operator-c-function-definition-syntax-list
   '(topmost-intro
     topmost-intro-cont
@@ -666,9 +681,11 @@ Using `cc-mode''s syntactic analysis."
   ;; There are similar but different symbols for objective-C, but I'm not
   ;; going to try to support that now.
 
-  (--> (c-guess-basic-syntax)
-    (-map #'car it)
-    (-intersection electric-operator-c-function-definition-syntax-list it)))
+  (if (derived-mode-p 'c-ts-mode 'c++-ts-mode)
+      (electric-operator-c-ts-is-function-parameter?)
+    (--> (c-guess-basic-syntax)
+         (-map #'car it)
+         (-intersection electric-operator-c-function-definition-syntax-list it))))
 
 (defun electric-operator-c-mode-include-line-opening-quote? ()
   (electric-operator-looking-back-locally "#\\s-*include\\s-*"))
@@ -859,6 +876,7 @@ Also handles C++ lambda capture by reference."
 				                      )
 
 (apply #'electric-operator-add-rules-for-mode 'inferior-python-mode (electric-operator-get-rules-for-mode 'python-mode))
+(apply #'electric-operator-add-rules-for-mode 'python-ts-mode (electric-operator-get-rules-for-mode 'python-mode))
 
 (defun electric-operator-python-mode-in-lambda-args? ()
   "Are we inside the arguments statement of a lambda?"
@@ -988,6 +1006,9 @@ Also handles C++ lambda capture by reference."
 				                      ;; pointer deref vs multiplication
 				                      (cons "*" nil)
 
+				                      (cons ":" ": ")
+				                      (cons "::" "::")
+
 				                      (cons "/" #'electric-operator-prog-mode-/)
 				                      (cons "/*" " /* ")
 				                      (cons "//" " // ")
@@ -1001,6 +1022,30 @@ Also handles C++ lambda capture by reference."
 				                      ;; Bar is used for lambdas as well as or
 				                      (cons "|" nil))
 
+(apply #'electric-operator-add-rules-for-mode 'rust-ts-mode (electric-operator-get-rules-for-mode 'rust-mode))
+
+(defun electric-operator-rust-mode-: ()
+  "Handle ':'"
+  (cond
+   ((electric-operator-python-mode-in-lambda-args?) ": ")
+
+   ;; A keyword statement (we can't use \\b here because it matches _)
+   ((electric-operator-looking-back-locally "\\(\\s-\\|^\\)\\(if\\|elif\\|else\\|for\\|while\\|class\\|def\\|try\\|except\\|with\\)") ":")
+
+   ;; type definition inside a function
+   ((and (eq (electric-operator-enclosing-paren) ?\() (electric-operator-looking-back-locally "def .*")) ": ")
+
+   ;; type definition on a variable declaration
+   ((electric-operator-looking-back-locally "^\\s-*[a-zA-Z0-9_.]+") ": ")
+
+   ;; A dictionary
+   ((eq (electric-operator-enclosing-paren) ?\{) ": ")
+
+
+   ;; Unsure, but probably a multiline declaration of some sort that we can't
+   ;; understand, leave it alone.
+   (t nil)))
+
 
 
 ;; R tweaks (ess mode)