restclient-jq.el

;;; restclient-jq.el --- support for setting resetclient vars from jq expressions
;;
;; Public domain.

;; Author: Cameron Dorrat <cdorrat@gmail.com>
;; Maintainer: Cameron Dorrat <cdorrat@gmail.com>
;; Created: 26 Apr 2020
;; Keywords: http jq
;; Package-Requires: ((restclient "0") (jq-mode "0.4.1"))

;; This file is not part of GNU Emacs.
;; This file is public domain software. Do what you want.

;;; Commentary:
;;
;; This is a companion to restclient.el to add support for setting variables from results using jq expressions

;;; Code:
;;
(require 'jq-mode)


;; --- jq support
(defun restclient-result-end-point ()
  (save-excursion
    (goto-char (point-max))
    (or (and (re-search-backward "^[^/].*" nil t)
	     (line-end-position))
	(point-max))))

(defun restclient-jq-get-var (jq-pattern)
  (with-temp-buffer
    (let ((output (current-buffer)))
      (with-current-buffer restclient-same-buffer-response-name
        (call-process-region
         (point-min)
         (restclient-result-end-point)
         shell-file-name
         nil
         output
         nil
         shell-command-switch
         (format "%s %s %s"
                 jq-interactive-command
		 "-r"
                 (shell-quote-argument jq-pattern))))      
      (string-trim (buffer-string)))))

(defun restclient-json-var-function (args args-offset)
  (save-match-data
   (and (string-match "\\(:[^: \n]+\\) \\(.*\\)$" args)
	(lexical-let ((var-name (match-string 1 args))
		      (jq-patt (match-string 2 args)))
	  (lambda ()
	    (let ((resp-val (restclient-jq-get-var jq-patt)))
	      (restclient-remove-var var-name)
	      (restclient-set-var var-name resp-val)
	      (message "restclient var [%s = \"%s\"] " var-name resp-val)))))))

(defun restclient-jq-interactive-result ()
  (interactive)
  (flush-lines "^//.*") ;; jq doesnt like comments
  (jq-interactively (point-min) (restclient-result-end-point)))


(provide 'restclient-jq)

(eval-after-load 'restclient
  '(progn
     (resetclient-register-result-func
      "jq-set-var" #'restclient-json-var-function
      "Set a resetclient variable with the value jq expression, 
       takes var & jq expression as args. 
       eg. -> jq-set-var :my-token .token")
     (define-key restclient-response-mode-map  (kbd "C-c C-j") #'restclient-jq-interactive-result)))