dify/web/app/components/datasets/documents/hooks/use-documents-page-state.ts

import type { SortType } from '@/service/datasets'
import { useDebounce } from 'ahooks'
import { useCallback, useState } from 'react'
import { normalizeStatusForQuery, sanitizeStatusValue } from '../status-filter'
import { useDocumentListQueryState } from './use-document-list-query-state'

export function useDocumentsPageState() {
  const { query, updateQuery } = useDocumentListQueryState()

  const inputValue = query.keyword
  const debouncedSearchValue = useDebounce(query.keyword, { wait: 500 })

  const statusFilterValue = sanitizeStatusValue(query.status)
  const sortValue = query.sort
  const normalizedStatusFilterValue = normalizeStatusForQuery(statusFilterValue)

  const currPage = query.page - 1
  const limit = query.limit

  const [selectedIds, setSelectedIds] = useState<string[]>([])

  const handlePageChange = useCallback(
    (newPage: number) => {
      updateQuery({ page: newPage + 1 })
    },
    [updateQuery],
  )

  const handleLimitChange = useCallback(
    (newLimit: number) => {
      updateQuery({ limit: newLimit, page: 1 })
    },
    [updateQuery],
  )

  const handleInputChange = useCallback(
    (value: string) => {
      if (value !== query.keyword) setSelectedIds([])
      updateQuery({ keyword: value, page: 1 })
    },
    [query.keyword, updateQuery],
  )

  const handleStatusFilterChange = useCallback(
    (value: string) => {
      const selectedValue = sanitizeStatusValue(value)
      setSelectedIds([])
      updateQuery({ status: selectedValue, page: 1 })
    },
    [updateQuery],
  )

  const handleStatusFilterClear = useCallback(() => {
    if (statusFilterValue === 'all') return
    setSelectedIds([])
    updateQuery({ status: 'all', page: 1 })
  }, [statusFilterValue, updateQuery])

  const handleSortChange = useCallback(
    (value: string) => {
      const next = value as SortType
      if (next === sortValue) return
      updateQuery({ sort: next, page: 1 })
    },
    [sortValue, updateQuery],
  )

  return {
    inputValue,
    debouncedSearchValue,
    handleInputChange,

    statusFilterValue,
    sortValue,
    normalizedStatusFilterValue,
    handleStatusFilterChange,
    handleStatusFilterClear,
    handleSortChange,

    currPage,
    limit,
    handlePageChange,
    handleLimitChange,

    selectedIds,
    setSelectedIds,
  }
}