@@ -1,8 +1,11 @@
11{-# OPTIONS_GHC -Wall #-}
2+{-# LANGUAGE OverloadedStrings #-}
23module AST.Pattern where
34
45import qualified Data.List as List
56import qualified Data.Set as Set
7+import qualified Data.Text as Text
8+import Data.Text (Text)
69
710import qualified AST.Literal as L
811import qualified AST.Variable as Var
@@ -15,10 +18,10 @@ type Pattern ann var =
1518
1619
1720data Pattern' ann var
18− = Data var [Pattern ann var]
19− | Record [String]
20− | Alias String (Pattern ann var)
21− | Var String
21+ = Ctor var [Pattern ann var]
22+ | Record [Text]
23+ | Alias Text (Pattern ann var)
24+ | Var Text
2225 | Anything
2326 | Literal L.Literal
2427
@@ -35,7 +38,7 @@ type Canonical =
3538 Pattern R.Region Var.Canonical
3639
3740
38−isVar :: String -> Pattern ann var -> Bool
41+isVar :: Text -> Pattern ann var -> Bool
3942isVar name (A.A _ pattern) =
4043 case pattern of
4144 Var pName ->
@@ -49,30 +52,26 @@ list :: R.Position -> [Raw] -> Raw
4952list end patterns =
5053 case patterns of
5154 [] ->
52− A.at end end (Data (Var.Raw "[]") [])
55+ A.at end end (Ctor (Var.Raw "[]") [])
5356
5457 pattern@(A.A (R.Region start _) _) : rest ->
55− A.at start end (Data (Var.Raw "::") [pattern, list end rest])
56−
57−
58−consMany :: R.Position -> [Raw] -> Raw
59−consMany end patterns =
60− let cons hd@(A.A (R.Region start _) _) tl =
61− A.at start end (Data (Var.Raw "::") [hd, tl])
62− in
63− foldr1 cons patterns
58+ A.at start end (Ctor (Var.Raw "::") [pattern, list end rest])
6459
6560
6661tuple :: [Raw] -> Raw'
6762tuple patterns =
68− Data (Var.Raw ("_Tuple" ++ show (length patterns))) patterns
63+ let
64+ name =
65+ Text.append "_Tuple" (Text.pack (show (length patterns)))
66+ in
67+ Ctor (Var.Raw name) patterns
6968
7069
7170
7271-- FIND VARIABLES
7372
7473
75−boundVars :: Pattern ann var -> [A.Annotated ann String]
74+boundVars :: Pattern ann var -> [A.Annotated ann Text]
7675boundVars (A.A ann pattern) =
7776 case pattern of
7877 Var x ->
@@ -81,7 +80,7 @@ boundVars (A.A ann pattern) =
8180 Alias name realPattern ->
8281 A.A ann name : boundVars realPattern
8382
84− Data _ patterns ->
83+ Ctor _ patterns ->
8584 concatMap boundVars patterns
8685
8786 Record fields ->
@@ -94,17 +93,17 @@ boundVars (A.A ann pattern) =
9493 []
9594
9695
97−member :: String -> Pattern ann var -> Bool
96+member :: Text -> Pattern ann var -> Bool
9897member name pattern =
99− any (name==) (map A.drop (boundVars pattern))
98+ elem name (map A.drop (boundVars pattern))
10099
101100
102−boundVarSet :: Pattern ann var -> Set.Set String
101+boundVarSet :: Pattern ann var -> Set.Set Text
103102boundVarSet pattern =
104103 Set.fromList (map A.drop (boundVars pattern))
105104
106105
107−boundVarList :: Pattern ann var -> [String]
106+boundVarList :: Pattern ann var -> [Text]
108107boundVarList pattern =
109108 Set.toList (boundVarSet pattern)
110109
@@ -117,15 +116,15 @@ toString :: Bool -> Canonical -> String
117116toString needsParens (A.A _ pattern) =
118117 case pattern of
119118 Var name ->
120− name
119+ Text.unpack name
121120
122− Data name [] ->
121+ Ctor name [] ->
123122 if Var.isTuple name then
124123 "()"
125124 else
126125 Var.toString name
127126
128− Data name args ->
127+ Ctor name args ->
129128 if Var.isTuple name then
130129 "( " ++ List.intercalate ", " (map (toString False) args) ++ " )"
131130 else
@@ -133,11 +132,11 @@ toString needsParens (A.A _ pattern) =
133132 Var.toString name ++ concatMap ((" "++) . toString True) args
134133
135134 Record fields ->
136− "{" ++ List.intercalate "," fields ++ "}"
135+ "{" ++ List.intercalate "," (map Text.unpack fields) ++ "}"
137136
138137 Alias alias subPattern ->
139138 parensIf needsParens $
140− toString False subPattern ++ " as " ++ alias
139+ toString False subPattern ++ " as " ++ Text.unpack alias
141140
142141 Anything ->
143142 "_"