blob: c26fa90a49dfd37a0f6585103ae68b9b05668308 (
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
(define-library (csc flag)
(export
*args*
bool-flag
define-bool-flag
define-flag
parse-error-flag
parse-error-msg
parse-error?
parse-flags)
(import (scheme base)
(only (scheme process-context)
command-line)
(only (csc hash-map)
compare-strings
insert
key-not-found-error?
lookup
make-map)
(only (csc loop)
loop)
(only (csc match)
match)
(only (csc strings)
contains?
has-prefix?
split))
(begin
(define (bool-flag s)
(cond
((member s '("1" "t" "T" "true" "TRUE" "True"))
#t)
((member s '("0" "f" "F" "false" "FALSE" "False"))
#f)
(else (error "Argument could not be parsed as a boolean" s))))
(define *parsers* (make-map compare-strings))
(define (add-parser flag bool? setter)
(set! *parsers* (insert *parsers* flag
(make-flag bool? setter))))
(define-record-type <flag>
(make-flag bool? setter)
flag?
(bool? flag-bool?)
(setter flag-setter))
(define-syntax define-flag
(syntax-rules ()
((define-flag name flag type default)
(begin
(define name default)
(add-parser flag (eq? type bool-flag)
(lambda (x) (set! name (type x))))))))
(define *args* '())
(define-record-type <parse-error>
(make-parse-error msg flag)
parse-error?
(msg parse-error-msg)
(flag parse-error-flag))
(define (parse-flags)
(define args (cdr (command-line)))
; Is this legal?
(define (parse-one)
(match args
('() #f)
((s . _) when (or (not (has-prefix? s "-"))
(string=? "-" s))
#f)
((s . rest) when (string=? "--" s)
(set! args rest)
#f)
((s . rest)
(define name (if (has-prefix? s "--")
(string-copy s 2)
(string-copy s 1)))
(when (or (string=? "" name)
(has-prefix? name "-")
(has-prefix? name "="))
(raise (make-parse-error "bad flag syntax" s)))
; It's a flag. Does it have an argument?
(set! args rest)
(define value (match (split name "=" 2)
((a b)
(set! name a)
b)
(_ #f)))
(define flag (guard (e ((key-not-found-error? e)
(raise (make-parse-error "flag provided but not defined" s))))
(lookup *parsers* name)))
(if (flag-bool? flag) ; Special case: doesn't need an arg.
(if value
((flag-setter flag) value)
((flag-setter flag) "true"))
(begin
; It must have a value, which might be the next argument.
(when (and (not value)
(not (null? args)))
; value is the next arg
(set! value (car args))
(set! args (cdr args)))
(unless value
(raise (make-parse-error "flag needs an argument" s)))
((flag-setter flag) value)))
#t)))
(loop while (parse-one))
(set! *args* args))))
|
