labelmain-demo/app/person/dashboard/components/PersonalProjectTable.tsx

"use client"

import { getPersonProjectDashBoard } from "@/components/label/api/project"
import { Project } from "@/components/label/api/project/typing"
import { Button, Flex, Group, Text } from "@mantine/core"
import { DatePickerInput, type DatesRangeValue } from "@mantine/dates"

import { IconRefresh } from "@tabler/icons-react"
import dayjs from "dayjs"
import "dayjs/locale/zh-cn"
import duration from "dayjs/plugin/duration"
import { DataTable, DataTableColumn } from "mantine-datatable"
import { useRouter } from "next/navigation"
import { useCallback, useEffect, useMemo, useState } from "react"

dayjs.extend(duration)

function formatDurationSeconds(seconds?: number | null) {
  const value = seconds ?? 0
  if (!value) return "-"
  const durationObj = dayjs.duration(value, "seconds")
  const hours = durationObj.hours()
  const minutes = durationObj.minutes()
  if (hours) return `${hours}小时${minutes}分钟`
  if (minutes) return `${minutes}分钟`
  return "-"
}

interface FilterFormRecord {
  query_date: DatesRangeValue
}

function createInitialFilters(): FilterFormRecord {
  return {
    query_date: [
      dayjs().subtract(1, "month").format("YYYY-MM-DD"),
      dayjs().format("YYYY-MM-DD"),
    ],
  }
}

export default function PersonalProjectTable() {
  const router = useRouter()
  const [records, setRecords] = useState<
    Project.PersonProjectDashboardResponseItem[]
  >([])
  const [loading, setLoading] = useState(false)

  const [page, setPage] = useState(1)
  const [pageSize, setPageSize] = useState(50)
  const [totalItems, setTotalItems] = useState(0)

  const [filters, setFilters] = useState(() => createInitialFilters())
  const [queryTrigger, setQueryTrigger] = useState(0)
  const [form, setForm] = useState<FilterFormRecord>(createInitialFilters())

  const totalPages = useMemo(() => {
    const totalItems = records.length
    return Math.max(1, Math.ceil(totalItems / pageSize))
  }, [pageSize, records.length])

  const pageRecords = useMemo(() => {
    const start = (page - 1) * pageSize
    return records.slice(start, start + pageSize)
  }, [page, pageSize, records])

  const appliedParams = useMemo(() => {
    const obj: Project.ReqPersonProjectDashBoard = {
      start_date: dayjs().format("YYYY-MM-DD"),
      end_date: dayjs().format("YYYY-MM-DD"),
    }
    if (form.query_date[0] && form.query_date[1]) {
      obj.start_date = dayjs(form.query_date[0]).format("YYYY-MM-DD")
      obj.end_date = dayjs(form.query_date[1]).format("YYYY-MM-DD")
    }

    return obj
  }, [form.query_date])

  const load = useCallback(async () => {
    if (!queryTrigger) return
    setLoading(true)
    try {
      const res = await getPersonProjectDashBoard(appliedParams)
      setRecords(res ?? [])
      setTotalItems(res?.length ?? 0)
      setPage(1)
    } finally {
      setLoading(false)
    }
  }, [appliedParams, queryTrigger])

  useEffect(() => {
    load()
  }, [load])

  useEffect(() => {
    queueMicrotask(() => {
      setQueryTrigger((v) => v + 1)
    })
  }, [])

  const handleSearch = () => {
    setPage(1)
    setForm({ ...filters })
    setQueryTrigger((v) => v + 1)
  }

  useEffect(() => {
    if (page > totalPages) setPage(totalPages)
  }, [page, totalPages])

  const [currentRowId, setCurrentRowId] = useState<number>(0)

  const originColumn = useMemo(() => {
    let column: DataTableColumn<Project.PersonProjectDashboardResponseItem>[] =
      [
        {
          accessor: "project_id",
          title: "项目ID",
          width: 150,
          render: (record: Project.PersonProjectDashboardResponseItem) => {
            return (
              <Text
                px={6}
                variant="transparent"
                onClick={async () => {
                  false && setCurrentRowId(record.project_id)
                  false &&
                    router.push(`/project/info/detail?id=${record.project_id}`)
                }}
                title={record.project_id.toString()}
                // c={"brand"}
                style={{
                  fontSize: "14px",
                  maxWidth: "100%",
                  whiteSpace: "nowrap",
                  overflow: "hidden",
                  textOverflow: "ellipsis",
                }}>
                {record.project_id}
              </Text>
            )
          },
        },
        {
          accessor: "project_name",
          title: "项目名称",
          width: 160,
          render: (record: Project.PersonProjectDashboardResponseItem) => {
            return (
              <Text
                title={record.project_name ?? "-"}
                style={{
                  fontSize: "14px",
                  maxWidth: "100%",
                  whiteSpace: "nowrap",
                  overflow: "hidden",
                  textOverflow: "ellipsis",
                }}>
                {record.project_name ?? "-"}
              </Text>
            )
          },
        },
        {
          accessor: "project_type",
          title: "所属业务",
          width: 160,
          render: (record: Project.PersonProjectDashboardResponseItem) => {
            return (
              <Text
                title={record.project_id.toString()}
                style={{
                  fontSize: "14px",
                  maxWidth: "100%",
                  whiteSpace: "nowrap",
                  overflow: "hidden",
                  textOverflow: "ellipsis",
                }}>
                {record.project_type ?? "-"}
              </Text>
            )
          },
        },
        {
          accessor: "work_time",
          title: "任务耗时",
          width: 140,
          render: (record: Project.PersonProjectDashboardResponseItem) => {
            return formatDurationSeconds(record.work_time)
          },
        },
        {
          accessor: "project_label_type",
          title: "标注类别",
          width: 160,
          render: (record: Project.PersonProjectDashboardResponseItem) => {
            return record.project_label_type ?? "-"
          },
        },
      ]
    return column
  }, [router])

  return (
    <Group
      gap="sm"
      h="100%"
      align="stretch"
      style={{ flexDirection: "column" }}>
      <Group justify="space-between" align="center">
        <Flex align={"center"} gap={8}>
          <Text fw={700} fz="lg">
            参与项目
          </Text>
          <DatePickerInput
            type="range"
            label=""
            placeholder="请选择时间范围，默认一周"
            miw={208}
            valueFormat="YYYY-MM-DD"
            locale="zh-cn"
            clearable
            value={filters.query_date}
            onChange={(value: DatesRangeValue) =>
              setFilters((s) => ({
                ...s,
                query_date: value,
              }))
            }
          />
        </Flex>

        <Button
          size={"xs"}
          fz={"sm"}
          fw={500}
          radius="sm"
          onClick={handleSearch}
          loading={loading}
          leftSection={<IconRefresh size={16} />}>
          更新
        </Button>
      </Group>
      <Group
        gap={0}
        align="stretch"
        style={{ flex: 1, minHeight: 0, width: "100%" }}>
        <DataTable<Project.PersonProjectDashboardResponseItem>
          width="100%"
          style={{ width: "100%" }}
          styles={{
            header: {
              color: "var(--mantine-color-grey-text)",
              backgroundColor: "var(--mantine-datatable-striped-color)",
            },
          }}
          withTableBorder
          withRowBorders
          pinFirstColumn
          rowBackgroundColor={(record) => {
            if (record.project_id === currentRowId)
              return "var(--mantine-datatable-highlight-on-hover-color-light, var(--mantine-datatable-highlight-on-hover-color))"
          }}
          idAccessor="project_id"
          scrollAreaProps={{ type: "auto" }}
          records={pageRecords}
          onRecordsPerPageChange={setPageSize}
          recordsPerPageOptions={[10, 15, 20]}
          columns={originColumn}
          totalRecords={totalItems}
          recordsPerPage={pageSize}
          page={page}
          onPageChange={setPage}
          noRecordsText="暂无数据"
          recordsPerPageLabel="当前页数"
        />
      </Group>
    </Group>
  )
}