From 5f2e4dd8ea9ecd22e9ab72393052f52264c601ca Mon Sep 17 00:00:00 2001
From: Pawan Kumar <pawan@appsmith.com>
Date: Fri, 21 Mar 2025 10:58:19 +0530
Subject: [PATCH 1/3] add ce files

---
 .../components/UQIEditor/FormRender.tsx       |  7 ++-
 .../formControls/DatasourceLinkControl.tsx    | 63 +++++++++++++++++++
 .../components/FunctionCallingConfigForm.tsx  |  1 +
 .../utils/formControl/FormControlRegistry.tsx | 14 +++++
 .../src/utils/formControl/formControlTypes.ts |  1 +
 5 files changed, 85 insertions(+), 1 deletion(-)
 create mode 100644 app/client/src/components/formControls/DatasourceLinkControl.tsx

diff --git a/app/client/src/PluginActionEditor/components/PluginActionForm/components/UQIEditor/FormRender.tsx b/app/client/src/PluginActionEditor/components/PluginActionForm/components/UQIEditor/FormRender.tsx
index 644c090f73a8..eb42337a0d11 100644
--- a/app/client/src/PluginActionEditor/components/PluginActionForm/components/UQIEditor/FormRender.tsx
+++ b/app/client/src/PluginActionEditor/components/PluginActionForm/components/UQIEditor/FormRender.tsx
@@ -183,6 +183,7 @@ const FormRender = (props: Props) => {
           <FieldWrapper
             className="uqi-form-wrapper"
             key={`${configProperty}_${idx}`}
+            style={{ ...section.zoneCustomStyle }}
           >
             <FormControl config={modifiedSection} formName={formName} />
           </FieldWrapper>
@@ -221,7 +222,11 @@ const FormRender = (props: Props) => {
             ? "single_column"
             : "double_column";
 
-        return <Zone layout={layout}>{children}</Zone>;
+        return (
+          <Zone layout={layout} style={{ ...section.zoneCustomStyle }}>
+            {children}
+          </Zone>
+        );
       }
       default:
         return children;
diff --git a/app/client/src/components/formControls/DatasourceLinkControl.tsx b/app/client/src/components/formControls/DatasourceLinkControl.tsx
new file mode 100644
index 000000000000..a102fe954415
--- /dev/null
+++ b/app/client/src/components/formControls/DatasourceLinkControl.tsx
@@ -0,0 +1,63 @@
+import React, { useCallback } from "react";
+import omit from "lodash/omit";
+import history from "utils/history";
+import { Button, type ButtonProps } from "@appsmith/ads";
+import type { ControlType } from "constants/PropertyControlConstants";
+
+import BaseControl from "./BaseControl";
+import type { ControlProps } from "./BaseControl";
+import { useParentEntityInfo } from "ee/IDE/hooks/useParentEntityInfo";
+import { getIDETypeByUrl } from "ee/entities/IDE/utils";
+import { datasourcesEditorIdURL } from "ee/RouteBuilder";
+import { getQueryParams } from "utils/URLUtils";
+
+export interface DatasourceLinkControlProps extends ControlProps {
+  href: string;
+  text: string;
+  size?: ButtonProps["size"];
+  kind?: ButtonProps["kind"];
+  icon?: ButtonProps["startIcon"];
+}
+
+class DatasourceLinkControl extends BaseControl<DatasourceLinkControlProps> {
+  getControlType(): ControlType {
+    return "DATASOURCE_LINK";
+  }
+  render() {
+    return <DatasourceLink {...this.props} />;
+  }
+}
+
+function DatasourceLink(props: DatasourceLinkControlProps) {
+  const { icon, kind = "secondary", size = "sm", text } = props;
+  const ideType = getIDETypeByUrl(location.pathname);
+  const { parentEntityId } = useParentEntityInfo(ideType);
+
+  const onPress = useCallback(() => {
+    const url = datasourcesEditorIdURL({
+      baseParentEntityId: parentEntityId,
+      datasourceId: props.datasourceId as string,
+      params: { ...omit(getQueryParams(), "viewMode"), viewMode: false },
+    });
+
+    history.push(url);
+  }, [parentEntityId, props.datasourceId]);
+
+  return (
+    <Button
+      UNSAFE_width="110px"
+      kind={kind}
+      onClick={onPress}
+      size={size}
+      startIcon={icon}
+    >
+      {text}
+    </Button>
+  );
+}
+
+export interface CheckboxControlProps extends ControlProps {
+  info?: string;
+}
+
+export { DatasourceLinkControl };
diff --git a/app/client/src/components/formControls/FunctionCallingConfigControl/components/FunctionCallingConfigForm.tsx b/app/client/src/components/formControls/FunctionCallingConfigControl/components/FunctionCallingConfigForm.tsx
index 26a7cf0e05d7..20054d4c3a88 100644
--- a/app/client/src/components/formControls/FunctionCallingConfigControl/components/FunctionCallingConfigForm.tsx
+++ b/app/client/src/components/formControls/FunctionCallingConfigControl/components/FunctionCallingConfigForm.tsx
@@ -59,6 +59,7 @@ export const FunctionCallingConfigForm = ({
         </div>
 
         <Button
+          UNSAFE_width="110px"
           kind="secondary"
           onClick={handleAddFunctionButtonClick}
           startIcon="plus"
diff --git a/app/client/src/utils/formControl/FormControlRegistry.tsx b/app/client/src/utils/formControl/FormControlRegistry.tsx
index 9ee6a471e94c..51307c97b877 100644
--- a/app/client/src/utils/formControl/FormControlRegistry.tsx
+++ b/app/client/src/utils/formControl/FormControlRegistry.tsx
@@ -47,6 +47,10 @@ import {
 } from "components/formControls/SliderControl";
 import { HybridSearchControl } from "components/formControls/HybridSearch";
 import FunctionCallingConfigControl from "components/formControls/FunctionCallingConfigControl";
+import {
+  DatasourceLinkControl,
+  type DatasourceLinkControlProps,
+} from "components/formControls/DatasourceLinkControl";
 
 /**
  * NOTE: If you are adding a component that uses FormControl
@@ -240,6 +244,16 @@ class FormControlRegistry {
         },
       },
     );
+    FormControlFactory.registerControlBuilder(
+      formControlTypes.DATASOURCE_LINK,
+      {
+        buildPropertyControl(
+          controlProps: DatasourceLinkControlProps,
+        ): JSX.Element {
+          return <DatasourceLinkControl {...controlProps} />;
+        },
+      },
+    );
   }
 }
 
diff --git a/app/client/src/utils/formControl/formControlTypes.ts b/app/client/src/utils/formControl/formControlTypes.ts
index 86554014d3e7..ba8143bc89dc 100644
--- a/app/client/src/utils/formControl/formControlTypes.ts
+++ b/app/client/src/utils/formControl/formControlTypes.ts
@@ -24,4 +24,5 @@ export default {
   RAG_DOCUMENTS_SELECTOR: "RAG_DOCUMENTS_SELECTOR",
   HYBRID_SEARCH: "HYBRID_SEARCH",
   FUNCTION_CALLING_CONFIG_FORM: "FUNCTION_CALLING_CONFIG_FORM",
+  DATASOURCE_LINK: "DATASOURCE_LINK",
 };

From 266e72a09a76c93895c96556b1009c123712477e Mon Sep 17 00:00:00 2001
From: Pawan Kumar <pawan@appsmith.com>
Date: Fri, 21 Mar 2025 12:00:18 +0530
Subject: [PATCH 2/3] check for emptyness

---
 .../PluginActionForm/components/UQIEditor/FormRender.tsx      | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/app/client/src/PluginActionEditor/components/PluginActionForm/components/UQIEditor/FormRender.tsx b/app/client/src/PluginActionEditor/components/PluginActionForm/components/UQIEditor/FormRender.tsx
index eb42337a0d11..68bad7447554 100644
--- a/app/client/src/PluginActionEditor/components/PluginActionForm/components/UQIEditor/FormRender.tsx
+++ b/app/client/src/PluginActionEditor/components/PluginActionForm/components/UQIEditor/FormRender.tsx
@@ -183,7 +183,7 @@ const FormRender = (props: Props) => {
           <FieldWrapper
             className="uqi-form-wrapper"
             key={`${configProperty}_${idx}`}
-            style={{ ...section.zoneCustomStyle }}
+            style={{ ...(section.zoneCustomStyle || {}) }}
           >
             <FormControl config={modifiedSection} formName={formName} />
           </FieldWrapper>
@@ -223,7 +223,7 @@ const FormRender = (props: Props) => {
             : "double_column";
 
         return (
-          <Zone layout={layout} style={{ ...section.zoneCustomStyle }}>
+          <Zone layout={layout} style={{ ...(section.zoneCustomStyle || {}) }}>
             {children}
           </Zone>
         );

From 456b2858fd9669ae8f683d34bf10261fc5a86228 Mon Sep 17 00:00:00 2001
From: Pawan Kumar <pawan@appsmith.com>
Date: Fri, 21 Mar 2025 12:06:57 +0530
Subject: [PATCH 3/3] remove unecessary code

---
 .../src/components/formControls/DatasourceLinkControl.tsx     | 4 ----
 1 file changed, 4 deletions(-)

diff --git a/app/client/src/components/formControls/DatasourceLinkControl.tsx b/app/client/src/components/formControls/DatasourceLinkControl.tsx
index a102fe954415..884c7ea15262 100644
--- a/app/client/src/components/formControls/DatasourceLinkControl.tsx
+++ b/app/client/src/components/formControls/DatasourceLinkControl.tsx
@@ -56,8 +56,4 @@ function DatasourceLink(props: DatasourceLinkControlProps) {
   );
 }
 
-export interface CheckboxControlProps extends ControlProps {
-  info?: string;
-}
-
 export { DatasourceLinkControl };