1 files changed, 33 insertions, 0 deletions

diff --git a/templates/data.hs.ede b/templates/data.hs.ede
index b2ad9de..fccd7dc 100644
--- a/templates/data.hs.ede
+++ b/templates/data.hs.ede
@@ -1,13 +1,20 @@
 -- generated file, do not modify!
 -- {{ dateTime }}
 
+{-# LANGUAGE OverloadedStrings, RecordWildCards #-}
 module {{ moduleName }} where
 
 import Data.Int
 import Data.Word
 import Data.Map
+import Data.Vector (Vector(..))
 import Linear
 
+import Data.Text
+import Data.Aeson hiding (Value,Bool)
+import Data.Aeson.Types hiding (Value,Bool)
+import Control.Monad
+
 {% for t in dataAndType %}
 {% case t.value | constType %}
 {% when "DataDef" %}
@@ -25,3 +32,29 @@ type {{ t.value.aliasName }} = {{ t.value.aliasType | hsType }}
 {% endcase %}
 
 {% endfor %}
+
+(.-) :: Text -> Text -> Pair
+a .- b = a .= b
+
+{% for t in definitions %}
+instance ToJSON {{ t.value.dataName }} where
+  toJSON v = case v of{% for c in t.value.constructors %}{% if c.value.fields | hasFieldNames %}
+    {{ c.value.name }}{..} -> object
+      [ "tag" .- "{{ c.value.name }}"{% for f in c.value.fields %}
+      , "{{ f.value.fieldName }}" .= {{ f.value.fieldName }}{% endfor %}
+      ]{% else %}
+    {{ c.value.name }}{% for f in c.value.fields %} arg{{ f.index0 }}{% endfor %} -> object [ "tag" .- "{{ c.value.name }}"{% for f in c.value.fields %}, "arg{{ f.index0 }}" .= arg{{ f.index0 }}{% endfor %}]{% endif %}{% endfor %}
+
+instance FromJSON {{ t.value.dataName }} where
+  parseJSON (Object obj) = do
+    tag <- obj .: "tag"
+    case tag :: Text of{% for c in t.value.constructors %}{% if c.value.fields | hasFieldNames %}
+      "{{ c.value.name }}" -> do{% for f in c.value.fields %}
+        {{ f.value.fieldName }} <- obj .: "{{ f.value.fieldName }}"{% endfor %}
+        pure $ {{ c.value.name }}{% for f in c.value.fields %}
+          {% if f.first %}{ {% else %}, {%endif%}{{ f.value.fieldName }} = {{ f.value.fieldName }}{% endfor %}
+          } {% else %}
+      "{{ c.value.name }}" -> {% for f in c.value.fields %}{% if f.first %}{{ c.value.name }} <$>{% else %} <*>{% endif %} obj .: "arg{{ f.index0 }}"{%else%}pure {{ c.value.name }}{% endfor %}{% endif %}{% endfor %}
+  parseJSON _ = mzero
+
+{% endfor %}
\ No newline at end of file