@go-avro/avro-js 0.0.2-beta.62 → 0.0.2-beta.63

package/dist/client/QueryClient.d.ts

@@ -1,10 +1,10 @@
+import { Socket } from 'socket.io-client';
 import { InfiniteData, UseInfiniteQueryResult, useMutation, UseQueryResult } from '@tanstack/react-query';
 import { AuthManager } from '../auth/AuthManager';
-import { _Event, ApiInfo, Bill, Break, Company, Job, LineItem, Route, ServiceMonth, Session, User, UserCompanyAssociation } from '../types/api';
+import { _Event, ApiInfo, Bill, Break, Company, Job, LineItem, LoginResponse, Route, ServiceMonth, Session, User, UserCompanyAssociation } from '../types/api';
 import { Tokens } from '../types/auth';
 import { CancelToken, RetryStrategy } from '../types/client';
 import { StandardError } from '../types/error';
-import { Socket } from 'socket.io-client';
 export interface AvroQueryClientConfig {
     baseUrl: string;
     authManager: AuthManager;
@@ -215,7 +215,7 @@ export declare class AvroQueryClient {
         username: string;
         password: string;
         code?: string;
-    }, cancelToken?: CancelToken): Promise<Boolean>;
+    }, cancelToken?: CancelToken): Promise<LoginResponse>;
     setTokens(tokens: Tokens): Promise<void>;
     clearTokens(): Promise<void>;
     useLogout(): ReturnType<typeof useMutation<void, StandardError, CancelToken | undefined>>;

package/dist/client/QueryClient.js

@@ -1,6 +1,7 @@
+import io from 'socket.io-client';
 import { useMutation, useQueryClient } from '@tanstack/react-query';
+import { LoginResponse } from '../types/api';
 import { StandardError } from '../types/error';
-import io from 'socket.io-client';
 export class AvroQueryClient {
     constructor(config) {
         this.config = {
@@ -62,15 +63,20 @@ export class AvroQueryClient {
     }
     login(data, cancelToken) {
         return this._fetch('POST', '/login', data, cancelToken)
-            .then(tokens => {
-            if (!tokens || !tokens.access_token || !tokens.refresh_token) {
+            .then(resp => {
+            if (!resp || !('access_token' in resp)) {
+                if (resp.msg === "TOTP email sent") {
+                    return LoginResponse.NEEDS_TOTP;
+                }
                 throw new StandardError(401, 'Invalid login response');
             }
-            this.socket.auth = { token: tokens.access_token };
+            this.socket.auth = { token: resp.access_token };
             if (!this.socket.connected) {
                 this.socket.connect();
             }
-            return this.config.authManager.setTokens(tokens).then(() => true);
+            return this.config.authManager.setTokens({ access_token: resp.access_token, refresh_token: resp.refresh_token }).then(() => {
+                return LoginResponse.SUCCESS;
+            });
         })
             .catch(err => {
             console.error('Login failed:', err);

package/dist/types/api.d.ts

@@ -257,12 +257,18 @@ export interface Group {
     time_created: number;
     time_updated: number;
 }
-export declare const NotificationLevel: Readonly<{
-    IN_APP: 0;
-    EMAIL: 1;
-    SMS: 2;
-    PUSH: 3;
-}>;
+export declare const NotificationLevel: {
+    readonly IN_APP: 0;
+    readonly EMAIL: 1;
+    readonly SMS: 2;
+    readonly PUSH: 3;
+};
+export type NotificationLevel = typeof NotificationLevel[keyof typeof NotificationLevel];
+export declare const LoginResponse: {
+    readonly SUCCESS: "SUCCESS";
+    readonly NEEDS_TOTP: "NEEDS_TOTP";
+};
+export type LoginResponse = typeof LoginResponse[keyof typeof LoginResponse];
 export interface UserCompanyAssociation {
     id: string;
     user: User;
@@ -271,7 +277,7 @@ export interface UserCompanyAssociation {
     effective_permissions: string[];
     time_created: number | null;
     time_updated: number | null;
-    notification_setting: number[];
+    notification_setting: (typeof NotificationLevel)[];
     share_email_company_wide: boolean;
     notifications: Notification[];
     groups: string[];

package/dist/types/api.js

@@ -1,6 +1,10 @@
-export const NotificationLevel = Object.freeze({
+export const NotificationLevel = {
     IN_APP: 0,
     EMAIL: 1,
     SMS: 2,
     PUSH: 3
-});
+};
+export const LoginResponse = {
+    SUCCESS: "SUCCESS",
+    NEEDS_TOTP: "NEEDS_TOTP",
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@go-avro/avro-js",
-  "version": "0.0.2-beta.62",
+  "version": "0.0.2-beta.63",
   "description": "JS client for Avro backend integration.",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",