emacs: Add an ‘mm-humanwave’ module

2 files changed, 81 insertions, 0 deletions

diff --git a/.config/emacs/init.el b/.config/emacs/init.el
index 97525f1..eb0f23e 100644
--- a/.config/emacs/init.el
+++ b/.config/emacs/init.el
@@ -317,6 +317,8 @@ the buffer without saving it."
   (require 'mm-darwin))                 ; MacOS
 (when mm-lsp-p
   (require 'mm-lsp))                    ; Language Server Protocol
+(when mm-humanwave-p
+  (require 'mm-humanwave))              ; Humanwave Extras
 
 ;; TODO: Put this somewhere reasonable
 
diff --git a/.config/emacs/modules/mm-humanwave.el b/.config/emacs/modules/mm-humanwave.el
new file mode 100644
index 0000000..178f7a8
--- /dev/null
+++ b/.config/emacs/modules/mm-humanwave.el
@@ -0,0 +1,79 @@
+;;; mm-humanwave.el --- Humanwave extras  -*- lexical-binding: t; -*-
+
+
+;;; Query the Backend
+
+(defvar mm--humanwave-query-history nil
+  "History for endpoints given to `mm-humanwave-query'.")
+
+(defun mm-humanwave-query (endpoint &optional method)
+  "Query and display the result of an HTTP reqiest on ENDPOINT.
+If METHOD is nil, a GET request is performed."
+  (interactive
+   (let* ((query (read-string (format-prompt "Query" nil)
+                              (car-safe mm--humanwave-query-history)
+                              'mm--humanwave-query-history))
+          (parts (string-split (string-trim query) " " :omit-nulls)))
+     (when (length> parts 2)
+       (user-error "Queries must be of the form `METHOD ENDPOINT' or `ENDPOINT'."))
+     (nreverse parts)))
+  (let* ((project-root (project-root (project-current :maybe-prompt)))
+         (qry-path (expand-file-name "qry" project-root))
+         extras)
+    (unless (file-executable-p qry-path)
+      (user-error "No `qry' executable found in the project root"))
+    (let ((output-buffer (get-buffer-create "*Query Response*")))
+      (with-current-buffer output-buffer
+        (delete-region (point-min) (point-max))
+        (call-process qry-path nil t nil
+                      (string-trim endpoint) "-X" (or method "GET"))
+        (unless (eq major-mode 'json-ts-mode)
+          (json-ts-mode))
+        (goto-char (point-min)))
+      (display-buffer output-buffer))))
+
+(keymap-set project-prefix-map "q" #'mm-humanwave-query)
+
+
+;;; IMenu Support for Handlers
+
+(defvar mm-humanwave-handler--regexp
+  (rx bol
+      (* blank)
+      (or "if" "elif")
+      (* blank)
+      (or "topic" "schedule")
+      (* blank)
+      "=="
+      (* blank)
+      (or (seq ?\' (* (not ?\')) ?\')
+          (seq ?\" (* (not ?\")) ?\"))
+      (* blank)
+      ?:
+      (* blank)
+      eol))
+
+(defun mm-humanwave-handler-topic-imenu-index ()
+  (let ((tree-index (when (fboundp 'treesit-simple-imenu--generic-function)
+                      (treesit-simple-imenu--generic-function
+                       treesit-simple-imenu-settings)))
+        (topic-index '()))
+    (save-excursion
+      (goto-char (point-min))
+      (while (re-search-forward mm-humanwave-handler--regexp nil :noerror)
+        (let ((label (match-string 0))
+              (pos (match-beginning 0)))
+          (push (cons (format "Topic: %s" (string-trim label)) pos) topic-index))))
+    (append (nreverse topic-index) tree-index)))
+
+(defun mm-humanwave-handler-topic-imenu-setup ()
+  "Setup custom imenu index for `python-ts-mode'."
+  (when (and (string-match-p "handlers?" (or (buffer-file-name) ""))
+             (derived-mode-p #'python-ts-mode))
+    (setq-local imenu-create-index-function
+                #'mm-humanwave-handler-topic-imenu-index)))
+
+(add-hook 'after-change-major-mode-hook
+          #'mm-humanwave-handler-topic-imenu-setup)
+
+(provide 'mm-humanwave)