adding ability to override copy function from client (#400)

* adding ability to override copy function from client * removing yarn file * fixing empty space * adding callback * adding focus prop
ClickHouse · Apr 24, 2024 · 15f0a0c · 15f0a0c
1 parent 006d5be
commit 15f0a0c
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 7 deletions.
diff --git a/src/components/Grid/Grid.tsx b/src/components/Grid/Grid.tsx
@@ -7,6 +7,7 @@ import {
   forwardRef,
   useCallback,
   useEffect,
+  useMemo,
   useRef,
   useState,
 } from "react";
@@ -136,6 +137,7 @@ export const Grid = forwardRef<HTMLDivElement, GridProps>(
       onMouseMove: onMouseMoveProp,
       showBorder = false,
       onCopy: onCopyProp,
+      onCopyCallback,
       onContextMenu: onContextMenuProp,
       forwardedGridRef,
       ...props
@@ -153,8 +155,8 @@ export const Grid = forwardRef<HTMLDivElement, GridProps>(
         type: "empty",
       }
     );
-    const onCopy = useCallback(async () => {
-      let isCopied = false;
+
+    const defaultOnCopy: () => Promise<void> = useCallback(async () => {
       try {
         await copyGridElements({
           cell,
@@ -164,7 +166,10 @@ export const Grid = forwardRef<HTMLDivElement, GridProps>(
           columnCount,
           outerRef: outerRef,
         });
-        isCopied = true;
+
+        if (onCopyCallback)  {
+          onCopyCallback(true);
+        }
         if (showToast) {
           createToast({
             title: "Copied successfully",
@@ -174,18 +179,31 @@ export const Grid = forwardRef<HTMLDivElement, GridProps>(
         }
       } catch (e) {
         console.error(e);
+
+        if (onCopyCallback)  {
+          onCopyCallback(false);
+        }
+
         if (showToast) {
           createToast({
             title: "Failed to copy",
             description: "Encountered an error while copying. Try again after sometime",
             type: "danger",
           });
         }
-        if (typeof onCopyProp === "function") {
-          onCopyProp(isCopied);
+      }
+    }, [cell, columnCount, focus, focusProp, rowCount, selection, showToast, onCopyCallback]);
+
+    const customOnCopy: () => Promise<void> = useMemo(() => {
+      const result = async () => {
+        if(onCopyProp) {
+          await  onCopyProp(selection, focus)
         }
       }
-    }, [cell, columnCount, focus, focusProp, onCopyProp, rowCount, selection, showToast]);
+      return result;
+    }, [onCopyProp, selection, focus]);
+
+    const onCopy: () => Promise<void> = typeof onCopyProp === "function" ? customOnCopy: defaultOnCopy;
 
     const defaultMenuOptions = [
       {

diff --git a/src/components/Grid/types.ts b/src/components/Grid/types.ts
@@ -198,7 +198,8 @@ export interface GridProps
   onMouseDown?: MouseEventHandler<HTMLDivElement>;
   onMouseMove?: MouseEventHandler<HTMLDivElement>;
   showBorder?: boolean;
-  onCopy?: (isCopied: boolean) => void | Promise<void>;
+  onCopy?: (selection: SelectedRegion, focus: SelectionFocus) => void | Promise<void>;
+  onCopyCallback?: (copied: boolean) => void;
   onContextMenu?: MouseEventHandler<HTMLDivElement>;
   forwardedGridRef?: MutableRefObject<VariableSizeGrid>;
 }

diff --git a/src/components/types.ts b/src/components/types.ts
@@ -56,6 +56,7 @@ export type {
   SelectionFocus,
   SelectionAction,
   GridContextMenuItemProps,
+  Rectangle
 } from "./Grid/types";
 
 export type States = "default" | "active" | "disabled" | "error" | "hover";