src/Language/SimpleShell/Parser/Name.hs


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

module Language.SimpleShell.Parser.Name
  ( varNameP
  , funNameP
  , keyword
  )
where


import Language.SimpleShell.AST.Name (FunName, VarName)
import Language.SimpleShell.Parser (Parser, lexeme)

import Control.Monad (void)
import Data.Text (Text, cons)
import Data.Char (isAlpha, isAlphaNum)
import Text.Megaparsec (takeWhileP, satisfy, notFollowedBy, try)
import Text.Megaparsec.Char (string)


isNameStartChar :: Char -> Bool
isNameStartChar c = isAlpha c || c == '_'

isNameChar :: Char -> Bool
isNameChar c = isAlphaNum c || c == '_'


nameP :: Parser Text
nameP =
  lexeme $ cons <$> satisfy isNameStartChar <*> takeWhileP Nothing isNameChar


keyword :: Text -> Parser ()
keyword kw =
  void $ lexeme $ try $ string kw <* notFollowedBy (satisfy isNameChar)


varNameP :: Parser VarName
varNameP = nameP

funNameP :: Parser FunName
funNameP = nameP