lean4-htt/library/init/lean/parser/reader/module.lean

141 lines
2.9 KiB
Text

/-
Copyright (c) 2018 Microsoft Corporation. All rights reserved.
Released under Apache 2.0 license as described in the file LICENSE.
Author: Sebastian Ullrich
Module-level readers and macros
-/
prelude
import init.lean.parser.reader.token init.lean.parser.reader.term
namespace lean.parser
namespace reader
open combinators
def «prelude» := {macro . name := `prelude}
def prelude.reader : reader :=
node «prelude» [keyword "prelude"]
def import_path := {macro . name := `import_path}
def import_path.reader : reader :=
-- use `raw_symbol` to ignore registered tokens like ".."
node import_path [many (raw_symbol "."), ident]
def «import» := {macro . name := `import}
def import.reader : reader :=
node «import» [keyword "import", many1 import_path.reader]
section commands
def «open» := {macro . name := `open}
def open_export.reader : reader :=
many1 $ seq [
ident,
optional $ seq [
keyword "as",
ident
],
optional $ seq [
try $ seq [symbol "(", ident],
many ident,
symbol ")"
],
optional $ seq [
try $ seq [symbol "(", keyword "renaming"],
many1 $ seq [ident, symbol "->", ident],
symbol ")"
],
optional $ seq [
symbol "(",
keyword "hiding",
many1 ident,
symbol ")"
]
]
def open.reader : reader :=
node «open» [keyword "open", open_export.reader]
def «section» := {macro . name := `section}
def section.reader : reader :=
node «section» [
keyword "section",
optional ident,
many recurse,
keyword "end",
optional ident
]
def «notation» := {macro . name := `notation}
def prec := seq [keyword ":", number/-TODO <|> expr-/]
def quoted_symbol : read_m syntax :=
do (s, info) ← with_source_info $ monad_parsec.take_until (= '`'),
pure $ syntax.atom ⟨info, atomic_val.string s⟩
def notation_tk :=
any_of [
seq [
raw_symbol "`",
{read := quoted_symbol},
raw_symbol "`",
optional prec
]
--TODO, {read := do tk ← token, /- check if reserved token-/}
]
def action :=
seq [
keyword ":",
any_of [
number,
keyword "max",
keyword "prev",
keyword "scoped"
/-TODO seq [
symbol "(",
any_of [keyword "foldl", keyword "foldr"],
optional prec,
notation_tk,-/]
]
def notation_reader : reader :=
any_of [
number,
seq [
optional ident,
many $ seq [
notation_tk,
optional $ any_of [
seq [keyword "binder", optional prec],
seq [keyword "binders", optional prec],
seq [ident, optional action]
]
]
]
]
def notation.reader :=
seq [keyword "notation", notation_reader, keyword ":=", term.reader]
def command.reader :=
with_recurse $ any_of [open.reader, section.reader, notation.reader] <?> "command"
end commands
def module := {macro . name := `module}
def module.reader : reader :=
node module [
optional prelude.reader,
many import.reader,
many command.reader
]
end reader
end lean.parser