aboutsummaryrefslogtreecommitdiffstats
path: root/format.csc
blob: 2c2da847251ab843dbc4b20a49599808aa3c4ac0 (plain) (blame)
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
(define-library (csc format)
  (export
    vfprintf
    fprintf
    vprintf
    printf
    vsprintf
    sprintf)
  (import (scheme base)
          (only (scheme write) display)
          (only (csc strings)
                str-find
                str-not-found-error?
                str-prefix?))
  (begin


    (define (vfprintf port format-string format-args)
      (let loop ((start 0)
                 (args format-args))
        (cond ((>= start (string-length format-string)))
              ((str-prefix? "{{" format-string start)
               (write-string "{" port)
               (loop (+ 2 start) args))
              ((str-prefix? "}}" format-string start)
               (write-string "}" port)
               (loop (+ 2 start) args))
              ((str-prefix? "{}" format-string start)
               (display (car args) port)
               (loop (+ 2 start) (cdr args)))
              ((str-prefix? "{" format-string start)
               (raise (error "invalid format string" format-string)))
              (else
                (let* ((open-brace-pos (guard (e
                                                ((str-not-found-error? e) (string-length format-string)))
                                              (str-find "{" format-string start)))
                       (close-brace-pos (guard (e
                                                 ((str-not-found-error? e) (string-length format-string)))
                                               (str-find "}" format-string start)))
                       (format-pos (min open-brace-pos close-brace-pos)))
                  (write-string format-string port start format-pos)
                  (loop format-pos args))))))


    (define-syntax fprintf
      (syntax-rules ()
                    ((_ port format-string format-args ...)
                     (vfprintf port format-string (list format-args ...)))))


    (define (vprintf format-string format-args)
      (vfprintf (current-output-port) format-string format-args))


    (define-syntax printf
      (syntax-rules ()
                    ((_ format-string format-args ...)
                     (vprintf format-string (list format-args ...)))))


    (define (vsprintf format-string format-args)
      (let ((string-builder (open-output-string)))
        (vfprintf string-builder format-string format-args)
        (get-output-string string-builder)))


    (define-syntax sprintf
      (syntax-rules ()
                    ((_ format-string format-args ...)
                     (vsprintf format-string (list format-args ...)))))))