module Data.DeriveTH(derive, derives, deriveFromDec, module Data.Derive.All) where
import Data.List
import Control.Monad
import Data.Derive.All
import Data.Derive.Internal.Derivation
import Language.Haskell.TH.All as TH hiding (Derivation,toName)
import Language.Haskell as HS
import Language.Haskell.Convert
derive :: Derivation -> TH.Name -> Q [Dec]
derive d name = do
x <- reify name
case x of
TyConI dec -> deriveFromDec d dec
_ -> error $ "Data.DeriveTH.derive: Expected a data type declaration, got:\n" ++ show x
derives :: [Derivation] -> [TH.Name] -> Q [Dec]
derives xs ys = liftM concat $ sequence [derive x y | y <- ys, x <- xs]
deriveFromDec :: Derivation -> Dec -> Q [Dec]
deriveFromDec d x = do
x <- liftM normData $ expandData x
case derivationOp d $ toFullDataDecl x of
Left y -> runIO (putStrLn $ "Warning, couldn't derive: " ++ y) >> return []
Right v -> return $ convert v
toFullDataDecl :: Dec -> FullDataDecl
toFullDataDecl x = (ModuleName "Todo", convert x)