You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
61 lines
1.8 KiB
Racket
61 lines
1.8 KiB
Racket
#lang br/quicklang
|
|
(require brag/support racket/contract)
|
|
|
|
(module+ test
|
|
(require rackunit))
|
|
|
|
(define (token? x)
|
|
(or (eof-object? x) (string? x) (token-struct? x)))
|
|
|
|
(module+ test
|
|
(check-true (token? eof))
|
|
(check-true (token? "a string"))
|
|
(check-true (token? (token 'A-TOKEN-STRUCT "hi")))
|
|
(check-false (token? 42)))
|
|
|
|
(define (tokenize port)
|
|
(port-count-lines! port)
|
|
(define (next-token)
|
|
(define our-lexer
|
|
(lexer
|
|
[(eof) eof]
|
|
[(from/to "//" "\n") (next-token)]
|
|
[(from/to "@$" "$@")
|
|
(token 'SEXP-TOK (trim-ends "@$" lexeme "$@")
|
|
#:position (+ (pos lexeme-start) 2)
|
|
#:line (line lexeme-start)
|
|
#:column (+ (col lexeme-start) 2)
|
|
#:span (- (pos lexeme-end)
|
|
(pos lexeme-start) 4))]
|
|
[any-char (token 'CHAR-TOK lexeme
|
|
#:position (pos lexeme-start)
|
|
#:line (line lexeme-start)
|
|
#:column (col lexeme-start)
|
|
#:span (- (pos lexeme-end)
|
|
(pos lexeme-start)))]))
|
|
(our-lexer port))
|
|
next-token)
|
|
(provide (contract-out
|
|
[tokenize (input-port? . -> . (-> token?))]))
|
|
|
|
(module+ test
|
|
(check-equal? (apply-tokenizer tokenize "// comment\n") empty)
|
|
(check-equal?
|
|
(apply-tokenizer tokenize "@$ (+ 6 7) $@")
|
|
(list (token 'SEXP-TOK " (+ 6 7) "
|
|
#:position 3
|
|
#:line 1
|
|
#:column 2
|
|
#:span 9)))
|
|
(check-equal?
|
|
(apply-tokenizer tokenize "hi")
|
|
(list (token 'CHAR-TOK "h"
|
|
#:position 1
|
|
#:line 1
|
|
#:column 0
|
|
#:span 1)
|
|
(token 'CHAR-TOK "i"
|
|
#:position 2
|
|
#:line 1
|
|
#:column 1
|
|
#:span 1)))) |