有没有一种简单的方法来引用具有约束参数的类型?

Is there an easy way to quote a type with constrained parameters?

> {-# LANGUAGE TemplateHaskell #-}
> import Language.Haskell.TH
> import Control.Monad

假设我有一个 class 像 Default

> class Default a where
>  def :: a

有一种直接的方法可以为也有 Monoid 实例的类型定义实例,例如

instance Monoid a => Default a where
  def = mempty

但由于存在重叠问题,并且为了使其更易于控制,我们可能会提供一个 TH 宏。

(是的,我知道我们可以使用-XDerivingVia,我对这里的这种解决方案不感兴趣。)

> makeMonoidDefault :: Q Type -> DecsQ
> makeMonoidDefault instT = sequence
>   [ InstanceD Nothing [] <$> [t| Default $instT |] <*> [d|
>         $(varP $ mkName "def") = mempty |] ]

这可以用引用类型轻松调用,例如

makeMonoidDefault [t| Maybe () |]

允许

*Main> def :: Maybe ()
Nothing

但这不允许像参数化实例这样的东西

instance Semigroup a => Default (Maybe a) where
  def = mempty

这可以用另一个宏来完成:

> makeMonoidDefault' :: Q (Cxt,Type) -> DecsQ
> makeMonoidDefault' cxtInstT = do
>  (cxt, instT) <- cxtInstT
>  sequence
>   [ InstanceD Nothing cxt <*> [t| Default $(pure instT) |] <*> [d|
>         $(varP $ mkName "def") = mempty |] ]

但是现在实际使用起来要尴尬得多:

makeMonoidDefault' $ do
   tParam <- pure . VarT <$> newName "a"
   sgcxt <- [t| Semigroup $tParam |]
   maybet <- [t| Maybe $tParam |]
   return ([sgcxt], maybet)

makeMonoidDefault' $ do
   tParam <- pure . VarT <$> newName "a"
   ((,) . (:[])) <$> [t| Semigroup $tParam |]
                 <*> [t| Maybe $tParam |]

我想出的一个 hacky 解决方案是这样的:

-- Quote.hs
{-# LANGUAGE TemplateHaskell #-}
module Quote where

import Language.Haskell.TH

class Default a where
  def :: a

makeMonoidDefault' :: Q Type -> DecsQ
makeMonoidDefault' q = do
  t <- q
  case t of
    ForallT _ cxt instT -> sequence
      [ InstanceD Nothing cxt <$> [t| Default $(pure instT) |] <*> [d|
            $(varP 'def) = mempty |] ]
    _ -> fail "<some nice error message>"

那么你可以这样使用它:

-- Main.hs
{-# LANGUAGE TemplateHaskell, ExplicitForAll #-}
import Quote

makeMonoidDefault' [t|forall a. Semigroup a => Maybe a|]

main = pure ()

一个选择是稍微改变一下设计。例如:

makeMonoidDefault [d| instance Semigroup a => Default (Maybe a) |]

这有点多余,因为它提到了两次 Default。 (你真的应该检查它是否确实声明了你期望的 class。)但从好的方面来说,它为你提供了很多你需要的结构作为参数的一部分。

基于@Noughtmare 的解决方案,您可以在引号括号中做更多的事情:

{-# LANGUAGE ConstraintKinds #-}

makeMonoidDefault :: Q Type -> DecsQ
makeMonoidDefault q = do
  t <- q
  let (ctx, instT) = case t of
        ForallT _ ctx instT -> (ctx, instT)
        _ -> ([], t)
  [d|
    instance $(pure $ foldCtx ctx) => Default $(pure instT) where
      def = mempty
    |]
  where
    foldCtx l = foldl AppT (TupleT $ length l) l

它可以通过以下任何一种方式使用:

makeMonoidDefault [t|forall a. Semigroup a => Maybe a|]
makeMonoidDefault [t|forall a b. (Monoid a, Monoid b) => (a,b)|]
makeMonoidDefault [t|Maybe ()|]

并且它不再有关于 def 的奇怪之处。 (顺便说一下,对于没有上下文的情况,需要使用 ConstraintKinds pragma,生成的上下文是 ()。)