feat: enable headers for event source (#189)

the-codegen-project · Feb 19, 2025 · 97300cb · 97300cb
1 parent 73370ea
commit 97300cb
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 8 deletions.
diff --git a/package.json b/package.json
@@ -109,6 +109,7 @@
     "pack:rename": "node scripts/releasePackagesRename.js",
     "prepublishOnly": "npm run build && npm run generate:assets",
     "test": "jest --coverage --testPathIgnorePatterns ./test/blackbox",
+    "test:update": "npm run test -- -u",
     "coverage": "nyc npm run test",
     "bump:version": "npm --no-git-tag-version --allow-same-version version $VERSION",
     "runtime:prepare": "npm link",

diff --git a/src/codegen/generators/typescript/channels/protocols/eventsource/fetch.ts b/src/codegen/generators/typescript/channels/protocols/eventsource/fetch.ts
@@ -41,7 +41,7 @@ export function renderFetch({
       : []),
     {
       parameter:
-        'options: {authorization?: string, onClose?: (err?: string) => void, baseUrl: string}',
+        'options: {authorization?: string, onClose?: (err?: string) => void, baseUrl: string, headers?: Record<string, string>}',
       jsDoc: ' * @param options additionally used to handle the event source'
     }
   ];
@@ -57,7 +57,8 @@ ${functionName}: async (
 	let eventsUrl: string = ${addressToUse};
 	const url = \`\${options.baseUrl}/\${eventsUrl}\`
   const headers: Record<string, string> = {
-    Accept: 'text/event-stream',
+	  ...options.headers ?? {},
+    Accept: 'text/event-stream'
   }
   if(options.authorization) {
     headers['authorization'] = \`Bearer \${options?.authorization}\`;

diff --git a/test/codegen/generators/typescript/__snapshots__/channels.spec.ts.snap b/test/codegen/generators/typescript/__snapshots__/channels.spec.ts.snap
@@ -317,12 +317,13 @@ event_source: {
  */
 listenForUserSignedup: async (
   callback: (messageEvent: MessageTypeModule.MessageType | null, error?: string) => void, 
-  options: {authorization?: string, onClose?: (err?: string) => void, baseUrl: string}
+  options: {authorization?: string, onClose?: (err?: string) => void, baseUrl: string, headers?: Record<string, string>}
 ) => {
 	let eventsUrl: string = 'user/signedup';
 	const url = \`\${options.baseUrl}/\${eventsUrl}\`
   const headers: Record<string, string> = {
-    Accept: 'text/event-stream',
+	  ...options.headers ?? {},
+    Accept: 'text/event-stream'
   }
   if(options.authorization) {
     headers['authorization'] = \`Bearer \${options?.authorization}\`;
@@ -696,12 +697,13 @@ event_source: {
  */
 listenForUserSignedup: async (
   callback: (messageEvent: MessageTypeModule.MessageType | null, error?: string) => void, 
-  options: {authorization?: string, onClose?: (err?: string) => void, baseUrl: string}
+  options: {authorization?: string, onClose?: (err?: string) => void, baseUrl: string, headers?: Record<string, string>}
 ) => {
 	let eventsUrl: string = 'user/signedup';
 	const url = \`\${options.baseUrl}/\${eventsUrl}\`
   const headers: Record<string, string> = {
-    Accept: 'text/event-stream',
+	  ...options.headers ?? {},
+    Accept: 'text/event-stream'
   }
   if(options.authorization) {
     headers['authorization'] = \`Bearer \${options?.authorization}\`;
@@ -1105,12 +1107,13 @@ event_source: {
  */
 listenForPing: async (
   callback: (messageEvent: MessageTypeModule.MessageType | null, error?: string) => void, 
-  options: {authorization?: string, onClose?: (err?: string) => void, baseUrl: string}
+  options: {authorization?: string, onClose?: (err?: string) => void, baseUrl: string, headers?: Record<string, string>}
 ) => {
 	let eventsUrl: string = '/ping';
 	const url = \`\${options.baseUrl}/\${eventsUrl}\`
   const headers: Record<string, string> = {
-    Accept: 'text/event-stream',
+	  ...options.headers ?? {},
+    Accept: 'text/event-stream'
   }
   if(options.authorization) {
     headers['authorization'] = \`Bearer \${options?.authorization}\`;