suzanne.soy/trivial
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[regexp] checkpoint: override regexp constructor, too

Browse Source
This commit is contained in:
ben 2015-12-08 02:57:43 -05:00
parent 2b41d7752b
commit 2b1ab27825
1 changed files with 37 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
regexp.rkt
View File

@ -1,6 +1,14 @@
#lang typed/racket/base #lang typed/racket/base
;; TODO byte-regexp
(provide (provide
regexp!
;; (-> String Regexp)
pregexp!
;; (-> String Regexp)
regexp-match! regexp-match!
;; (-> Pattern String Any * (U #f (List String *N+1))) ;; (-> Pattern String Any * (U #f (List String *N+1)))
;; Match the regular expression pattern against a string. ;; Match the regular expression pattern against a string.
@ -16,10 +24,27 @@
;; ============================================================================= ;; =============================================================================
(define-for-syntax num-groups-key 'num-groups)
;; TODO abstract these macros?
(define-syntax regexp!
(syntax-parser
[(_ x:str)
(syntax-property #'(regexp x) num-groups-key (count-groups/regexp (syntax-e #'x)))]
[(_ arg* ...)
#'(regexp arg* ...)]))
(define-syntax pregexp!
(syntax-parser
[(_ x:str)
(syntax-property #'(pregexp x) num-groups-key (count-groups/pregexp (syntax-e #'x)))]
[(_ arg* ...)
#'(pregexp arg* ...)]))
(define-syntax regexp-match! (define-syntax regexp-match!
(syntax-parser (syntax-parser
[(f pat-stx arg* ...) [(f pat-stx arg* ...)
#:with num-groups (count-groups (format "~a" (syntax-e #'pat-stx)) #:src #'f) #:with num-groups (count-groups #'pat-stx #:src #'f)
#:with ((index* . group-id*) ...) #:with ((index* . group-id*) ...)
#`#,(for/list ([i (in-range (syntax-e #'num-groups))]) #`#,(for/list ([i (in-range (syntax-e #'num-groups))])
(cons i (format-id #'f "group-~a" i))) (cons i (format-id #'f "group-~a" i)))
@ -32,12 +57,16 @@
[(f arg* ...) [(f arg* ...)
(syntax/loc #'f (regexp-match arg* ...))])) (syntax/loc #'f (regexp-match arg* ...))]))
(define-for-syntax (count-groups v #:src stx) (define-for-syntax (count-groups v-stx #:src stx)
(cond (or (syntax-property v-stx num-groups-key)
[(string? v) (count-groups/string v #:src stx)] (let ([v (format "~a" (syntax-e v-stx))])
;[(regexp? v) (count-groups/regexp v #:src stx)] (cond
;[(pregexp? v) (count-groups/pregexp v #:src stx)] [(string? v) (count-groups/string v #:src stx)]
[else (error 'regexp-match! "Internal error on input" v)])) ;[(regexp? v) (count-groups/regexp v #:src stx)]
;[(pregexp? v) (count-groups/pregexp v #:src stx)]
;[(byte-regexp? v) (count-groups/byte-regexp v #:src stx)]
;[(byte-pregexp? v) (count-groups/byte-pregexp v #:src stx)]
[else (error 'regexp-match! "Internal error on input" v)]))))
;; Count the number of matched parentheses in a regexp pattern. ;; Count the number of matched parentheses in a regexp pattern.
;; Raise an exception if there are unmatched parens. ;; Raise an exception if there are unmatched parens.
@ -65,7 +94,7 @@
(define-for-syntax (count-groups/regexp rxp #:src stx) (define-for-syntax (count-groups/regexp rxp #:src stx)
(error 'regexp-match! "Not implemented")) (error 'regexp-match! "Not implemented"))
(define-for-syntax (count-groups/regexp pxp #:src stx) (define-for-syntax (count-groups/pregexp pxp #:src stx)
(error 'regexp-match! "Not implemented")) (error 'regexp-match! "Not implemented"))
(define-for-syntax (group-error str reason) (define-for-syntax (group-error str reason)