{-# LANGUAGE CPP #-}
{-# LANGUAGE GADTs #-}
{-# LANGUAGE NamedFieldPuns #-}
{-# LANGUAGE TemplateHaskellQuotes #-}
{-# LANGUAGE ViewPatterns #-}

module Language.Haskell.Meta.Translate (toExp) where

#if MIN_VERSION_ghc(9,2,0)
import GHC.Hs.Type (HsWildCardBndrs (..), HsType (..), HsSigType(HsSig), sig_body)
#elif MIN_VERSION_ghc(9,0,0)
import GHC.Hs.Type (HsWildCardBndrs (..), HsType (..), HsImplicitBndrs(HsIB), hsib_body)
#elif MIN_VERSION_ghc(8,10,0)
import GHC.Hs.Types (HsWildCardBndrs (..), HsType (..), HsImplicitBndrs (HsIB, hsib_body))
#else
import HsTypes (HsWildCardBndrs (..), HsType (..), HsImplicitBndrs (HsIB), hsib_body)
#endif

#if MIN_VERSION_ghc(8,10,0)
import GHC.Hs.Expr as Expr
import GHC.Hs.Extension as Ext
import GHC.Hs.Pat as Pat
import GHC.Hs.Lit
#else
import HsExpr as Expr
import HsExtension as Ext
import HsPat as Pat
import HsLit
#endif

import qualified Data.ByteString as B
import qualified Language.Haskell.TH.Syntax as GhcTH
import qualified Language.Haskell.TH.Syntax as TH

#if MIN_VERSION_ghc(9,0,0)
import GHC.Types.SrcLoc
import GHC.Types.Name
import GHC.Types.Name.Reader
import GHC.Data.FastString
#if MIN_VERSION_ghc(9,2,0)
import GHC.Utils.Outputable (ppr)
import GHC.Types.Basic (Boxity(..))
import GHC.Types.SourceText (il_value, rationalFromFractionalLit)
import GHC.Driver.Ppr (showSDocDebug)
#else
import GHC.Utils.Outputable (ppr, showSDocDebug)
import GHC.Types.Basic (il_value, fl_value, Boxity(..))
#endif
import GHC.Driver.Session (DynFlags, xopt_set, defaultDynFlags)
import qualified GHC.Unit.Module as Module
#else
import SrcLoc
import Name
import RdrName
import FastString
import Outputable (ppr, showSDocDebug)
import BasicTypes (il_value, fl_value, Boxity(..))
import DynFlags (DynFlags, xopt_set, defaultDynFlags)
import qualified Module
#endif

import GHC.Stack
import qualified Language.Haskell.Meta.Settings as Settings

import qualified Data.List.NonEmpty as NonEmpty

#if MIN_VERSION_ghc(9,2,0)
-- TODO: why this disapears in GHC >= 9.2?
fl_value = rationalFromFractionalLit
#endif



-----------------------------

toLit :: HsLit GhcPs -> TH.Lit
toLit :: HsLit GhcPs -> Lit
toLit (HsChar XHsChar GhcPs
_ Char
c) = Char -> Lit
TH.CharL Char
c
toLit (HsCharPrim XHsCharPrim GhcPs
_ Char
c) = Char -> Lit
TH.CharPrimL Char
c
toLit (HsString XHsString GhcPs
_ FastString
s) = String -> Lit
TH.StringL (FastString -> String
unpackFS FastString
s)
toLit (HsStringPrim XHsStringPrim GhcPs
_ ByteString
s) = [Word8] -> Lit
TH.StringPrimL (ByteString -> [Word8]
B.unpack ByteString
s)
toLit (HsInt XHsInt GhcPs
_ IntegralLit
i) = Integer -> Lit
TH.IntegerL (IntegralLit -> Integer
il_value IntegralLit
i)
toLit (HsIntPrim XHsIntPrim GhcPs
_ Integer
i) = Integer -> Lit
TH.IntPrimL Integer
i
toLit (HsWordPrim XHsWordPrim GhcPs
_ Integer
i) = Integer -> Lit
TH.WordPrimL Integer
i
toLit (HsInt64Prim XHsInt64Prim GhcPs
_ Integer
i) = Integer -> Lit
TH.IntegerL Integer
i
toLit (HsWord64Prim XHsWord64Prim GhcPs
_ Integer
i) = Integer -> Lit
TH.WordPrimL Integer
i
toLit (HsInteger XHsInteger GhcPs
_ Integer
i Type
_) = Integer -> Lit
TH.IntegerL Integer
i
toLit (HsRat XHsRat GhcPs
_ FractionalLit
f Type
_) = Rational -> Lit
TH.FloatPrimL (FractionalLit -> Rational
fl_value FractionalLit
f)
toLit (HsFloatPrim XHsFloatPrim GhcPs
_ FractionalLit
f) = Rational -> Lit
TH.FloatPrimL (FractionalLit -> Rational
fl_value FractionalLit
f)
toLit (HsDoublePrim XHsDoublePrim GhcPs
_ FractionalLit
f) = Rational -> Lit
TH.DoublePrimL (FractionalLit -> Rational
fl_value FractionalLit
f)

#if !MIN_VERSION_ghc(9,0,0)
toLit (XLit _) = noTH "toLit" "XLit"
#endif

toLit' :: OverLitVal -> TH.Lit
toLit' :: OverLitVal -> Lit
toLit' (HsIntegral IntegralLit
i) = Integer -> Lit
TH.IntegerL (IntegralLit -> Integer
il_value IntegralLit
i)
toLit' (HsFractional FractionalLit
f) = Rational -> Lit
TH.RationalL (FractionalLit -> Rational
fl_value FractionalLit
f)
toLit' (HsIsString SourceText
_ FastString
fs) = String -> Lit
TH.StringL (FastString -> String
unpackFS FastString
fs)

toType :: HsType GhcPs -> TH.Type
toType :: HsType GhcPs -> Type
toType (HsWildCardTy XWildCardTy GhcPs
_) = Type
TH.WildCardT
toType (HsTyVar XTyVar GhcPs
_ PromotionFlag
_ Located (IdP GhcPs)
n) =
  let n' :: RdrName
n' = GenLocated SrcSpan RdrName -> RdrName
forall l e. GenLocated l e -> e
unLoc Located (IdP GhcPs)
GenLocated SrcSpan RdrName
n
   in if RdrName -> Bool
isRdrTyVar RdrName
n'
        then Name -> Type
TH.VarT (RdrName -> Name
toName RdrName
n')
        else Name -> Type
TH.ConT (RdrName -> Name
toName RdrName
n')
toType HsType GhcPs
t = String -> String -> Type
forall e a. (HasCallStack, Show e) => String -> e -> a
todo String
"toType" (DynFlags -> SDoc -> String
showSDocDebug ([Extension] -> DynFlags
Settings.baseDynFlags []) (SDoc -> String)
-> (HsType GhcPs -> SDoc) -> HsType GhcPs -> String
forall b c a. (b -> c) -> (a -> b) -> a -> c
. HsType GhcPs -> SDoc
forall a. Outputable a => a -> SDoc
ppr (HsType GhcPs -> String) -> HsType GhcPs -> String
forall a b. (a -> b) -> a -> b
$ HsType GhcPs
t)

toName :: RdrName -> TH.Name
toName :: RdrName -> Name
toName RdrName
n = case RdrName
n of
  (Unqual OccName
o) -> String -> Name
TH.mkName (OccName -> String
occNameString OccName
o)
  (Qual ModuleName
m OccName
o) -> String -> Name
TH.mkName (ModuleName -> String
Module.moduleNameString ModuleName
m String -> String -> String
forall a. Semigroup a => a -> a -> a
<> String
"." String -> String -> String
forall a. Semigroup a => a -> a -> a
<> OccName -> String
occNameString OccName
o)
  (Orig Module
_ OccName
_) -> String -> Name
forall a. HasCallStack => String -> a
error String
"orig"
  (Exact Name
_) -> String -> Name
forall a. HasCallStack => String -> a
error String
"exact"

toFieldExp :: a
toFieldExp :: forall a. a
toFieldExp = a
forall a. HasCallStack => a
undefined

toPat :: DynFlags -> Pat.Pat GhcPs -> TH.Pat
toPat :: DynFlags -> Pat GhcPs -> Pat
toPat DynFlags
_dynFlags (Pat.VarPat XVarPat GhcPs
_ (Located (IdP GhcPs) -> RdrName
forall l e. GenLocated l e -> e
unLoc -> RdrName
name)) = Name -> Pat
TH.VarP (RdrName -> Name
toName RdrName
name)
toPat DynFlags
dynFlags Pat GhcPs
p = String -> String -> Pat
forall e a. (HasCallStack, Show e) => String -> e -> a
todo String
"toPat" (DynFlags -> SDoc -> String
showSDocDebug DynFlags
dynFlags (SDoc -> String) -> (Pat GhcPs -> SDoc) -> Pat GhcPs -> String
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Pat GhcPs -> SDoc
forall a. Outputable a => a -> SDoc
ppr (Pat GhcPs -> String) -> Pat GhcPs -> String
forall a b. (a -> b) -> a -> b
$ Pat GhcPs
p)

toExp :: DynFlags -> Expr.HsExpr GhcPs -> TH.Exp
toExp :: DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
_ (Expr.HsVar XVar GhcPs
_ Located (IdP GhcPs)
n) =
  let n' :: RdrName
n' = GenLocated SrcSpan RdrName -> RdrName
forall l e. GenLocated l e -> e
unLoc Located (IdP GhcPs)
GenLocated SrcSpan RdrName
n
   in if RdrName -> Bool
isRdrDataCon RdrName
n'
        then Name -> Exp
TH.ConE (RdrName -> Name
toName RdrName
n')
        else Name -> Exp
TH.VarE (RdrName -> Name
toName RdrName
n')

#if MIN_VERSION_ghc(9,0,0)
toExp DynFlags
_ (Expr.HsUnboundVar XUnboundVar GhcPs
_ OccName
n)              = Name -> Exp
TH.UnboundVarE (String -> Name
TH.mkName (String -> Name) -> (OccName -> String) -> OccName -> Name
forall b c a. (b -> c) -> (a -> b) -> a -> c
. OccName -> String
occNameString (OccName -> Name) -> OccName -> Name
forall a b. (a -> b) -> a -> b
$ OccName
n)
#else
toExp _ (Expr.HsUnboundVar _ n)              = TH.UnboundVarE (TH.mkName . occNameString . Expr.unboundVarOcc $ n)
#endif

toExp DynFlags
_ Expr.HsIPVar {}
  = String -> String -> Exp
forall e a. (HasCallStack, Show e) => String -> e -> a
noTH String
"toExp" String
"HsIPVar"

toExp DynFlags
_ (Expr.HsLit XLitE GhcPs
_ HsLit GhcPs
l)
  = Lit -> Exp
TH.LitE (HsLit GhcPs -> Lit
toLit HsLit GhcPs
l)

toExp DynFlags
_ (Expr.HsOverLit XOverLitE GhcPs
_ OverLit {OverLitVal
ol_val :: forall p. HsOverLit p -> OverLitVal
ol_val :: OverLitVal
ol_val})
  = Lit -> Exp
TH.LitE (OverLitVal -> Lit
toLit' OverLitVal
ol_val)

toExp DynFlags
d (Expr.HsApp XApp GhcPs
_ LHsExpr GhcPs
e1 LHsExpr GhcPs
e2)
  = Exp -> Exp -> Exp
TH.AppE (DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
d (HsExpr GhcPs -> Exp)
-> (LHsExpr GhcPs -> HsExpr GhcPs) -> LHsExpr GhcPs -> Exp
forall b c a. (b -> c) -> (a -> b) -> a -> c
. LHsExpr GhcPs -> HsExpr GhcPs
forall l e. GenLocated l e -> e
unLoc (LHsExpr GhcPs -> Exp) -> LHsExpr GhcPs -> Exp
forall a b. (a -> b) -> a -> b
$ LHsExpr GhcPs
e1) (DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
d (HsExpr GhcPs -> Exp)
-> (LHsExpr GhcPs -> HsExpr GhcPs) -> LHsExpr GhcPs -> Exp
forall b c a. (b -> c) -> (a -> b) -> a -> c
. LHsExpr GhcPs -> HsExpr GhcPs
forall l e. GenLocated l e -> e
unLoc (LHsExpr GhcPs -> Exp) -> LHsExpr GhcPs -> Exp
forall a b. (a -> b) -> a -> b
$ LHsExpr GhcPs
e2)

#if MIN_VERSION_ghc(9,2,0)
toExp d (Expr.HsAppType _ e HsWC {hswc_body}) = TH.AppTypeE (toExp d . unLoc $ e) (toType . unLoc $ hswc_body)
toExp d (Expr.ExprWithTySig _ e HsWC{hswc_body=unLoc -> HsSig{sig_body}}) = TH.SigE (toExp d . unLoc $ e) (toType . unLoc $ sig_body)
#elif MIN_VERSION_ghc(8,8,0)
toExp DynFlags
d (Expr.HsAppType XAppTypeE GhcPs
_ LHsExpr GhcPs
e HsWC {LHsType (NoGhcTc GhcPs)
hswc_body :: forall pass thing. HsWildCardBndrs pass thing -> thing
hswc_body :: LHsType (NoGhcTc GhcPs)
hswc_body}) = Exp -> Type -> Exp
TH.AppTypeE (DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
d (HsExpr GhcPs -> Exp)
-> (LHsExpr GhcPs -> HsExpr GhcPs) -> LHsExpr GhcPs -> Exp
forall b c a. (b -> c) -> (a -> b) -> a -> c
. LHsExpr GhcPs -> HsExpr GhcPs
forall l e. GenLocated l e -> e
unLoc (LHsExpr GhcPs -> Exp) -> LHsExpr GhcPs -> Exp
forall a b. (a -> b) -> a -> b
$ LHsExpr GhcPs
e) (HsType GhcPs -> Type
toType (HsType GhcPs -> Type)
-> (GenLocated SrcSpan (HsType GhcPs) -> HsType GhcPs)
-> GenLocated SrcSpan (HsType GhcPs)
-> Type
forall b c a. (b -> c) -> (a -> b) -> a -> c
. GenLocated SrcSpan (HsType GhcPs) -> HsType GhcPs
forall l e. GenLocated l e -> e
unLoc (GenLocated SrcSpan (HsType GhcPs) -> Type)
-> GenLocated SrcSpan (HsType GhcPs) -> Type
forall a b. (a -> b) -> a -> b
$ LHsType (NoGhcTc GhcPs)
GenLocated SrcSpan (HsType GhcPs)
hswc_body)
toExp DynFlags
d (Expr.ExprWithTySig XExprWithTySig GhcPs
_ LHsExpr GhcPs
e HsWC{hswc_body :: forall pass thing. HsWildCardBndrs pass thing -> thing
hswc_body=HsIB{LHsType (NoGhcTc GhcPs)
hsib_body :: LHsType (NoGhcTc GhcPs)
hsib_body :: forall pass thing. HsImplicitBndrs pass thing -> thing
hsib_body}}) = Exp -> Type -> Exp
TH.SigE (DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
d (HsExpr GhcPs -> Exp)
-> (LHsExpr GhcPs -> HsExpr GhcPs) -> LHsExpr GhcPs -> Exp
forall b c a. (b -> c) -> (a -> b) -> a -> c
. LHsExpr GhcPs -> HsExpr GhcPs
forall l e. GenLocated l e -> e
unLoc (LHsExpr GhcPs -> Exp) -> LHsExpr GhcPs -> Exp
forall a b. (a -> b) -> a -> b
$ LHsExpr GhcPs
e) (HsType GhcPs -> Type
toType (HsType GhcPs -> Type)
-> (GenLocated SrcSpan (HsType GhcPs) -> HsType GhcPs)
-> GenLocated SrcSpan (HsType GhcPs)
-> Type
forall b c a. (b -> c) -> (a -> b) -> a -> c
. GenLocated SrcSpan (HsType GhcPs) -> HsType GhcPs
forall l e. GenLocated l e -> e
unLoc (GenLocated SrcSpan (HsType GhcPs) -> Type)
-> GenLocated SrcSpan (HsType GhcPs) -> Type
forall a b. (a -> b) -> a -> b
$ LHsType (NoGhcTc GhcPs)
GenLocated SrcSpan (HsType GhcPs)
hsib_body)
#else
toExp d (Expr.HsAppType HsWC {hswc_body} e) = TH.AppTypeE (toExp d . unLoc $ e) (toType . unLoc $ hswc_body)
#endif

toExp DynFlags
d (Expr.OpApp XOpApp GhcPs
_ LHsExpr GhcPs
e1 LHsExpr GhcPs
o LHsExpr GhcPs
e2)
  = Exp -> Exp -> Exp -> Exp
TH.UInfixE (DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
d (HsExpr GhcPs -> Exp)
-> (LHsExpr GhcPs -> HsExpr GhcPs) -> LHsExpr GhcPs -> Exp
forall b c a. (b -> c) -> (a -> b) -> a -> c
. LHsExpr GhcPs -> HsExpr GhcPs
forall l e. GenLocated l e -> e
unLoc (LHsExpr GhcPs -> Exp) -> LHsExpr GhcPs -> Exp
forall a b. (a -> b) -> a -> b
$ LHsExpr GhcPs
e1) (DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
d (HsExpr GhcPs -> Exp)
-> (LHsExpr GhcPs -> HsExpr GhcPs) -> LHsExpr GhcPs -> Exp
forall b c a. (b -> c) -> (a -> b) -> a -> c
. LHsExpr GhcPs -> HsExpr GhcPs
forall l e. GenLocated l e -> e
unLoc (LHsExpr GhcPs -> Exp) -> LHsExpr GhcPs -> Exp
forall a b. (a -> b) -> a -> b
$ LHsExpr GhcPs
o) (DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
d (HsExpr GhcPs -> Exp)
-> (LHsExpr GhcPs -> HsExpr GhcPs) -> LHsExpr GhcPs -> Exp
forall b c a. (b -> c) -> (a -> b) -> a -> c
. LHsExpr GhcPs -> HsExpr GhcPs
forall l e. GenLocated l e -> e
unLoc (LHsExpr GhcPs -> Exp) -> LHsExpr GhcPs -> Exp
forall a b. (a -> b) -> a -> b
$ LHsExpr GhcPs
e2)

toExp DynFlags
d (Expr.NegApp XNegApp GhcPs
_ LHsExpr GhcPs
e SyntaxExpr GhcPs
_)
  = Exp -> Exp -> Exp
TH.AppE (Name -> Exp
TH.VarE 'negate) (DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
d (HsExpr GhcPs -> Exp)
-> (LHsExpr GhcPs -> HsExpr GhcPs) -> LHsExpr GhcPs -> Exp
forall b c a. (b -> c) -> (a -> b) -> a -> c
. LHsExpr GhcPs -> HsExpr GhcPs
forall l e. GenLocated l e -> e
unLoc (LHsExpr GhcPs -> Exp) -> LHsExpr GhcPs -> Exp
forall a b. (a -> b) -> a -> b
$ LHsExpr GhcPs
e)

-- NOTE: for lambda, there is only one match
toExp DynFlags
d (Expr.HsLam XLam GhcPs
_ (Expr.MG XMG GhcPs (LHsExpr GhcPs)
_ (Located [LMatch GhcPs (LHsExpr GhcPs)]
-> [LMatch GhcPs (LHsExpr GhcPs)]
forall l e. GenLocated l e -> e
unLoc -> ((LMatch GhcPs (LHsExpr GhcPs) -> Match GhcPs (LHsExpr GhcPs))
-> [LMatch GhcPs (LHsExpr GhcPs)] -> [Match GhcPs (LHsExpr GhcPs)]
forall a b. (a -> b) -> [a] -> [b]
map LMatch GhcPs (LHsExpr GhcPs) -> Match GhcPs (LHsExpr GhcPs)
forall l e. GenLocated l e -> e
unLoc -> [Expr.Match XCMatch GhcPs (LHsExpr GhcPs)
_ HsMatchContext (NoGhcTc GhcPs)
_ ((GenLocated SrcSpan (Pat GhcPs) -> Pat GhcPs)
-> [GenLocated SrcSpan (Pat GhcPs)] -> [Pat GhcPs]
forall a b. (a -> b) -> [a] -> [b]
map GenLocated SrcSpan (Pat GhcPs) -> Pat GhcPs
forall l e. GenLocated l e -> e
unLoc -> [Pat GhcPs]
ps) (Expr.GRHSs XCGRHSs GhcPs (LHsExpr GhcPs)
_ [LGRHS GhcPs (LHsExpr GhcPs) -> GRHS GhcPs (LHsExpr GhcPs)
forall l e. GenLocated l e -> e
unLoc -> Expr.GRHS XCGRHS GhcPs (LHsExpr GhcPs)
_ [GuardLStmt GhcPs]
_ (LHsExpr GhcPs -> HsExpr GhcPs
forall l e. GenLocated l e -> e
unLoc -> HsExpr GhcPs
e)] LHsLocalBinds GhcPs
_)])) Origin
_))
  = [Pat] -> Exp -> Exp
TH.LamE ((Pat GhcPs -> Pat) -> [Pat GhcPs] -> [Pat]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap (DynFlags -> Pat GhcPs -> Pat
toPat DynFlags
d) [Pat GhcPs]
ps) (DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
d HsExpr GhcPs
e)

-- toExp (Expr.Let _ bs e)                       = TH.LetE (toDecs bs) (toExp e)
--
#if MIN_VERSION_ghc(9, 0, 0)
toExp DynFlags
d (Expr.HsIf XIf GhcPs
_ LHsExpr GhcPs
a LHsExpr GhcPs
b LHsExpr GhcPs
c)                   = Exp -> Exp -> Exp -> Exp
TH.CondE (DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
d (LHsExpr GhcPs -> HsExpr GhcPs
forall l e. GenLocated l e -> e
unLoc LHsExpr GhcPs
a)) (DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
d (LHsExpr GhcPs -> HsExpr GhcPs
forall l e. GenLocated l e -> e
unLoc LHsExpr GhcPs
b)) (DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
d (LHsExpr GhcPs -> HsExpr GhcPs
forall l e. GenLocated l e -> e
unLoc LHsExpr GhcPs
c))
#else
toExp d (Expr.HsIf _ _ a b c)                   = TH.CondE (toExp d (unLoc a)) (toExp d (unLoc b)) (toExp d (unLoc c))
#endif

-- toExp (Expr.MultiIf _ ifs)                    = TH.MultiIfE (map toGuard ifs)
-- toExp (Expr.Case _ e alts)                    = TH.CaseE (toExp e) (map toMatch alts)
-- toExp (Expr.Do _ ss)                          = TH.DoE (map toStmt ss)
-- toExp e@Expr.MDo{}                            = noTH "toExp" e
--
#if MIN_VERSION_ghc(9, 2, 0)
toExp d (Expr.ExplicitTuple _ args boxity) = ctor tupArgs
#else
toExp DynFlags
d (Expr.ExplicitTuple XExplicitTuple GhcPs
_ ((LHsTupArg GhcPs -> HsTupArg GhcPs)
-> [LHsTupArg GhcPs] -> [HsTupArg GhcPs]
forall a b. (a -> b) -> [a] -> [b]
map LHsTupArg GhcPs -> HsTupArg GhcPs
forall l e. GenLocated l e -> e
unLoc -> [HsTupArg GhcPs]
args) Boxity
boxity) = [Maybe Exp] -> Exp
ctor [Maybe Exp]
tupArgs
#endif
  where
    toTupArg :: HsTupArg id -> Maybe (HsExpr id)
toTupArg (Expr.Present XPresent id
_ LHsExpr id
e) = HsExpr id -> Maybe (HsExpr id)
forall a. a -> Maybe a
Just (HsExpr id -> Maybe (HsExpr id)) -> HsExpr id -> Maybe (HsExpr id)
forall a b. (a -> b) -> a -> b
$ LHsExpr id -> HsExpr id
forall l e. GenLocated l e -> e
unLoc LHsExpr id
e
    toTupArg (Expr.Missing XMissing id
_) = Maybe (HsExpr id)
forall a. Maybe a
Nothing
    toTupArg HsTupArg id
_ = String -> Maybe (HsExpr id)
forall a. HasCallStack => String -> a
error String
"impossible case"

    ctor :: [Maybe Exp] -> Exp
ctor = case Boxity
boxity of
      Boxity
Boxed -> [Maybe Exp] -> Exp
TH.TupE
      Boxity
Unboxed -> [Maybe Exp] -> Exp
TH.UnboxedTupE

#if MIN_VERSION_ghc(8,10,0)
    tupArgs :: [Maybe Exp]
tupArgs = (HsTupArg GhcPs -> Maybe Exp) -> [HsTupArg GhcPs] -> [Maybe Exp]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap (((HsExpr GhcPs -> Exp) -> Maybe (HsExpr GhcPs) -> Maybe Exp
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap (DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
d)) (Maybe (HsExpr GhcPs) -> Maybe Exp)
-> (HsTupArg GhcPs -> Maybe (HsExpr GhcPs))
-> HsTupArg GhcPs
-> Maybe Exp
forall b c a. (b -> c) -> (a -> b) -> a -> c
. HsTupArg GhcPs -> Maybe (HsExpr GhcPs)
forall {id}. HsTupArg id -> Maybe (HsExpr id)
toTupArg) [HsTupArg GhcPs]
args
#else
    tupArgs = case traverse toTupArg args of
      Nothing -> error "Tuple section are not supported by template haskell < 8.10"
      Just args -> fmap (toExp d) args
#endif

-- toExp (Expr.List _ xs)                        = TH.ListE (fmap toExp xs)
toExp DynFlags
d (Expr.HsPar XPar GhcPs
_ LHsExpr GhcPs
e)
  = Exp -> Exp
TH.ParensE (DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
d (HsExpr GhcPs -> Exp)
-> (LHsExpr GhcPs -> HsExpr GhcPs) -> LHsExpr GhcPs -> Exp
forall b c a. (b -> c) -> (a -> b) -> a -> c
. LHsExpr GhcPs -> HsExpr GhcPs
forall l e. GenLocated l e -> e
unLoc (LHsExpr GhcPs -> Exp) -> LHsExpr GhcPs -> Exp
forall a b. (a -> b) -> a -> b
$ LHsExpr GhcPs
e)

toExp DynFlags
d (Expr.SectionL XSectionL GhcPs
_ (LHsExpr GhcPs -> HsExpr GhcPs
forall l e. GenLocated l e -> e
unLoc -> HsExpr GhcPs
a) (LHsExpr GhcPs -> HsExpr GhcPs
forall l e. GenLocated l e -> e
unLoc -> HsExpr GhcPs
b))
  = Maybe Exp -> Exp -> Maybe Exp -> Exp
TH.InfixE (Exp -> Maybe Exp
forall a. a -> Maybe a
Just (Exp -> Maybe Exp)
-> (HsExpr GhcPs -> Exp) -> HsExpr GhcPs -> Maybe Exp
forall b c a. (b -> c) -> (a -> b) -> a -> c
. DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
d (HsExpr GhcPs -> Maybe Exp) -> HsExpr GhcPs -> Maybe Exp
forall a b. (a -> b) -> a -> b
$ HsExpr GhcPs
a) (DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
d HsExpr GhcPs
b) Maybe Exp
forall a. Maybe a
Nothing

toExp DynFlags
d (Expr.SectionR XSectionR GhcPs
_ (LHsExpr GhcPs -> HsExpr GhcPs
forall l e. GenLocated l e -> e
unLoc -> HsExpr GhcPs
a) (LHsExpr GhcPs -> HsExpr GhcPs
forall l e. GenLocated l e -> e
unLoc -> HsExpr GhcPs
b))
  = Maybe Exp -> Exp -> Maybe Exp -> Exp
TH.InfixE Maybe Exp
forall a. Maybe a
Nothing (DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
d HsExpr GhcPs
a) (Exp -> Maybe Exp
forall a. a -> Maybe a
Just (Exp -> Maybe Exp)
-> (HsExpr GhcPs -> Exp) -> HsExpr GhcPs -> Maybe Exp
forall b c a. (b -> c) -> (a -> b) -> a -> c
. DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
d (HsExpr GhcPs -> Maybe Exp) -> HsExpr GhcPs -> Maybe Exp
forall a b. (a -> b) -> a -> b
$ HsExpr GhcPs
b)

toExp DynFlags
_ (Expr.RecordCon XRecordCon GhcPs
_ Located (IdP GhcPs)
name HsRecFields {[LHsRecField GhcPs (LHsExpr GhcPs)]
rec_flds :: forall p arg. HsRecFields p arg -> [LHsRecField p arg]
rec_flds :: [LHsRecField GhcPs (LHsExpr GhcPs)]
rec_flds})
  = Name -> [FieldExp] -> Exp
TH.RecConE (RdrName -> Name
toName (RdrName -> Name)
-> (GenLocated SrcSpan RdrName -> RdrName)
-> GenLocated SrcSpan RdrName
-> Name
forall b c a. (b -> c) -> (a -> b) -> a -> c
. GenLocated SrcSpan RdrName -> RdrName
forall l e. GenLocated l e -> e
unLoc (GenLocated SrcSpan RdrName -> Name)
-> GenLocated SrcSpan RdrName -> Name
forall a b. (a -> b) -> a -> b
$ Located (IdP GhcPs)
GenLocated SrcSpan RdrName
name) ((LHsRecField GhcPs (LHsExpr GhcPs) -> FieldExp)
-> [LHsRecField GhcPs (LHsExpr GhcPs)] -> [FieldExp]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap LHsRecField GhcPs (LHsExpr GhcPs) -> FieldExp
forall a. a
toFieldExp [LHsRecField GhcPs (LHsExpr GhcPs)]
rec_flds)

-- toExp (Expr.RecUpdate _ e xs)                 = TH.RecUpdE (toExp e) (fmap toFieldExp xs)
-- toExp (Expr.ListComp _ e ss)                  = TH.CompE $ map convert ss ++ [TH.NoBindS (toExp e)]
--  where
--   convert (Expr.QualStmt _ st)                = toStmt st
--   convert s                                   = noTH "toExp ListComp" s
-- toExp (Expr.ExpTypeSig _ e t)                 = TH.SigE (toExp e) (toType t)
--
#if MIN_VERSION_ghc(9, 2, 0)
toExp d (Expr.ExplicitList _ (map unLoc -> args)) = TH.ListE (map (toExp d) args)
#else
toExp DynFlags
d (Expr.ExplicitList XExplicitList GhcPs
_ Maybe (SyntaxExpr GhcPs)
_ ((LHsExpr GhcPs -> HsExpr GhcPs)
-> [LHsExpr GhcPs] -> [HsExpr GhcPs]
forall a b. (a -> b) -> [a] -> [b]
map LHsExpr GhcPs -> HsExpr GhcPs
forall l e. GenLocated l e -> e
unLoc -> [HsExpr GhcPs]
args)) = [Exp] -> Exp
TH.ListE ((HsExpr GhcPs -> Exp) -> [HsExpr GhcPs] -> [Exp]
forall a b. (a -> b) -> [a] -> [b]
map (DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
d) [HsExpr GhcPs]
args)
#endif

toExp DynFlags
d (Expr.ArithSeq XArithSeq GhcPs
_ Maybe (SyntaxExpr GhcPs)
_ ArithSeqInfo GhcPs
e)
  = Range -> Exp
TH.ArithSeqE (Range -> Exp) -> Range -> Exp
forall a b. (a -> b) -> a -> b
$ case ArithSeqInfo GhcPs
e of
    (From LHsExpr GhcPs
a) -> Exp -> Range
TH.FromR (DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
d (HsExpr GhcPs -> Exp) -> HsExpr GhcPs -> Exp
forall a b. (a -> b) -> a -> b
$ LHsExpr GhcPs -> HsExpr GhcPs
forall l e. GenLocated l e -> e
unLoc LHsExpr GhcPs
a)
    (FromThen LHsExpr GhcPs
a LHsExpr GhcPs
b) -> Exp -> Exp -> Range
TH.FromThenR (DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
d (HsExpr GhcPs -> Exp) -> HsExpr GhcPs -> Exp
forall a b. (a -> b) -> a -> b
$ LHsExpr GhcPs -> HsExpr GhcPs
forall l e. GenLocated l e -> e
unLoc LHsExpr GhcPs
a) (DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
d (HsExpr GhcPs -> Exp) -> HsExpr GhcPs -> Exp
forall a b. (a -> b) -> a -> b
$ LHsExpr GhcPs -> HsExpr GhcPs
forall l e. GenLocated l e -> e
unLoc LHsExpr GhcPs
b)
    (FromTo LHsExpr GhcPs
a LHsExpr GhcPs
b) -> Exp -> Exp -> Range
TH.FromToR (DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
d (HsExpr GhcPs -> Exp) -> HsExpr GhcPs -> Exp
forall a b. (a -> b) -> a -> b
$ LHsExpr GhcPs -> HsExpr GhcPs
forall l e. GenLocated l e -> e
unLoc LHsExpr GhcPs
a) (DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
d (HsExpr GhcPs -> Exp) -> HsExpr GhcPs -> Exp
forall a b. (a -> b) -> a -> b
$ LHsExpr GhcPs -> HsExpr GhcPs
forall l e. GenLocated l e -> e
unLoc LHsExpr GhcPs
b)
    (FromThenTo LHsExpr GhcPs
a LHsExpr GhcPs
b LHsExpr GhcPs
c) -> Exp -> Exp -> Exp -> Range
TH.FromThenToR (DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
d (HsExpr GhcPs -> Exp) -> HsExpr GhcPs -> Exp
forall a b. (a -> b) -> a -> b
$ LHsExpr GhcPs -> HsExpr GhcPs
forall l e. GenLocated l e -> e
unLoc LHsExpr GhcPs
a) (DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
d (HsExpr GhcPs -> Exp) -> HsExpr GhcPs -> Exp
forall a b. (a -> b) -> a -> b
$ LHsExpr GhcPs -> HsExpr GhcPs
forall l e. GenLocated l e -> e
unLoc LHsExpr GhcPs
b) (DynFlags -> HsExpr GhcPs -> Exp
toExp DynFlags
d (HsExpr GhcPs -> Exp) -> HsExpr GhcPs -> Exp
forall a b. (a -> b) -> a -> b
$ LHsExpr GhcPs -> HsExpr GhcPs
forall l e. GenLocated l e -> e
unLoc LHsExpr GhcPs
c)

#if MIN_VERSION_ghc(9, 2, 0)
toExp _ (Expr.HsProjection _ locatedFields) =
  let
    extractFieldLabel (HsFieldLabel _ locatedStr) = locatedStr
    extractFieldLabel _ = error "Don't know how to handle XHsFieldLabel constructor..."
  in
    TH.ProjectionE (NonEmpty.map (unpackFS . unLoc . extractFieldLabel . unLoc) locatedFields)

toExp d (Expr.HsGetField _ expr locatedField) =
  let
    extractFieldLabel (HsFieldLabel _ locatedStr) = locatedStr
    extractFieldLabel _ = error "Don't know how to handle XHsFieldLabel constructor..."
  in
    TH.GetFieldE (toExp d (unLoc expr)) (unpackFS . unLoc . extractFieldLabel . unLoc $ locatedField)

toExp _ (Expr.HsOverLabel _ fastString) = TH.LabelE (unpackFS fastString)
#endif

toExp DynFlags
dynFlags HsExpr GhcPs
e = String -> String -> Exp
forall e a. (HasCallStack, Show e) => String -> e -> a
todo String
"toExp" (DynFlags -> SDoc -> String
showSDocDebug DynFlags
dynFlags (SDoc -> String)
-> (HsExpr GhcPs -> SDoc) -> HsExpr GhcPs -> String
forall b c a. (b -> c) -> (a -> b) -> a -> c
. HsExpr GhcPs -> SDoc
forall a. Outputable a => a -> SDoc
ppr (HsExpr GhcPs -> String) -> HsExpr GhcPs -> String
forall a b. (a -> b) -> a -> b
$ HsExpr GhcPs
e)

todo :: (HasCallStack, Show e) => String -> e -> a
todo :: forall e a. (HasCallStack, Show e) => String -> e -> a
todo String
fun e
thing = String -> a
forall a. HasCallStack => String -> a
error (String -> a) -> ([String] -> String) -> [String] -> a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. [String] -> String
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
concat ([String] -> a) -> [String] -> a
forall a b. (a -> b) -> a -> b
$ [String
moduleName, String
".", String
fun, String
": not implemented: ", e -> String
forall a. Show a => a -> String
show e
thing]

noTH :: (HasCallStack, Show e) => String -> e -> a
noTH :: forall e a. (HasCallStack, Show e) => String -> e -> a
noTH String
fun e
thing = String -> a
forall a. HasCallStack => String -> a
error (String -> a) -> ([String] -> String) -> [String] -> a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. [String] -> String
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
concat ([String] -> a) -> [String] -> a
forall a b. (a -> b) -> a -> b
$ [String
moduleName, String
".", String
fun, String
": no TemplateHaskell for: ", e -> String
forall a. Show a => a -> String
show e
thing]

moduleName :: String
moduleName :: String
moduleName = String
"Language.Haskell.Meta.Translate"