forked from mirrors/thatmattlove-hyperglass
128 lines
3.7 KiB
TypeScript
128 lines
3.7 KiB
TypeScript
import { useCallback } from 'react';
|
|
import dayjs from 'dayjs';
|
|
import relativeTimePlugin from 'dayjs/plugin/relativeTime';
|
|
import utcPlugin from 'dayjs/plugin/utc';
|
|
import { useConfig } from '~/context';
|
|
import { isStructuredOutput } from '~/types';
|
|
|
|
type TableToStringFormatter =
|
|
| ((v: string) => string)
|
|
| ((v: number) => string)
|
|
| ((v: number[]) => string)
|
|
| ((v: string[]) => string)
|
|
| ((v: boolean) => string);
|
|
|
|
interface TableToStringFormatted {
|
|
age: (v: number) => string;
|
|
active: (v: boolean) => string;
|
|
as_path: (v: number[]) => string;
|
|
communities: (v: string[]) => string;
|
|
rpki_state: (v: number, n: RPKIState) => string;
|
|
}
|
|
|
|
dayjs.extend(relativeTimePlugin);
|
|
dayjs.extend(utcPlugin);
|
|
|
|
function formatAsPath(path: number[]): string {
|
|
return path.join(' → ');
|
|
}
|
|
|
|
function formatCommunities(comms: string[]): string {
|
|
const commsStr = comms.map(c => ` - ${c}`);
|
|
return '\n' + commsStr.join('\n');
|
|
}
|
|
|
|
function formatBool(val: boolean): string {
|
|
let fmt = '';
|
|
if (val === true) {
|
|
fmt = 'yes';
|
|
} else if (val === false) {
|
|
fmt = 'no';
|
|
}
|
|
return fmt;
|
|
}
|
|
|
|
function formatTime(val: number): string {
|
|
const now = dayjs.utc();
|
|
const then = now.subtract(val, 'second');
|
|
const timestamp = then.toString().replace('GMT', 'UTC');
|
|
const relative = now.to(then, true);
|
|
return `${relative} (${timestamp})`;
|
|
}
|
|
|
|
/**
|
|
* Get a function to convert table data to string, for use in the copy button component.
|
|
*/
|
|
export function useTableToString(
|
|
target: string,
|
|
data: QueryResponse | undefined,
|
|
...deps: unknown[]
|
|
): () => string {
|
|
const { web, parsedDataFields, messages } = useConfig();
|
|
|
|
function formatRpkiState(val: number): string {
|
|
const rpkiStates = [
|
|
web.text.rpkiInvalid,
|
|
web.text.rpkiValid,
|
|
web.text.rpkiUnknown,
|
|
web.text.rpkiUnverified,
|
|
];
|
|
return rpkiStates[val];
|
|
}
|
|
|
|
const tableFormatMap = {
|
|
age: formatTime,
|
|
active: formatBool,
|
|
as_path: formatAsPath,
|
|
communities: formatCommunities,
|
|
rpki_state: formatRpkiState,
|
|
};
|
|
|
|
function isFormatted(key: string): key is keyof TableToStringFormatted {
|
|
return key in tableFormatMap;
|
|
}
|
|
|
|
function getFmtFunc(accessor: keyof Route): TableToStringFormatter {
|
|
if (isFormatted(accessor)) {
|
|
return tableFormatMap[accessor];
|
|
} else {
|
|
return String;
|
|
}
|
|
}
|
|
|
|
function doFormat(target: string, data: QueryResponse | undefined): string {
|
|
let result = messages.noOutput;
|
|
try {
|
|
if (typeof data !== 'undefined' && isStructuredOutput(data)) {
|
|
const tableStringParts = [`Routes For: ${target}`, `Timestamp: ${data.timestamp} UTC`];
|
|
for (const route of data.output.routes) {
|
|
for (const field of parsedDataFields) {
|
|
const [header, accessor, align] = field;
|
|
if (align !== null) {
|
|
let value = route[accessor];
|
|
const fmtFunc = getFmtFunc(accessor) as (v: typeof value) => string;
|
|
value = fmtFunc(value);
|
|
if (accessor === 'prefix') {
|
|
tableStringParts.push(` - ${header}: ${value}`);
|
|
} else {
|
|
tableStringParts.push(` - ${header}: ${value}`);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
result = tableStringParts.join('\n');
|
|
}
|
|
return result;
|
|
} catch (err) {
|
|
console.error(err);
|
|
let error = String(err);
|
|
if (err instanceof Error) {
|
|
error = err.message;
|
|
}
|
|
return `An error occurred while parsing the output: '${error}'`;
|
|
}
|
|
}
|
|
const formatCallback = useCallback(doFormat, [target, data, doFormat]);
|
|
// eslint-disable-next-line react-hooks/exhaustive-deps
|
|
return useCallback(() => formatCallback(target, data), [target, data, formatCallback, ...deps]);
|
|
}
|