From 838f28629c5b70ca8a2ef84104ce832e8490b68d Mon Sep 17 00:00:00 2001
From: Grant Gurvis <grangurv@amazon.com>
Date: Thu, 30 May 2024 09:14:47 -0700
Subject: [PATCH] Add export default

---
 src/aws_cli_plugin/__init__.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/src/aws_cli_plugin/__init__.py b/src/aws_cli_plugin/__init__.py
index 2e1f92b..02b368b 100644
--- a/src/aws_cli_plugin/__init__.py
+++ b/src/aws_cli_plugin/__init__.py
@@ -154,9 +154,9 @@ def parseBasicCommand(command):
 
 def saveJsonAsSpec(d, path):
     prefix = (
-        "export const completionSpec: Fig.Spec ="
+        "export const completionSpec: Fig.Spec = "
         if exportTypescript
-        else "var completionSpec ="
+        else "var completionSpec = "
     )
     extension = ".ts" if exportTypescript else ".js"
 
@@ -165,7 +165,13 @@ def saveJsonAsSpec(d, path):
     )
     file_path.parent.mkdir(parents=True, exist_ok=True)
 
-    final = f"{prefix} {json.dumps(d, indent=4)}"
+    suffix = (
+        ";\n\nexport default completionSpec;"
+        if exportTypescript
+        else ";\n\nmodule.exports = completionSpec;"
+    )
+
+    final = f"{prefix}{json.dumps(d, indent=4)}{suffix}"
     file_path.write_text(final)