signel fixes

1 files changed, 8 insertions, 5 deletions

diff --git a/net/signel.org b/net/signel.org
index 924ecf6..89ba415 100644
--- a/net/signel.org
+++ b/net/signel.org
@@ -371,6 +371,11 @@ buffers:
 #+begin_src emacs-lisp
 (defvar signel-report-deliveries t)
 
+(defun signel--prompt-and-notify ()
+  (insert (signel--prompt))
+  (when (fboundp 'tracking-add-buffer)
+    (tracking-add-buffer (current-buffer) '(signel-notification-face))))
+
 (defun signel--update-chat-buffer (source data stamp rec-stamp msg)
   (when-let ((b (signel--contact-buffer source)))
     (with-current-buffer b
@@ -385,7 +390,7 @@ buffers:
                     data
                     "\n")
             (fill-region p (point))
-            (insert (signel--prompt)))
+            (signel--prompt-and-notify))
         (let ((is-delivery (or (signel--msg-is-delivery msg)
                                (signel--msg-is-receipt msg)))
               (is-read (signel--msg-is-read msg)))
@@ -395,10 +400,8 @@ buffers:
             (insert (signel--timestamp "*" (or rec-stamp stamp) "* ")
                     (signel--notice (if is-read "(read)" "(delivered)"))
                     "\n")
-            (insert (signel--prompt)))))
-      (end-of-line))
-    (when (fboundp 'tracking-add-buffer)
-      (tracking-add-buffer b '(signel-notification-face)))))
+            (signel--prompt-and-notify))))
+      (end-of-line))))
 #+end_src
 
 There are some rough edges in the above implementation that must be