utils.ts

import specialCodeDictionary from './consts';
import type { TimetableItemResult } from './types';

/** Filter timetable items by their names and labels */
export const keepItem = (item: TimetableItemResult, searchString: string): boolean => {
  if (searchString) {
    const searchStringInName = item.trainName.toLowerCase().includes(searchString.toLowerCase());
    const searchStringInTags = item.labels
      ? item.labels.join('').toLowerCase().includes(searchString.toLowerCase())
      : false;
    return searchStringInName || searchStringInTags;
  }
  return true;
};

export const extractTagCode = (tag?: string | null) => {
  if (!tag) {
    return 'NO CODE';
  }
  if (tag in specialCodeDictionary) {
    return specialCodeDictionary[tag];
  }

  const matches = tag.match(/\w+$/);
  return matches ? matches[0] : tag;
};

export const timetableHasInvalidItem = (timetableItems: TimetableItemResult[]) =>
  timetableItems.some((timetableItem) => timetableItem.invalidReason);