Files
umami/src/lib/data.ts
T

97 lines
2.4 KiB
TypeScript

import { DATA_TYPE, DATETIME_REGEX, FIELD_LENGTH } from './constants';
import { truncateString } from './format';
import type { DynamicDataType } from './types';
export interface KeyValueData {
key: string;
value: any;
dataType: DynamicDataType;
}
export function flattenJSON(eventData: Record<string, any>): KeyValueData[] {
function flatten(obj: Record<string, any>, parentKey: string): KeyValueData[] {
return Object.entries(obj).flatMap(([key, value]) => {
const fullKey = parentKey ? `${parentKey}.${key}` : key;
if (value && typeof value === 'object' && !Array.isArray(value) && !isValidDateValue(value)) {
return flatten(value, fullKey);
}
return [createKeyValue(fullKey, value)];
});
}
return flatten(eventData, '');
}
export function isValidDateValue(value: string) {
return typeof value === 'string' && DATETIME_REGEX.test(value);
}
export function getDataType(value: any): string {
let type: string = typeof value;
if (isValidDateValue(value)) {
type = 'date';
}
return type;
}
export function getStringValue(value: string, dataType: number) {
if (dataType === DATA_TYPE.number) {
return parseFloat(value).toFixed(4);
}
if (dataType === DATA_TYPE.date) {
return new Date(value).toISOString();
}
return value;
}
export function getStoredStringValue(value: string, dataType: number) {
const stringValue = getStringValue(value, dataType);
if (dataType === DATA_TYPE.array && stringValue.length > FIELD_LENGTH.stringValue) {
return null;
}
return truncateString(stringValue, FIELD_LENGTH.stringValue);
}
export function createKeyValue(key: string, value: any): KeyValueData {
const type = getDataType(value);
let dataType: DynamicDataType;
let processedValue = value;
switch (type) {
case 'number':
dataType = DATA_TYPE.number;
break;
case 'string':
dataType = DATA_TYPE.string;
break;
case 'boolean':
dataType = DATA_TYPE.boolean;
processedValue = value ? 'true' : 'false';
break;
case 'date':
dataType = DATA_TYPE.date;
break;
case 'object':
dataType = DATA_TYPE.array;
processedValue = JSON.stringify(value);
break;
default:
dataType = DATA_TYPE.string;
break;
}
return { key, value: processedValue, dataType };
}
export function objectToArray(obj: object) {
return Object.keys(obj).map(key => obj[key]);
}