From db25d123da29da72c4d28610c9163ceb604cb0a1 Mon Sep 17 00:00:00 2001
From: Pawan Kumar <pawan@appsmith.com>
Date: Fri, 21 Mar 2025 13:23:44 +0530
Subject: [PATCH] chore: Add datasource link control (#39841)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

/ok-to-test tags="@tag.Sanity"

This PR adds an datasource link control that is required to on query
page of AI datasource.

<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->
## Summary by CodeRabbit

## Summary by CodeRabbit

- **New Features**
- Introduced a new form control that allows users to navigate directly
to datasource-related views.
- Enhanced forms with dynamic styling capabilities, enabling sections to
apply custom visual configurations.

- **Style**
- Adjusted the width of a key configuration button to improve layout
consistency.

- **Bug Fixes**
- Resolved issues related to the application of inline styles in form
components.
<!-- end of auto-generated comment: release notes by coderabbit.ai -->

<!-- This is an auto-generated comment: Cypress test results  -->
> [!TIP]
> 🟢 🟢 🟢 All cypress tests have passed! 🎉 🎉 🎉
> Workflow run:
<https://github.com/appsmithorg/appsmith/actions/runs/13986086739>
> Commit: 456b2858fd9669ae8f683d34bf10261fc5a86228
> <a
href="https://internal.appsmith.com/app/cypress-dashboard/rundetails-65890b3c81d7400d08fa9ee5?branch=master&workflowId=13986086739&attempt=1"
target="_blank">Cypress dashboard</a>.
> Tags: `@tag.Sanity`
> Spec:
> <hr>Fri, 21 Mar 2025 07:20:48 UTC
<!-- end of auto-generated comment: Cypress test results  -->
---
 .../components/UQIEditor/FormRender.tsx       |  7 ++-
 .../formControls/DatasourceLinkControl.tsx    | 59 +++++++++++++++++++
 .../components/FunctionCallingConfigForm.tsx  |  1 +
 .../utils/formControl/FormControlRegistry.tsx | 14 +++++
 .../src/utils/formControl/formControlTypes.ts |  1 +
 5 files changed, 81 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 644c090f73..68bad74475 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 0000000000..884c7ea152
--- /dev/null
+++ b/app/client/src/components/formControls/DatasourceLinkControl.tsx
@@ -0,0 +1,59 @@
+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 { DatasourceLinkControl };
diff --git a/app/client/src/components/formControls/FunctionCallingConfigControl/components/FunctionCallingConfigForm.tsx b/app/client/src/components/formControls/FunctionCallingConfigControl/components/FunctionCallingConfigForm.tsx
index 26a7cf0e05..20054d4c3a 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 9ee6a471e9..51307c97b8 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 86554014d3..ba8143bc89 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",
 };