@atlaskit/editor-plugin-card 2.6.2 → 2.6.3

package/dist/es2019/ui/EditToolbarButton/index.js

@@ -0,0 +1,278 @@
+/** @jsx jsx */
+import { useCallback, useMemo, useRef, useState } from 'react';
+
+// eslint-disable-next-line @atlaskit/ui-styling-standard/use-compiled -- Ignored via go/DSP-18766
+import { css, jsx } from '@emotion/react';
+import { FormattedMessage } from 'react-intl-next';
+import { linkToolbarMessages, cardMessages as messages } from '@atlaskit/editor-common/messages';
+import { FloatingToolbarButton as Button, FloatingToolbarSeparator as Separator } from '@atlaskit/editor-common/ui';
+import { ArrowKeyNavigationType, DropdownContainer as UiDropdown } from '@atlaskit/editor-common/ui-menu';
+import ExpandIcon from '@atlaskit/icon/glyph/chevron-down';
+import { ButtonItem } from '@atlaskit/menu';
+import { fg } from '@atlaskit/platform-feature-flags';
+import { Flex } from '@atlaskit/primitives';
+import { editDatasource } from '../../ui/EditDatasourceButton';
+import { focusEditorView, isDatasourceConfigEditable } from '../../utils';
+import { CardContextProvider } from '../CardContextProvider';
+import { useFetchDatasourceDataInfo } from '../useFetchDatasourceDataInfo';
+import { useFetchDatasourceInfo } from '../useFetchDatasourceInfo';
+import EditToolbarButtonPresentation from './EditToolbarButtonPresentation';
+const dropdownExpandContainer = css({
+  margin: `0px ${"var(--ds-space-negative-050, -4px)"}`
+});
+const EditToolbarButtonWithCardContext = props => {
+  var _response$datasourceI;
+  const {
+    cardContext,
+    currentAppearance,
+    editorAnalyticsApi,
+    editorView,
+    intl,
+    onLinkEditClick,
+    url
+  } = props;
+  const {
+    extensionKey,
+    ...response
+  } = useFetchDatasourceInfo({
+    isRegularCardNode: true,
+    url,
+    cardContext
+  });
+  const datasourceId = (_response$datasourceI = response.datasourceId) !== null && _response$datasourceI !== void 0 ? _response$datasourceI : props.datasourceId;
+  const [isOpen, setIsOpen] = useState(false);
+  const containerRef = useRef();
+  const toggleOpen = () => setIsOpen(open => !open);
+  const onClose = () => setIsOpen(false);
+  const onEditLink = useCallback(() => {
+    if (editorView) {
+      onLinkEditClick(editorView.state, editorView.dispatch);
+      focusEditorView(editorView);
+    }
+  }, [editorView, onLinkEditClick]);
+  const editVariant = useMemo(() => {
+    const hasUrl = url !== null && url !== undefined;
+    if (!datasourceId || !isDatasourceConfigEditable(datasourceId)) {
+      if (hasUrl) {
+        return 'edit-link';
+      }
+      return 'none';
+    }
+    if (hasUrl) {
+      var _cardContext$store, _urlState$error;
+      const urlState = cardContext === null || cardContext === void 0 ? void 0 : (_cardContext$store = cardContext.store) === null || _cardContext$store === void 0 ? void 0 : _cardContext$store.getState()[url];
+      if ((urlState === null || urlState === void 0 ? void 0 : (_urlState$error = urlState.error) === null || _urlState$error === void 0 ? void 0 : _urlState$error.kind) === 'fatal') {
+        return 'edit-link';
+      }
+      return 'edit-dropdown';
+    } else {
+      return 'edit-datasource';
+    }
+  }, [cardContext === null || cardContext === void 0 ? void 0 : cardContext.store, datasourceId, url]);
+  const onEditDatasource = useCallback(() => {
+    if (editorView && datasourceId) {
+      editDatasource(datasourceId, editorAnalyticsApi, currentAppearance, extensionKey)(editorView.state, editorView.dispatch);
+      focusEditorView(editorView);
+    }
+  }, [currentAppearance, datasourceId, editorAnalyticsApi, editorView, extensionKey]);
+  switch (editVariant) {
+    case 'edit-link':
+      {
+        return jsx(Flex, {
+          gap: "space.050"
+        }, jsx(Button, {
+          testId: "edit-link",
+          onClick: onEditLink
+        }, jsx(FormattedMessage, linkToolbarMessages.editLink)), jsx(Separator, null));
+      }
+    case 'edit-datasource':
+      {
+        return jsx(Flex, {
+          gap: "space.050"
+        }, jsx(Button, {
+          testId: "edit-datasource",
+          tooltipContent: intl.formatMessage(linkToolbarMessages.editDatasourceStandaloneTooltip),
+          onClick: onEditDatasource
+        }, jsx(FormattedMessage, linkToolbarMessages.editDatasourceStandalone)), jsx(Separator, null));
+      }
+    case 'edit-dropdown':
+      {
+        const trigger = jsx(Flex, {
+          gap: "space.050"
+        }, jsx(Button, {
+          testId: "edit-dropdown-trigger",
+          iconAfter: jsx("span", {
+            css: dropdownExpandContainer
+          }, jsx(ExpandIcon, {
+            label: intl.formatMessage(messages.editDropdownTriggerTitle)
+          })),
+          onClick: toggleOpen,
+          selected: isOpen,
+          disabled: false,
+          ariaHasPopup: true
+        }, jsx(FormattedMessage, messages.editDropdownTriggerTitle)), jsx(Separator, null));
+        return jsx(Flex, {
+          ref: containerRef
+        }, jsx(UiDropdown, {
+          mountTo: containerRef.current,
+          isOpen: isOpen,
+          handleClickOutside: onClose,
+          handleEscapeKeydown: onClose,
+          trigger: trigger,
+          scrollableElement: containerRef.current,
+          arrowKeyNavigationProviderOptions: {
+            type: ArrowKeyNavigationType.MENU
+          }
+        }, jsx(ButtonItem, {
+          key: "edit.link",
+          onClick: onEditLink,
+          testId: "edit-dropdown-edit-link-item"
+        }, jsx(FormattedMessage, messages.editDropdownEditLinkTitle)), jsx(ButtonItem, {
+          key: "edit.datasource",
+          onClick: onEditDatasource,
+          testId: "edit-dropdown-edit-datasource-item"
+        }, jsx(FormattedMessage, messages.editDropdownEditDatasourceTitle))));
+      }
+    case 'none':
+    default:
+      return null;
+  }
+};
+const EditToolbarButtonWithUrl = props => {
+  const {
+    cardContext,
+    currentAppearance,
+    editorAnalyticsApi,
+    editorView,
+    intl,
+    onLinkEditClick,
+    url
+  } = props;
+  const {
+    extensionKey,
+    datasourceId
+  } = useFetchDatasourceInfo({
+    isRegularCardNode: true,
+    url,
+    cardContext
+  });
+  const editVariant = useMemo(() => {
+    var _cardContext$store2, _urlState$error2;
+    if (!datasourceId || !isDatasourceConfigEditable(datasourceId)) {
+      return 'edit-link';
+    }
+    const urlState = cardContext === null || cardContext === void 0 ? void 0 : (_cardContext$store2 = cardContext.store) === null || _cardContext$store2 === void 0 ? void 0 : _cardContext$store2.getState()[url];
+    if ((urlState === null || urlState === void 0 ? void 0 : (_urlState$error2 = urlState.error) === null || _urlState$error2 === void 0 ? void 0 : _urlState$error2.kind) === 'fatal') {
+      return 'edit-link';
+    }
+    return 'edit-dropdown';
+  }, [cardContext === null || cardContext === void 0 ? void 0 : cardContext.store, datasourceId, url]);
+  return jsx(EditToolbarButtonPresentation, {
+    datasourceId: datasourceId,
+    currentAppearance: currentAppearance,
+    editorAnalyticsApi: editorAnalyticsApi,
+    editVariant: editVariant,
+    editorView: editorView,
+    extensionKey: extensionKey,
+    onLinkEditClick: onLinkEditClick,
+    intl: intl
+  });
+};
+const EditToolbarButtonWithDatasourceId = props => {
+  const {
+    currentAppearance,
+    editorAnalyticsApi,
+    editorView,
+    intl,
+    onLinkEditClick,
+    datasourceId,
+    node
+  } = props;
+  const fetchData = useMemo(() => {
+    try {
+      var _attrs$datasource$vie, _attrs$datasource$vie2;
+      const attrs = node.attrs;
+      const parameters = attrs.datasource.parameters;
+      const visibleColumnKeys = (_attrs$datasource$vie = attrs.datasource.views[0]) === null || _attrs$datasource$vie === void 0 ? void 0 : (_attrs$datasource$vie2 = _attrs$datasource$vie.properties) === null || _attrs$datasource$vie2 === void 0 ? void 0 : _attrs$datasource$vie2.columns.map(c => c.key);
+      return {
+        parameters,
+        visibleColumnKeys
+      };
+    } catch (e) {
+      // eslint-disable-next-line no-console
+      console.error(e);
+    }
+  }, [node.attrs]);
+  const {
+    extensionKey
+  } = useFetchDatasourceDataInfo({
+    datasourceId,
+    parameters: fetchData === null || fetchData === void 0 ? void 0 : fetchData.parameters,
+    visibleColumnKeys: fetchData === null || fetchData === void 0 ? void 0 : fetchData.visibleColumnKeys
+  });
+  const editVariant = useMemo(() => {
+    if (!datasourceId || !isDatasourceConfigEditable(datasourceId)) {
+      return 'none';
+    }
+    return 'edit-datasource';
+  }, [datasourceId]);
+  return jsx(EditToolbarButtonPresentation, {
+    datasourceId: datasourceId,
+    currentAppearance: currentAppearance,
+    editorAnalyticsApi: editorAnalyticsApi,
+    editVariant: editVariant,
+    editorView: editorView,
+    extensionKey: extensionKey,
+    onLinkEditClick: onLinkEditClick,
+    intl: intl
+  });
+};
+export const EditToolbarButton = props => {
+  const {
+    currentAppearance,
+    datasourceId,
+    editorAnalyticsApi,
+    editorView,
+    intl,
+    onLinkEditClick,
+    url
+  } = props;
+  return jsx(CardContextProvider, null, ({
+    cardContext
+  }) => {
+    if (fg('enable_datasource_nourl_edit_dropdown_datafetch')) {
+      if (props.url) {
+        return jsx(EditToolbarButtonWithUrl, {
+          url: props.url,
+          intl: intl,
+          editorAnalyticsApi: editorAnalyticsApi,
+          editorView: editorView,
+          cardContext: cardContext,
+          onLinkEditClick: onLinkEditClick,
+          currentAppearance: currentAppearance
+        });
+      }
+      if (props.datasourceId && props.node) {
+        return jsx(EditToolbarButtonWithDatasourceId, {
+          datasourceId: props.datasourceId,
+          node: props.node,
+          intl: intl,
+          editorAnalyticsApi: editorAnalyticsApi,
+          editorView: editorView,
+          onLinkEditClick: onLinkEditClick,
+          currentAppearance: currentAppearance
+        });
+      }
+    }
+    return jsx(EditToolbarButtonWithCardContext, {
+      datasourceId: datasourceId,
+      url: url,
+      intl: intl,
+      editorAnalyticsApi: editorAnalyticsApi,
+      editorView: editorView,
+      cardContext: cardContext,
+      onLinkEditClick: onLinkEditClick,
+      currentAppearance: currentAppearance
+    });
+  });
+};

package/dist/es2019/ui/EditToolbarButton/types.js

@@ -0,0 +1 @@
+export {};

package/dist/es2019/ui/useFetchDatasourceDataInfo.js

@@ -0,0 +1,41 @@
+/** @jsx jsx */
+import { useEffect, useState } from 'react';
+import { DEFAULT_GET_DATASOURCE_DATA_PAGE_SIZE, useDatasourceClientExtension } from '@atlaskit/link-client-extension';
+export const useFetchDatasourceDataInfo = ({
+  datasourceId,
+  parameters,
+  visibleColumnKeys
+}) => {
+  const [extensionKey, setExtensionKey] = useState(undefined);
+  const {
+    getDatasourceData
+  } = useDatasourceClientExtension();
+  useEffect(() => {
+    const fetchDatasource = async () => {
+      try {
+        if (!datasourceId || !parameters || !visibleColumnKeys) {
+          return;
+        }
+        const datasourceDataRequest = {
+          parameters,
+          pageSize: DEFAULT_GET_DATASOURCE_DATA_PAGE_SIZE,
+          pageCursor: undefined,
+          fields: visibleColumnKeys,
+          includeSchema: true
+        };
+        const {
+          meta
+        } = await getDatasourceData(datasourceId, datasourceDataRequest, false);
+        setExtensionKey(meta.extensionKey);
+      } catch (e) {
+        // eslint-disable-next-line no-console
+        console.error(e);
+        setExtensionKey(undefined);
+      }
+    };
+    void fetchDatasource();
+  }, [getDatasourceData, visibleColumnKeys, parameters, datasourceId]);
+  return {
+    extensionKey
+  };
+};

package/dist/esm/toolbar.js

@@ -466,6 +466,7 @@ var getDatasourceButtonGroup = function getDatasourceButtonGroup(metadata, intl,
       render: function render(editorView) {
         return /*#__PURE__*/React.createElement(EditToolbarButton, {
           datasourceId: datasourceId,
+          node: node,
           key: "edit-toolbar-item",
           url: metadata.url,
           intl: intl,

package/dist/esm/ui/EditToolbarButton/EditToolbarButtonPresentation.js

@@ -0,0 +1,116 @@
+import _slicedToArray from "@babel/runtime/helpers/slicedToArray";
+/** @jsx jsx */
+import { useCallback, useRef, useState } from 'react';
+
+// eslint-disable-next-line @atlaskit/ui-styling-standard/use-compiled -- Ignored via go/DSP-18766
+import { css, jsx } from '@emotion/react';
+import { FormattedMessage } from 'react-intl-next';
+import { linkToolbarMessages, cardMessages as messages } from '@atlaskit/editor-common/messages';
+import { FloatingToolbarButton as Button, FloatingToolbarSeparator as Separator } from '@atlaskit/editor-common/ui';
+import { ArrowKeyNavigationType, DropdownContainer as UiDropdown } from '@atlaskit/editor-common/ui-menu';
+import ExpandIcon from '@atlaskit/icon/glyph/chevron-down';
+import { ButtonItem } from '@atlaskit/menu';
+import { Flex } from '@atlaskit/primitives';
+import { focusEditorView } from '../../utils';
+import { editDatasource } from '../EditDatasourceButton';
+var dropdownExpandContainer = css({
+  margin: "0px ".concat("var(--ds-space-negative-050, -4px)")
+});
+var EditToolbarButtonPresentation = function EditToolbarButtonPresentation(_ref) {
+  var datasourceId = _ref.datasourceId,
+    currentAppearance = _ref.currentAppearance,
+    editorAnalyticsApi = _ref.editorAnalyticsApi,
+    editVariant = _ref.editVariant,
+    editorView = _ref.editorView,
+    extensionKey = _ref.extensionKey,
+    onLinkEditClick = _ref.onLinkEditClick,
+    intl = _ref.intl;
+  var _useState = useState(false),
+    _useState2 = _slicedToArray(_useState, 2),
+    isOpen = _useState2[0],
+    setIsOpen = _useState2[1];
+  var containerRef = useRef();
+  var toggleOpen = function toggleOpen() {
+    return setIsOpen(function (open) {
+      return !open;
+    });
+  };
+  var onClose = function onClose() {
+    return setIsOpen(false);
+  };
+  var onEditLink = useCallback(function () {
+    if (editorView) {
+      onLinkEditClick(editorView.state, editorView.dispatch);
+      focusEditorView(editorView);
+    }
+  }, [editorView, onLinkEditClick]);
+  var onEditDatasource = useCallback(function () {
+    if (editorView && datasourceId) {
+      editDatasource(datasourceId, editorAnalyticsApi, currentAppearance, extensionKey)(editorView.state, editorView.dispatch);
+      focusEditorView(editorView);
+    }
+  }, [currentAppearance, datasourceId, editorAnalyticsApi, editorView, extensionKey]);
+  switch (editVariant) {
+    case 'edit-link':
+      {
+        return jsx(Flex, {
+          gap: "space.050"
+        }, jsx(Button, {
+          testId: "edit-link",
+          onClick: onEditLink
+        }, jsx(FormattedMessage, linkToolbarMessages.editLink)), jsx(Separator, null));
+      }
+    case 'edit-datasource':
+      {
+        return jsx(Flex, {
+          gap: "space.050"
+        }, jsx(Button, {
+          testId: "edit-datasource",
+          tooltipContent: intl.formatMessage(linkToolbarMessages.editDatasourceStandaloneTooltip),
+          onClick: onEditDatasource
+        }, jsx(FormattedMessage, linkToolbarMessages.editDatasourceStandalone)), jsx(Separator, null));
+      }
+    case 'edit-dropdown':
+      {
+        var trigger = jsx(Flex, {
+          gap: "space.050"
+        }, jsx(Button, {
+          testId: "edit-dropdown-trigger",
+          iconAfter: jsx("span", {
+            css: dropdownExpandContainer
+          }, jsx(ExpandIcon, {
+            label: intl.formatMessage(messages.editDropdownTriggerTitle)
+          })),
+          onClick: toggleOpen,
+          selected: isOpen,
+          disabled: false,
+          ariaHasPopup: true
+        }, jsx(FormattedMessage, messages.editDropdownTriggerTitle)), jsx(Separator, null));
+        return jsx(Flex, {
+          ref: containerRef
+        }, jsx(UiDropdown, {
+          mountTo: containerRef.current,
+          isOpen: isOpen,
+          handleClickOutside: onClose,
+          handleEscapeKeydown: onClose,
+          trigger: trigger,
+          scrollableElement: containerRef.current,
+          arrowKeyNavigationProviderOptions: {
+            type: ArrowKeyNavigationType.MENU
+          }
+        }, jsx(ButtonItem, {
+          key: "edit.link",
+          onClick: onEditLink,
+          testId: "edit-dropdown-edit-link-item"
+        }, jsx(FormattedMessage, messages.editDropdownEditLinkTitle)), jsx(ButtonItem, {
+          key: "edit.datasource",
+          onClick: onEditDatasource,
+          testId: "edit-dropdown-edit-datasource-item"
+        }, jsx(FormattedMessage, messages.editDropdownEditDatasourceTitle))));
+      }
+    case 'none':
+    default:
+      return null;
+  }
+};
+export default EditToolbarButtonPresentation;

package/dist/esm/ui/{EditToolbarButton.js → EditToolbarButton/index.js}

@@ -12,11 +12,14 @@ import { FloatingToolbarButton as Button, FloatingToolbarSeparator as Separator
 import { ArrowKeyNavigationType, DropdownContainer as UiDropdown } from '@atlaskit/editor-common/ui-menu';
 import ExpandIcon from '@atlaskit/icon/glyph/chevron-down';
 import { ButtonItem } from '@atlaskit/menu';
+import { fg } from '@atlaskit/platform-feature-flags';
 import { Flex } from '@atlaskit/primitives';
-import { editDatasource } from '../ui/EditDatasourceButton';
-import { focusEditorView, isDatasourceConfigEditable } from '../utils';
-import { CardContextProvider } from './CardContextProvider';
-import { useFetchDatasourceInfo } from './useFetchDatasourceInfo';
+import { editDatasource } from '../../ui/EditDatasourceButton';
+import { focusEditorView, isDatasourceConfigEditable } from '../../utils';
+import { CardContextProvider } from '../CardContextProvider';
+import { useFetchDatasourceDataInfo } from '../useFetchDatasourceDataInfo';
+import { useFetchDatasourceInfo } from '../useFetchDatasourceInfo';
+import EditToolbarButtonPresentation from './EditToolbarButtonPresentation';
 var dropdownExpandContainer = css({
   margin: "0px ".concat("var(--ds-space-negative-050, -4px)")
 });
@@ -144,6 +147,91 @@ var EditToolbarButtonWithCardContext = function EditToolbarButtonWithCardContext
       return null;
   }
 };
+var EditToolbarButtonWithUrl = function EditToolbarButtonWithUrl(props) {
+  var cardContext = props.cardContext,
+    currentAppearance = props.currentAppearance,
+    editorAnalyticsApi = props.editorAnalyticsApi,
+    editorView = props.editorView,
+    intl = props.intl,
+    onLinkEditClick = props.onLinkEditClick,
+    url = props.url;
+  var _useFetchDatasourceIn2 = useFetchDatasourceInfo({
+      isRegularCardNode: true,
+      url: url,
+      cardContext: cardContext
+    }),
+    extensionKey = _useFetchDatasourceIn2.extensionKey,
+    datasourceId = _useFetchDatasourceIn2.datasourceId;
+  var editVariant = useMemo(function () {
+    var _cardContext$store2, _urlState$error2;
+    if (!datasourceId || !isDatasourceConfigEditable(datasourceId)) {
+      return 'edit-link';
+    }
+    var urlState = cardContext === null || cardContext === void 0 || (_cardContext$store2 = cardContext.store) === null || _cardContext$store2 === void 0 ? void 0 : _cardContext$store2.getState()[url];
+    if ((urlState === null || urlState === void 0 || (_urlState$error2 = urlState.error) === null || _urlState$error2 === void 0 ? void 0 : _urlState$error2.kind) === 'fatal') {
+      return 'edit-link';
+    }
+    return 'edit-dropdown';
+  }, [cardContext === null || cardContext === void 0 ? void 0 : cardContext.store, datasourceId, url]);
+  return jsx(EditToolbarButtonPresentation, {
+    datasourceId: datasourceId,
+    currentAppearance: currentAppearance,
+    editorAnalyticsApi: editorAnalyticsApi,
+    editVariant: editVariant,
+    editorView: editorView,
+    extensionKey: extensionKey,
+    onLinkEditClick: onLinkEditClick,
+    intl: intl
+  });
+};
+var EditToolbarButtonWithDatasourceId = function EditToolbarButtonWithDatasourceId(props) {
+  var currentAppearance = props.currentAppearance,
+    editorAnalyticsApi = props.editorAnalyticsApi,
+    editorView = props.editorView,
+    intl = props.intl,
+    onLinkEditClick = props.onLinkEditClick,
+    datasourceId = props.datasourceId,
+    node = props.node;
+  var fetchData = useMemo(function () {
+    try {
+      var _attrs$datasource$vie;
+      var attrs = node.attrs;
+      var parameters = attrs.datasource.parameters;
+      var visibleColumnKeys = (_attrs$datasource$vie = attrs.datasource.views[0]) === null || _attrs$datasource$vie === void 0 || (_attrs$datasource$vie = _attrs$datasource$vie.properties) === null || _attrs$datasource$vie === void 0 ? void 0 : _attrs$datasource$vie.columns.map(function (c) {
+        return c.key;
+      });
+      return {
+        parameters: parameters,
+        visibleColumnKeys: visibleColumnKeys
+      };
+    } catch (e) {
+      // eslint-disable-next-line no-console
+      console.error(e);
+    }
+  }, [node.attrs]);
+  var _useFetchDatasourceDa = useFetchDatasourceDataInfo({
+      datasourceId: datasourceId,
+      parameters: fetchData === null || fetchData === void 0 ? void 0 : fetchData.parameters,
+      visibleColumnKeys: fetchData === null || fetchData === void 0 ? void 0 : fetchData.visibleColumnKeys
+    }),
+    extensionKey = _useFetchDatasourceDa.extensionKey;
+  var editVariant = useMemo(function () {
+    if (!datasourceId || !isDatasourceConfigEditable(datasourceId)) {
+      return 'none';
+    }
+    return 'edit-datasource';
+  }, [datasourceId]);
+  return jsx(EditToolbarButtonPresentation, {
+    datasourceId: datasourceId,
+    currentAppearance: currentAppearance,
+    editorAnalyticsApi: editorAnalyticsApi,
+    editVariant: editVariant,
+    editorView: editorView,
+    extensionKey: extensionKey,
+    onLinkEditClick: onLinkEditClick,
+    intl: intl
+  });
+};
 export var EditToolbarButton = function EditToolbarButton(props) {
   var currentAppearance = props.currentAppearance,
     datasourceId = props.datasourceId,
@@ -154,6 +242,30 @@ export var EditToolbarButton = function EditToolbarButton(props) {
     url = props.url;
   return jsx(CardContextProvider, null, function (_ref) {
     var cardContext = _ref.cardContext;
+    if (fg('enable_datasource_nourl_edit_dropdown_datafetch')) {
+      if (props.url) {
+        return jsx(EditToolbarButtonWithUrl, {
+          url: props.url,
+          intl: intl,
+          editorAnalyticsApi: editorAnalyticsApi,
+          editorView: editorView,
+          cardContext: cardContext,
+          onLinkEditClick: onLinkEditClick,
+          currentAppearance: currentAppearance
+        });
+      }
+      if (props.datasourceId && props.node) {
+        return jsx(EditToolbarButtonWithDatasourceId, {
+          datasourceId: props.datasourceId,
+          node: props.node,
+          intl: intl,
+          editorAnalyticsApi: editorAnalyticsApi,
+          editorView: editorView,
+          onLinkEditClick: onLinkEditClick,
+          currentAppearance: currentAppearance
+        });
+      }
+    }
     return jsx(EditToolbarButtonWithCardContext, {
       datasourceId: datasourceId,
       url: url,

package/dist/esm/ui/EditToolbarButton/types.js

@@ -0,0 +1 @@
+export {};

package/dist/esm/ui/useFetchDatasourceDataInfo.js

@@ -0,0 +1,67 @@
+import _asyncToGenerator from "@babel/runtime/helpers/asyncToGenerator";
+import _slicedToArray from "@babel/runtime/helpers/slicedToArray";
+import _regeneratorRuntime from "@babel/runtime/regenerator";
+/** @jsx jsx */
+import { useEffect, useState } from 'react';
+import { DEFAULT_GET_DATASOURCE_DATA_PAGE_SIZE, useDatasourceClientExtension } from '@atlaskit/link-client-extension';
+export var useFetchDatasourceDataInfo = function useFetchDatasourceDataInfo(_ref) {
+  var datasourceId = _ref.datasourceId,
+    parameters = _ref.parameters,
+    visibleColumnKeys = _ref.visibleColumnKeys;
+  var _useState = useState(undefined),
+    _useState2 = _slicedToArray(_useState, 2),
+    extensionKey = _useState2[0],
+    setExtensionKey = _useState2[1];
+  var _useDatasourceClientE = useDatasourceClientExtension(),
+    getDatasourceData = _useDatasourceClientE.getDatasourceData;
+  useEffect(function () {
+    var fetchDatasource = /*#__PURE__*/function () {
+      var _ref2 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee() {
+        var datasourceDataRequest, _yield$getDatasourceD, meta;
+        return _regeneratorRuntime.wrap(function _callee$(_context) {
+          while (1) switch (_context.prev = _context.next) {
+            case 0:
+              _context.prev = 0;
+              if (!(!datasourceId || !parameters || !visibleColumnKeys)) {
+                _context.next = 3;
+                break;
+              }
+              return _context.abrupt("return");
+            case 3:
+              datasourceDataRequest = {
+                parameters: parameters,
+                pageSize: DEFAULT_GET_DATASOURCE_DATA_PAGE_SIZE,
+                pageCursor: undefined,
+                fields: visibleColumnKeys,
+                includeSchema: true
+              };
+              _context.next = 6;
+              return getDatasourceData(datasourceId, datasourceDataRequest, false);
+            case 6:
+              _yield$getDatasourceD = _context.sent;
+              meta = _yield$getDatasourceD.meta;
+              setExtensionKey(meta.extensionKey);
+              _context.next = 15;
+              break;
+            case 11:
+              _context.prev = 11;
+              _context.t0 = _context["catch"](0);
+              // eslint-disable-next-line no-console
+              console.error(_context.t0);
+              setExtensionKey(undefined);
+            case 15:
+            case "end":
+              return _context.stop();
+          }
+        }, _callee, null, [[0, 11]]);
+      }));
+      return function fetchDatasource() {
+        return _ref2.apply(this, arguments);
+      };
+    }();
+    void fetchDatasource();
+  }, [getDatasourceData, visibleColumnKeys, parameters, datasourceId]);
+  return {
+    extensionKey: extensionKey
+  };
+};

package/dist/types/ui/EditToolbarButton/EditToolbarButtonPresentation.d.ts

@@ -0,0 +1,9 @@
+import { jsx } from '@emotion/react';
+import type { EditDatasourceToolbarButtonWithCommonProps, EditVariant } from './types';
+interface EditToolbarPresentationProps extends EditDatasourceToolbarButtonWithCommonProps {
+    extensionKey?: string;
+    datasourceId?: string;
+    editVariant: EditVariant;
+}
+declare const EditToolbarButtonPresentation: ({ datasourceId, currentAppearance, editorAnalyticsApi, editVariant, editorView, extensionKey, onLinkEditClick, intl, }: EditToolbarPresentationProps) => jsx.JSX.Element | null;
+export default EditToolbarButtonPresentation;

package/dist/types/ui/EditToolbarButton/index.d.ts

@@ -0,0 +1,3 @@
+import { jsx } from '@emotion/react';
+import type { EditDatasourceToolbarButtonProps } from './types';
+export declare const EditToolbarButton: (props: EditDatasourceToolbarButtonProps) => jsx.JSX.Element;