Files
zeekay 1b367e2b62 chore: rebase the tree onto upstream master
The tree was a debranded snapshot of upstream taken around 2026-02-20, and
upstream has moved about five months since. This replaces it with upstream
master, mechanically re-derived: bin/debrand does the rename, and the
pipeline and squashed migrations are restored from the shipped tree.

Not a merge. The previous import replaced the tree wholesale in a single
commit, so there is no shared content history to merge against. Reproducing
the rename on the new tree is what actually converges.

What survives: .hanzo/workflows, the Dockerfiles and bin/docker* entrypoints,
compose files, LLM.md, and insights/migrations. The squashed initial
migration is load-bearing — the live database records it by name, so
upstream's ~918 individual migrations cannot come back.

Nine products are dropped as unreferenced. Twenty-one more that we do not run
are kept anyway because the core app still imports them; unpicking those is a
refactor and does not belong in a rebase.

This base does not build yet, and the reasons are not in the rename:
upstream now wants Python 3.13.13 against our 3.12.12 and Django 5.2 against
our 4.2, and the dependency manifests still name upstream packages that we
republish under our own names. Those are decisions, not renames.
2026-08-01 15:44:40 -07:00

520 lines
19 KiB
TypeScript

import { MakeLogicType, actions, connect, kea, listeners, path, reducers, selectors } from 'kea'
import { router, urlToAction } from 'kea-router'
import { dayjs } from 'lib/dayjs'
import { trackedActionToUrl } from 'lib/logic/scenes/trackedActionToUrl'
import { dateStringToDayJs } from 'lib/utils/dateFilters'
import { urls } from 'scenes/urls'
import {
EndpointsUsageBreakdown,
EndpointsUsageOverviewQuery,
EndpointsUsageTableQuery,
EndpointsUsageTrendsQuery,
NodeKind,
} from '~/queries/schema/schema-general'
import { EndpointType, IntervalType } from '~/types'
import { endpointsLogic } from './endpointsLogic'
export const INITIAL_DATE_FROM = '-7d'
export const INITIAL_DATE_TO = null as string | null
export const INITIAL_INTERVAL: IntervalType = 'day'
// Generated by kea-typegen. Update if you're an agent, ignore if you're human.
export interface endpointsUsageLogicValues {
allEndpoints: EndpointType[] // endpointsLogic
allEndpointsLoading: boolean // endpointsLogic
activeEndpointNames: Set<string>
breakdownBy: EndpointsUsageBreakdown | null
bytesReadTrendsQuery: EndpointsUsageTrendsQuery
canRefresh: boolean
cooldownActive: boolean
cpuSecondsTrendsQuery: EndpointsUsageTrendsQuery
dateFilter: {
dateFrom: string | null
dateTo: string | null
}
dateRange: {
date_from: string | null
date_to: string | null
}
endpointFilter: string[]
endpointNames: string[]
endpointNamesLoading: boolean
endpointTableQuery: EndpointsUsageTableQuery
errorRateTrendsQuery: EndpointsUsageTrendsQuery
interval: IntervalType
materializationType: 'inline' | 'materialized' | null
overviewQuery: EndpointsUsageOverviewQuery
queryDurationTrendsQuery: EndpointsUsageTrendsQuery
refreshKey: number
requestsTrendsQuery: EndpointsUsageTrendsQuery
}
// Generated by kea-typegen. Update if you're an agent, ignore if you're human.
export interface endpointsUsageLogicActions {
clearCooldown: () => {
value: true
}
refresh: () => {
value: true
}
setBreakdownBy: (breakdownBy: EndpointsUsageBreakdown | null) => {
breakdownBy: EndpointsUsageBreakdown | null
}
setDates: (
dateFrom: string | null,
dateTo: string | null
) => {
dateFrom: string | null
dateTo: string | null
}
setEndpointFilter: (endpointFilter: string[]) => {
endpointFilter: string[]
}
setInterval: (interval: IntervalType) => {
interval: IntervalType
}
setMaterializationType: (materializationType: 'inline' | 'materialized' | null) => {
materializationType: 'inline' | 'materialized' | null
}
}
// Generated by kea-typegen. Update if you're an agent, ignore if you're human.
export interface endpointsUsageLogicMeta {
__keaTypeGenInternalSelectorTypes: {
endpointNames: (allEndpoints: EndpointType[]) => string[]
activeEndpointNames: (endpointNames: string[]) => Set<string>
endpointNamesLoading: (allEndpointsLoading: boolean) => boolean
canRefresh: (cooldownActive: boolean) => boolean
dateRange: (dateFilter: { dateFrom: string | null; dateTo: string | null }) => {
date_from: string | null
date_to: string | null
}
overviewQuery: (
dateRange: {
date_from: string | null
date_to: string | null
},
endpointFilter: string[],
materializationType: 'inline' | 'materialized' | null
) => EndpointsUsageOverviewQuery
requestsTrendsQuery: (
dateRange: {
date_from: string | null
date_to: string | null
},
endpointFilter: string[],
materializationType: 'inline' | 'materialized' | null,
interval: IntervalType,
breakdownBy: EndpointsUsageBreakdown | null
) => EndpointsUsageTrendsQuery
bytesReadTrendsQuery: (
dateRange: {
date_from: string | null
date_to: string | null
},
endpointFilter: string[],
materializationType: 'inline' | 'materialized' | null,
interval: IntervalType,
breakdownBy: EndpointsUsageBreakdown | null
) => EndpointsUsageTrendsQuery
cpuSecondsTrendsQuery: (
dateRange: {
date_from: string | null
date_to: string | null
},
endpointFilter: string[],
materializationType: 'inline' | 'materialized' | null,
interval: IntervalType,
breakdownBy: EndpointsUsageBreakdown | null
) => EndpointsUsageTrendsQuery
queryDurationTrendsQuery: (
dateRange: {
date_from: string | null
date_to: string | null
},
endpointFilter: string[],
materializationType: 'inline' | 'materialized' | null,
interval: IntervalType,
breakdownBy: EndpointsUsageBreakdown | null
) => EndpointsUsageTrendsQuery
errorRateTrendsQuery: (
dateRange: {
date_from: string | null
date_to: string | null
},
endpointFilter: string[],
materializationType: 'inline' | 'materialized' | null,
interval: IntervalType,
breakdownBy: EndpointsUsageBreakdown | null
) => EndpointsUsageTrendsQuery
endpointTableQuery: (
dateRange: {
date_from: string | null
date_to: string | null
},
endpointFilter: string[],
materializationType: 'inline' | 'materialized' | null
) => EndpointsUsageTableQuery
}
}
export type endpointsUsageLogicType = MakeLogicType<
endpointsUsageLogicValues,
endpointsUsageLogicActions,
Record<string, any>,
endpointsUsageLogicMeta
>
export const endpointsUsageLogic = kea<endpointsUsageLogicType>([
path(['products', 'endpoints', 'frontend', 'endpointsUsageLogic']),
connect(() => ({
values: [endpointsLogic(), ['allEndpoints', 'allEndpointsLoading']],
})),
actions({
setDates: (dateFrom: string | null, dateTo: string | null) => ({ dateFrom, dateTo }),
setEndpointFilter: (endpointFilter: string[]) => ({ endpointFilter }),
setMaterializationType: (materializationType: 'materialized' | 'inline' | null) => ({ materializationType }),
setInterval: (interval: IntervalType) => ({ interval }),
setBreakdownBy: (breakdownBy: EndpointsUsageBreakdown | null) => ({ breakdownBy }),
refresh: true,
clearCooldown: true,
}),
reducers({
dateFilter: [
{
dateFrom: INITIAL_DATE_FROM,
dateTo: INITIAL_DATE_TO,
} as { dateFrom: string | null; dateTo: string | null },
{
setDates: (_, { dateFrom, dateTo }) => ({ dateFrom, dateTo }),
},
],
endpointFilter: [
[] as string[],
{
setEndpointFilter: (_, { endpointFilter }) => endpointFilter,
},
],
materializationType: [
null as 'materialized' | 'inline' | null,
{
setMaterializationType: (_, { materializationType }) => materializationType,
},
],
interval: [
INITIAL_INTERVAL as IntervalType,
{
setInterval: (_, { interval }) => interval,
},
],
breakdownBy: [
null as EndpointsUsageBreakdown | null,
{
setBreakdownBy: (_, { breakdownBy }) => breakdownBy,
},
],
cooldownActive: [
false,
{
refresh: () => true,
clearCooldown: () => false,
},
],
refreshKey: [
0,
{
refresh: (state) => state + 1,
},
],
}),
selectors({
endpointNames: [
(s) => [s.allEndpoints],
(allEndpoints: EndpointType[]): string[] => allEndpoints.map((e) => e.name).sort(),
],
activeEndpointNames: [
(s) => [s.endpointNames],
(endpointNames: string[]): Set<string> => new Set(endpointNames),
],
endpointNamesLoading: [(s) => [s.allEndpointsLoading], (loading: boolean): boolean => loading],
canRefresh: [(s) => [s.cooldownActive], (cooldownActive: boolean): boolean => !cooldownActive],
dateRange: [
(s) => [s.dateFilter],
(dateFilter: {
dateFrom: string | null
dateTo: string | null
}): { date_from: string | null; date_to: string | null } => {
const dateFromDayjs = dateStringToDayJs(dateFilter.dateFrom)
const dateToDayjs = dateFilter.dateTo ? dateStringToDayJs(dateFilter.dateTo) : null
return {
date_from: dateFromDayjs?.format('YYYY-MM-DD') ?? dayjs().subtract(7, 'day').format('YYYY-MM-DD'),
date_to: dateToDayjs?.format('YYYY-MM-DD') ?? null,
}
},
],
overviewQuery: [
(s) => [s.dateRange, s.endpointFilter, s.materializationType],
(
dateRange: {
date_from: string | null
date_to: string | null
},
endpointFilter: string[],
materializationType: 'inline' | 'materialized' | null
): EndpointsUsageOverviewQuery => ({
kind: NodeKind.EndpointsUsageOverviewQuery,
dateRange,
endpointNames: endpointFilter.length > 0 ? endpointFilter : undefined,
materializationType,
compareFilter: { compare: true },
}),
],
requestsTrendsQuery: [
(s) => [s.dateRange, s.endpointFilter, s.materializationType, s.interval, s.breakdownBy],
(
dateRange: {
date_from: string | null
date_to: string | null
},
endpointFilter: string[],
materializationType: 'inline' | 'materialized' | null,
interval: IntervalType,
breakdownBy: EndpointsUsageBreakdown | null
): EndpointsUsageTrendsQuery => ({
kind: NodeKind.EndpointsUsageTrendsQuery,
dateRange,
endpointNames: endpointFilter.length > 0 ? endpointFilter : undefined,
materializationType,
metric: 'requests',
interval,
breakdownBy: breakdownBy ?? undefined,
}),
],
bytesReadTrendsQuery: [
(s) => [s.dateRange, s.endpointFilter, s.materializationType, s.interval, s.breakdownBy],
(
dateRange: {
date_from: string | null
date_to: string | null
},
endpointFilter: string[],
materializationType: 'inline' | 'materialized' | null,
interval: IntervalType,
breakdownBy: EndpointsUsageBreakdown | null
): EndpointsUsageTrendsQuery => ({
kind: NodeKind.EndpointsUsageTrendsQuery,
dateRange,
endpointNames: endpointFilter.length > 0 ? endpointFilter : undefined,
materializationType,
metric: 'bytes_read',
interval,
breakdownBy: breakdownBy ?? undefined,
}),
],
cpuSecondsTrendsQuery: [
(s) => [s.dateRange, s.endpointFilter, s.materializationType, s.interval, s.breakdownBy],
(
dateRange: {
date_from: string | null
date_to: string | null
},
endpointFilter: string[],
materializationType: 'inline' | 'materialized' | null,
interval: IntervalType,
breakdownBy: EndpointsUsageBreakdown | null
): EndpointsUsageTrendsQuery => ({
kind: NodeKind.EndpointsUsageTrendsQuery,
dateRange,
endpointNames: endpointFilter.length > 0 ? endpointFilter : undefined,
materializationType,
metric: 'cpu_seconds',
interval,
breakdownBy: breakdownBy ?? undefined,
}),
],
queryDurationTrendsQuery: [
(s) => [s.dateRange, s.endpointFilter, s.materializationType, s.interval, s.breakdownBy],
(
dateRange: {
date_from: string | null
date_to: string | null
},
endpointFilter: string[],
materializationType: 'inline' | 'materialized' | null,
interval: IntervalType,
breakdownBy: EndpointsUsageBreakdown | null
): EndpointsUsageTrendsQuery => ({
kind: NodeKind.EndpointsUsageTrendsQuery,
dateRange,
endpointNames: endpointFilter.length > 0 ? endpointFilter : undefined,
materializationType,
metric: 'query_duration',
interval,
breakdownBy: breakdownBy ?? undefined,
}),
],
errorRateTrendsQuery: [
(s) => [s.dateRange, s.endpointFilter, s.materializationType, s.interval, s.breakdownBy],
(
dateRange: {
date_from: string | null
date_to: string | null
},
endpointFilter: string[],
materializationType: 'inline' | 'materialized' | null,
interval: IntervalType,
breakdownBy: EndpointsUsageBreakdown | null
): EndpointsUsageTrendsQuery => ({
kind: NodeKind.EndpointsUsageTrendsQuery,
dateRange,
endpointNames: endpointFilter.length > 0 ? endpointFilter : undefined,
materializationType,
metric: 'error_rate',
interval,
breakdownBy: breakdownBy ?? undefined,
}),
],
endpointTableQuery: [
(s) => [s.dateRange, s.endpointFilter, s.materializationType],
(
dateRange: {
date_from: string | null
date_to: string | null
},
endpointFilter: string[],
materializationType: 'inline' | 'materialized' | null
): EndpointsUsageTableQuery => ({
kind: NodeKind.EndpointsUsageTableQuery,
dateRange,
endpointNames: endpointFilter.length > 0 ? endpointFilter : undefined,
materializationType,
breakdownBy: EndpointsUsageBreakdown.Endpoint,
orderBy: ['requests', 'DESC'],
limit: 100,
}),
],
}),
listeners(({ actions }) => ({
refresh: () => {
setTimeout(
() => {
actions.clearCooldown()
},
15 * 60 * 1000
)
},
})),
trackedActionToUrl(({ values }) => {
const actionToUrl = ({
dateFilter = values.dateFilter,
endpointFilter = values.endpointFilter,
materializationType = values.materializationType,
interval = values.interval,
breakdownBy = values.breakdownBy,
}): [string, Record<string, any> | undefined, string | undefined] | undefined => {
const { dateFrom, dateTo } = dateFilter
const searchParams = { ...router.values.searchParams }
if (dateFrom !== INITIAL_DATE_FROM) {
searchParams.dateFrom = dateFrom
} else {
delete searchParams.dateFrom
}
if (dateTo !== INITIAL_DATE_TO) {
searchParams.dateTo = dateTo
} else {
delete searchParams.dateTo
}
if (endpointFilter.length > 0) {
searchParams.endpointFilter = endpointFilter.join(',')
} else {
delete searchParams.endpointFilter
}
if (materializationType !== null) {
searchParams.materializationType = materializationType
} else {
delete searchParams.materializationType
}
if (interval !== INITIAL_INTERVAL) {
searchParams.interval = interval
} else {
delete searchParams.interval
}
if (breakdownBy !== null) {
searchParams.breakdownBy = breakdownBy
} else {
delete searchParams.breakdownBy
}
return [router.values.location.pathname, searchParams, router.values.location.hash]
}
return {
setDates: actionToUrl,
setEndpointFilter: actionToUrl,
setMaterializationType: actionToUrl,
setInterval: actionToUrl,
setBreakdownBy: actionToUrl,
}
}),
urlToAction(({ actions, values }) => ({
[urls.endpoints()]: (_, searchParams) => {
if (searchParams.tab !== 'usage') {
return
}
const { dateFrom, dateTo, endpointFilter, materializationType, interval, breakdownBy } = searchParams
const nextDateFrom = dateFrom ?? INITIAL_DATE_FROM
const nextDateTo = dateTo ?? INITIAL_DATE_TO
if (nextDateFrom !== values.dateFilter.dateFrom || nextDateTo !== values.dateFilter.dateTo) {
actions.setDates(nextDateFrom, nextDateTo)
}
const nextEndpointFilter: string[] = endpointFilter ? endpointFilter.split(',') : []
if (
nextEndpointFilter.length !== values.endpointFilter.length ||
nextEndpointFilter.some((v: string, i: number) => v !== values.endpointFilter[i])
) {
actions.setEndpointFilter(nextEndpointFilter)
}
const nextMaterializationType = materializationType ?? null
if (nextMaterializationType !== values.materializationType) {
actions.setMaterializationType(nextMaterializationType)
}
const nextInterval = interval ?? INITIAL_INTERVAL
if (nextInterval !== values.interval) {
actions.setInterval(nextInterval)
}
const nextBreakdownBy = breakdownBy ?? null
if (nextBreakdownBy !== values.breakdownBy) {
actions.setBreakdownBy(nextBreakdownBy)
}
},
})),
])