(defpackage :alimenta.format
(:use :cl :alexandria :serapeum :fw.lu)
(:export format-document format-title format-link format-paragraph
#:indent-feed))
(in-package :alimenta.format)
(defclass document-formatter ()
())
;;; Formatting protocol
;; Generic entrypoint
(defgeneric format-document (formatter stream document)
(:documentation "Format a document with the given formatter to the given stream"))
;; Semantic formatters
(defgeneric format-title (formatter title)
(:documentation "Format a title according to FORMATTER"))
(defgeneric format-link (formatter link)
(:documentation "Format a link according to FORMATTER"))
(defgeneric format-paragraph (formatter paragraph)
(:documentation "Format a paragraph according to FORMATTER"))
;;; Make alimenta's classes formattable
(defmethod format-document (formatter stream (document alimenta::feed-entity))
(format stream "~&~v,4@t~a~%~v,4@t~a~%"
(level formatter) (format-title formatter (alimenta:title document))
(level formatter) (format-link formatter (alimenta:link document))))
(defmethod format-document (formatter stream (document alimenta:item))
(call-next-method)
(let ((paragraphs (remove-if (op (every #'whitespacep _))
(lquery:$ (initialize (alimenta:content document))
(children)
(text)))))
(format stream "~&~{~a~%~}~2&"
(map 'list (op (format-paragraph formatter _))
paragraphs))))
;;; Define some output formats
;;;; Indentation-based
(defclass indent-formatter (document-formatter)
((%level :initarg :level :accessor level :initform 0)))
(defmethod format-title ((formatter indent-formatter) (title string))
(format nil "~v,1,0,'*a Title: ~a" (1+ (level formatter)) "" title))
(defmethod format-link ((formatter indent-formatter) (link string))
(format nil "~vt Link: ~a" (1+ (level formatter)) link))
(defun pp-fill (stream string &optional (colon? t) atsign?)
(declare (ignore atsign?))
(pprint-logical-block (stream (tokens string)
:prefix (if colon? "(" "")
:suffix (if colon? ")" ""))
(pprint-exit-if-list-exhausted)
(loop
(princ (pprint-pop) stream)
(pprint-exit-if-list-exhausted)
(write-char #\space stream)
(pprint-newline :fill stream))))
(defmethod format-paragraph ((formatter indent-formatter) (paragraph string))
(format nil "~&~v,4@t ~/alimenta.format::pp-fill/~%" (level formatter) paragraph))
(defmethod format-paragraph ((formatter indent-formatter) (paragraph list))
(format nil "~&~{ ~a~%~}" paragraph))
(defmethod format-document ((formatter indent-formatter) stream (document alimenta:feed))
(call-next-method)
(incf (level formatter))
(for:for ((item over document))
(format-document formatter stream item)))
;;;; HTML
(defclass html-formatter (document-formatter)
((%level :initarg :level :accessor level :initform 0)))
(defmethod format-title ((formatter html-formatter) (title string))
(format nil "~a~2:*" (1+ (level formatter)) title))
(defmethod format-link ((formatter html-formatter) (link string))
(format nil "~:*~a" link))
(defmethod format-paragraph ((formatter html-formatter) (paragraph string))
(format nil "~&~a
" paragraph))
(defmethod format-paragraph ((formatter html-formatter) (paragraph list))
(format nil "~%~{~a
~}" paragraph))
(defmethod format-document ((formatter html-formatter) stream (document alimenta:feed))
(let ((ostream (or stream (make-string-output-stream))))
(unwind-protect
(progn (format ostream "~&~%")
(call-next-method formatter ostream document)
(incf (level formatter))
(for:for ((item over document))
(format ostream "~&~%")
(format-document formatter ostream item)
(format ostream "~&~%"))
(format ostream "~&~%")
(finish-output ostream)
(get-output-stream-string ostream))
(unless stream
(close ostream)))))
;;;; Org
(defclass org-formatter (document-formatter)
((%level :initarg :level :accessor level :initform 0)))
(defmethod format-document ((formatter org-formatter) stream (document alimenta::feed-entity))
(format stream "~&~a~%~a~2%"
(format-title formatter (alimenta:title document))
(format-link formatter (alimenta:link document))))
(defmethod format-document ((formatter org-formatter) stream (document alimenta:item))
(call-next-method)
(let ((paragraphs (remove-if (op (every #'whitespacep _))
(lquery:$ (initialize (alimenta:content document))
(children)
(text)))))
(format stream "~&~{~a~%~}~2&"
(map 'list (op (format-paragraph formatter _))
paragraphs))))
(defmethod format-title ((formatter org-formatter) (title string))
(with-output-to-string (s)
(loop repeat (1+ (level formatter))
do (princ #\* s))
(princ #\space s)
(princ (serapeum:trim-whitespace title) s)))
(defmethod format-link ((formatter org-formatter) (link string))
(format nil "~vt[[~a]]" (+ 2 (level formatter)) link))
(defmethod format-paragraph ((formatter org-formatter) (paragraph string))
(format nil "~&~v,1@t~/alimenta.format::pp-fill/~%" (+ 2 (level formatter)) paragraph))
(defmethod format-paragraph ((formatter org-formatter) (paragraph list))
(format nil "~&~{ ~a~%~}" paragraph))
(defmethod format-document ((formatter org-formatter) stream (document alimenta:feed))
(call-next-method)
(incf (level formatter))
(for:for ((item over document))
(format-document formatter stream item)))
(defun indent-feed (feed &optional (stream *standard-output*))
(format-document (make-instance 'alimenta.format::indent-formatter)
stream
feed))