dify/web/app/components/plugins/reference-setting-modal/index.tsx

'use client'
import type { FC } from 'react'
import type { AutoUpdateConfig } from './auto-update-setting/types'
import type { Permissions, ReferenceSetting } from '@/app/components/plugins/types'
import { Button } from '@langgenius/dify-ui/button'
import { Dialog, DialogCloseButton, DialogContent } from '@langgenius/dify-ui/dialog'
import { useSuspenseQuery } from '@tanstack/react-query'
import * as React from 'react'
import { useCallback, useState } from 'react'
import { useTranslation } from 'react-i18next'
import { PermissionType } from '@/app/components/plugins/types'
import OptionCard from '@/app/components/workflow/nodes/_base/components/option-card'
import { systemFeaturesQueryOptions } from '@/features/system-features/client'
import AutoUpdateSetting from './auto-update-setting'
import { AUTO_UPDATE_MODE, AUTO_UPDATE_STRATEGY } from './auto-update-setting/types'
import Label from './label'

const i18nPrefix = 'privilege'
const autoUpdateDefaultValue: AutoUpdateConfig = {
  strategy_setting: AUTO_UPDATE_STRATEGY.fixOnly,
  upgrade_time_of_day: 36000,
  upgrade_mode: AUTO_UPDATE_MODE.update_all,
  exclude_plugins: [],
  include_plugins: [],
}

type Props = Readonly<{
  payload: ReferenceSetting
  canSetPermissions?: boolean
  canSetAutoUpdate?: boolean
  onHide: () => void
  onSave: (payload: ReferenceSetting) => void
}>

const PluginSettingModal: FC<Props> = ({
  payload,
  canSetPermissions = false,
  canSetAutoUpdate = false,
  onHide,
  onSave,
}) => {
  const { t } = useTranslation()
  const { auto_upgrade: autoUpdateConfig, permission: privilege } = payload || {}
  const [tempPrivilege, setTempPrivilege] = useState<Permissions>(privilege)
  const [tempAutoUpdateConfig, setTempAutoUpdateConfig] = useState<AutoUpdateConfig>(
    autoUpdateConfig || autoUpdateDefaultValue,
  )
  const { data: systemFeatures } = useSuspenseQuery(systemFeaturesQueryOptions())
  const isPermissionDisabledByRBAC = systemFeatures.rbac_enabled
  const permissionDisabledTip = t(($) => $[`${i18nPrefix}.configurePermissionsInSettings`], {
    ns: 'plugin',
  })
  const handlePrivilegeChange = useCallback(
    (key: string) => {
      return (value: PermissionType) => {
        if (isPermissionDisabledByRBAC) return
        setTempPrivilege({
          ...tempPrivilege,
          [key]: value,
        })
      }
    },
    [isPermissionDisabledByRBAC, tempPrivilege],
  )

  const handleSave = useCallback(async () => {
    await onSave({
      permission: canSetPermissions ? tempPrivilege : privilege || tempPrivilege,
      auto_upgrade: canSetAutoUpdate
        ? tempAutoUpdateConfig
        : autoUpdateConfig || tempAutoUpdateConfig,
    })
    onHide()
  }, [
    autoUpdateConfig,
    canSetAutoUpdate,
    canSetPermissions,
    onHide,
    onSave,
    privilege,
    tempAutoUpdateConfig,
    tempPrivilege,
  ])

  return (
    <Dialog
      open
      onOpenChange={(open) => {
        if (!open) onHide()
      }}
    >
      <DialogContent className="w-155 max-w-155 overflow-hidden! border-none p-0! text-left align-middle">
        <DialogCloseButton />

        <div className="shadows-shadow-xl flex w-full flex-col items-start rounded-2xl border border-components-panel-border bg-components-panel-bg">
          <div className="flex items-start gap-2 self-stretch pt-6 pr-14 pb-3 pl-6">
            <span className="self-stretch title-2xl-semi-bold text-text-primary">
              {t(($) => $[`${i18nPrefix}.title`], { ns: 'plugin' })}
            </span>
          </div>
          {canSetPermissions && (
            <div className="flex flex-col items-start justify-center gap-4 self-stretch px-6 py-3">
              {[
                {
                  title: t(($) => $[`${i18nPrefix}.whoCanInstall`], { ns: 'plugin' }),
                  key: 'install_permission',
                  value: tempPrivilege?.install_permission || PermissionType.noOne,
                },
                {
                  title: t(($) => $[`${i18nPrefix}.whoCanDebug`], { ns: 'plugin' }),
                  key: 'debug_permission',
                  value: tempPrivilege?.debug_permission || PermissionType.noOne,
                },
              ].map(({ title, key, value }) => (
                <div key={key} className="flex flex-col items-start gap-1 self-stretch">
                  <Label
                    label={title}
                    tooltip={isPermissionDisabledByRBAC ? permissionDisabledTip : undefined}
                  />
                  <div className="flex w-full items-start justify-between gap-2">
                    {[PermissionType.everyone, PermissionType.admin, PermissionType.noOne].map(
                      (option) => (
                        <OptionCard
                          key={option}
                          title={t(($) => $[`${i18nPrefix}.${option}`], { ns: 'plugin' })}
                          onSelect={() => handlePrivilegeChange(key)(option)}
                          selected={value === option}
                          disabled={isPermissionDisabledByRBAC}
                          className="flex-1"
                        />
                      ),
                    )}
                  </div>
                </div>
              ))}
            </div>
          )}
          {systemFeatures.enable_marketplace && canSetAutoUpdate && (
            <AutoUpdateSetting payload={tempAutoUpdateConfig} onChange={setTempAutoUpdateConfig} />
          )}
          <div className="flex h-19 items-center justify-end gap-2 self-stretch p-6 pt-5">
            <Button className="min-w-18" onClick={onHide}>
              {t(($) => $['operation.cancel'], { ns: 'common' })}
            </Button>
            <Button className="min-w-18" variant="primary" onClick={handleSave}>
              {t(($) => $['operation.save'], { ns: 'common' })}
            </Button>
          </div>
        </div>
      </DialogContent>
    </Dialog>
  )
}

export default React.memo(PluginSettingModal)