module Language.PureScript.Names where
import Prelude.Compat
import Control.Monad.Supply.Class
import Data.Aeson
import Data.Aeson.TH
import Data.List
data Name
= IdentName Ident
| ValOpName (OpName 'ValueOpName)
| TyName (ProperName 'TypeName)
| TyOpName (OpName 'TypeOpName)
| DctorName (ProperName 'ConstructorName)
| TyClassName (ProperName 'ClassName)
| ModName ModuleName
deriving (Eq, Show)
getIdentName :: Name -> Maybe Ident
getIdentName (IdentName name) = Just name
getIdentName _ = Nothing
getValOpName :: Name -> Maybe (OpName 'ValueOpName)
getValOpName (ValOpName name) = Just name
getValOpName _ = Nothing
getTypeName :: Name -> Maybe (ProperName 'TypeName)
getTypeName (TyName name) = Just name
getTypeName _ = Nothing
getTypeOpName :: Name -> Maybe (OpName 'TypeOpName)
getTypeOpName (TyOpName name) = Just name
getTypeOpName _ = Nothing
getDctorName :: Name -> Maybe (ProperName 'ConstructorName)
getDctorName (DctorName name) = Just name
getDctorName _ = Nothing
getClassName :: Name -> Maybe (ProperName 'ClassName)
getClassName (TyClassName name) = Just name
getClassName _ = Nothing
getModName :: Name -> Maybe ModuleName
getModName (ModName name) = Just name
getModName _ = Nothing
data Ident
= Ident String
| GenIdent (Maybe String) Integer
deriving (Show, Eq, Ord)
runIdent :: Ident -> String
runIdent (Ident i) = i
runIdent (GenIdent Nothing n) = "$" ++ show n
runIdent (GenIdent (Just name) n) = "$" ++ name ++ show n
showIdent :: Ident -> String
showIdent = runIdent
freshIdent :: MonadSupply m => String -> m Ident
freshIdent name = GenIdent (Just name) <$> fresh
freshIdent' :: MonadSupply m => m Ident
freshIdent' = GenIdent Nothing <$> fresh
newtype OpName (a :: OpNameType) = OpName { runOpName :: String }
deriving (Show, Eq, Ord)
instance ToJSON (OpName a) where
toJSON = toJSON . runOpName
instance FromJSON (OpName a) where
parseJSON = fmap OpName . parseJSON
showOp :: OpName a -> String
showOp op = '(' : runOpName op ++ ")"
data OpNameType = ValueOpName | TypeOpName
newtype ProperName (a :: ProperNameType) = ProperName { runProperName :: String }
deriving (Show, Eq, Ord)
instance ToJSON (ProperName a) where
toJSON = toJSON . runProperName
instance FromJSON (ProperName a) where
parseJSON = fmap ProperName . parseJSON
data ProperNameType = TypeName | ConstructorName | ClassName | Namespace
coerceProperName :: ProperName a -> ProperName b
coerceProperName = ProperName . runProperName
newtype ModuleName = ModuleName [ProperName 'Namespace]
deriving (Show, Eq, Ord)
runModuleName :: ModuleName -> String
runModuleName (ModuleName pns) = intercalate "." (runProperName `map` pns)
moduleNameFromString :: String -> ModuleName
moduleNameFromString = ModuleName . splitProperNames
where
splitProperNames s = case dropWhile (== '.') s of
"" -> []
s' -> ProperName w : splitProperNames s''
where (w, s'') = break (== '.') s'
data Qualified a = Qualified (Maybe ModuleName) a
deriving (Show, Eq, Ord, Functor)
showQualified :: (a -> String) -> Qualified a -> String
showQualified f (Qualified Nothing a) = f a
showQualified f (Qualified (Just name) a) = runModuleName name ++ "." ++ f a
getQual :: Qualified a -> Maybe ModuleName
getQual (Qualified mn _) = mn
qualify :: ModuleName -> Qualified a -> (ModuleName, a)
qualify m (Qualified Nothing a) = (m, a)
qualify _ (Qualified (Just m) a) = (m, a)
mkQualified :: a -> ModuleName -> Qualified a
mkQualified name mn = Qualified (Just mn) name
disqualify :: Qualified a -> a
disqualify (Qualified _ a) = a
disqualifyFor :: Maybe ModuleName -> Qualified a -> Maybe a
disqualifyFor mn (Qualified mn' a) | mn == mn' = Just a
disqualifyFor _ _ = Nothing
isQualified :: Qualified a -> Bool
isQualified (Qualified Nothing _) = False
isQualified _ = True
isUnqualified :: Qualified a -> Bool
isUnqualified = not . isQualified
isQualifiedWith :: ModuleName -> Qualified a -> Bool
isQualifiedWith mn (Qualified (Just mn') _) = mn == mn'
isQualifiedWith _ _ = False
$(deriveJSON (defaultOptions { sumEncoding = ObjectWithSingleField }) ''Qualified)
$(deriveJSON (defaultOptions { sumEncoding = ObjectWithSingleField }) ''Ident)
$(deriveJSON (defaultOptions { sumEncoding = ObjectWithSingleField }) ''ModuleName)