git.iko.soy/elk
on 33a95f4 help sign in
elk/compiler/SourceSyntax/Declaration.hs
116 lines 3.7 kB history raw
1{-# OPTIONS_GHC -Wall #-}
2module SourceSyntax.Declaration where
4import Data.Binary
5import qualified SourceSyntax.Expression as Expr
6import qualified SourceSyntax.Type as T
7import SourceSyntax.PrettyPrint
8import Text.PrettyPrint as P
10data Declaration' port def
11 = Definition def
12 | Datatype String [String] [(String,[T.Type])] [Derivation]
13 | TypeAlias String [String] T.Type [Derivation]
14 | Port port
15 | Fixity Assoc Int String
16 deriving (Eq, Show)
18data Assoc = L | N | R
19 deriving (Eq)
21data Derivation = Json | JS | Binary | New
22 deriving (Eq, Show)
24data ParsePort
25 = PortAnnotation String T.Type
26 | SendDefinition String Expr.LParseExpr
27 | RecvDefinition String Expr.LParseExpr
28 deriving (Eq,Show)
30data Port
31 = Send String Expr.LExpr T.Type
32 | Recv String Expr.LExpr T.Type
33 deriving (Eq,Show)
35type ParseDeclaration = Declaration' ParsePort Expr.ParseDef
36type Declaration = Declaration' Port Expr.Def
38instance Binary Derivation where
39 get = do n <- getWord8
40 return $ case n of
41 0 -> Json
42 1 -> JS
43 2 -> Binary
44 3 -> New
45 _ -> error "Unable to decode Derivation. You may have corrupted binary files,\n\
46 \so please report an issue at <https://github.com/evancz/Elm/issues>"
48 put derivation =
49 putWord8 $ case derivation of
50 Json -> 0
51 JS -> 1
52 Binary -> 2
53 New -> 3
55instance Show Assoc where
56 show assoc =
57 case assoc of
58 L -> "left"
59 N -> "non"
60 R -> "right"
62instance (Pretty port, Pretty def) => Pretty (Declaration' port def) where
63 pretty decl =
64 case decl of
65 Definition def -> pretty def
67 Datatype tipe tvars ctors deriveables ->
68 P.hang (P.text "data" <+> P.text tipe <+> P.hsep (map P.text tvars)) 4
69 (P.sep $ zipWith join ("=" : repeat "|") ctors) <+> prettyDeriving deriveables
70 where
71 join c ctor = P.text c <+> prettyCtor ctor
72 prettyCtor (name, tipes) =
73 P.hang (P.text name) 2 (P.sep (map T.prettyParens tipes))
75 TypeAlias name tvars tipe deriveables ->
76 alias <+> prettyDeriving deriveables
77 where
78 name' = P.text name <+> P.hsep (map P.text tvars)
79 alias = P.hang (P.text "type" <+> name' <+> P.equals) 4 (pretty tipe)
81 Port port -> pretty port
83 Fixity assoc prec op -> P.text "infix" <> assoc' <+> P.int prec <+> P.text op
84 where
85 assoc' = case assoc of
86 L -> P.text "l"
87 N -> P.empty
88 R -> P.text "r"
90instance Pretty ParsePort where
91 pretty port =
92 case port of
93 PortAnnotation name tipe -> prettyPort name ":" tipe
94 SendDefinition name expr -> prettyPort name "<-" expr
95 RecvDefinition name expr -> prettyPort name "->" expr
97instance Pretty Port where
98 pretty port =
99 case port of
100 Send name expr tipe -> mkPort "<-" name expr tipe
101 Recv name expr tipe -> mkPort "->" name expr tipe
102 where
103 mkPort arrow name expr tipe =
104 P.vcat [ prettyPort name ":" tipe
105 , prettyPort name arrow expr ]
107prettyPort :: (Pretty a) => String -> String -> a -> Doc
108prettyPort name op e = P.text "port" <+> P.text name <+> P.text op <+> pretty e
110prettyDeriving :: [Derivation] -> Doc
111prettyDeriving deriveables =
112 case deriveables of
113 [] -> P.empty
114 [d] -> P.text "deriving" <+> P.text (show d)
115 ds -> P.text "deriving" <+>
116 P.parens (P.hsep $ P.punctuate P.comma $ map (P.text . show) ds)