1 files changed, 65 insertions, 0 deletions
diff --git a/packages/bun-macro-relay/bun-macro-relay.tsx b/packages/bun-macro-relay/bun-macro-relay.tsx
new file mode 100644
index 000000000..eaeca062a
--- /dev/null
+++ b/packages/bun-macro-relay/bun-macro-relay.tsx
@@ -0,0 +1,65 @@
+import { parse, print } from "graphql";
+
+//
+// 1. Parse the GraphQL tag.
+// 2. From the parsed GraphQL query, get the AST definition.
+// 3. From the AST definition, inject an import to that file inside the artifact directory
+// 4. MD5 the printed source text
+// 5. At runtime, if md5 !== import.md5, then warn the user that the query has changed
+//    but the file hasn't been updated so it must be reloaded.
+// 6. Replace the TemplateLiteral with the default identifier from the injected import
+let artifactDirectory: string =
+  process?.env?.BUN_MACRO_RELAY_ARTIFACT_DIRECTORY ??
+  process?.env?.RELAY_ARTIFACT_DIRECTORY ??
+  `__generated__`;
+
+artifactDirectory = artifactDirectory.startsWith("/")
+  ? artifactDirectory
+  : Bun.cwd + artifactDirectory;
+
+export function graphql(node) {
+  const [templateLiteral] = node.arguments;
+
+  const query = templateLiteral?.toString();
+  if (typeof query !== "string" || query.length === 0) {
+    throw new Error("BunMacroRelay: Unexpected empty graphql string.");
+  }
+
+  const ast = parse(query);
+
+  if (ast.definitions.length === 0) {
+    throw new Error("BunMacroRelay: Unexpected empty graphql tag.");
+  }
+
+  const definition = ast.definitions[0];
+
+  if (
+    definition.kind !== "FragmentDefinition" &&
+    definition.kind !== "OperationDefinition"
+  ) {
+    throw new Error(
+      `BunMacroRelay: Expected a fragment, mutation, query, or subscription, got "${definition.kind}"`
+    );
+  }
+
+  const graphqlDefinition = definition;
+
+  const definitionName = graphqlDefinition.name && graphqlDefinition.name.value;
+  if (!definitionName) {
+    throw new Error("GraphQL operations and fragments must contain names");
+  }
+
+  const importStmt = (
+    <import
+      default={definitionName}
+      path={`${artifactDirectory}/${definitionName}`}
+    />
+  );
+
+  return (
+    <>
+      <inject>{importStmt}</inject>,
+      <id to={importStmt.symbols.default} pure />,
+    </>
+  );
+}