import type { CustomDataMap } from '../core/CoreNode.js';

export function santizeCustomDataMap(d: CustomDataMap): CustomDataMap {
  const validTypes = {
    boolean: true,
    string: true,
    number: true,
    undefined: true,
  };

  const keys = Object.keys(d);
  for (let i = 0; i < keys.length; i++) {
    const key = keys[i];
    if (!key) {
      continue;
    }

    const value = d[key];
    const valueType = typeof value;

    // Typescript doesn't understand the above const valueType ¯\_(ツ)_/¯
    // eslint-disable-next-line @typescript-eslint/ban-ts-comment
    // @ts-ignore-next-line
    if (valueType === 'string' && value.length > 2048) {
      console.warn(
        `Custom Data value for ${key} is too long, it will be truncated to 2048 characters`,
      );

      // same here, see above comment, this can only be a string at this point
      // eslint-disable-next-line @typescript-eslint/ban-ts-comment
      // @ts-ignore-next-line
      // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment, @typescript-eslint/no-unsafe-call
      d[key] = value.substring(0, 2048);
    }

    if (!validTypes[valueType as keyof typeof validTypes]) {
      console.warn(
        `Custom Data value for ${key} is not a boolean, string, or number, it will be ignored`,
      );
      delete d[key];
    }
  }

  return d;
}