aboutsummaryrefslogtreecommitdiffhomepage
path: root/arith/Parser.hs
blob: bf5eecb5d7c34b6279a9aee313f8862e48f0e4db (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
module Parser
    ( parseProgram
    ) where

import Text.Parsec
import Text.Parsec.String (Parser)

import Core

lexeme :: Parser a -> Parser a
lexeme p = p <* spaces

symbol :: String -> Parser String
symbol = lexeme . string

keyword :: String -> Parser ()
keyword s = try (string s *> notFollowedBy alphaNum) *> spaces

parens :: Parser a -> Parser a
parens = between (symbol "(") (symbol ")")

parseTerm :: Parser Term
parseTerm = parseIf
        <|> parseSucc
        <|> parsePred
        <|> parseIsZero
        <|> parseTrue
        <|> parseFalse
        <|> parseZero
        <|> parens parseTerm

parseTrue :: Parser Term
parseTrue = TmTrue <$ keyword "true"

parseFalse :: Parser Term
parseFalse = TmFalse <$ keyword "false"

parseZero :: Parser Term
parseZero = TmZero <$ symbol "0"

parseIf :: Parser Term
parseIf = do
    keyword "if"
    t1 <- parseTerm
    keyword "then"
    t2 <- parseTerm
    keyword "else"
    t3 <- parseTerm
    return $ TmIf t1 t2 t3

parseSucc :: Parser Term
parseSucc = do
    keyword "succ"
    t <- parseTerm
    return $ TmSucc t

parsePred :: Parser Term
parsePred = do
    keyword "pred"
    t <- parseTerm
    return $ TmPred t

parseIsZero :: Parser Term
parseIsZero = do
    keyword "iszero"
    t <- parseTerm
    return $ TmIsZero t

parseProgram :: Parser Term
parseProgram = spaces *> parseTerm <* eof