mes: string-join: Support optional delimiter, support grammar.
[mes.git] / module / srfi / srfi-13.mes
1 ;;; -*-scheme-*-
2
3 ;;; Mes --- Maxwell Equations of Software
4 ;;; Copyright © 2016,2017,2018 Jan (janneke) Nieuwenhuizen <janneke@gnu.org>
5 ;;;
6 ;;; This file is part of Mes.
7 ;;;
8 ;;; Mes is free software; you can redistribute it and/or modify it
9 ;;; under the terms of the GNU General Public License as published by
10 ;;; the Free Software Foundation; either version 3 of the License, or (at
11 ;;; your option) any later version.
12 ;;;
13 ;;; Mes is distributed in the hope that it will be useful, but
14 ;;; WITHOUT ANY WARRANTY; without even the implied warranty of
15 ;;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16 ;;; GNU General Public License for more details.
17 ;;;
18 ;;; You should have received a copy of the GNU General Public License
19 ;;; along with Mes.  If not, see <http://www.gnu.org/licenses/>.
20
21 ;;; Commentary:
22
23 ;;; srfi-13.mes is the minimal srfi-13
24
25 ;;; Code:
26
27 (mes-use-module (srfi srfi-1))
28 (mes-use-module (srfi srfi-14))
29
30 (define (string-join lst . delimiter+grammar)
31   (let ((delimiter (or (and (pair? delimiter+grammar) (car delimiter+grammar))
32                        " "))
33         (grammar (or (and (pair? delimiter+grammar) (pair? (cdr delimiter+grammar)) (cadr delimiter+grammar))
34                      'infix)))
35     (if (null? lst) ""
36         (case grammar
37          ((infix) (if (null? (cdr lst)) (car lst)
38                                    (string-append (car lst) delimiter (string-join (cdr lst) delimiter))))
39          ((prefix) (string-append delimiter (car lst) (apply string-join (cdr lst) delimiter+grammar)))
40          ((suffix) (string-append (car lst) delimiter (apply string-join (cdr lst) delimiter+grammar)))))))
41
42 (define (string-copy s)
43   (list->string (string->list s)))
44
45 (define (string=? a b)
46     (eq? (string->symbol a)
47          (string->symbol b)))
48
49 (define (string= a b . rest)
50   (let* ((start1 (and (pair? rest) (car rest)))
51          (end1 (and start1 (pair? (cdr rest)) (cadr rest)))
52          (start2 (and end1 (pair? (cddr rest)) (caddr rest)))
53          (end2 (and start2 (pair? (cdddr rest)) (cadddr rest))))
54     (string=? (if start1 (if end1 (substring a start1 end1)
55                              (substring a start1))
56                   a)
57               (if start2 (if end2 (substring b start2 end2)
58                              (substring b start2))
59                   b))))
60
61 (define (string-split s c)
62   (let loop ((lst (string->list s)) (result '()))
63     (let ((rest (memq c lst)))
64       (if (not rest) (append result (list (list->string lst)))
65           (loop (cdr rest)
66                 (append result
67                         (list (list->string (list-head lst (- (length lst) (length rest)))))))))))
68
69 (define (string-take s n)
70   (cond ((zero? n) s)
71         ((> n 0) (list->string (list-head (string->list s) n)))
72         (else (error "string-take: not supported: n=" n))))
73
74 (define (string-drop s n)
75   (cond ((zero? n) s)
76         ((> n 0) (list->string (list-tail (string->list s) n)))
77         (else s (error "string-drop: not supported: (n s)=" (cons n s)))))
78
79 (define (string-drop-right s n)
80   (cond ((zero? n) s)
81         ((> n 0) (let ((length (string-length s)))
82                    (list->string (list-head (string->list s) (- length n)))))
83         (else (error "string-drop-right: not supported: n=" n))))
84
85 (define (string-delete pred s)
86   (let ((p (if (procedure? pred) pred
87                (lambda (c) (not (eq? pred c))))))
88     (list->string (filter p (string->list s)))))
89
90 (define (string-index s pred . rest)
91   (let* ((start (and (pair? rest) (car rest)))
92          (end (and start (pair? (cdr rest)) (cadr rest)))
93          (pred (if (char? pred) (lambda (c) (eq? c pred)) pred)))
94     (if start (error "string-index: not supported: start=" start))
95     (if end (error "string-index: not supported: end=" end))
96     (let loop ((lst (string->list s)) (i 0))
97       (if (null? lst) #f
98           (if (pred (car lst)) i
99               (loop (cdr lst) (1+ i)))))))
100
101 (define (string-rindex s pred . rest)
102   (let* ((start (and (pair? rest) (car rest)))
103          (end (and start (pair? (cdr rest)) (cadr rest)))
104          (pred (if (char? pred) (lambda (c) (eq? c pred)) pred)))
105     (if start (error "string-rindex: not supported: start=" start))
106     (if end (error "string-rindex: not supported: end=" end))
107     (let loop ((lst (reverse (string->list s))) (i (1- (string-length s))))
108       (if (null? lst) #f
109           (if (pred (car lst)) i
110               (loop (cdr lst) (1- i)))))))
111
112 (define reverse-list->string (compose list->string reverse))
113
114 (define substring/copy substring)
115 (define substring/shared substring)
116
117 (define string-null? (compose null? string->list))
118
119 (define (string-fold cons' nil' s . rest)
120   (let* ((start (and (pair? rest) (car rest)))
121          (end (and start (pair? (cdr rest)) (cadr rest))))
122     (if start (error "string-fold: not supported: start=" start))
123     (if end (error "string-fold: not supported: end=" end))
124     (let loop ((lst (string->list s)) (prev nil'))
125       (if (null? lst) prev
126           (loop (cdr lst) (cons' (car lst) prev))))))
127
128 (define (string-fold-right cons' nil' s . rest)
129   (let* ((start (and (pair? rest) (car rest)))
130          (end (and start (pair? (cdr rest)) (cadr rest))))
131     (if start (error "string-fold-right: not supported: start=" start))
132     (if end (error "string-fold-right: not supported: end=" end))
133     (let loop ((lst (reverse (string->list s))) (prev nil'))
134       (if (null? lst) prev
135           (loop (cdr lst) (cons' (car lst) prev))))))
136
137 (define (string-contains string needle)
138   (let ((needle (string->list needle)))
139     (let loop ((string (string->list string)) (i 0))
140       (and (pair? string)
141            (let match ((start string) (needle needle) (n i))
142              (if (null? needle) i
143                  (and (pair? start)
144                       (if (eq? (car start) (car needle))
145                           (or (match (cdr start) (cdr needle) (1+ n))
146                               (loop (cdr string) (1+ i)))
147                           (loop (cdr string) (1+ i))))))))))
148
149 (define (string-trim string . pred)
150   (list->string
151    (if (pair? pred) (error "string-trim: not supported: PRED=" pred)
152        (let loop ((lst (string->list string)))
153          (if (or (null? lst)
154                  (not (char-whitespace? (car lst)))) lst
155                  (loop (cdr lst)))))))
156
157 (define (string-trim-right string . pred)
158   (list->string
159    (reverse!
160     (if (pair? pred) (error "string-trim-right: not supported: PRED=" pred)
161         (let loop ((lst (reverse (string->list string))))
162           (if (or (null? lst)
163                   (not (char-whitespace? (car lst)))) lst
164                   (loop (cdr lst))))))))
165
166 (define (string-trim-both string . pred)
167   ((compose string-trim string-trim-right) string))
168
169 (define (string-map f string)
170   (list->string (map f (string->list string))))
171
172 (define (string-replace string replace . rest)
173   (let* ((start1 (and (pair? rest) (car rest)))
174          (end1 (and start1 (pair? (cdr rest)) (cadr rest)))
175          (start2 (and end1 (pair? (cddr rest)) (caddr rest)))
176          (end2 (and start2 (pair? (cdddr rest)) (cadddr rest))))
177     (if start2 (error "string-replace: not supported: START2=" start2))
178     (if end2 (error "string-replace: not supported: END2=" end2))
179     (list->string
180      (append
181       (string->list (string-take string (or start1 0)))
182       (string->list replace)
183       (string->list (string-drop string (or end1 (string-length string))))))))