#lang br
(require brag/support)

(define-lex-abbrev digits (:+ (char-set "0123456789")))

(define basic-lexer
  (lexer-srcloc
   [(eof) (return-without-srcloc eof)]
   [whitespace (token lexeme #:skip? #t)]
   [(from/to "rem" "\n") (token 'REM lexeme)]
   [(:or "print" "goto" "end" "+" ":") lexeme]
   [digits (token 'INTEGER (string->number lexeme))]
   [(:or (:seq digits ".") (:seq (:? digits) "." digits))
    (token 'DECIMAL (string->number lexeme))]
   [(from/to "\"" "\"")
    (token 'STRING (trim-ends "\"" lexeme "\""))]))

(provide basic-lexer)