@go-avro/avro-js 0.0.2-beta.76 → 0.0.2-beta.77

package/dist/client/QueryClient.d.ts

@@ -254,11 +254,12 @@ export declare class AvroQueryClient {
     post<T>(path: string, data: any, cancelToken?: CancelToken, headers?: Record<string, string>, progressUpdateCallback?: (loaded: number, total: number) => void): Promise<T>;
     put<T>(path: string, data: any, cancelToken?: CancelToken, headers?: Record<string, string>, progressUpdateCallback?: (loaded: number, total: number) => void): Promise<T>;
     delete<T>(path: string, cancelToken?: CancelToken, headers?: Record<string, string>, progressUpdateCallback?: (loaded: number, total: number) => void): Promise<T>;
-    login(data: {
+    useLogin(): ReturnType<typeof useMutation<LoginResponse, StandardError, {
         username: string;
         password: string;
         code?: string;
-    }, cancelToken?: CancelToken): Promise<LoginResponse>;
+        cancelToken?: CancelToken;
+    }>>;
     setTokens(tokens: Tokens): Promise<void>;
     clearTokens(): Promise<void>;
     useLogout(): ReturnType<typeof useMutation<void, StandardError, CancelToken | undefined>>;

package/dist/client/QueryClient.js

@@ -61,26 +61,32 @@ export class AvroQueryClient {
     delete(path, cancelToken, headers = {}, progressUpdateCallback) {
         return this._xhr('DELETE', path, null, cancelToken, headers, false, this.config.maxRetries, progressUpdateCallback);
     }
-    login(data, cancelToken) {
-        return this._fetch('POST', '/login', data, cancelToken)
-            .then(resp => {
-            if (!resp || !('access_token' in resp)) {
-                if (resp.msg === "TOTP email sent") {
-                    return LoginResponse.NEEDS_TOTP;
+    useLogin() {
+        const queryClient = useQueryClient();
+        return useMutation({
+            mutationFn: async ({ username, password, code, cancelToken }) => {
+                const resp = await this._fetch('POST', '/login', { username, password, code }, cancelToken);
+                if (!resp || !('access_token' in resp)) {
+                    if (resp.msg === "TOTP email sent") {
+                        return LoginResponse.NEEDS_TOTP;
+                    }
+                    throw new StandardError(401, 'Invalid login response');
                 }
-                throw new StandardError(401, 'Invalid login response');
-            }
-            this.socket.auth = { token: resp.access_token };
-            if (!this.socket.connected) {
-                this.socket.connect();
-            }
-            return this.config.authManager.setTokens({ access_token: resp.access_token, refresh_token: resp.refresh_token }).then(() => {
+                this.socket.auth = { token: resp.access_token };
+                if (!this.socket.connected) {
+                    this.socket.connect();
+                }
+                await this.config.authManager.setTokens({ access_token: resp.access_token, refresh_token: resp.refresh_token });
                 return LoginResponse.SUCCESS;
-            });
-        })
-            .catch(err => {
-            console.error('Login failed:', err);
-            throw new StandardError(401, 'Login failed');
+            },
+            onSettled: () => {
+                queryClient.invalidateQueries();
+            },
+            onError: (err) => {
+                this.config.authManager.clearTokens();
+                console.error('Login failed:', err);
+                throw new StandardError(401, 'Login failed');
+            }
         });
     }
     setTokens(tokens) {

package/package.json

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