dify/web/app/components/workflow/skill/file-tree/artifacts-section.tsx

'use client'

import type { FC } from 'react'
import type { SandboxFileTreeNode } from '@/types/sandbox-file'
import { RiArrowDownSLine, RiArrowRightSLine } from '@remixicon/react'
import * as React from 'react'
import { useCallback, useState } from 'react'
import { useTranslation } from 'react-i18next'
import FolderSpark from '@/app/components/base/icons/src/vender/workflow/FolderSpark'
import { useAppContext } from '@/context/app-context'
import { useDownloadSandboxFile, useSandboxFilesTree } from '@/service/use-sandbox-file'
import { cn } from '@/utils/classnames'
import ArtifactsTree from './artifacts-tree'

type ArtifactsSectionProps = {
  className?: string
}

const ArtifactsSection: FC<ArtifactsSectionProps> = ({ className }) => {
  const { t } = useTranslation('workflow')
  const { userProfile } = useAppContext()
  const sandboxId = userProfile?.id

  const [isExpanded, setIsExpanded] = useState(false)

  const { data: treeData, hasFiles, isLoading } = useSandboxFilesTree(sandboxId, {
    enabled: isExpanded,
  })

  const downloadMutation = useDownloadSandboxFile(sandboxId)

  const handleToggle = useCallback(() => {
    setIsExpanded(prev => !prev)
  }, [])

  const handleDownload = useCallback(async (node: SandboxFileTreeNode) => {
    try {
      const ticket = await downloadMutation.mutateAsync(node.path)
      window.open(ticket.download_url, '_blank')
    }
    catch (error) {
      console.error('Download failed:', error)
    }
  }, [downloadMutation])

  const showBlueDot = !isExpanded && hasFiles

  return (
    <div className={cn('shrink-0 border-t border-divider-regular p-1', className)}>
      <button
        type="button"
        onClick={handleToggle}
        className={cn(
          'flex w-full items-center rounded-md py-1 pl-2 pr-1.5',
          'hover:bg-state-base-hover',
          'focus-visible:outline-none focus-visible:ring-2 focus-visible:ring-inset focus-visible:ring-components-input-border-active',
        )}
        aria-expanded={isExpanded}
        aria-label={t('skillSidebar.artifacts.openArtifacts')}
      >
        <div className="flex flex-1 items-center gap-1 py-0.5">
          <div className="flex size-5 items-center justify-center">
            <FolderSpark className="size-4 text-text-secondary" aria-hidden="true" />
          </div>
          <span className="system-sm-semibold uppercase text-text-secondary">
            {t('skillSidebar.artifacts.title')}
          </span>
        </div>

        <div className="relative flex items-center">
          {showBlueDot && (
            <div className="absolute -left-2 size-[7px] rounded-full border border-white bg-state-accent-solid" />
          )}
          {isExpanded
            ? <RiArrowDownSLine className="size-4 text-text-tertiary" aria-hidden="true" />
            : <RiArrowRightSLine className="size-4 text-text-tertiary" aria-hidden="true" />}
        </div>
      </button>

      {isExpanded && (
        <div className="flex flex-col gap-px">
          {isLoading
            ? (
                <div className="px-2.5 py-3">
                  <div className="h-4 w-full animate-pulse rounded bg-components-panel-bg" />
                </div>
              )
            : hasFiles
              ? (
                  <ArtifactsTree
                    data={treeData}
                    onDownload={handleDownload}
                    isDownloading={downloadMutation.isPending}
                  />
                )
              : (
                  <div className="px-2.5 pb-1.5 pt-0.5">
                    <div className="rounded-lg bg-background-section p-3">
                      <p className="system-xs-regular text-text-tertiary">
                        {t('skillSidebar.artifacts.emptyState')}
                      </p>
                    </div>
                  </div>
                )}
        </div>
      )}
    </div>
  )
}

export default React.memo(ArtifactsSection)