|
| 1 | +"use client"; |
| 2 | + |
| 3 | +import { type Row } from "@tanstack/react-table"; |
| 4 | +import { useParams, usePathname, useRouter, useSearchParams } from "next/navigation"; |
| 5 | +import { useCallback, useEffect, useMemo, useState } from "react"; |
| 6 | + |
| 7 | +import AdvancedSearch from "@/components/common/advanced-search"; |
| 8 | +import { filters as traceFilters } from "@/components/traces/traces-table/columns"; |
| 9 | +import DateRangeFilter from "@/components/ui/date-range-filter"; |
| 10 | +import { InfiniteDataTable } from "@/components/ui/infinite-datatable"; |
| 11 | +import { useInfiniteScroll } from "@/components/ui/infinite-datatable/hooks"; |
| 12 | +import RefreshButton from "@/components/ui/infinite-datatable/ui/refresh-button"; |
| 13 | +import type { Filter } from "@/lib/actions/common/filters"; |
| 14 | +import type { TraceRow } from "@/lib/traces/types"; |
| 15 | + |
| 16 | +import { FETCH_SIZE, tracePickerColumns } from "./columns"; |
| 17 | + |
| 18 | +export interface TracePickerProps { |
| 19 | + onTraceSelect: (trace: TraceRow) => void; |
| 20 | + focusedTraceId?: string | null; |
| 21 | + excludeTraceId?: string; |
| 22 | + description?: string; |
| 23 | + fetchParams?: Record<string, string>; |
| 24 | + className?: string; |
| 25 | + mode?: "url" | "state"; |
| 26 | +} |
| 27 | + |
| 28 | +const TracePickerContent = ({ |
| 29 | + onTraceSelect, |
| 30 | + focusedTraceId, |
| 31 | + excludeTraceId, |
| 32 | + description, |
| 33 | + fetchParams, |
| 34 | + className, |
| 35 | + mode = "state", |
| 36 | +}: TracePickerProps) => { |
| 37 | + const { projectId } = useParams<{ projectId: string }>(); |
| 38 | + const searchParams = useSearchParams(); |
| 39 | + const router = useRouter(); |
| 40 | + const pathname = usePathname(); |
| 41 | + |
| 42 | + // State-mode local state |
| 43 | + const [stateFilters, setStateFilters] = useState<{ filters: Filter[]; search: string }>({ filters: [], search: "" }); |
| 44 | + const [stateDateRange, setStateDateRange] = useState<{ |
| 45 | + pastHours?: string; |
| 46 | + startDate?: string; |
| 47 | + endDate?: string; |
| 48 | + }>({ pastHours: "24" }); |
| 49 | + |
| 50 | + // URL-mode: set default pastHours=24 if no date params present |
| 51 | + useEffect(() => { |
| 52 | + if (mode !== "url") return; |
| 53 | + const hasPastHours = searchParams.has("pastHours"); |
| 54 | + const hasStartDate = searchParams.has("startDate"); |
| 55 | + const hasEndDate = searchParams.has("endDate"); |
| 56 | + if (!hasPastHours && !hasStartDate && !hasEndDate) { |
| 57 | + const params = new URLSearchParams(searchParams.toString()); |
| 58 | + params.set("pastHours", "24"); |
| 59 | + router.replace(`${pathname}?${params.toString()}`); |
| 60 | + } |
| 61 | + }, [mode, searchParams, router, pathname]); |
| 62 | + |
| 63 | + // Resolve effective values based on mode |
| 64 | + const effectiveFilter = useMemo(() => { |
| 65 | + if (mode === "url") return searchParams.getAll("filter"); |
| 66 | + return stateFilters.filters.map((f) => JSON.stringify(f)); |
| 67 | + }, [mode, searchParams, stateFilters.filters]); |
| 68 | + |
| 69 | + const effectiveSearch = useMemo(() => { |
| 70 | + if (mode === "url") return searchParams.get("search") ?? ""; |
| 71 | + return stateFilters.search; |
| 72 | + }, [mode, searchParams, stateFilters.search]); |
| 73 | + |
| 74 | + const effectiveDateRange = useMemo(() => { |
| 75 | + if (mode === "url") { |
| 76 | + const startDate = searchParams.get("startDate") ?? undefined; |
| 77 | + const endDate = searchParams.get("endDate") ?? undefined; |
| 78 | + // If explicit start/end dates are set, ignore pastHours to avoid conflicts |
| 79 | + const pastHours = startDate && endDate ? undefined : (searchParams.get("pastHours") ?? undefined); |
| 80 | + return { pastHours, startDate, endDate }; |
| 81 | + } |
| 82 | + return stateDateRange; |
| 83 | + }, [mode, searchParams, stateDateRange]); |
| 84 | + |
| 85 | + const fetchTraces = useCallback( |
| 86 | + async (pageNumber: number) => { |
| 87 | + const urlParams = new URLSearchParams(); |
| 88 | + |
| 89 | + if (fetchParams) { |
| 90 | + for (const [key, value] of Object.entries(fetchParams)) { |
| 91 | + urlParams.set(key, value); |
| 92 | + } |
| 93 | + } |
| 94 | + |
| 95 | + if (effectiveDateRange.pastHours) urlParams.set("pastHours", effectiveDateRange.pastHours); |
| 96 | + if (effectiveDateRange.startDate) urlParams.set("startDate", effectiveDateRange.startDate); |
| 97 | + if (effectiveDateRange.endDate) urlParams.set("endDate", effectiveDateRange.endDate); |
| 98 | + |
| 99 | + effectiveFilter.forEach((filter) => { |
| 100 | + urlParams.append("filter", filter); |
| 101 | + }); |
| 102 | + |
| 103 | + if (excludeTraceId) { |
| 104 | + urlParams.append("filter", JSON.stringify({ column: "id", operator: "ne", value: excludeTraceId })); |
| 105 | + } |
| 106 | + |
| 107 | + if (effectiveSearch.length > 0) { |
| 108 | + urlParams.set("search", effectiveSearch); |
| 109 | + } |
| 110 | + |
| 111 | + urlParams.set("pageNumber", pageNumber.toString()); |
| 112 | + urlParams.set("pageSize", FETCH_SIZE.toString()); |
| 113 | + |
| 114 | + const res = await fetch(`/api/projects/${projectId}/traces?${urlParams.toString()}`); |
| 115 | + if (!res.ok) { |
| 116 | + const text = (await res.json()) as { error: string }; |
| 117 | + throw new Error(text.error); |
| 118 | + } |
| 119 | + |
| 120 | + const data = (await res.json()) as { items: TraceRow[] }; |
| 121 | + return { items: data.items ?? [], count: undefined }; |
| 122 | + }, |
| 123 | + [projectId, effectiveFilter, effectiveSearch, effectiveDateRange, fetchParams, excludeTraceId] |
| 124 | + ); |
| 125 | + |
| 126 | + const { |
| 127 | + data: traces, |
| 128 | + hasMore, |
| 129 | + isFetching, |
| 130 | + isLoading, |
| 131 | + fetchNextPage, |
| 132 | + refetch, |
| 133 | + } = useInfiniteScroll<TraceRow>({ |
| 134 | + fetchFn: fetchTraces, |
| 135 | + enabled: !!(effectiveDateRange.pastHours || (effectiveDateRange.startDate && effectiveDateRange.endDate)), |
| 136 | + deps: [effectiveFilter, effectiveSearch, effectiveDateRange, projectId, fetchParams, excludeTraceId], |
| 137 | + }); |
| 138 | + |
| 139 | + const handleRowClick = useCallback( |
| 140 | + (row: Row<TraceRow>) => { |
| 141 | + onTraceSelect(row.original); |
| 142 | + }, |
| 143 | + [onTraceSelect] |
| 144 | + ); |
| 145 | + |
| 146 | + return ( |
| 147 | + <div className={className ?? "flex flex-col flex-1 gap-3 px-4 py-2 overflow-hidden"}> |
| 148 | + {description && <span className="text-secondary-foreground text-xs px-1">{description}</span>} |
| 149 | + |
| 150 | + <InfiniteDataTable<TraceRow> |
| 151 | + className="w-full flex-1" |
| 152 | + columns={tracePickerColumns} |
| 153 | + data={traces} |
| 154 | + getRowId={(t) => t.id} |
| 155 | + onRowClick={handleRowClick} |
| 156 | + focusedRowId={focusedTraceId} |
| 157 | + hasMore={!effectiveSearch && hasMore} |
| 158 | + isFetching={isFetching} |
| 159 | + isLoading={isLoading} |
| 160 | + fetchNextPage={fetchNextPage} |
| 161 | + estimatedRowHeight={36} |
| 162 | + lockedColumns={["status"]} |
| 163 | + > |
| 164 | + <div className="flex gap-2 w-full items-center"> |
| 165 | + {mode === "url" ? ( |
| 166 | + <DateRangeFilter mode="url" /> |
| 167 | + ) : ( |
| 168 | + <DateRangeFilter mode="state" value={stateDateRange} onChange={setStateDateRange} /> |
| 169 | + )} |
| 170 | + <RefreshButton onClick={refetch} variant="outline" /> |
| 171 | + </div> |
| 172 | + <div className="w-full px-px"> |
| 173 | + {mode === "url" ? ( |
| 174 | + <AdvancedSearch |
| 175 | + mode="url" |
| 176 | + filters={traceFilters} |
| 177 | + resource="traces" |
| 178 | + placeholder="Search traces..." |
| 179 | + className="w-full flex-1" |
| 180 | + options={{ disableHotKey: true }} |
| 181 | + /> |
| 182 | + ) : ( |
| 183 | + <AdvancedSearch |
| 184 | + mode="state" |
| 185 | + filters={traceFilters} |
| 186 | + resource="traces" |
| 187 | + value={stateFilters} |
| 188 | + onSubmit={(f, search) => setStateFilters({ filters: f, search })} |
| 189 | + placeholder="Search traces..." |
| 190 | + className="w-full flex-1" |
| 191 | + options={{ disableHotKey: true }} |
| 192 | + /> |
| 193 | + )} |
| 194 | + </div> |
| 195 | + </InfiniteDataTable> |
| 196 | + </div> |
| 197 | + ); |
| 198 | +}; |
| 199 | + |
| 200 | +export default TracePickerContent; |
0 commit comments