gradio-app--gradio/js/dataframe/shared/utils/utils.ts

import type { CellValue } from "../types";

export type Headers = string[];
export type Data = CellValue[][];

export type Datatype =
	| "str"
	| "number"
	| "bool"
	| "date"
	| "markdown"
	| "html"
	| "image";

export type Metadata = {
	[key: string]: string[][] | null;
} | null;
export type HeadersWithIDs = { value: string; id: string }[];
export type DataframeValue = {
	data: Data;
	headers: Headers;
	metadata?: Metadata;
};

/**
 * Coerce a value to a given type.
 * @param v - The value to coerce.
 * @param t - The type to coerce to.
 * @returns The coerced value.
 */
export function cast_value_to_type(v: any, t: Datatype): CellValue {
	if (v === null || v === undefined) {
		return v;
	}
	if (t === "number") {
		const n = Number(v);
		return isNaN(n) ? v : n;
	}
	if (t === "bool") {
		if (typeof v === "boolean") return v;
		if (typeof v === "number") return v !== 0;
		const s = String(v).toLowerCase();
		if (s === "true" || s === "1") return true;
		if (s === "false" || s === "0") return false;
		return v;
	}
	if (t === "date") {
		const d = new Date(v);
		return isNaN(d.getTime()) ? v : d.toISOString();
	}
	return v;
}

export interface EditData {
	index: number | [number, number];
	value: string;
	previous_value: string;
}