git.iko.soy/elk
on master help sign in
Track whether a module name was specified
3a02ad4Evan Czaplicki
compiler/src/AST/Source.hs+16−1
compare with master
unifiedsplit
@@ -1,10 +1,13 @@
1{-# OPTIONS_GHC -Wall #-}
1{-# OPTIONS_GHC -Wall #-}
2+{-# LANGUAGE OverloadedStrings #-}
2module AST.Source
3module AST.Source
3 ( Expr, Expr_(..), VarType(..)
4 ( Expr, Expr_(..), VarType(..)
4 , Def(..)
5 , Def(..)
5 , Pattern, Pattern_(..)
6 , Pattern, Pattern_(..)
6 , Type, Type_(..)
7 , Type, Type_(..)
7 , Module(..)
8 , Module(..)
9+ , getName
10+ , getImport
8 , Import(..)
11 , Import(..)
9 , Value(..)
12 , Value(..)
10 , Union(..)
13 , Union(..)
@@ -119,7 +122,7 @@ data Type_
119
122
120data Module =
123data Module =
121 Module
124 Module
122− { _name :: Name
125+ { _name :: Maybe Name
123 , _exports :: A.Located Exposing
126 , _exports :: A.Located Exposing
124 , _imports :: [Import]
127 , _imports :: [Import]
125 , _values :: [A.Located Value]
128 , _values :: [A.Located Value]
@@ -130,6 +133,18 @@ data Module =
130 }
133 }
131
134
132
135
136+getName :: Module -> Name
137+getName (Module maybeName _ _ _ _ _ _ _) =
138+ case maybeName of
139+ Just name -> name
140+ Nothing -> "Main"
141+
142+
143+getImport :: Import -> Name
144+getImport (Import (A.At _ name) _ _) =
145+ name
146+
147+
133data Import =
148data Import =
134 Import
149 Import
135 { _import :: A.Located Name
150 { _import :: A.Located Name