ement.el/ement-room.el
2020-11-30 13:25:05 -06:00

254 lines
11 KiB
EmacsLisp

;;; ement-room.el --- Ement room buffers -*- lexical-binding: t; -*-
;; Copyright (C) 2020 Adam Porter
;; Author: Adam Porter <adam@alphapapa.net>
;; Keywords:
;; This program is free software; you can redistribute it and/or modify
;; it under the terms of the GNU General Public License as published by
;; the Free Software Foundation, either version 3 of the License, or
;; (at your option) any later version.
;; This program is distributed in the hope that it will be useful,
;; but WITHOUT ANY WARRANTY; without even the implied warranty of
;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
;; GNU General Public License for more details.
;; You should have received a copy of the GNU General Public License
;; along with this program. If not, see <https://www.gnu.org/licenses/>.
;;; Commentary:
;;
;;; Code:
;;;; Requirements
(require 'ewoc)
(require 'subr-x)
(require 'ement-macros)
(require 'ement-structs)
;;;; Variables
(defvar-local ement-room-ewoc nil
"EWOC for Ement room buffers.")
;;;; Customization
(defgroup ement-room nil
"Options for room buffers."
:group 'ement)
(defcustom ement-room-buffer-prefix "*Ement: "
"Prefix for Ement room buffer names."
:type 'string)
(defcustom ement-room-buffer-suffix "*"
"Suffix for Ement room buffer names."
:type 'string)
(defcustom ement-room-timestamp-format "%Y-%m-%d %H:%M:%S" ;; "%H:%M:%S"
"Format string for event timestamps.
See function `format-time-string'."
:type 'string)
(defface ement-room-timestamp
'((t (:inherit font-lock-variable-name-face)))
"Event timestamps.")
(defface ement-room-user
'((t (:inherit font-lock-keyword-face)))
"Usernames.")
;;;; Commands
;;;; Functions
(defun ement-room--buffer (room name)
"Return a buffer named NAME showing ROOM's events."
(or (get-buffer name)
(with-current-buffer (get-buffer-create name)
(ement-room-mode)
;; FIXME: Move visual-line-mode to a hook.
(visual-line-mode 1)
(setf ement-room room)
;; Move new events to main list.
(setf (ement-room-timeline room) (append (ement-room-timeline* room) (ement-room-timeline room))
(ement-room-timeline* room) nil)
(ement-room--insert-events room)
(current-buffer)))) ; Return the buffer!
(define-derived-mode ement-room-mode fundamental-mode "Ement Room"
"Major mode for Ement room buffers.
This mode initializes a buffer to be used for showing events in
an Ement room. It kills all local variables, removes overlays,
and erases the buffer."
(kill-all-local-variables)
(let ((inhibit-read-only t))
(erase-buffer))
(remove-overlays)
(setf buffer-read-only t
ement-room-ewoc (ewoc-create #'ement-room--pp-event "HEADER" "FOOTER")))
;;;;; EWOC
(defvar-local ement-room nil
"The room displayed in the current buffer.")
(defun ement-room--insert-events (room)
"Insert events for ROOM into current buffer."
(mapc #'ement-room--insert-event (ement-room-timeline room))
;; Move new events to timeline slot. FIXME: This belongs elsewhere.
(setf (ement-room-timeline room) (append (ement-room-timeline* room) (ement-room-timeline room))
(ement-room-timeline* room) nil)
)
(defun ement-room--pp-event (struct)
"Pretty-print STRUCT.
To be used as the pretty-printer for `ewoc-create'."
(cl-etypecase struct
;; FIXME: Null probably not needed anymore.
;; (null (insert ""))
(ement-event (insert " " (ement-room--format-event struct)))
(ement-user (insert (ement-room--format-user struct)))
;; FIXME: Function probably not needed anymore.
;; (function (insert ""))
))
(defface ement-room-membership
'((t (:inherit font-lock-comment-face)))
"Membership events (join/part).")
(defun ement-room--format-event (event)
"Format `ement-event' EVENT."
(pcase-let* (((cl-struct ement-event type content origin-server-ts) event)
((map body) content)
(ts (/ origin-server-ts 1000)) ; Matrix timestamps are in milliseconds.
(timestamp
(propertize (format "[%s] " (format-time-string ement-room-timestamp-format ts))
'face 'ement-room-timestamp))
(body-face (pcase type
("m.room.member" 'ement-room-membership)
(_ 'default)))
(string (propertize (pcase type
("m.room.message" body)
("m.room.member" (alist-get 'membership content)))
'face body-face)))
(concat timestamp string)))
(defun ement-room--format-user (user)
"Format `ement-user' USER for current buffer's room."
(propertize (or (gethash ement-room (ement-user-room-display-names user))
(puthash ement-room (ement-room--user-display-name user ement-room)
(ement-user-room-display-names user)))
'face 'ement-room-user))
(defun ement-room--user-display-name (user room)
"Return the displayname for USER in ROOM."
;; SPEC: <https://matrix.org/docs/spec/client_server/r0.6.1#calculating-the-display-name-for-a-user>.
(if-let ((member-state-event (cl-loop for event in (ement-room-state room)
when (and (equal "m.room.member" (ement-event-type event))
(equal user (ement-event-sender event)))
return event)))
(or (alist-get 'displayname (ement-event-content member-state-event))
;; FIXME: Add step 3 of the spec. For now we skip to step 4.
;; No displayname given: use raw user ID.
(ement-user-id user))
;; No membership state event: use pre-calculated displayname or ID.
(or (ement-user-displayname user)
(ement-user-id user))))
(defun ement-room--insert-event (event)
"Insert EVENT into current buffer."
(let* ((ewoc ement-room-ewoc)
(event< (lambda (a b)
"Return non-nil if event A's timestamp is before B's."
(< (ement-event-origin-server-ts a)
(ement-event-origin-server-ts b))))
(node-before (ement-room--ewoc-node-before ewoc event event< :pred #'ement-event-p))
new-node)
(setf new-node (if (not node-before)
(progn
(debug-warn "No event before it: add first.")
(if-let ((first-node (ewoc-nth ewoc 0)))
(progn
(debug-warn "EWOC not empty.")
(if (and (ement-user-p (ewoc-data first-node))
(equal (ement-event-sender event)
(ewoc-data first-node)))
(progn
(debug-warn "First node is header for this sender: insert after it, instead.")
(setf node-before first-node)
(ewoc-enter-after ewoc first-node event))
(debug-warn "First node is not header for this sender: insert first.")
(ewoc-enter-first ewoc event)))
(debug-warn "EWOC empty: add first.")
(ewoc-enter-first ewoc event)))
(debug-warn "Found event before new event: insert after it.")
(when-let ((next-node (ewoc-next ewoc node-before)))
(when (and (ement-user-p (ewoc-data next-node))
(equal (ement-event-sender event)
(ewoc-data next-node)))
(debug-warn "Next node is header for this sender: insert after it, instead.")
(setf node-before next-node)))
(ewoc-enter-after ewoc node-before event)))
;; Insert sender where necessary.
(if (not node-before)
(progn
(debug-warn "No event before: Add sender before new node.")
(ewoc-enter-before ewoc new-node (ement-event-sender event)))
(debug-warn "Event before: compare sender.")
(if (equal (ement-event-sender event)
(cl-typecase (ewoc-data node-before)
(ement-event (ement-event-sender (ewoc-data node-before)))
(ement-user (ewoc-data node-before))))
(debug-warn "Same sender.")
(debug-warn "Different sender: insert new sender node.")
(ewoc-enter-before ewoc new-node (ement-event-sender event))
(when-let* ((next-node (ewoc-next ewoc new-node)))
(when (ement-event-p (ewoc-data next-node))
(debug-warn "Event after from different sender: insert its sender before it.")
(ewoc-enter-before ewoc next-node (ement-event-sender (ewoc-data next-node)))))))))
(cl-defun ement-room--ewoc-node-before (ewoc data <-fn
&key (from 'last) (pred #'identity))
"Return node in EWOC that matches PRED and belongs before DATA according to COMPARATOR."
(cl-assert (member from '(first last)))
(if (null (ewoc-nth ewoc 0))
(debug-warn "EWOC is empty: returning nil.")
(debug-warn "EWOC has data: add at appropriate place.")
(cl-labels ((next-matching
(ewoc node next-fn pred) (cl-loop do (setf node (funcall next-fn ewoc node))
until (or (null node)
(funcall pred (ewoc-data node)))
finally return node)))
(let* ((next-fn (pcase from ('first #'ewoc-next) ('last #'ewoc-prev)))
(start-node (ewoc-nth ewoc (pcase from ('first 0) ('last -1)))))
(unless (funcall pred (ewoc-data start-node))
(setf start-node (next-matching ewoc start-node next-fn pred)))
(if (funcall <-fn (ewoc-data start-node) data)
(progn
(debug-warn "New data goes before start node.")
start-node)
(debug-warn "New data goes after start node: find node before new data.")
(let ((compare-node start-node))
(cl-loop while (setf compare-node (next-matching ewoc compare-node next-fn pred))
until (funcall <-fn (ewoc-data compare-node) data)
finally return (if compare-node
(progn
(debug-warn "Found place: enter there.")
compare-node)
(debug-warn "Reached end of collection: insert there.")
(pcase from
('first (ewoc-nth ewoc -1))
('last nil))))))))))
;;;; Footer
(provide 'ement-room)
;;; ement-room.el ends here