Ehnance control oauth (#74)

cloudquery · Nov 26, 2024 · 6e893b6 · 6e893b6
1 parent f2d09eb
commit 6e893b6
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 6 deletions.
diff --git a/etc/plugin-config-ui-lib.api.md b/etc/plugin-config-ui-lib.api.md
@@ -185,10 +185,11 @@ export interface ControlNumberFieldProps {
 }
 
 // @public
-export function ControlOAuthField({ getConnectPayloadSpec, getFinishPayloadSpec, }: ControlOAuthFieldProps): JSX_2.Element;
+export function ControlOAuthField({ shouldAuthenticate, getConnectPayloadSpec, getFinishPayloadSpec, }: ControlOAuthFieldProps): JSX_2.Element;
 
 // @public (undocumented)
 export type ControlOAuthFieldProps = {
+    shouldAuthenticate?: () => Promise<boolean>;
     getConnectPayloadSpec?: (formValues: any) => Promise<Record<string, any>>;
     getFinishPayloadSpec?: (formValues: any) => Promise<Record<string, any>>;
 };

diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -1,7 +1,7 @@
 {
   "name": "@cloudquery/plugin-config-ui-lib",
   "description": "Plugin configuration UI library for CloudQuery Cloud App",
-  "version": "4.2.2",
+  "version": "4.2.3",
   "private": false,
   "main": "dist/index.cjs.js",
   "module": "dist/index.esm.js",
@@ -21,6 +21,7 @@
       "require": "./dist/components/devWrapper.cjs.js"
     },
     "./e2e-utils": {
+      "types": "./dist/e2e-utils/index.d.ts",
       "import": "./dist/e2e-utils/index.esm.js",
       "require": "./dist/e2e-utils/index.cjs.js"
     },

diff --git a/src/components/form/controls/controlOAuthField.tsx b/src/components/form/controls/controlOAuthField.tsx
@@ -20,6 +20,7 @@ import { cloudQueryOauthConnectorUrl } from '../../../utils';
  * @public
  */
 export type ControlOAuthFieldProps = {
+  shouldAuthenticate?: () => Promise<boolean>;
   getConnectPayloadSpec?: (formValues: any) => Promise<Record<string, any>>;
   getFinishPayloadSpec?: (formValues: any) => Promise<Record<string, any>>;
 };
@@ -30,6 +31,7 @@ export type ControlOAuthFieldProps = {
  * @public
  */
 export function ControlOAuthField({
+  shouldAuthenticate,
   getConnectPayloadSpec,
   getFinishPayloadSpec,
 }: ControlOAuthFieldProps) {
@@ -55,6 +57,16 @@ export function ControlOAuthField({
     getFinishPayloadSpec: async () => await getFinishPayloadSpec?.(getValues()),
   });
 
+  const handleAuthenticate = async () => {
+    let proceed = true;
+    if (shouldAuthenticate) {
+      proceed = await shouldAuthenticate();
+    }
+    if (proceed) {
+      authenticate();
+    }
+  };
+
   useEffect(() => {
     if (authConnectorResult && connectorId) {
       setValue('connectorId', connectorId);
@@ -86,7 +98,7 @@ export function ControlOAuthField({
         <LoadingButton
           size="large"
           variant="contained"
-          onClick={authenticate}
+          onClick={handleAuthenticate}
           loading={isLoading}
           fullWidth={false}
           disabled={!!connectorIdValue}
@@ -106,7 +118,7 @@ export function ControlOAuthField({
         <Typography variant="body2" color="textSecondary">
           To reconnect CloudQuery via {config.label}{' '}
           {/* eslint-disable-next-line jsx-a11y/anchor-is-valid */}
-          <Link underline="always" sx={{ cursor: 'pointer' }} onClick={authenticate}>
+          <Link underline="always" sx={{ cursor: 'pointer' }} onClick={handleAuthenticate}>
             click here
           </Link>
         </Typography>