Feat: Add LinkToDatasetDialog #3221 (#4500)

### What problem does this PR solve? Feat: Add LinkToDatasetDialog #3221 ### Type of change - [x] New Feature (non-breaking change which adds functionality)
infiniflow · Jan 16, 2025 · a75cda4 · a75cda4
1 parent 961e8c4
commit a75cda4
Show file tree

Hide file tree

Showing 10 changed files with 267 additions and 45 deletions.
diff --git a/web/src/components/ui/badge.tsx b/web/src/components/ui/badge.tsx
@@ -16,7 +16,7 @@ const badgeVariants = cva(
           'border-transparent bg-destructive text-destructive-foreground hover:bg-destructive/80',
         outline: 'text-foreground',
         tertiary:
-          'border-transparent bg-colors-text-core-standard text-foreground hover:bg-colors-text-core-standard/80',
+          'border-transparent bg-colors-background-core-strong text-colors-text-persist-light hover:bg-colors-background-core-strong/80',
       },
     },
     defaultVariants: {

diff --git a/web/src/components/ui/button.tsx b/web/src/components/ui/button.tsx
@@ -19,14 +19,15 @@ const buttonVariants = cva(
         ghost: 'hover:bg-accent hover:text-accent-foreground',
         link: 'text-primary underline-offset-4 hover:underline',
         tertiary:
-          'bg-colors-text-core-standard text-foreground hover:bg-colors-text-core-standard/80',
+          'bg-colors-background-sentiment-solid-primary text-colors-text-persist-light hover:bg-colors-background-sentiment-solid-primary/80',
         icon: 'bg-colors-background-inverse-standard text-foreground hover:bg-colors-background-inverse-standard/80',
       },
       size: {
         default: 'h-10 px-4 py-2',
         sm: 'h-9 rounded-md px-3',
         lg: 'h-11 rounded-md px-8',
         icon: 'h-10 w-10',
+        auto: 'h-full px-1',
       },
     },
     defaultVariants: {

diff --git a/web/src/components/ui/loading-button.tsx b/web/src/components/ui/loading-button.tsx
@@ -20,6 +20,8 @@ const buttonVariants = cva(
           'bg-secondary text-secondary-foreground hover:bg-secondary/80',
         ghost: 'hover:bg-accent hover:text-accent-foreground',
         link: 'text-primary underline-offset-4 hover:underline',
+        tertiary:
+          'bg-colors-background-sentiment-solid-primary text-colors-text-persist-light hover:bg-colors-background-sentiment-solid-primary/80',
       },
       size: {
         default: 'h-10 px-4 py-2',

diff --git a/web/src/hooks/knowledge-hooks.ts b/web/src/hooks/knowledge-hooks.ts
@@ -72,6 +72,17 @@ export const useFetchKnowledgeList = (
   return { list: data, loading };
 };
 
+export const useSelectKnowledgeOptions = () => {
+  const { list } = useFetchKnowledgeList();
+
+  const options = list?.map((item) => ({
+    label: item.name,
+    value: item.id,
+  }));
+
+  return options;
+};
+
 export const useInfiniteFetchKnowledgeList = () => {
   const { searchString, handleInputChange } = useHandleSearchChange();
   const debouncedSearchString = useDebounce(searchString, { wait: 500 });

diff --git a/web/src/pages/files/action-cell.tsx b/web/src/pages/files/action-cell.tsx
@@ -0,0 +1,61 @@
+import { ConfirmDeleteDialog } from '@/components/confirm-delete-dialog';
+import { Button } from '@/components/ui/button';
+import {
+  DropdownMenu,
+  DropdownMenuContent,
+  DropdownMenuItem,
+  DropdownMenuLabel,
+  DropdownMenuSeparator,
+  DropdownMenuTrigger,
+} from '@/components/ui/dropdown-menu';
+import { IFile } from '@/interfaces/database/file-manager';
+import { CellContext } from '@tanstack/react-table';
+import { EllipsisVertical, Link2, Trash2 } from 'lucide-react';
+import { useCallback } from 'react';
+import { UseHandleConnectToKnowledgeReturnType } from './hooks';
+
+type IProps = Pick<CellContext<IFile, unknown>, 'row'> &
+  Pick<UseHandleConnectToKnowledgeReturnType, 'showConnectToKnowledgeModal'>;
+
+export function ActionCell({ row, showConnectToKnowledgeModal }: IProps) {
+  const record = row.original;
+
+  const handleShowConnectToKnowledgeModal = useCallback(() => {
+    showConnectToKnowledgeModal(record);
+  }, [record, showConnectToKnowledgeModal]);
+
+  return (
+    <section className="flex gap-4 items-center">
+      <Button
+        variant="secondary"
+        size={'icon'}
+        onClick={handleShowConnectToKnowledgeModal}
+      >
+        <Link2 />
+      </Button>
+      <ConfirmDeleteDialog>
+        <Button variant="secondary" size={'icon'}>
+          <Trash2 />
+        </Button>
+      </ConfirmDeleteDialog>
+      <DropdownMenu>
+        <DropdownMenuTrigger asChild>
+          <Button variant="secondary" size={'icon'}>
+            <EllipsisVertical />
+          </Button>
+        </DropdownMenuTrigger>
+        <DropdownMenuContent align="end">
+          <DropdownMenuLabel>Actions</DropdownMenuLabel>
+          <DropdownMenuItem
+            onClick={() => navigator.clipboard.writeText(record.id)}
+          >
+            Copy payment ID
+          </DropdownMenuItem>
+          <DropdownMenuSeparator />
+          <DropdownMenuItem>View customer</DropdownMenuItem>
+          <DropdownMenuItem>View payment details</DropdownMenuItem>
+        </DropdownMenuContent>
+      </DropdownMenu>
+    </section>
+  );
+}
diff --git a/web/src/pages/files/files-table.tsx b/web/src/pages/files/files-table.tsx
@@ -11,22 +11,14 @@ import {
   getSortedRowModel,
   useReactTable,
 } from '@tanstack/react-table';
-import { ArrowUpDown, MoreHorizontal, Pencil } from 'lucide-react';
+import { ArrowUpDown } from 'lucide-react';
 import * as React from 'react';
 
 import SvgIcon from '@/components/svg-icon';
 import { TableEmpty, TableSkeleton } from '@/components/table-skeleton';
+import { Badge } from '@/components/ui/badge';
 import { Button } from '@/components/ui/button';
 import { Checkbox } from '@/components/ui/checkbox';
-import {
-  DropdownMenu,
-  DropdownMenuContent,
-  DropdownMenuItem,
-  DropdownMenuLabel,
-  DropdownMenuSeparator,
-  DropdownMenuTrigger,
-} from '@/components/ui/dropdown-menu';
-import { Switch } from '@/components/ui/switch';
 import {
   Table,
   TableBody,
@@ -48,7 +40,9 @@ import { formatDate } from '@/utils/date';
 import { getExtension } from '@/utils/document-util';
 import { useMemo } from 'react';
 import { useTranslation } from 'react-i18next';
-import { useNavigateToOtherFolder } from './hooks';
+import { ActionCell } from './action-cell';
+import { useHandleConnectToKnowledge, useNavigateToOtherFolder } from './hooks';
+import { LinkToDatasetDialog } from './link-to-dataset-dialog';
 
 export function FilesTable() {
   const [sorting, setSorting] = React.useState<SortingState>([]);
@@ -62,6 +56,14 @@ export function FilesTable() {
     keyPrefix: 'fileManager',
   });
   const navigateToOtherFolder = useNavigateToOtherFolder();
+  const {
+    connectToKnowledgeVisible,
+    hideConnectToKnowledgeModal,
+    showConnectToKnowledgeModal,
+    initialConnectedIds,
+    onConnectToKnowledgeOk,
+    connectToKnowledgeLoading,
+  } = useHandleConnectToKnowledge();
 
   const { pagination, data, loading, setPagination } = useFetchFileList();
 
@@ -176,44 +178,37 @@ export function FilesTable() {
     {
       accessorKey: 'kbs_info',
       header: t('knowledgeBase'),
-      cell: ({ row }) => (
-        <Button variant="destructive" size={'sm'}>
-          {row.getValue('kbs_info')}
-        </Button>
-      ),
+      cell: ({ row }) => {
+        const value = row.getValue('kbs_info');
+        return Array.isArray(value) ? (
+          <section className="flex gap-2 items-center">
+            {value?.slice(0, 2).map((x) => (
+              <Badge key={x.kb_id} className="" variant={'tertiary'}>
+                {x.kb_name}
+              </Badge>
+            ))}
+
+            {value.length > 2 && (
+              <Button variant={'icon'} size={'auto'}>
+                +{value.length - 2}
+              </Button>
+            )}
+          </section>
+        ) : (
+          ''
+        );
+      },
     },
     {
       id: 'actions',
       header: t('action'),
       enableHiding: false,
       cell: ({ row }) => {
-        const payment = row.original;
-
         return (
-          <section className="flex gap-4 items-center">
-            <Switch id="airplane-mode" />
-            <Button variant="secondary" size={'icon'}>
-              <Pencil />
-            </Button>
-            <DropdownMenu>
-              <DropdownMenuTrigger asChild>
-                <Button variant="secondary" size={'icon'}>
-                  <MoreHorizontal />
-                </Button>
-              </DropdownMenuTrigger>
-              <DropdownMenuContent align="end">
-                <DropdownMenuLabel>Actions</DropdownMenuLabel>
-                <DropdownMenuItem
-                  onClick={() => navigator.clipboard.writeText(payment.id)}
-                >
-                  Copy payment ID
-                </DropdownMenuItem>
-                <DropdownMenuSeparator />
-                <DropdownMenuItem>View customer</DropdownMenuItem>
-                <DropdownMenuItem>View payment details</DropdownMenuItem>
-              </DropdownMenuContent>
-            </DropdownMenu>
-          </section>
+          <ActionCell
+            row={row}
+            showConnectToKnowledgeModal={showConnectToKnowledgeModal}
+          ></ActionCell>
         );
       },
     },
@@ -338,6 +333,14 @@ export function FilesTable() {
           </Button>
         </div>
       </div>
+      {connectToKnowledgeVisible && (
+        <LinkToDatasetDialog
+          hideModal={hideConnectToKnowledgeModal}
+          initialConnectedIds={initialConnectedIds}
+          onConnectToKnowledgeOk={onConnectToKnowledgeOk}
+          loading={connectToKnowledgeLoading}
+        ></LinkToDatasetDialog>
+      )}
     </div>
   );
 }
diff --git a/web/src/pages/files/hooks.ts b/web/src/pages/files/hooks.ts
@@ -224,7 +224,7 @@ export const useHandleConnectToKnowledge = () => {
   );
 
   return {
-    initialValue,
+    initialConnectedIds: initialValue,
     connectToKnowledgeLoading: loading,
     onConnectToKnowledgeOk,
     connectToKnowledgeVisible,
@@ -233,6 +233,10 @@ export const useHandleConnectToKnowledge = () => {
   };
 };
 
+export type UseHandleConnectToKnowledgeReturnType = ReturnType<
+  typeof useHandleConnectToKnowledge
+>;
+
 export const useHandleBreadcrumbClick = () => {
   const navigate = useNavigate();