@kitlangton/motel 0.1.0 → 0.1.2

package/src/ui/state.ts

@@ -106,6 +106,32 @@ export const autoRefreshAtom = Atom.make(false).pipe(Atom.keepAlive)
 export const filterModeAtom = Atom.make(false).pipe(Atom.keepAlive)
 export const filterTextAtom = Atom.make("").pipe(Atom.keepAlive)
 
+// Attribute filter (F key): pick a span-attribute key + exact value to restrict the trace list.
+export type AttrPickerMode = "off" | "keys" | "values"
+export const attrPickerModeAtom = Atom.make<AttrPickerMode>("off").pipe(Atom.keepAlive)
+export const attrPickerInputAtom = Atom.make("").pipe(Atom.keepAlive)
+export const attrPickerIndexAtom = Atom.make(0).pipe(Atom.keepAlive)
+
+export interface AttrFacetState {
+	readonly status: LoadStatus
+	readonly key: string | null // null when loading keys; set when loading values
+	readonly data: readonly { readonly value: string; readonly count: number }[]
+	readonly error: string | null
+}
+
+export const initialAttrFacetState: AttrFacetState = {
+	status: "ready",
+	key: null,
+	data: [],
+	error: null,
+}
+
+export const attrFacetStateAtom = Atom.make(initialAttrFacetState).pipe(Atom.keepAlive)
+
+// Applied filter (drives trace list query)
+export const activeAttrKeyAtom = Atom.make<string | null>(null).pipe(Atom.keepAlive)
+export const activeAttrValueAtom = Atom.make<string | null>(null).pipe(Atom.keepAlive)
+
 const lastThemePath = `${dirname(config.otel.databasePath)}/last-theme.txt`
 const readLastTheme = (): ThemeName => {
 	try {
@@ -132,6 +158,12 @@ export const collapsedSpanIdsAtom = Atom.make(new Set<string>() as ReadonlySet<s
 
 export const loadTraceServices = () => queryRuntime.runPromise(Effect.flatMap(TraceQueryService.asEffect(), (service) => service.listServices))
 export const loadRecentTraceSummaries = (serviceName: string) => queryRuntime.runPromise(Effect.flatMap(TraceQueryService.asEffect(), (service) => service.listTraceSummaries(serviceName)))
+export const loadFilteredTraceSummaries = (serviceName: string, attributeFilters: Readonly<Record<string, string>>) =>
+	queryRuntime.runPromise(Effect.flatMap(TraceQueryService.asEffect(), (service) => service.searchTraceSummaries({ serviceName, attributeFilters, limit: config.otel.traceFetchLimit })))
+export const loadTraceAttributeKeys = (serviceName: string) =>
+	queryRuntime.runPromise(Effect.flatMap(TraceQueryService.asEffect(), (service) => service.listFacets({ type: "traces", field: "attribute_keys", serviceName, limit: 200 })))
+export const loadTraceAttributeValues = (serviceName: string, key: string) =>
+	queryRuntime.runPromise(Effect.flatMap(TraceQueryService.asEffect(), (service) => service.listFacets({ type: "traces", field: "attribute_values", serviceName, key, limit: 200 })))
 export const loadTraceDetail = (traceId: string) => queryRuntime.runPromise(Effect.flatMap(TraceQueryService.asEffect(), (service) => service.getTrace(traceId)))
 export const loadTraceLogs = (traceId: string) => queryRuntime.runPromise(Effect.flatMap(LogQueryService.asEffect(), (service) => service.listTraceLogs(traceId)))
 export const loadServiceLogs = (serviceName: string) => queryRuntime.runPromise(Effect.flatMap(LogQueryService.asEffect(), (service) => service.listRecentLogs(serviceName)))

package/src/ui/theme.ts

@@ -32,32 +32,37 @@ export interface ThemeDefinition {
 	readonly waterfall: ThemeWaterfallColors
 }
 
+// motel-default palette derived in OKLCH. All "surface" tokens share hue
+// 282 (twilight purple) at varying lightness so depth is communicated by
+// lightness alone (footer < screen < selected < bar track). The amber
+// accent (hue 73) sits almost complementary to the surfaces, giving the
+// motel-sign neon maximum contrast without color clash.
 const motelDefaultTheme: ThemeDefinition = {
 	name: "motel-default",
 	label: "Motel Default",
 	colors: {
-		screenBg: "#1c1b29",
-		text: "#ede7da",
-		muted: "#9f9788",
-		separator: "#6f685d",
-		accent: "#f4a51c",
-		error: "#f97316",
-		selectedBg: "#263044",
-		warning: "#facc15",
+		screenBg: "#111120",       // oklch(0.185 0.030 282)
+		text: "#eee5d6",           // oklch(0.925 0.022 82)  — warm cream
+		muted: "#9a9181",          // oklch(0.660 0.025 82)
+		separator: "#686155",      // oklch(0.495 0.020 81)
+		accent: "#f5a41a",         // oklch(0.780 0.161 73)  — motel neon
+		error: "#f97312",          // oklch(0.705 0.187 48)
+		selectedBg: "#2b2c48",     // oklch(0.305 0.050 282) — same hue as screen
+		warning: "#facc16",        // oklch(0.861 0.173 92)
 		selectedText: "#f8fafc",
-		count: "#d7c5a1",
-		passing: "#7dd3a3",
-		defaultService: "#93c5fd",
-		footerBg: "#000000",
-		treeLine: "#524d45",
-		previewKey: "#6a6358",
+		count: "#d7c5a1",          // oklch(0.830 0.052 85)
+		passing: "#7ed5a4",        // oklch(0.805 0.110 158)
+		defaultService: "#93c5fe", // oklch(0.810 0.096 252)
+		footerBg: "#04040e",       // oklch(0.115 0.025 282) — deeper than screen
+		treeLine: "#48433b",       // oklch(0.385 0.015 80)
+		previewKey: "#645d51",     // oklch(0.480 0.020 80)
 	},
 	waterfall: {
-		bar: "#f4a51c",
-		barError: "#f97316",
-		barBg: "#2a2520",
-		barLane: "#4a4338",
-		barSelected: "#e8c547",
+		bar: "#f5a41a",            // = accent
+		barError: "#f97312",       // = error
+		barBg: "#1f1f34",          // oklch(0.250 0.040 282) — purple track (was warm)
+		barLane: "#3d3e5b",        // oklch(0.375 0.050 282)
+		barSelected: "#f3c048",    // oklch(0.832 0.145 85) — warmer amber
 		barSelectedError: "#ff8c42",
 	},
 }

package/src/ui/traceSortNav.repro.test.ts

@@ -60,8 +60,9 @@ const listRows = (snap: string): { readonly rows: readonly string[]; readonly se
 		const leftHalf = raw.split("\u2502")[0] ?? raw
 		const rightHalf = raw.includes("\u2502") ? raw.split("\u2502").slice(1).join("\u2502") : ""
 
-		// Trace rows: left pane, `·` then `op #hash`.
-		const rowMatch = leftHalf.match(/^\s+\u00b7\s+(\S+)\s+#/)
+		// Trace rows: left pane, `·` then the operation name as the first
+		// token. (Earlier versions appended `#<hash>`; that's been removed.)
+		const rowMatch = leftHalf.match(/^\s+\u00b7\s+(op[A-Z])\b/)
 		if (rowMatch) rows.push(rowMatch[1]!)
 
 		// Selected trace: right pane, line immediately after `TRACE DETAILS`

package/src/ui/useAttrFilterPicker.ts

@@ -0,0 +1,47 @@
+import { useAtom } from "@effect/atom-react"
+import { useEffect } from "react"
+import {
+	attrFacetStateAtom,
+	attrPickerModeAtom,
+	initialAttrFacetState,
+	loadTraceAttributeKeys,
+	loadTraceAttributeValues,
+	selectedTraceServiceAtom,
+} from "./state.ts"
+
+// When the picker is open, load the current facet page (keys, or values for
+// a specific key) and keep it in sync with the selected service. We key the
+// effect off picker mode + service + target key so refetches happen on drill
+// in/out and when the user switches services mid-pick.
+export const useAttrFilterPicker = (selectedKey: string | null) => {
+	const [pickerMode] = useAtom(attrPickerModeAtom)
+	const [service] = useAtom(selectedTraceServiceAtom)
+	const [, setFacetState] = useAtom(attrFacetStateAtom)
+
+	useEffect(() => {
+		if (pickerMode === "off" || !service) {
+			setFacetState(initialAttrFacetState)
+			return
+		}
+		let cancelled = false
+		setFacetState({ status: "loading", key: pickerMode === "values" ? selectedKey : null, data: [], error: null })
+		const load = async () => {
+			try {
+				const rows = pickerMode === "keys"
+					? await loadTraceAttributeKeys(service)
+					: selectedKey
+						? await loadTraceAttributeValues(service, selectedKey)
+						: []
+				if (cancelled) return
+				setFacetState({ status: "ready", key: pickerMode === "values" ? selectedKey : null, data: rows, error: null })
+			} catch (err) {
+				if (cancelled) return
+				setFacetState({ status: "error", key: pickerMode === "values" ? selectedKey : null, data: [], error: err instanceof Error ? err.message : String(err) })
+			}
+		}
+		void load()
+		return () => {
+			cancelled = true
+		}
+	}, [pickerMode, service, selectedKey, setFacetState])
+}

package/src/ui/useKeyboardNav.ts

@@ -5,6 +5,12 @@ import type { TraceItem, TraceSummaryItem } from "../domain.ts"
 import { otelServerInstructions } from "../instructions.ts"
 import { copyToClipboard, traceUiUrl, webUiUrl } from "./format.ts"
 import {
+	activeAttrKeyAtom,
+	activeAttrValueAtom,
+	attrFacetStateAtom,
+	attrPickerIndexAtom,
+	attrPickerInputAtom,
+	attrPickerModeAtom,
 	autoRefreshAtom,
 	collapsedSpanIdsAtom,
 	detailViewAtom,
@@ -22,6 +28,7 @@ import {
 	type TraceSortMode,
 	traceStateAtom,
 } from "./state.ts"
+import { filterFacets } from "./AttrFilterModal.tsx"
 import { G_PREFIX_TIMEOUT_MS } from "./theme.ts"
 import { cycleThemeName, themeLabel } from "./theme.ts"
 import { getVisibleSpans } from "./Waterfall.tsx"
@@ -65,6 +72,12 @@ export const useKeyboardNav = (params: KeyboardNavParams) => {
 	const [filterMode, setFilterMode] = useAtom(filterModeAtom)
 	const [filterText, setFilterText] = useAtom(filterTextAtom)
 	const [traceSort, setTraceSort] = useAtom(traceSortAtom)
+	const [pickerMode, setPickerMode] = useAtom(attrPickerModeAtom)
+	const [pickerInput, setPickerInput] = useAtom(attrPickerInputAtom)
+	const [pickerIndex, setPickerIndex] = useAtom(attrPickerIndexAtom)
+	const [attrFacets] = useAtom(attrFacetStateAtom)
+	const [activeAttrKey, setActiveAttrKey] = useAtom(activeAttrKeyAtom)
+	const [activeAttrValue, setActiveAttrValue] = useAtom(activeAttrValueAtom)
 
 	const pendingGRef = useRef(false)
 	const pendingGTimeoutRef = useRef<ReturnType<typeof setTimeout> | null>(null)
@@ -73,12 +86,12 @@ export const useKeyboardNav = (params: KeyboardNavParams) => {
 	const spanNavActive = detailView !== "service-logs" && selectedSpanIndex !== null
 	const serviceLogNavActive = detailView === "service-logs"
 
-	const stateRef = useRef({ traceState, serviceLogState, selectedServiceLogIndex, selectedTheme, selectedTraceIndex, selectedSpanIndex, selectedTraceService, detailView, showHelp, collapsedSpanIds, spanNavActive, serviceLogNavActive, filterMode, filterText, autoRefresh, traceSort, ...params })
+	const stateRef = useRef({ traceState, serviceLogState, selectedServiceLogIndex, selectedTheme, selectedTraceIndex, selectedSpanIndex, selectedTraceService, detailView, showHelp, collapsedSpanIds, spanNavActive, serviceLogNavActive, filterMode, filterText, autoRefresh, traceSort, pickerMode, pickerInput, pickerIndex, attrFacets, activeAttrKey, activeAttrValue, ...params })
 	// Keep the keyboard handler's state mirror in sync before the next paint.
 	// OpenTUI's own effect-event helper uses useLayoutEffect for this same reason:
 	// rapid repeated keypresses can otherwise observe stale selection state.
 	useLayoutEffect(() => {
-		stateRef.current = { traceState, serviceLogState, selectedServiceLogIndex, selectedTheme, selectedTraceIndex, selectedSpanIndex, selectedTraceService, detailView, showHelp, collapsedSpanIds, spanNavActive, serviceLogNavActive, filterMode, filterText, autoRefresh, traceSort, ...params }
+		stateRef.current = { traceState, serviceLogState, selectedServiceLogIndex, selectedTheme, selectedTraceIndex, selectedSpanIndex, selectedTraceService, detailView, showHelp, collapsedSpanIds, spanNavActive, serviceLogNavActive, filterMode, filterText, autoRefresh, traceSort, pickerMode, pickerInput, pickerIndex, attrFacets, activeAttrKey, activeAttrValue, ...params }
 	})
 
 	const clearPendingG = () => {
@@ -230,7 +243,8 @@ export const useKeyboardNav = (params: KeyboardNavParams) => {
 				if (s.serviceLogState.data.length === 0) return 0
 				return Math.max(0, Math.min(current + direction * serviceLogPageSize, s.serviceLogState.data.length - 1))
 			})
-		} else if (s.spanNavActive && s.selectedTrace) {
+		} else if (s.spanNavActive) {
+			if (!s.selectedTrace) return
 			const visibleCount = getVisibleSpans(s.selectedTrace.spans, s.collapsedSpanIds).length
 			setSelectedSpanIndex((current) => {
 				if (visibleCount === 0) return null
@@ -257,6 +271,66 @@ export const useKeyboardNav = (params: KeyboardNavParams) => {
 	useKeyboard((key) => {
 		const s = $()
 
+		// Attribute picker modal owns the keyboard while open.
+		if (s.pickerMode !== "off") {
+			const rows = filterFacets(s.attrFacets.data, s.pickerInput)
+			const clampedIndex = rows.length === 0 ? 0 : Math.max(0, Math.min(s.pickerIndex, rows.length - 1))
+			const move = (delta: number) => {
+				if (rows.length === 0) return
+				setPickerIndex(Math.max(0, Math.min(clampedIndex + delta, rows.length - 1)))
+			}
+			if (key.name === "escape") {
+				setPickerMode("off")
+				setPickerInput("")
+				setPickerIndex(0)
+				return
+			}
+			if (key.name === "up" || (key.ctrl && key.name === "p")) { move(-1); return }
+			if (key.name === "down" || (key.ctrl && key.name === "n")) { move(1); return }
+			if (key.name === "pageup") { move(-10); return }
+			if (key.name === "pagedown") { move(10); return }
+			if (key.name === "return" || key.name === "enter") {
+				const row = rows[clampedIndex]
+				if (!row) return
+				if (s.pickerMode === "keys") {
+					// Drill from keys → values for this key.
+					setActiveAttrKey(row.value)
+					setPickerMode("values")
+					setPickerInput("")
+					setPickerIndex(0)
+				} else {
+					// Apply: activeAttrKey is already set, now pin the value.
+					setActiveAttrValue(row.value)
+					setPickerMode("off")
+					setPickerInput("")
+					setPickerIndex(0)
+					s.flashNotice(`Filter: ${s.activeAttrKey}=${row.value}`)
+				}
+				return
+			}
+			if (key.name === "backspace") {
+				if (s.pickerInput.length > 0) {
+					setPickerInput(s.pickerInput.slice(0, -1))
+					setPickerIndex(0)
+					return
+				}
+				// At empty input in values mode, backspace walks back to keys.
+				if (s.pickerMode === "values") {
+					setPickerMode("keys")
+					setActiveAttrKey(null)
+					setPickerIndex(0)
+					return
+				}
+				return
+			}
+			if (key.name.length === 1 && !key.ctrl && !key.meta) {
+				setPickerInput(s.pickerInput + key.name)
+				setPickerIndex(0)
+				return
+			}
+			return
+		}
+
 		// Filter mode: capture text input
 		if (s.filterMode) {
 			if (key.name === "escape") {
@@ -365,6 +439,15 @@ export const useKeyboardNav = (params: KeyboardNavParams) => {
 				setSelectedSpanIndex(null)
 				return
 			}
+			// At the trace list, `esc` clears any applied attribute filter so
+			// there's a clean way back to the unfiltered list without hunting
+			// for the picker key.
+			if (s.activeAttrKey || s.activeAttrValue) {
+				setActiveAttrKey(null)
+				setActiveAttrValue(null)
+				s.flashNotice("Cleared attribute filter")
+				return
+			}
 			return
 		}
 		if (key.name === "return" || key.name === "enter") {
@@ -416,6 +499,15 @@ export const useKeyboardNav = (params: KeyboardNavParams) => {
 			setFilterMode(true)
 			return
 		}
+		if ((key.name === "f" || key.name === "F") && !key.ctrl && !key.meta) {
+			// Open attribute picker at the keys step. If a filter is already
+			// applied, reopening lets the user refine or switch.
+			setPickerMode("keys")
+			setPickerInput("")
+			setPickerIndex(0)
+			setActiveAttrKey(null)
+			return
+		}
 		if (key.name === "tab") {
 			toggleServiceLogsView()
 			return
@@ -431,12 +523,17 @@ export const useKeyboardNav = (params: KeyboardNavParams) => {
 		if (key.name === "up" || key.name === "k") {
 			if (s.serviceLogNavActive) {
 				moveServiceLogBy(-1)
-			} else if (s.spanNavActive && s.selectedTrace) {
-				const visibleCount = getVisibleSpans(s.selectedTrace.spans, s.collapsedSpanIds).length
-				setSelectedSpanIndex((current) => {
-					if (current === null || visibleCount === 0) return 0
-					return Math.max(0, current - 1)
-				})
+			} else if (s.spanNavActive) {
+				// Locked to span nav; never fall through to trace-list nav while
+				// drilled in. If the trace detail is still loading, swallow the
+				// key instead of silently leaking it to the trace list.
+				if (s.selectedTrace) {
+					const visibleCount = getVisibleSpans(s.selectedTrace.spans, s.collapsedSpanIds).length
+					setSelectedSpanIndex((current) => {
+						if (current === null || visibleCount === 0) return 0
+						return Math.max(0, current - 1)
+					})
+				}
 			} else {
 				moveTraceBy(-1)
 			}
@@ -445,12 +542,14 @@ export const useKeyboardNav = (params: KeyboardNavParams) => {
 		if (key.name === "down" || key.name === "j") {
 			if (s.serviceLogNavActive) {
 				moveServiceLogBy(1)
-			} else if (s.spanNavActive && s.selectedTrace) {
-				const visibleCount = getVisibleSpans(s.selectedTrace.spans, s.collapsedSpanIds).length
-				setSelectedSpanIndex((current) => {
-					if (current === null || visibleCount === 0) return 0
-					return Math.min(current + 1, visibleCount - 1)
-				})
+			} else if (s.spanNavActive) {
+				if (s.selectedTrace) {
+					const visibleCount = getVisibleSpans(s.selectedTrace.spans, s.collapsedSpanIds).length
+					setSelectedSpanIndex((current) => {
+						if (current === null || visibleCount === 0) return 0
+						return Math.min(current + 1, visibleCount - 1)
+					})
+				}
 			} else {
 				moveTraceBy(1)
 			}

package/src/ui/waterfallNav.test.ts

@@ -3,7 +3,8 @@ import type { TraceSpanItem } from "../domain.ts"
 import {
 	findFirstChildIndex,
 	findParentIndex,
-	getWaterfallColumns,
+	getWaterfallLayout,
+	getWaterfallSuffixMetrics,
 	getVisibleSpans,
 } from "./Waterfall.tsx"
 import { resolveCollapseStep } from "./waterfallNav.ts"
@@ -130,13 +131,27 @@ describe("getVisibleSpans", () => {
 	})
 })
 
-describe("getWaterfallColumns", () => {
-	it("pads duration and log columns to fill the reserved width", () => {
+describe("getWaterfallSuffixMetrics", () => {
+	it("uses the widest visible duration as the shared suffix width", () => {
+		const spans = [
+			{ spanId: "a", durationMs: 1 },
+			{ spanId: "b", durationMs: 57_000 },
+			{ spanId: "c", durationMs: 120 },
+		]
+		const metrics = getWaterfallSuffixMetrics(spans)
+		// `120ms` = 5 is the widest
+		expect(metrics.maxDurationWidth).toBe(5)
+		expect(metrics.suffixWidth).toBe(5)
+	})
+
+	it("layout reserves the suffix once and leaves the rest for the bar", () => {
 		const contentWidth = 72
-		const columns = getWaterfallColumns(contentWidth, 153_000, 1, 0)
-		expect(columns.durationCell.length).toBe(columns.durationWidth)
-		expect(columns.logCell.length).toBe(columns.logWidth)
-		expect(columns.labelMaxWidth + 1 + columns.barWidth + 1 + columns.durationCell.length + columns.logCell.length).toBe(contentWidth)
+		const metrics = getWaterfallSuffixMetrics(
+			[{ spanId: "a", durationMs: 57_000 }, { spanId: "b", durationMs: 1 }],
+		)
+		const { labelMaxWidth, barWidth } = getWaterfallLayout(contentWidth, metrics.suffixWidth)
+		// label + 1 (gap before bar) + bar + 1 (gap before suffix) + suffix = contentWidth
+		expect(labelMaxWidth + 1 + barWidth + 1 + metrics.suffixWidth).toBe(contentWidth)
 	})
 })