dify/web/app/components/header/account-setting/access-rules-page/access-rule-row-menu.tsx

'use client'

import type { AccessPolicy } from '@/models/access-control'
import {
  AlertDialog,
  AlertDialogActions,
  AlertDialogCancelButton,
  AlertDialogConfirmButton,
  AlertDialogContent,
  AlertDialogDescription,
  AlertDialogTitle,
} from '@langgenius/dify-ui/alert-dialog'
import {
  DropdownMenu,
  DropdownMenuContent,
  DropdownMenuItem,
  DropdownMenuSeparator,
  DropdownMenuTrigger,
} from '@langgenius/dify-ui/dropdown-menu'
import { toast } from '@langgenius/dify-ui/toast'
import { useCallback, useState } from 'react'
import { useTranslation } from 'react-i18next'
import ActionButton from '@/app/components/base/action-button'
import {
  useCopyAccessRule,
  useDeleteAccessRule,
} from '@/service/access-control/use-workspace-access-rules'

type AccessRuleRowMenuProps = {
  rule: AccessPolicy
  onView?: () => void
  onEdit?: () => void
}

const AccessRuleRowMenu = ({ rule, onView, onEdit }: AccessRuleRowMenuProps) => {
  const { t } = useTranslation()
  const [open, setOpen] = useState(false)
  const [showDeleteConfirm, setShowDeleteConfirm] = useState(false)

  const { mutateAsync: copyAccessRule } = useCopyAccessRule(rule.resource_type)
  const { mutateAsync: deleteAccessRule, isPending: isDeletingAccessRule } = useDeleteAccessRule(
    rule.resource_type,
  )

  const handleView = useCallback(() => {
    onView?.()
  }, [onView])

  const handleCopyRules = useCallback(() => {
    copyAccessRule(rule.id, {
      onSuccess: () => {
        toast.success(t(($) => $['accessRule.copied'], { ns: 'permission' }))
        setOpen(false)
      },
    })
  }, [copyAccessRule, rule.id, t])

  const openDeleteConfirm = useCallback(() => {
    setShowDeleteConfirm(true)
    setOpen(false)
  }, [])

  const handleDelete = useCallback(() => {
    deleteAccessRule(rule.id, {
      onSuccess: () => {
        toast.success(t(($) => $['accessRule.deleted'], { ns: 'permission' }))
        setShowDeleteConfirm(false)
      },
    })
  }, [deleteAccessRule, rule.id, t])

  const isBuiltIn = rule.is_builtin

  return (
    <>
      <DropdownMenu open={open} onOpenChange={setOpen}>
        <DropdownMenuTrigger
          render={
            <ActionButton
              size="l"
              className={open ? 'bg-state-base-hover' : ''}
              aria-label={t(($) => $['operation.moreActions'], { ns: 'common' })}
            />
          }
        >
          <span aria-hidden className="i-ri-more-fill h-4 w-4 text-text-tertiary" />
        </DropdownMenuTrigger>
        <DropdownMenuContent placement="bottom-end" sideOffset={4} popupClassName="min-w-[140px]">
          {isBuiltIn ? (
            <DropdownMenuItem
              className="system-sm-semibold text-text-secondary"
              onClick={handleView}
            >
              {t(($) => $['operation.view'], { ns: 'common' })}
            </DropdownMenuItem>
          ) : (
            <DropdownMenuItem className="system-sm-semibold text-text-secondary" onClick={onEdit}>
              {t(($) => $['operation.edit'], { ns: 'common' })}
            </DropdownMenuItem>
          )}
          <DropdownMenuItem
            className="system-sm-semibold text-text-secondary"
            onClick={handleCopyRules}
          >
            {t(($) => $['common.duplicateAction'], { ns: 'permission' })}
          </DropdownMenuItem>
          {!isBuiltIn && (
            <>
              <DropdownMenuSeparator />
              <DropdownMenuItem
                variant="destructive"
                className="system-sm-semibold"
                onClick={openDeleteConfirm}
              >
                {t(($) => $['operation.delete'], { ns: 'common' })}
              </DropdownMenuItem>
            </>
          )}
        </DropdownMenuContent>
      </DropdownMenu>
      <AlertDialog
        open={showDeleteConfirm}
        onOpenChange={(open) => !open && setShowDeleteConfirm(false)}
      >
        <AlertDialogContent backdropProps={{ forceRender: true }}>
          <div className="flex flex-col gap-2 px-6 pt-6 pb-4">
            <AlertDialogTitle className="w-full truncate title-2xl-semi-bold text-text-primary">
              {t(($) => $['accessRule.deleteTitle'], { ns: 'permission', name: rule.name })}
            </AlertDialogTitle>
            <AlertDialogDescription className="w-full system-md-regular wrap-break-word whitespace-pre-wrap text-text-tertiary">
              {t(($) => $['accessRule.deleteDescription'], { ns: 'permission' })}
            </AlertDialogDescription>
          </div>
          <AlertDialogActions>
            <AlertDialogCancelButton>
              {t(($) => $['operation.cancel'], { ns: 'common' })}
            </AlertDialogCancelButton>
            <AlertDialogConfirmButton disabled={isDeletingAccessRule} onClick={handleDelete}>
              {t(($) => $['operation.delete'], { ns: 'common' })}
            </AlertDialogConfirmButton>
          </AlertDialogActions>
        </AlertDialogContent>
      </AlertDialog>
    </>
  )
}

export default AccessRuleRowMenu