Last active
December 22, 2015 13:29
-
-
Save zsol/6479547 to your computer and use it in GitHub Desktop.
Generating function to parse plain data constructors without arguments.
Useful for @abesto and the SDL types
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
{-# LANGUAGE TemplateHaskell #-} | |
module Abesto(mkRead) where | |
import Language.Haskell.TH | |
import Data.Maybe | |
constructors :: Info -> Maybe [Name] | |
constructors (TyConI (DataD _cxt _name _bnds cons _names)) = Just $ catMaybes $ map getName cons | |
where getName (NormalC name _types) = Just name | |
getName _ = Nothing | |
constructors _ = Nothing | |
consToClause :: Name -> Q Clause | |
consToClause cons = do | |
lit <- litP $ stringL $ nameBase cons | |
body <- normalB $ appE [| Just |] (conE cons) | |
return $ Clause [lit] body [] | |
fallbackClause :: Q Clause | |
fallbackClause = do | |
pat <- wildP | |
body <- normalB $ [| Nothing |] | |
return $ Clause [pat] body [] | |
mkRead :: Name -> Q [Dec] | |
mkRead dataName = do | |
tyInfo <- reify dataName | |
let Just conss = constructors tyInfo | |
clauses = map consToClause conss | |
fun <- funD (mkName $ "read" ++ (nameBase dataName)) (clauses ++ [fallbackClause]) | |
return [fun] |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
{-# LANGUAGE TemplateHaskell #-} | |
module Main where | |
import Abesto | |
data Alma = Korte | Barack deriving (Show) | |
$(mkRead ''Alma) | |
instance Read Alma where | |
readsPrec _ x = case readAlma x of | |
Just alma -> [(alma, "")] | |
Nothing -> [] | |
korte :: Alma | |
korte = read "Korte" | |
main :: IO () | |
main = interact $ show . readAlma |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment