aboutsummaryrefslogtreecommitdiffstats
path: root/format.csc
blob: 48909a65a44410c2cd0f8bb7a45fe0c85af08869 (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
(define-library (csc format)
  (export
    fprintf
    printf
    sprintf)
  (import (scheme base)
          (only (scheme write) display)
          (only (csc strings)
                str-find
                str-not-found-error?
                str-prefix?))
  (begin


    (define (fprintf 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 (printf format-string . format-args)
      (apply fprintf (current-output-port) format-string format-args))


    (define (sprintf format-string . format-args)
      (let ((string-builder (open-output-string)))
        (apply fprintf string-builder format-string format-args)
        (get-output-string string-builder)))))