blob: 5ade71441a8b4e1624e2c2f0418f1975b9eed4dc (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
;;; Commentary:
;;; Small utility functions which helps the rest of the text processing.
;;; Code:
(define-module (text util)
:use-module ((hnh util) :select (intersperse))
:export (words unwords lines unlines
true-string-length
true-string-pad
trim-to-width
add-enumeration-punctuation))
(define (words str) (string-split str #\space))
(define (unwords list) (string-join list " " 'infix))
(define (lines str) (string-split str #\newline))
(define (unlines list) (string-join list "\n" 'infix))
;; Alternative string-length whith counts ANSI escapes as 0-length.
;; NOTE Some way to opt in and out of different features would be nice.
(define (true-string-length word)
(let loop ((chars (string->list word)))
(if (null? chars)
0
(let ((char (car chars)))
(if (eqv? #\escape char)
(loop (cdr (memv #\m chars)))
(1+ (loop (cdr chars))))))))
(define* (true-string-pad str len optional: (chr #\space))
(let ((strlen (true-string-length str)))
(if (> strlen len)
str
(string-append (make-string (- len strlen) chr)
str))))
(define (trim-to-width str len)
(let ((trimmed (string-pad-right str len)))
(if (< (string-length trimmed)
(string-length str))
(string-append (string-drop-right trimmed 1)
"…")
trimmed)))
;; TODO more options for infix strings
(define* (add-enumeration-punctuation
list optional: (final-delim "&"))
(cond [(null? list) ""]
[(= 1 (length list)) (car list)]
[else
(let* ((rev (reverse list))
(tail (car rev))
(rest (cdr rev)))
(reverse (cons* tail " " final-delim " "
(intersperse ", " rest))))]))
|