feat: Add custom header support to Express-style cloud functions via res.header()

Co-authored-by: mtrezza <[email protected]>

Author: Copilot

spec/CloudCode.spec.js

@@ -4946,5 +4946,73 @@ describe('beforePasswordResetRequest hook', () => {
 
       await Parse.User.logOut();
     });
+
+    it('should support setting custom headers with res.header()', async () => {
+      Parse.Cloud.define('customHeaderFunction', (req, res) => {
+        res.header('X-Custom-Header', 'custom-value').success({ message: 'OK' });
+      });
+
+      const response = await request({
+        method: 'POST',
+        url: 'http://localhost:8378/1/functions/customHeaderFunction',
+        headers: {
+          'X-Parse-Application-Id': 'test',
+          'X-Parse-REST-API-Key': 'rest',
+        },
+        json: true,
+        body: {},
+      });
+
+      expect(response.status).toBe(200);
+      expect(response.headers['x-custom-header']).toBe('custom-value');
+      expect(response.data.result.message).toBe('OK');
+    });
+
+    it('should support setting multiple custom headers', async () => {
+      Parse.Cloud.define('multipleHeadersFunction', (req, res) => {
+        res.header('X-Header-One', 'value1')
+           .header('X-Header-Two', 'value2')
+           .success({ message: 'Multiple headers' });
+      });
+
+      const response = await request({
+        method: 'POST',
+        url: 'http://localhost:8378/1/functions/multipleHeadersFunction',
+        headers: {
+          'X-Parse-Application-Id': 'test',
+          'X-Parse-REST-API-Key': 'rest',
+        },
+        json: true,
+        body: {},
+      });
+
+      expect(response.status).toBe(200);
+      expect(response.headers['x-header-one']).toBe('value1');
+      expect(response.headers['x-header-two']).toBe('value2');
+      expect(response.data.result.message).toBe('Multiple headers');
+    });
+
+    it('should support combining status code and custom headers', async () => {
+      Parse.Cloud.define('statusAndHeaderFunction', (req, res) => {
+        res.status(201)
+           .header('X-Resource-Id', '12345')
+           .success({ created: true });
+      });
+
+      const response = await request({
+        method: 'POST',
+        url: 'http://localhost:8378/1/functions/statusAndHeaderFunction',
+        headers: {
+          'X-Parse-Application-Id': 'test',
+          'X-Parse-REST-API-Key': 'rest',
+        },
+        json: true,
+        body: {},
+      });
+
+      expect(response.status).toBe(201);
+      expect(response.headers['x-resource-id']).toBe('12345');
+      expect(response.data.result.created).toBe(true);
+    });
   });
 });

src/Routers/FunctionsRouter.js

@@ -105,6 +105,7 @@ export class FunctionsRouter extends PromiseRouter {
 
   static createResponseObject(resolve, reject, statusCode = null) {
     let httpStatusCode = statusCode;
+    let customHeaders = {};
     let responseSent = false;
     const responseObject = {
       success: function (result) {
@@ -120,6 +121,9 @@ export class FunctionsRouter extends PromiseRouter {
         if (httpStatusCode !== null) {
           response.status = httpStatusCode;
         }
+        if (Object.keys(customHeaders).length > 0) {
+          response.headers = customHeaders;
+        }
         resolve(response);
       },
       error: function (message) {
@@ -138,6 +142,10 @@ export class FunctionsRouter extends PromiseRouter {
         httpStatusCode = code;
         return responseObject;
       },
+      header: function (key, value) {
+        customHeaders[key] = value;
+        return responseObject;
+      },
       _isResponseSent: () => responseSent,
     };
     return responseObject;

src/cloud-code/Parse.Cloud.js

@@ -137,6 +137,10 @@ var ParseCloud = {};
  *   }
  * });
  *
+ * Parse.Cloud.define('withCustomHeaders', (request, response) => {
+ *   response.header('X-Custom-Header', 'value').success({ data: 'OK' });
+ * });
+ *
  * Parse.Cloud.define('errorFunction', (request, response) => {
  *   response.error('Something went wrong');
  * });
@@ -824,6 +828,7 @@ module.exports = ParseCloud;
  * @property {function} success Call this function to return a successful response with an optional result. Usage: `response.success(result)`
  * @property {function} error Call this function to return an error response with an error message. Usage: `response.error(message)`
  * @property {function} status Call this function to set a custom HTTP status code for the response. Returns the response object for chaining. Usage: `response.status(code).success(result)` or `response.status(code).error(message)`
+ * @property {function} header Call this function to set a custom HTTP header for the response. Returns the response object for chaining. Usage: `response.header('X-Custom-Header', 'value').success(result)`
  */
 
 /**