feat(bruno): add support for setting http headers

Author: mvantellingen

.changeset/twenty-years-wonder.md

@@ -0,0 +1,5 @@
+---
+"@labdigital/graphql-codegen-bruno": minor
+---
+
+Add support for setting http headers

packages/bruno/package.json

@@ -42,7 +42,7 @@
 		"typescript": "^5.7.2"
 	},
 	"peerDependencies": {
-		"graphql": "^16.9.0"
+		"graphql": ">= 16"
 	},
 	"pnpm": {
 		"overrides": {

packages/bruno/src/__output__/queries/GetCustomer.bru

@@ -1,4 +1,3 @@
-
 meta {
   name: GetCustomer
   type: graphql
@@ -53,5 +52,4 @@ body:graphql:vars {
       "authMethod": "EMAIL"
     }
   }
-}
-  
+}

packages/bruno/src/bruno.ts

@@ -1,36 +1,44 @@
 import prettier from "prettier";
 import type { FileContent } from "./operations";
 
+export interface BrunoPluginConfig {
+	defaults: Record<string, unknown>;
+	headers: Record<string, string>;
+	clean: boolean;
+}
+
 export const asBruno = async (
 	operation: FileContent,
-	defaults: Record<string, unknown>,
+	config: BrunoPluginConfig,
 ) => {
 	const formattedContent = await prettier.format(operation.content, {
 		parser: "graphql",
 	});
 
-	const vars = mergeDefaults(operation.vars, defaults);
+	const vars = mergeDefaults(operation.vars, config.defaults);
 
-	return `
-meta {
-  name: ${operation.name}
-  type: graphql
-}
+	const file = new FileCreator();
+	file.addElement("meta", [`name: ${operation.name}`, "type: graphql"]);
+	file.addElement("post", [
+		"url: {{graphql-gateway}}/graphql",
+		"body: graphql",
+		"auth: none",
+	]);
+	file.addElement(
+		"headers",
+		Object.entries(config.headers ?? {}).map(
+			([key, value]) => `${key}: "${value}"`,
+		),
+	);
 
-post {
-  url: {{graphql-gateway}}/graphql
-  body: graphql
-  auth: none
-}
+	file.addElement("body:graphql", formattedContent.split("\n"));
 
-body:graphql {
-  ${formattedContent.split("\n").join("\n  ")}
-}
+	file.addElement(
+		"body:graphql:vars",
+		JSON.stringify(vars, null, 2).split("\n"),
+	);
 
-body:graphql:vars {
-  ${JSON.stringify(vars, null, 2).split("\n").join("\n  ")}
-}
-  `;
+	return file.toString();
 };
 
 const mergeDefaults = (
@@ -51,3 +59,29 @@ const mergeDefaults = (
 
 	return mergedVars;
 };
+
+class FileCreator {
+	elements: string[];
+
+	constructor() {
+		this.elements = [];
+	}
+
+	addElement(name: string, lines: string[]) {
+		if (lines.length === 0) {
+			return;
+		}
+		const snippet: string[] = [];
+		snippet.push(`${name} {`);
+
+		for (const line of lines) {
+			snippet.push(`  ${line}`);
+		}
+		snippet.push("}");
+		this.elements.push(snippet.join("\n"));
+	}
+
+	toString() {
+		return this.elements.join("\n\n");
+	}
+}

packages/bruno/src/index.ts

@@ -2,14 +2,9 @@ import path from "node:path";
 import type { PluginFunction, Types } from "@graphql-codegen/plugin-helpers";
 import fs from "fs-extra";
 import type { GraphQLSchema } from "graphql";
-import { asBruno } from "./bruno";
+import { type BrunoPluginConfig, asBruno } from "./bruno";
 import { extractOperations } from "./operations";
 
-export interface BrunoPluginConfig {
-	defaults: Record<string, unknown>;
-	clean: boolean;
-}
-
 export const plugin: PluginFunction<BrunoPluginConfig> = async (
 	schema: GraphQLSchema,
 	documents: Types.DocumentFile[],
@@ -31,7 +26,7 @@ export const plugin: PluginFunction<BrunoPluginConfig> = async (
 		const fileName = `${operation.name}.bru`;
 		const outputPath = path.join(outputDir, subpath, fileName);
 
-		const formattedContent = await asBruno(operation, config.defaults);
+		const formattedContent = await asBruno(operation, config);
 
 		fs.outputFileSync(outputPath, formattedContent);
 		result[operation.name] = {