Merge pull request #41 from dgraph-io/apoorv/support-old-login

Support old login method

Author: mbj36

lib/client.d.ts

@@ -12,6 +12,7 @@ export declare class DgraphClient {
     setAlphaAuthToken(authToken: string): void;
     setSlashApiKey(apiKey: string): void;
     login(userid: string, password: string): Promise<boolean>;
+    loginIntoNamespace(userid: string, password: string, namespace?: number): Promise<boolean>;
     logout(): void;
     newTxn(options?: TxnOptions): Txn;
     setDebugMode(mode?: boolean): void;

lib/client.js

@@ -88,6 +88,16 @@ var DgraphClient = (function () {
             });
         });
     };
+    DgraphClient.prototype.loginIntoNamespace = function (userid, password, namespace) {
+        return __awaiter(this, void 0, void 0, function () {
+            var c;
+            return __generator(this, function (_a) {
+                this.debug("Login request:\nuserid: " + userid);
+                c = this.anyClient();
+                return [2, c.loginIntoNamespace(userid, password, namespace)];
+            });
+        });
+    };
     DgraphClient.prototype.logout = function () {
         this.debug("Logout");
         this.clients.forEach(function (c) { return c.logout(); });

lib/clientStub.js

@@ -261,8 +261,35 @@ var DgraphClientStub = (function () {
     };
     DgraphClientStub.prototype.login = function (userid, password, refreshToken) {
         return __awaiter(this, void 0, void 0, function () {
+            var body, res;
             return __generator(this, function (_a) {
-                return [2, this.loginIntoNamespace(userid, password, 0, refreshToken)];
+                switch (_a.label) {
+                    case 0:
+                        if (this.legacyApi) {
+                            throw new Error("Pre v1.1 clients do not support Login methods");
+                        }
+                        body = {};
+                        if (userid === undefined &&
+                            refreshToken === undefined &&
+                            this.refreshToken === undefined) {
+                            throw new Error("Cannot find login details: neither userid/password nor refresh token are specified");
+                        }
+                        if (userid === undefined) {
+                            body.refresh_token =
+                                refreshToken !== undefined ? refreshToken : this.refreshToken;
+                        }
+                        else {
+                            body.userid = userid;
+                            body.password = password;
+                        }
+                        return [4, this.callAPI("login", __assign(__assign({}, this.options), { method: "POST", body: JSON.stringify(body) }))];
+                    case 1:
+                        res = _a.sent();
+                        this.accessToken = res.data.accessJWT;
+                        this.refreshToken = res.data.refreshJWT;
+                        this.maybeStartRefreshTimer(this.accessToken);
+                        return [2, true];
+                }
             });
         });
     };

src/client.ts

@@ -73,6 +73,20 @@ export class DgraphClient {
         return c.login(userid, password); // tslint:disable-line no-unsafe-any
     }
 
+    /**
+     * loginIntoNamespace obtains access tokens from Dgraph Server for the particular userid & namespace
+     */
+    public async loginIntoNamespace(
+        userid: string,
+        password: string,
+        namespace?: number,
+    ): Promise<boolean> {
+        this.debug(`Login request:\nuserid: ${userid}`);
+
+        const c = this.anyClient();
+        return c.loginIntoNamespace(userid, password, namespace); // tslint:disable-line no-unsafe-any
+    }
+
     /**
      * logout - forget all access tokens.
      */

src/clientStub.ts

@@ -296,7 +296,38 @@ export class DgraphClientStub {
         password?: string,
         refreshToken?: string,
     ): Promise<boolean> {
-        return this.loginIntoNamespace(userid, password, 0, refreshToken);
+         if (this.legacyApi) {
+             throw new Error("Pre v1.1 clients do not support Login methods");
+         }
+
+         const body: { [k: string]: string } = {};
+         if (
+             userid === undefined &&
+             refreshToken === undefined &&
+             this.refreshToken === undefined
+         ) {
+             throw new Error(
+                 "Cannot find login details: neither userid/password nor refresh token are specified",
+             );
+         }
+         if (userid === undefined) {
+             body.refresh_token =
+                 refreshToken !== undefined ? refreshToken : this.refreshToken;
+         } else {
+             body.userid = userid;
+             body.password = password;
+         }
+
+         const res: LoginResponse = await this.callAPI("login", {
+             ...this.options,
+             method: "POST",
+             body: JSON.stringify(body),
+         });
+         this.accessToken = res.data.accessJWT;
+         this.refreshToken = res.data.refreshJWT;
+
+         this.maybeStartRefreshTimer(this.accessToken);
+         return true;
     }
 
     public async loginIntoNamespace(
@@ -309,7 +340,7 @@ export class DgraphClientStub {
             throw new Error("Pre v1.1 clients do not support Login methods");
         }
 
-        const body: { [k: string ]: string | number} = {};
+        const body: { [k: string]: string | number } = {};
         if (
             userid === undefined &&
             refreshToken === undefined &&