Allow custom HTTP headers

Author: Paul Korzhyk

lib/clientStub.js

@@ -54,18 +54,14 @@ var AUTO_REFRESH_PREFETCH_TIME = 5000;
 var DgraphClientStub = (function () {
     function DgraphClientStub(addr, stubConfig, options) {
         if (stubConfig === void 0) { stubConfig = {}; }
+        if (options === void 0) { options = {}; }
         if (addr === undefined) {
             this.addr = "http://localhost:8080";
         }
         else {
             this.addr = addr;
         }
-        if (options === undefined) {
-            this.options = {};
-        }
-        else {
-            this.options = options;
-        }
+        this.options = options;
         this.legacyApi = !!stubConfig.legacyApi;
         this.jsonParser =
             stubConfig.jsonParser !== undefined
@@ -108,7 +104,7 @@ var DgraphClientStub = (function () {
         return this.callAPI("alter", __assign(__assign({}, this.options), { method: "POST", body: body }));
     };
     DgraphClientStub.prototype.query = function (req) {
-        var headers = {};
+        var headers = Object.assign({}, this.options.headers !== undefined ? this.options.headers : {});
         if (req.vars !== undefined) {
             if (this.legacyApi) {
                 headers["X-Dgraph-Vars"] = JSON.stringify(req.vars);
@@ -217,9 +213,9 @@ var DgraphClientStub = (function () {
         else {
             return Promise.reject("Mutation has no data");
         }
-        var headers = {
+        var headers = Object.assign({}, this.options.headers !== undefined ? this.options.headers : {}, {
             "Content-Type": "application/" + (usingJSON ? "json" : "rdf"),
-        };
+        });
         if (usingJSON && this.legacyApi) {
             headers["X-Dgraph-MutationType"] = "json";
         }

package.json

@@ -1,6 +1,6 @@
 {
   "name": "dgraph-js-http",
-  "version": "20.7.0",
+  "version": "20.7.1-rc1",
   "description": "A javascript HTTP client for Dgraph",
   "license": "Apache-2.0",
   "repository": {

src/clientStub.ts

@@ -41,7 +41,7 @@ export class DgraphClientStub {
             // tslint:disable-next-line no-any
             jsonParser?(text: string): any;
         } = {},
-        options?: Options,
+        options: Options = {},
     ) {
         if (addr === undefined) {
             // tslint:disable-next-line no-http-string
@@ -50,11 +50,7 @@ export class DgraphClientStub {
             this.addr = addr;
         }
 
-        if (options === undefined) {
-            this.options = {};
-        } else {
-            this.options = options;
-        }
+        this.options = options;
 
         this.legacyApi = !!stubConfig.legacyApi;
         this.jsonParser =
@@ -95,7 +91,10 @@ export class DgraphClientStub {
     }
 
     public query(req: Request): Promise<Response> {
-        const headers: { [k: string]: string } = {};
+        const headers = Object.assign(
+            {},
+            this.options.headers !== undefined ? this.options.headers : {},
+        );
         if (req.vars !== undefined) {
             if (this.legacyApi) {
                 headers["X-Dgraph-Vars"] = JSON.stringify(req.vars);
@@ -228,9 +227,14 @@ export class DgraphClientStub {
             return Promise.reject("Mutation has no data");
         }
 
-        const headers: { [k: string]: string } = {
-            "Content-Type": `application/${usingJSON ? "json" : "rdf"}`,
-        };
+        const headers = Object.assign(
+            {},
+            this.options.headers !== undefined ? this.options.headers : {},
+            {
+                "Content-Type": `application/${usingJSON ? "json" : "rdf"}`,
+            },
+        );
+
         if (usingJSON && this.legacyApi) {
             headers["X-Dgraph-MutationType"] = "json";
         }
@@ -314,6 +318,7 @@ export class DgraphClientStub {
         }
 
         const res: LoginResponse = await this.callAPI("login", {
+            ...this.options,
             method: "POST",
             body: JSON.stringify(body),
         });