144 lines
4.8 KiB
Racket
144 lines
4.8 KiB
Racket
#lang scheme/base
|
||
(require scheme/class
|
||
lang/private/rewrite-error-message)
|
||
|
||
;; --------------------------------------------------------------------------
|
||
(provide check-arg check-arity check-proc check-result
|
||
check-list-list check-color
|
||
check-fun-res check-dependencies
|
||
natural?
|
||
find-non tp-exn? number->ord
|
||
tp-error)
|
||
|
||
(define (natural? w)
|
||
(and (number? w) (integer? w) (>= w 0)))
|
||
|
||
;; (_ -> Boolean) (listof X) -> (union X false)
|
||
(define (find-non pred? l)
|
||
(let ([r (filter (compose not pred?) l)])
|
||
(if (null? r) #f (car r))))
|
||
|
||
|
||
;(: check-fun-res (∀ (γ) (∀ (β α ...) (α ...α -> β)) (_ -γ-> boolean) _ -> γ))
|
||
(define (check-fun-res f pred? type)
|
||
(lambda x
|
||
(define r (apply f x))
|
||
(check-result (object-name f) pred? type r)
|
||
r))
|
||
|
||
;; check-dependencies : Symbol x Boolean x FormatString x Any* -> Void
|
||
(define (check-dependencies pname condition fmt . args)
|
||
(unless condition
|
||
(tp-error pname (apply format fmt args))))
|
||
|
||
#| Tests ------------------------------------------------------------------
|
||
(not (find-non list? '((1 2 3) (a b c))))
|
||
(symbol? (find-non number? '(1 2 3 a)))
|
||
(symbol? (find-non list? '((1 2 3) a (b c))))
|
||
|#
|
||
|
||
(define-struct (tp-exn exn) ())
|
||
|
||
(define (tp-error name fmt . args)
|
||
(raise
|
||
(make-exn:fail:contract #; make-tp-exn
|
||
(string-append (format "~a: " name) (apply format fmt args))
|
||
(current-continuation-marks))))
|
||
|
||
(define (number->ord i)
|
||
(if (= i 0)
|
||
"zeroth"
|
||
(case (modulo i 10)
|
||
[(0 4 5 6 7 8 9) (format "~ath" i)]
|
||
[(1) (format "~ast" i)]
|
||
[(2) (format "~and" i)]
|
||
[(3) (format "~ard" i)])))
|
||
|
||
;; spell-out : number-or-string -> string
|
||
(define (spell-out arg-posn)
|
||
(cond
|
||
[(string? arg-posn) arg-posn]
|
||
[(number? arg-posn)
|
||
(case arg-posn
|
||
[(1) "first"]
|
||
[(2) "second"]
|
||
[(3) "third"]
|
||
[(4) "fourth"]
|
||
[(5) "fifth"]
|
||
[(6) "sixth"]
|
||
[(7) "seventh"]
|
||
[(8) "eighth"]
|
||
[(9) "ninth"]
|
||
[(10) "tenth"]
|
||
[else (number->ord arg-posn)])]))
|
||
|
||
;; Symbol (union true String) String X -> void
|
||
(define (check-list-list pname condition pred given)
|
||
(when (string? condition)
|
||
(tp-error pname (string-append condition (format "\nin ~e" given)))))
|
||
|
||
;; Symbol (_ -> Boolean) String X X *-> X
|
||
(define (check-result pname pred? expected given . other-given)
|
||
(if (pred? given)
|
||
given
|
||
(tp-error pname "is expected to return a ~a, but it returned ~v" expected
|
||
(if (pair? other-given)
|
||
(car other-given)
|
||
given))))
|
||
|
||
;; check-color : symbol (or/c str non-negative-integer) TST -> void
|
||
(define (check-color pname arg-pos given)
|
||
(check-arg pname
|
||
(or (string? given)
|
||
(symbol? given))
|
||
'color
|
||
arg-pos given)
|
||
;; this would be good to check, but it isn't possible, since this
|
||
;; file is not allowed to rely on mred.
|
||
;; also nice would be to allow color% objects here, but that's
|
||
;; not possible for the same reason (but that is why
|
||
;; the '[else color]' case is below in the cond.
|
||
#;
|
||
(let ([color
|
||
(cond
|
||
[(symbol? given)
|
||
(send the-color-database find-color (symbol->string given))]
|
||
[(string? given)
|
||
(send the-color-database find-color given)]
|
||
[else given])])
|
||
(unless color
|
||
(tp-error pname
|
||
"expected the name ~e to be a color, but did not recognize it"
|
||
given))))
|
||
|
||
;; check-arg : sym bool str (or/c str non-negative-integer) TST -> void
|
||
(define (check-arg pname condition expected arg-posn given)
|
||
(unless condition
|
||
(tp-error pname "expects a ~a as ~a argument, given ~e"
|
||
expected
|
||
(spell-out arg-posn)
|
||
given)))
|
||
|
||
;; check-arity : sym num (list-of TST) -> void
|
||
(define (check-arity name arg# args)
|
||
(if (= (length args) arg#)
|
||
(void)
|
||
(tp-error name (argcount-error-message arg# (length args)))))
|
||
|
||
;; check-proc :
|
||
;; sym (... *->* ...) num (union sym str) (union sym str) -> void
|
||
(define (check-proc proc f exp-arity arg# arg-err)
|
||
(unless (procedure? f)
|
||
(tp-error proc "expected a function as ~a argument; given ~e" arg# f))
|
||
(let ([arity-of-f (procedure-arity f)])
|
||
(unless (procedure-arity-includes? f exp-arity) ; (and (number? arity-of-f) (>= arity-of-f exp-arity))
|
||
(tp-error proc "expected function of ~a as ~a argument; given function of ~a "
|
||
arg-err arg#
|
||
(cond
|
||
[(number? arity-of-f)
|
||
(if (= arity-of-f 1)
|
||
(format "1 argument")
|
||
(format "~s arguments" arity-of-f))]
|
||
[(arity-at-least? arity-of-f) "variable number of arguments"]
|
||
[else (format "multiple arities (~s)" arity-of-f)])))))
|