KnowledgeOptions.tsx

import { Knowledge } from "@/api";
import useDeleteKnowledgeApi from "@/hooks/api/useDeleteKnowledgeApi";
import {
  DeleteIcon,
  FileLock2,
  FilePenIcon,
  MoreVertical,
  RefreshCw,
  Tag,
} from "lucide-react";
import { Button } from "@/components/ui/button";
import {
  DropdownMenu,
  DropdownMenuContent,
  DropdownMenuItem,
  DropdownMenuLabel,
  DropdownMenuSeparator,
  DropdownMenuTrigger,
} from "@/components/ui/dropdown-menu";
import { Dialog, DialogTrigger } from "@/components/ui/dialog";
import { useTranslation } from "react-i18next";
import UploadFileDialog from "./dialog/UploadFileDialog";
import { useState } from "react";
import { useGetKnowledgeList } from "@/services/GetKnowledgeListService";
import useFileStatus from "@/hooks/useFileStatus";
import useRetryIngestionApi from "@/hooks/api/useRetryIngestionApi";
import { useUserContext } from "@/services/UserContextService";
import { USER_ANY } from "@/views/knowledge/KnowledgePermissions.tsx";

interface KnowledgeOptionsProps {
  onClickTagEdit: (knowledgeId: string | null) => void;
  onClickPermissionEdit: (knowledgeId: string | null) => void;
  knowledge: Readonly<Knowledge>;
}

// Dropdown menu item of the data table
export default function KnowledgeOptions({
  onClickTagEdit,
  onClickPermissionEdit,
  knowledge,
}: Readonly<KnowledgeOptionsProps>) {
  const { t } = useTranslation();
  const { user } = useUserContext();
  const { handleDeleteKnowledge } = useDeleteKnowledgeApi();
  const { deleteLocalKnowledge } = useGetKnowledgeList();
  const [isDialogOpen, setIsDialogOpen] = useState(false);
  const { registerFiles } = useFileStatus();
  const { retryIngestion } = useRetryIngestionApi();

  //Function to handle the Deletion of a file
  async function handleFileDelete() {
    if (!knowledge.id) return;
    await handleDeleteKnowledge(knowledge.id);
    deleteLocalKnowledge(knowledge);
  }

  async function handleRetryIngestion() {
    if (!knowledge.id) return;
    await retryIngestion(knowledge.id);
    await registerFiles([knowledge.id]);
  }

  const hasUserPermission = () => {
    if (!knowledge.permissions) return false;
    if (knowledge.permissions[USER_ANY] === "READWRITE") return true;
    if (!user?.username) return false;
    return (
      knowledge.permissions[user.username] === "READONLY" ||
      knowledge.permissions[user.username] === "OWNER"
    );
  };

  return (
    <DropdownMenu>
      <DropdownMenuTrigger asChild>
        <Button variant="ghost" className="h-8 w-8 p-0">
          <MoreVertical className="h-4 w-4" />
        </Button>
      </DropdownMenuTrigger>
      <DropdownMenuContent align="end">
        <DropdownMenuLabel>
          {t("knowledgePage.options.actions")}
        </DropdownMenuLabel>
        {hasUserPermission() ? (
          <>
            <DropdownMenuItem
              onClick={() => {
                onClickTagEdit(knowledge.id ?? null);
              }}
            >
              <Tag className="mr-2 h-4 w-4" />
              {t("knowledgePage.options.editTags")}
            </DropdownMenuItem>

            <DropdownMenuItem
              onClick={() => {
                onClickPermissionEdit(knowledge.id ?? null);
              }}
            >
              <FileLock2 className="mr-2 h-4 w-4" />
              {t("knowledgePage.options.editPermissions")}
            </DropdownMenuItem>

            <DropdownMenuSeparator />

            {knowledge.ingestionStatus === "FAILED" && (
              <DropdownMenuItem
                onClick={() => {
                  void handleRetryIngestion();
                }}
              >
                <RefreshCw className="mr-2 h-4 w-4" />
                {t("knowledgePage.options.retry")}
              </DropdownMenuItem>
            )}
            <Dialog open={isDialogOpen} onOpenChange={setIsDialogOpen}>
              <DialogTrigger asChild>
                <DropdownMenuItem
                  onSelect={(e) => {
                    e.preventDefault();
                  }}
                >
                  <FilePenIcon className="mr-2 h-4 w-4" />
                  {t("knowledgePage.options.update")}
                </DropdownMenuItem>
              </DialogTrigger>
              <UploadFileDialog
                knowledge={knowledge}
                onClose={() => {
                  setIsDialogOpen(false);
                }}
                resetSelectedFiles={!isDialogOpen}
              />
            </Dialog>
            <DropdownMenuItem
              onClick={() => {
                void handleFileDelete();
              }}
            >
              <DeleteIcon className="mr-2 h-4 w-4" />
              {t("knowledgePage.options.delete")}
            </DropdownMenuItem>
          </>
        ) : (
          <DropdownMenuItem disabled>
            {t("knowledgePage.options.noPermission")}
          </DropdownMenuItem>
        )}
      </DropdownMenuContent>
    </DropdownMenu>
  );
}