git.iko.soy/rowtype-yoga/purescript-native
on 9ee79c6 help sign in
rowtype-yoga/purescript-native/src/PureScript/Parser/Values.hs
287 lines 14 kB history raw
1-----------------------------------------------------------------------------
2--
3-- Module : PureScript.Parser.Values
4-- Copyright : (c) Phil Freeman 2013
5-- License : MIT
6--
7-- Maintainer : Phil Freeman <paf31@cantab.net>
8-- Stability : experimental
9-- Portability :
10--
11-- |
12--
13-----------------------------------------------------------------------------
15module PureScript.Parser.Values (
16 parseValue,
17 parseBinder
18) where
20import PureScript.Values
21import PureScript.Names
22import PureScript.Declarations
23import PureScript.Parser.State
24import Data.Function (on)
25import Data.List
26import Data.Functor.Identity
27import qualified Data.Map as M
28import qualified PureScript.Parser.Common as C
29import Control.Applicative
30import qualified Text.Parsec as P
31import Text.Parsec.Expr
32import Control.Monad
33import Control.Arrow (Arrow(..))
34import PureScript.Parser.Types
35import PureScript.Types
37booleanLiteral :: P.Parsec String ParseState Bool
38booleanLiteral = (C.reserved "true" >> return True) P.<|> (C.reserved "false" >> return False)
40parseNumericLiteral :: P.Parsec String ParseState Value
41parseNumericLiteral = NumericLiteral <$> C.integerOrFloat
43parseStringLiteral :: P.Parsec String ParseState Value
44parseStringLiteral = StringLiteral <$> C.stringLiteral
46parseBooleanLiteral :: P.Parsec String ParseState Value
47parseBooleanLiteral = BooleanLiteral <$> booleanLiteral
49parseArrayLiteral :: P.Parsec String ParseState Value
50parseArrayLiteral = ArrayLiteral <$> (C.squares $ parseValue `P.sepBy` (C.indented *> C.comma))
52parseObjectLiteral :: P.Parsec String ParseState Value
53parseObjectLiteral = ObjectLiteral <$> (C.braces $ parseIdentifierAndValue `P.sepBy` (C.indented *> C.comma))
55parseIdentifierAndValue :: P.Parsec String ParseState (String, Value)
56parseIdentifierAndValue = (,) <$> (C.indented *> C.identifier <* C.indented <* C.colon)
57 <*> (C.indented *> parseValue)
59parseAbs :: P.Parsec String ParseState Value
60parseAbs = do
61 C.lexeme $ P.char '\\'
62 args <- (C.indented *> C.parseIdent) `P.sepBy` (C.indented *> C.comma)
63 C.lexeme $ C.indented *> P.string "->"
64 value <- parseValue
65 return $ Abs args value
67parseApp :: P.Parsec String ParseState Value
68parseApp = App <$> parseValue
69 <*> (C.indented *> C.parens (parseValue `P.sepBy` (C.indented *> C.comma)))
71parseVar :: P.Parsec String ParseState Value
72parseVar = Var <$> C.parseIdent
74parseConstructor :: P.Parsec String ParseState Value
75parseConstructor = Constructor <$> C.properName
77parseCase :: P.Parsec String ParseState Value
78parseCase = Case <$> (P.between (C.reserved "case") (C.indented *> C.reserved "of") parseValue)
79 <*> (C.indented *> C.mark (P.many (C.same *> C.mark parseCaseAlternative)))
81parseCaseAlternative :: P.Parsec String ParseState (Binder, Value)
82parseCaseAlternative = (,) <$> (parseGuardedBinder <* C.lexeme (P.string "->"))
83 <*> parseValue
84 P.<?> "case alternative"
86parseIfThenElse :: P.Parsec String ParseState Value
87parseIfThenElse = IfThenElse <$> (C.reserved "if" *> C.indented *> parseValue)
88 <*> (C.indented *> C.reserved "then" *> C.indented *> parseValue)
89 <*> (C.indented *> C.reserved "else" *> C.indented *> parseValue)
91parseBlock :: P.Parsec String ParseState Value
92parseBlock = Block <$> (C.reserved "do" *> parseManyStatements)
94parseManyStatements :: P.Parsec String ParseState [Statement]
95parseManyStatements = C.indented *> (C.mark $ P.many $ C.same *> C.mark parseStatement) P.<?> "block"
97parseValueAtom :: P.Parsec String ParseState Value
98parseValueAtom = C.indented *> P.choice (map P.try
99 [ parseNumericLiteral
100 , parseStringLiteral
101 , parseBooleanLiteral
102 , parseArrayLiteral
103 , parseObjectLiteral
104 , parseAbs
105 , parseVar
106 , parseConstructor
107 , parseBlock
108 , parseCase
109 , parseIfThenElse
110 , C.parens parseValue ])
112parseValue :: P.Parsec String ParseState Value
113parseValue = do
114 customOps <- fixities <$> P.getState
115 (buildExpressionParser (operators customOps)
116 . C.buildPostfixParser postfixTable2
117 $ indexersAndAccessors) P.<?> "expression"
118 where
119 indexersAndAccessors = C.buildPostfixParser postfixTable1 $ parseValueAtom
120 postfixTable1 = [ Accessor <$> (C.indented *> C.dot *> C.indented *> C.identifier)
121 , Indexer <$> (C.indented *> C.squares parseValue) ]
122 postfixTable2 = [ C.indented *> indexersAndAccessors >>= \t2 -> return (\t1 -> App t1 [t2])
123 , flip App <$> (C.indented *> C.parens (parseValue `P.sepBy` (C.indented *> C.comma)))
124 , flip TypedValue <$> (P.try $ C.lexeme (C.indented *> P.string "::") *> parsePolyType) ]
125 operators user =
126 [ [ Prefix $ C.lexeme (P.try $ C.indented *> C.reservedOp "!") >> return (Unary Not)
127 , Prefix $ C.lexeme (P.try $ C.indented *> C.reservedOp "~") >> return (Unary BitwiseNot)
128 , Prefix $ C.lexeme (P.try $ C.indented *> C.reservedOp "-") >> return (Unary Negate)
129 , Prefix $ C.lexeme (P.try $ C.indented *> C.reservedOp "+") >> return id ]
130 ] ++ customOperatorTable user ++
131 [ [ Infix (C.lexeme (P.try (C.indented *> C.parseIdentInfix P.<?> "operator") >>= \ident -> return $ \t1 t2 -> App (App (Var ident) [t1]) [t2])) AssocRight ]
132 , [ Infix (C.lexeme (P.try $ C.indented *> C.reservedOp "<=") >> return (Binary LessThanOrEqualTo)) AssocRight
133 , Infix (C.lexeme (P.try $ C.indented *> C.reservedOp ">=") >> return (Binary GreaterThanOrEqualTo)) AssocRight ]
134 , [ Infix (C.lexeme (P.try $ C.indented *> C.reservedOp "<") >> return (Binary LessThan)) AssocRight
135 , Infix (C.lexeme (P.try $ C.indented *> C.reservedOp ">") >> return (Binary GreaterThan)) AssocRight ]
136 , [ Infix (C.lexeme (P.try $ C.indented *> C.reservedOp "*") >> return (Binary Multiply)) AssocRight
137 , Infix (C.lexeme (P.try $ C.indented *> C.reservedOp "/") >> return (Binary Divide)) AssocRight
138 , Infix (C.lexeme (P.try $ C.indented *> C.reservedOp "%") >> return (Binary Modulus)) AssocRight ]
139 , [ Infix (C.lexeme (P.try $ C.indented *> C.reservedOp "++") >> return (Binary Concat)) AssocRight
140 , Infix (C.lexeme (P.try $ C.indented *> C.reservedOp "+") >> return (Binary Add)) AssocRight
141 , Infix (C.lexeme (P.try $ C.indented *> C.reservedOp "-") >> return (Binary Subtract)) AssocRight ]
142 , [ Infix (C.lexeme (P.try $ C.indented *> C.reservedOp "<<") >> return (Binary ShiftLeft)) AssocRight
143 , Infix (C.lexeme (P.try $ C.indented *> C.reservedOp ">>>") >> return (Binary ZeroFillShiftRight)) AssocRight
144 , Infix (C.lexeme (P.try $ C.indented *> C.reservedOp ">>") >> return (Binary ShiftRight)) AssocRight ]
145 , [ Infix (C.lexeme (P.try $ C.indented *> C.reservedOp "==") >> return (Binary EqualTo)) AssocRight
146 , Infix (C.lexeme (P.try $ C.indented *> C.reservedOp "!=") >> return (Binary NotEqualTo)) AssocRight ]
147 , [ Infix (C.lexeme (P.try $ C.indented *> C.reservedOp "&") >> return (Binary BitwiseAnd)) AssocRight ]
148 , [ Infix (C.lexeme (P.try $ C.indented *> C.reservedOp "^") >> return (Binary BitwiseXor)) AssocRight ]
149 , [ Infix (C.lexeme (P.try $ C.indented *> C.reservedOp "|") >> return (Binary BitwiseOr)) AssocRight ]
150 , [ Infix (C.lexeme (P.try $ C.indented *> C.reservedOp "&&") >> return (Binary And)) AssocRight ]
151 , [ Infix (C.lexeme (P.try $ C.indented *> C.reservedOp "||") >> return (Binary Or)) AssocRight ]
152 ]
154customOperatorTable :: M.Map String Fixity -> OperatorTable String ParseState Identity Value
155customOperatorTable fixities =
156 let
157 ops = map (\(name, Fixity a p) -> (name, (a, p))) . M.toList $ fixities
158 sorted = sortBy (compare `on` (snd . snd)) ops
159 levels = groupBy ((==) `on` (snd . snd)) sorted
160 in
161 flip map levels (map (\(name, (a, _)) -> flip Infix (toAssoc a) $ do
162 C.lexeme $ P.try $ do
163 C.indented
164 C.reservedOp name P.<?> "operator"
165 return $ \t1 t2 -> App (App (Var (Op name)) [t1]) [t2]))
167toAssoc :: Associativity -> Assoc
168toAssoc Infixl = AssocLeft
169toAssoc Infixr = AssocRight
171parseVariableIntroduction :: P.Parsec String ParseState Statement
172parseVariableIntroduction = do
173 C.reserved "var"
174 name <- C.indented *> C.parseIdent
175 C.lexeme $ C.indented *> P.char '='
176 value <- parseValue
177 return $ VariableIntroduction name value
179parseAssignment :: P.Parsec String ParseState Statement
180parseAssignment = do
181 tgt <- parseAssignmentTarget
182 C.lexeme $ C.indented *> P.char '='
183 value <- parseValue
184 return $ Assignment tgt value
186parseWhile :: P.Parsec String ParseState Statement
187parseWhile = While <$> (C.reserved "while" *> C.indented *> parseValue <* C.indented <* C.colon)
188 <*> parseManyStatements
190parseFor :: P.Parsec String ParseState Statement
191parseFor = For <$> (C.reserved "for" *> C.indented *> C.parseIdent)
192 <*> (C.indented *> C.lexeme (P.string "<-") *> parseValue)
193 <*> (C.indented *> C.reserved "until" *> parseValue <* C.colon)
194 <*> parseManyStatements
196parseForEach :: P.Parsec String ParseState Statement
197parseForEach = ForEach <$> (C.reserved "foreach" *> C.indented *> C.parseIdent)
198 <*> (C.indented *> C.reserved "in" *> parseValue <* C.colon)
199 <*> parseManyStatements
201parseIf :: P.Parsec String ParseState Statement
202parseIf = If <$> parseIfStatement
204parseIfStatement :: P.Parsec String ParseState IfStatement
205parseIfStatement =
206 IfStatement <$> (C.reserved "if" *> C.indented *> parseValue <* C.indented <* C.colon)
207 <*> parseManyStatements
208 <*> P.optionMaybe (C.same *> parseElseStatement)
210parseElseStatement :: P.Parsec String ParseState ElseStatement
211parseElseStatement = C.reserved "else" >> (ElseIf <$> (C.indented *> parseIfStatement)
212 <|> Else <$> (C.indented *> C.colon *> parseManyStatements))
214parseReturn :: P.Parsec String ParseState Statement
215parseReturn = Return <$> (C.reserved "return" *> parseValue)
217parseStatement :: P.Parsec String ParseState Statement
218parseStatement = P.choice (map P.try
219 [ parseVariableIntroduction
220 , parseAssignment
221 , parseWhile
222 , parseFor
223 , parseForEach
224 , parseIf
225 , parseReturn ]) P.<?> "statement"
227parseStringBinder :: P.Parsec String ParseState Binder
228parseStringBinder = StringBinder <$> C.stringLiteral
230parseBooleanBinder :: P.Parsec String ParseState Binder
231parseBooleanBinder = BooleanBinder <$> booleanLiteral
233parseNumberBinder :: P.Parsec String ParseState Binder
234parseNumberBinder = NumberBinder <$> C.integerOrFloat
236parseVarBinder :: P.Parsec String ParseState Binder
237parseVarBinder = VarBinder <$> C.parseIdent
239parseNullaryBinder :: P.Parsec String ParseState Binder
240parseNullaryBinder = NullaryBinder <$> C.lexeme C.properName
242parseUnaryBinder :: P.Parsec String ParseState Binder
243parseUnaryBinder = UnaryBinder <$> C.lexeme C.properName <*> (C.indented *> parseBinder)
245parseObjectBinder :: P.Parsec String ParseState Binder
246parseObjectBinder = ObjectBinder <$> C.braces ((C.indented *> parseIdentifierAndBinder) `P.sepBy` (C.indented *> C.comma))
248parseArrayBinder :: P.Parsec String ParseState Binder
249parseArrayBinder = C.squares $ ArrayBinder <$> ((C.indented *> parseBinder) `P.sepBy` (C.indented *> C.comma))
250 <*> P.optionMaybe (C.indented *> C.colon *> C.indented *> parseBinder)
252parseNamedBinder :: P.Parsec String ParseState Binder
253parseNamedBinder = NamedBinder <$> (C.parseIdent <* C.indented <* C.lexeme (P.char '@'))
254 <*> (C.indented *> parseBinder)
256parseNullBinder :: P.Parsec String ParseState Binder
257parseNullBinder = C.lexeme (P.char '_') *> P.notFollowedBy C.identLetter *> return NullBinder
259parseIdentifierAndBinder :: P.Parsec String ParseState (String, Binder)
260parseIdentifierAndBinder = do
261 name <- C.lexeme C.identifier
262 C.lexeme $ C.indented *> P.char '='
263 binder <- C.indented *> parseBinder
264 return (name, binder)
266parseBinder :: P.Parsec String ParseState Binder
267parseBinder = P.choice (map P.try
268 [ parseNullBinder
269 , parseStringBinder
270 , parseBooleanBinder
271 , parseNumberBinder
272 , parseNamedBinder
273 , parseVarBinder
274 , parseUnaryBinder
275 , parseNullaryBinder
276 , parseObjectBinder
277 , parseArrayBinder
278 , C.parens parseBinder ]) P.<?> "binder"
280parseGuardedBinder :: P.Parsec String ParseState Binder
281parseGuardedBinder = flip ($) <$> parseBinder <*> P.option id (GuardedBinder <$> (C.indented *> C.lexeme (P.char '|') *> C.indented *> parseValue))
283parseAssignmentTarget :: P.Parsec String ParseState AssignmentTarget
284parseAssignmentTarget = buildExpressionParser operators (AssignVariable <$> C.parseIdent)
285 where
286 operators = [ [ Postfix $ AssignArrayIndex <$> (C.indented *> C.squares parseValue)
287 , Postfix $ AssignObjectProperty <$> (C.indented *> C.dot *> C.indented *> C.identifier) ] ]