module Language.PureScript.Names where
import Data.List
import Data.Data
import Data.List.Split (splitOn)
import qualified Data.Aeson as A
import qualified Data.Text as T
data Ident
= Ident String
| Op String deriving (Eq, Ord, Data, Typeable)
runIdent :: Ident -> String
runIdent (Ident i) = i
runIdent (Op op) = op
instance Show Ident where
show (Ident s) = s
show (Op op) = '(':op ++ ")"
newtype ProperName = ProperName { runProperName :: String } deriving (Eq, Ord, Data, Typeable)
instance Show ProperName where
show = runProperName
data ModuleName = ModuleName [ProperName] deriving (Eq, Ord, Data, Typeable)
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'
instance Show ModuleName where
show = runModuleName
data Qualified a = Qualified (Maybe ModuleName) a deriving (Eq, Ord, Data, Typeable, Functor)
instance (Show a) => Show (Qualified a) where
show (Qualified Nothing a) = show a
show (Qualified (Just name) a) = show name ++ "." ++ show a
instance (a ~ ProperName) => A.ToJSON (Qualified a) where
toJSON = A.toJSON . show
instance (a ~ ProperName) => A.FromJSON (Qualified a) where
parseJSON =
A.withText "Qualified ProperName" $ \str ->
return $ case reverse (splitOn "." (T.unpack str)) of
[name] -> Qualified Nothing (ProperName name)
(name:rest) -> Qualified (Just (reconstructModuleName rest)) (ProperName name)
_ -> Qualified Nothing (ProperName "")
where
reconstructModuleName = moduleNameFromString . intercalate "." . reverse
qualify :: ModuleName -> Qualified a -> (ModuleName, a)
qualify m (Qualified Nothing a) = (m, a)
qualify _ (Qualified (Just m) a) = (m, a)