From 85c6dd16023bfeba0a1ca1bf74192e7143af6848 Mon Sep 17 00:00:00 2001
From: Micah <git@dekkonot.com>
Date: Sun, 6 Oct 2024 18:54:23 -0700
Subject: [PATCH] Include attributes in JsonEncoder for functions

---
 Analysis/src/AstJsonEncoder.cpp | 29 ++++++++++++++++++++++++++++-
 1 file changed, 28 insertions(+), 1 deletion(-)

diff --git a/Analysis/src/AstJsonEncoder.cpp b/Analysis/src/AstJsonEncoder.cpp
index 45a0e8f2..e7469792 100644
--- a/Analysis/src/AstJsonEncoder.cpp
+++ b/Analysis/src/AstJsonEncoder.cpp
@@ -425,6 +425,7 @@ struct AstJsonEncoder : public AstVisitor
             "AstExprFunction",
             [&]()
             {
+                PROP(attributes);
                 PROP(generics);
                 PROP(genericPacks);
                 if (node->self)
@@ -894,7 +895,7 @@ struct AstJsonEncoder : public AstVisitor
             "AstStatDeclareFunction",
             [&]()
             {
-                // TODO: attributes
+                PROP(attributes);
                 PROP(name);
                 PROP(nameLocation);
                 PROP(params);
@@ -1042,6 +1043,7 @@ struct AstJsonEncoder : public AstVisitor
             "AstTypeFunction",
             [&]()
             {
+                PROP(attributes);
                 PROP(generics);
                 PROP(genericPacks);
                 PROP(argTypes);
@@ -1136,6 +1138,31 @@ struct AstJsonEncoder : public AstVisitor
         );
     }
 
+    void write(AstAttr::Type type)
+    {
+        switch (type)
+        {
+        case AstAttr::Type::Checked:
+            return writeString("checked");
+        case AstAttr::Type::Native:
+            return writeString("native");
+        default:
+            return writeRaw("null");
+        }
+    }
+
+    void write(class AstAttr* node)
+    {
+        writeNode(
+            node,
+            "AstAttr",
+            [&]()
+            {
+                write("name", node->type);
+            }
+        );
+    }
+
     bool visit(class AstTypeSingletonBool* node) override
     {
         writeNode(