Como trabalhar com AST com anotação Cofree?
Eu tenho esse simplesExpr
AST e eu podemos facilmente convertê-lo emString
.
import Prelude hiding (Foldable)
import qualified Prelude
import Data.Foldable as F
import Data.Functor.Foldable
import Data.Monoid
import Control.Comonad.Cofree
data ExprF r = Const Int
| Add r r
deriving ( Show, Eq, Ord, Functor, Prelude.Foldable )
type Expr = Fix ExprF
testExpr = Fix $ Add (Fix (Const 1)) (Fix (Const 2))
convertToString :: Expr -> String
convertToString = cata $ \case
e@(Const x) -> show x
e@(Add x y) -> unwords [x, "+", y]
Agora eu quero adicionar dados adicionais a ele. Então, eu estou tentando usarCofree
type LineNumber = Int
type Expr2 = Cofree ExprF LineNumber
Eu posso converterExpr
paraExpr2
addLineNumbers :: Expr -> Expr2
addLineNumbers = cata $ \case
e@(Const _) -> 1 :< e
e -> 2 :< e
Mas não consigo descobrir como converterExpr2
paraString
convertToString2 :: Expr2 -> String
convertToString2 = cata $ \case
e@(_ :< (Const x)) -> show x
e@(_ :< (Add x y)) -> unwords [x, "+", y]
Além disso, o Cofree é a melhor maneira de resolver esse problema de anotação?