diff options
Diffstat (limited to 'org-roam-ui.el')
| -rw-r--r-- | org-roam-ui.el | 65 |
1 files changed, 44 insertions, 21 deletions
diff --git a/org-roam-ui.el b/org-roam-ui.el index 8a3a5b3..877a4d6 100644 --- a/org-roam-ui.el +++ b/org-roam-ui.el @@ -86,6 +86,8 @@ E.g. '((bg . '#1E2029') :type 'list) (defvar org-roam-ui--ws-current-node nil) +(defvar oru-ws nil + "The websocket for org-roam-ui.") ;;;###autoload (define-minor-mode @@ -105,22 +107,24 @@ This serves the web-build and API over HTTP." (websocket-server 35903 :host 'local - :on-open (lambda (ws) (progn (setq oru-ws ws) (org-roam-ui--send-graphdata) (org-roam-ui-sync-theme--advice) (message "Connection established with org-roam-ui"))) - :on-close (lambda (_websocket) (setq oru-ws nil) (message "Connection with org-roam-ui closed succesfully.")))) + :on-open (lambda (ws) (progn (setq oru-ws + ws) (org-roam-ui--send-graphdata) (message "Connection established with org-roam-ui") + (add-hook 'post-command-hook #'org-roam-ui--update-current-node))) + :on-close (lambda (_websocket) (setq oru-ws + nil) (message "Connection with org-roam-ui closed succesfully.")))) (if (boundp 'counsel-load-theme) (advice-add 'counsel-load-theme :after #'org-roam-ui-sync-theme--advice) - (advice-add 'load-theme :after #'org-roam-ui-sync-theme-manually)) - (add-hook 'post-command-hook #'org-roam-ui--update-current-node) - (add-hook 'post-command-hook #'org-roam-ui-update)) + (advice-add 'load-theme :around #'org-roam-ui-sync-theme--advice)) + (add-hook 'post-command-hook #'org-roam-ui--update-current-node)) + ;(add-hook 'post-command-hook #'org-roam-ui-update)) (t (progn + (websocket-server-close org-roam-ui-ws) (remove-hook 'post-command-hook #'org-roam-ui-update) (remove-hook 'post-command-hook #'org-roam-ui--update-current-node) (if (boundp 'counsel-load-theme) (advice-remove 'counsel-load-theme #'org-roam-ui-sync-theme--advice) (advice-remove 'load-theme #'org-roam-ui-sync-theme--advice)) - (websocket-server-close org-roam-ui-ws) - (delete-process org-roam-ui-ws) (httpd-stop))))) @@ -135,11 +139,13 @@ This serves the web-build and API over HTTP." (websocket-send-text oru-ws (json-encode `((type . "graphdata") (data . ,response)))))) (defun org-roam-ui--update-current-node () + (when (websocket-openp oru-ws) (let* ((node (org-roam-id-at-point))) (unless (string-match-p (regexp-quote "Minibuf") (buffer-name (current-buffer))) (unless (string= org-roam-ui--ws-current-node node) (setq org-roam-ui--ws-current-node node) - (websocket-send-text oru-ws (json-encode `((type . "command") (data . ((commandName . "follow") (id . ,node)))))))))) + (websocket-send-text oru-ws (json-encode `((type . "command") (data +. ((commandName . "follow") (id . ,node))))))))))) (defun org-roam-ui-show-node () "Open the current org-roam node in org-roam-ui." @@ -148,6 +154,7 @@ This serves the web-build and API over HTTP." (defun org-roam-ui-sync-theme--advice () "Function which is called after load-theme to sync your current theme with org-roam-ui." + (message "Syncing theme") (websocket-send-text oru-ws (json-encode `((type . "theme") (data . ,(org-roam-ui--update-theme)))))) (defun org-roam-ui-sync-theme-manually () @@ -245,19 +252,35 @@ This function is added to `post-command-hook'." )) -;; (defservlet* theme text/stream () -;; (progn) -;; (if org-roam-ui-sync-theme -;; (if (boundp 'doom-themes--colors) -;; (let* -;; ((colors (butlast doom-themes--colors (- (length doom-themes--colors) 25))) ui-theme (list nil)) -;; (progn -;; (dolist (color colors) (push (cons (car color) (car (cdr color))) ui-theme)) -;; ui-theme)) -;; (insert (format "data: %s\n\n" (json-encode (org-roam-ui-get-theme))))) -;; (when org-roam-ui-custom-theme -;; (insert (format "data %s\n\n" (json-encode org-roam-ui-custom-theme))))) -;; (httpd-send-header t "text/event-stream" 200 :Access-Control-Allow-Origin "*")) +;;;; commands +(defun orui-zoom-to-node (&optional id speed padding) + "Move the view of the graph to the node at points, or optionally a node of your choosing. +Optionally takes three arguments: +The id of the node you want to travel to. +The time in ms it takes to make the transition. +The padding around the nodes in the viewport." + (interactive) + (if-let ((node (or id (org-roam-id-at-point)))) + (websocket-send-text oru-ws (json-encode `((type . "command") (data . + ((commandName . "zoom") (id . ,node) (speed . ,speed) (padding . ,padding))))))) + (message "No node found.")) + +(defun orui-toggle-following () + "Set whether ORUI should follow your every move in emacs. Default yes." + (interactive) + (if (member #'org-roam-ui--update-current-node post-command-hook) + (progn + (remove-hook 'post-command-hook #'org-roam-ui--update-current-node) + (message "Org-Roam-UI will now leave you alone.") + (add-hook 'post-command-hook #'org-roam-ui--update-current-node) + (message "Org-Roam-UI will now follow you around.")) + )) + +(defun orui-toggle-local-zoom () + "Toggles whether org-roam-ui should go to the local view of a given node or zoom to it. +Defaults to local." + (interactive) + (org-roam-ui--send-command "toggle" `(id . yes))) (provide 'org-roam-ui) ;;; org-roam-ui.el ends here |