import {z} from 'zod' export type TraceSearchFilterDurationFilter = {min_secs?: number | undefined, max_secs?: number | undefined}; export type TraceSearchFilterBoolFilter = {value?: boolean | undefined}; export type TraceSearchFilterTraceFilter = {duration?: TraceSearchFilterDurationFilter | undefined, error?: TraceSearchFilterBoolFilter | undefined}; export type SpanFilterSpanFilterMatchType = 'INCLUDE' | 'EXCLUDE'; export type StringFilterStringFilterMatchType = 'EXACT' | 'REGEX' | 'EXACT_NEGATION' | 'REGEX_NEGATION' | 'IN' | 'NOT_IN'; export type TraceSearchFilterStringFilter = {match?: StringFilterStringFilterMatchType | undefined, value?: string | undefined, in_values?: Array | undefined}; export type NumericFilterComparisonType = 'EQUAL' | 'NOT_EQUAL' | 'GREATER_THAN' | 'GREATER_THAN_OR_EQUAL' | 'LESS_THAN' | 'LESS_THAN_OR_EQUAL'; export type TraceSearchFilterNumericFilter = {comparison?: NumericFilterComparisonType | undefined, value?: number | undefined}; export type TraceSearchFilterTagFilter = {key?: string | undefined, value?: TraceSearchFilterStringFilter | undefined, numeric_value?: TraceSearchFilterNumericFilter | undefined}; export type TraceSearchFilterCountFilter = {min?: number | undefined, max?: number | undefined}; export type TraceSearchFilterSpanFilter = {match_type?: SpanFilterSpanFilterMatchType | undefined, service?: TraceSearchFilterStringFilter | undefined, operation?: TraceSearchFilterStringFilter | undefined, parent_service?: TraceSearchFilterStringFilter | undefined, parent_operation?: TraceSearchFilterStringFilter | undefined, duration?: TraceSearchFilterDurationFilter | undefined, error?: TraceSearchFilterBoolFilter | undefined, tags?: Array | undefined, span_count?: TraceSearchFilterCountFilter | undefined, is_root_span?: TraceSearchFilterBoolFilter | undefined}; export type Configv1TraceSearchFilter = {trace?: TraceSearchFilterTraceFilter | undefined, span?: Array | undefined}; export type GroupByKeyGroupByKeyType = 'SERVICE' | 'OPERATION' | 'TAG'; export type GroupByGroupByKey = {type?: GroupByKeyGroupByKeyType | undefined, named_key?: string | undefined}; export type Configv1TraceMetricsRuleGroupBy = {label?: string | undefined, key?: GroupByGroupByKey | undefined}; export type Configv1TraceMetricsRule = {name?: string | undefined, slug?: string | undefined, created_at?: string | undefined, updated_at?: string | undefined, metric_name?: string | undefined, metric_labels?: {[key: string]: string} | undefined, histogram_buckets_secs?: Array | undefined, trace_filter?: Configv1TraceSearchFilter | undefined, group_by?: Array | undefined}; export type Configv1CreateTraceMetricsRuleRequest = {trace_metrics_rule?: Configv1TraceMetricsRule | undefined, dry_run?: boolean | undefined}; export type CreateApiApiV1ConfigTraceMetricsRulesArgs = {body: Configv1CreateTraceMetricsRuleRequest}; export type CreateApiApiV1ConfigTraceMetricsRulesOptions = {baseUrl: string, fetch?: typeof fetch | undefined}; export const traceSearchFilterDurationFilter = z.object({min_secs: z.number().optional(), max_secs: z.number().optional()}); export const traceSearchFilterBoolFilter = z.object({value: z.boolean().optional()}); export const traceSearchFilterTraceFilter = z.object({duration: traceSearchFilterDurationFilter.optional(), error: traceSearchFilterBoolFilter.optional()}); export const spanFilterSpanFilterMatchType = z.enum(["INCLUDE", "EXCLUDE"]); export const stringFilterStringFilterMatchType = z.enum(["EXACT", "REGEX", "EXACT_NEGATION", "REGEX_NEGATION", "IN", "NOT_IN"]); export const traceSearchFilterStringFilter = z.object({match: stringFilterStringFilterMatchType.optional(), value: z.string().optional(), in_values: z.array(z.string()).optional()}); export const numericFilterComparisonType = z.enum(["EQUAL", "NOT_EQUAL", "GREATER_THAN", "GREATER_THAN_OR_EQUAL", "LESS_THAN", "LESS_THAN_OR_EQUAL"]); export const traceSearchFilterNumericFilter = z.object({comparison: numericFilterComparisonType.optional(), value: z.number().optional()}); export const traceSearchFilterTagFilter = z.object({key: z.string().optional(), value: traceSearchFilterStringFilter.optional(), numeric_value: traceSearchFilterNumericFilter.optional()}); export const traceSearchFilterCountFilter = z.object({min: z.number().int().optional(), max: z.number().int().optional()}); export const traceSearchFilterSpanFilter = z.object({match_type: spanFilterSpanFilterMatchType.optional(), service: traceSearchFilterStringFilter.optional(), operation: traceSearchFilterStringFilter.optional(), parent_service: traceSearchFilterStringFilter.optional(), parent_operation: traceSearchFilterStringFilter.optional(), duration: traceSearchFilterDurationFilter.optional(), error: traceSearchFilterBoolFilter.optional(), tags: z.array(traceSearchFilterTagFilter).optional(), span_count: traceSearchFilterCountFilter.optional(), is_root_span: traceSearchFilterBoolFilter.optional()}); export const configv1TraceSearchFilter = z.object({trace: traceSearchFilterTraceFilter.optional(), span: z.array(traceSearchFilterSpanFilter).optional()}); export const groupByKeyGroupByKeyType = z.enum(["SERVICE", "OPERATION", "TAG"]); export const groupByGroupByKey = z.object({type: groupByKeyGroupByKeyType.optional(), named_key: z.string().optional()}); export const configv1TraceMetricsRuleGroupBy = z.object({label: z.string().optional(), key: groupByGroupByKey.optional()}); export const configv1TraceMetricsRule = z.object({name: z.string().optional(), slug: z.string().optional(), created_at: z.string().optional(), updated_at: z.string().optional(), metric_name: z.string().optional(), metric_labels: z.record(z.string(), z.string()).optional(), histogram_buckets_secs: z.array(z.number()).optional(), trace_filter: configv1TraceSearchFilter.optional(), group_by: z.array(configv1TraceMetricsRuleGroupBy).optional()}); export const configv1CreateTraceMetricsRuleResponse = z.object({trace_metrics_rule: configv1TraceMetricsRule.optional()}); export const validateResponse = async ( schema: { readonly '~standard': { readonly validate: (value: unknown) => ({ readonly value: Output; readonly issues?: undefined } | { readonly issues: ReadonlyArray<{ readonly message: string; readonly path?: ReadonlyArray | undefined }> }) | Promise<{ readonly value: Output; readonly issues?: undefined } | { readonly issues: ReadonlyArray<{ readonly message: string; readonly path?: ReadonlyArray | undefined }> }> } }, value: unknown ): Promise => { const result = await schema['~standard'].validate(value) if (result.issues !== undefined) { const detail = result.issues .map(({ message, path }) => path?.length ? `${path.map((segment) => String(typeof segment === 'object' ? segment.key : segment)).join('.')}: ${message}` : message ) .join('; ') throw new Error(`Response does not match the schema: ${detail}`, { cause: { issues: result.issues } }) } return result.value }; /** * *** * * POST /api/v1/config/trace-metrics-rules */ export const createApiApiV1ConfigTraceMetricsRules = async (args: CreateApiApiV1ConfigTraceMetricsRulesArgs, options: CreateApiApiV1ConfigTraceMetricsRulesOptions) => { const {baseUrl, fetch: fetchFn = fetch} = options ?? {} const query = new URLSearchParams() const headers = new Headers() headers.set('content-type', "application/json") headers.set('accept', "application/json") const url = `${baseUrl.replace(/\/+$/, '')}/api/v1/config/trace-metrics-rules${query.size > 0 ? `?${query}` : ''}` const res = await fetchFn(url, {method: "POST", headers, body: JSON.stringify(args.body)}) if (!res.ok) throw new Error(`Request failed with status ${res.status}`, {cause: {status: res.status, body: await res.text()}}) return validateResponse(configv1CreateTraceMetricsRuleResponse, await res.json()) };