feat: Update agent device token api. (TabbyML#244)

* feat: Update agent device token api. * fix: typedef for DeviceTokenRequest.
javadeath · Jun 16, 2023 · 6f7a6af · 6f7a6af
1 parent 7cb9f2f
commit 6f7a6af
Show file tree

Hide file tree

Showing 11 changed files with 27 additions and 20 deletions.
diff --git a/clients/tabby-agent/dist/cli.js b/clients/tabby-agent/dist/cli.js
diff --git a/clients/tabby-agent/dist/cli.js.map b/clients/tabby-agent/dist/cli.js.map
diff --git a/clients/tabby-agent/dist/index.global.js b/clients/tabby-agent/dist/index.global.js
diff --git a/clients/tabby-agent/dist/index.global.js.map b/clients/tabby-agent/dist/index.global.js.map
diff --git a/clients/tabby-agent/dist/index.js b/clients/tabby-agent/dist/index.js
diff --git a/clients/tabby-agent/dist/index.js.map b/clients/tabby-agent/dist/index.js.map
diff --git a/clients/tabby-agent/dist/index.mjs b/clients/tabby-agent/dist/index.mjs
@@ -13183,10 +13183,11 @@ var ApiService = class {
    * @returns DeviceTokenResponse Success
    * @throws ApiError
    */
-  deviceToken() {
+  deviceToken(body) {
     return this.httpRequest.request({
       method: "POST",
-      url: "/device-token"
+      url: "/device-token",
+      body
     });
   }
   /**
@@ -13341,7 +13342,7 @@ var _Auth = class extends EventEmitter {
   async requestToken() {
     try {
       await this.reset();
-      const deviceToken = await this.authApi.api.deviceToken();
+      const deviceToken = await this.authApi.api.deviceToken({ auth_url: this.endpoint });
       this.logger.debug({ deviceToken }, "Request device token response");
       const authUrl = new URL(_Auth.authPageUrl);
       authUrl.searchParams.append("code", deviceToken.data.code);

diff --git a/clients/tabby-agent/dist/index.mjs.map b/clients/tabby-agent/dist/index.mjs.map
diff --git a/clients/tabby-agent/src/Auth.ts b/clients/tabby-agent/src/Auth.ts
@@ -32,7 +32,7 @@ export class Auth extends EventEmitter {
     this.endpoint = options.endpoint;
     this.dataStore = options.dataStore || dataStore;
 
-    // From tabby endpoint: http[s]://{namespace}.app.tabbyml.com/tabby[/] 
+    // From tabby endpoint: http[s]://{namespace}.app.tabbyml.com/tabby[/]
     // To auth endpoint: http[s]://{namespace}.app.tabbyml.com/api
     const authApiBase = this.endpoint.replace(/\/tabby\/?$/, "/api");
     this.authApi = new CloudApi({ BASE: authApiBase });
@@ -92,7 +92,7 @@ export class Auth extends EventEmitter {
   async requestToken(): Promise<string> {
     try {
       await this.reset();
-      const deviceToken = await this.authApi.api.deviceToken();
+      const deviceToken = await this.authApi.api.deviceToken({ auth_url: this.endpoint });
       this.logger.debug({ deviceToken }, "Request device token response");
       const authUrl = new URL(Auth.authPageUrl);
       authUrl.searchParams.append("code", deviceToken.data.code);

diff --git a/clients/tabby-agent/src/cloud/models/DeviceTokenRequest.ts b/clients/tabby-agent/src/cloud/models/DeviceTokenRequest.ts
@@ -0,0 +1,3 @@
+export type DeviceTokenResponse = {
+  auth_url: string;
+};
diff --git a/clients/tabby-agent/src/cloud/services/ApiService.ts b/clients/tabby-agent/src/cloud/services/ApiService.ts
@@ -1,6 +1,7 @@
 import type { CancelablePromise } from "../../generated/core/CancelablePromise";
 import type { BaseHttpRequest } from "../../generated/core/BaseHttpRequest";
 
+import { DeviceTokenRequest } from "../models/DeviceTokenRequest";
 import { DeviceTokenResponse } from "../models/DeviceTokenResponse";
 import { DeviceTokenAcceptResponse } from "../models/DeviceTokenAcceptResponse";
 
@@ -11,10 +12,11 @@ export class ApiService {
    * @returns DeviceTokenResponse Success
    * @throws ApiError
    */
-  public deviceToken(): CancelablePromise<DeviceTokenResponse> {
+  public deviceToken(body: DeviceTokenRequest): CancelablePromise<DeviceTokenResponse> {
     return this.httpRequest.request({
       method: "POST",
       url: "/device-token",
+      body,
     });
   }