#lang racket/base

(require syntax/parse
         "annotate-classes.rkt"
         (for-template racket/base
                       "base-types.rkt"))

(provide (all-defined-out))

(define-splicing-syntax-class for-clause
    ;; single-valued seq-expr
    (pattern (~and c (var:annotated-name seq-expr:expr))
             #:with (expand ...) (list (syntax/loc
                                           #'c
                                         (var.ann-name seq-expr))))
    ;; multi-valued seq-expr
    ;; currently disabled because it triggers an internal error in the typechecker
    #;(pattern (~and c (((v:annotated-name) ...) seq-expr:expr))
             #:with (expand ...) (list (syntax/loc
                                           #'c
                                         ((v.ann-name ...) seq-expr))))
    ;; when clause
    (pattern (~seq #:when guard:expr)
             #:with (expand ...) (list #'#:when #'guard)))

;; intersperses "#:when #t" clauses to emulate the for* variants' semantics
(define-splicing-syntax-class for*-clause
  ;; single-valued seq-expr
    (pattern (~and c (var:annotated-name seq-expr:expr))
             #:with (expand ...) (list (syntax/loc
                                           #'c
                                         (var.ann-name seq-expr))
                                       #'#:when #'#t))
    ;; multi-valued seq-expr
    ;; currently disabled because it triggers an internal error in the typechecker
    #;(pattern (~and c (((v:annotated-name) ...) seq-expr:expr))
             #:with (expand ...) (list (quasisyntax/loc
                                           #'c
                                         ((v.ann-name ...) seq-expr))
                                       #'#:when #'#t))
    ;; when clause
    (pattern (~seq #:when guard:expr)
             #:with (expand ...) (list #'#:when #'guard)))
