utils.ts

import Papa from 'papaparse';
import { getEmojiFlag, type TCountryCode, languages } from 'countries-list';
import { type paths, type components } from './types';

export type AllFeedsType =
  paths['/v1/search']['get']['responses'][200]['content']['application/json'];

export type SearchFeedItem = components['schemas']['SearchFeedItemResult'];

export type EntityLocations = components['schemas']['Locations'];

export type AllFeedsParams = paths['/v1/search']['get']['parameters'];

export type AllFeedType =
  | paths['/v1/feeds/{id}']['get']['responses'][200]['content']['application/json']
  | paths['/v1/gtfs_feeds/{id}']['get']['responses'][200]['content']['application/json']
  | paths['/v1/gtfs_rt_feeds/{id}']['get']['responses'][200]['content']['application/json']
  | paths['/v1/gbfs_feeds/{id}']['get']['responses'][200]['content']['application/json']
  | undefined;

export type BasicFeedType =
  | paths['/v1/feeds/{id}']['get']['responses'][200]['content']['application/json']
  | undefined;

export const isBasicFeedType = (
  data: AllFeedType,
): data is paths['/v1/feeds/{id}']['get']['responses'][200]['content']['application/json'] => {
  return data !== undefined;
};

export type GTFSFeedType =
  | paths['/v1/gtfs_feeds/{id}']['get']['responses'][200]['content']['application/json']
  | undefined;

export const isGtfsFeedType = (
  data: AllFeedType,
): data is paths['/v1/gtfs_feeds/{id}']['get']['responses'][200]['content']['application/json'] => {
  return data !== undefined && data.data_type === 'gtfs';
};

export type GTFSRTFeedType =
  | paths['/v1/gtfs_rt_feeds/{id}']['get']['responses'][200]['content']['application/json']
  | undefined;

export const isGtfsRtFeedType = (
  data: AllFeedType,
): data is paths['/v1/gtfs_rt_feeds/{id}']['get']['responses'][200]['content']['application/json'] => {
  return data !== undefined && data.data_type === 'gtfs_rt';
};

export type GBFSFeedType =
  | paths['/v1/gbfs_feeds/{id}']['get']['responses'][200]['content']['application/json']
  | undefined;

export type GBFSVersionType = components['schemas']['GbfsVersion'];

export type GBFSVersionEndpointType = components['schemas']['GbfsEndpoint'];

export const isGbfsFeedType = (
  data: AllFeedType,
): data is paths['/v1/gbfs_feeds/{id}']['get']['responses'][200]['content']['application/json'] => {
  return data !== undefined && data.data_type === 'gbfs';
};

export type AllDatasetType =
  | paths['/v1/gtfs_feeds/{id}/datasets']['get']['responses'][200]['content']['application/json']
  | paths['/v1/datasets/gtfs/{id}']['get']['responses'][200]['content']['application/json']
  | undefined;

export function getLocationName(
  locations: EntityLocations | undefined | null,
): string {
  if (locations === undefined || locations === null) {
    return '';
  }
  let displayLocation = '';
  locations.forEach((location, index) => {
    if (location.country_code !== undefined && location.country_code !== null) {
      displayLocation += `${getEmojiFlag(
        location.country_code as TCountryCode,
      )} `;
    }
    if (location.country !== undefined && location.country !== null) {
      displayLocation += `${location.country}`;
    }
    if (
      location.subdivision_name !== undefined &&
      location.subdivision_name !== null
    ) {
      displayLocation += `, ${location.subdivision_name}`;
    }
    if (location.municipality !== undefined && location.municipality !== null) {
      displayLocation += `, ${location.municipality}`;
    }

    if (index < locations.length - 1) {
      displayLocation += ' | ';
    }
  });
  return displayLocation;
}

export function isValidFeedLink(feedLink: string): boolean {
  const lowercaseFeedLink = feedLink.toLowerCase();
  const urlPattern = /^(https?:\/\/)([^\s$.?#].[^\s]*)$/;
  return urlPattern.test(lowercaseFeedLink ?? '');
}

export function getCountryLocationSummaries(locations: EntityLocations): Array<{
  country_code: string | undefined;
  country: string | undefined;
  subdivisions: Set<string>;
  municipalities: Set<string>;
}> {
  const uniqueCountries = Array.from(
    new Map(locations.map((loc) => [loc.country_code, loc])).values(),
  );

  return uniqueCountries
    .map((uniqueLoc) => {
      const subdivisions = new Set<string>();
      const municipalities = new Set<string>();

      locations
        .filter((loc) => loc.country_code === uniqueLoc.country_code)
        .forEach((loc) => {
          if (loc.subdivision_name != null)
            subdivisions.add(loc.subdivision_name);
          if (loc.municipality != null) municipalities.add(loc.municipality);
        });

      return {
        country_code: uniqueLoc.country_code,
        country: uniqueLoc.country,
        subdivisions,
        municipalities,
      };
    })
    .sort((a, b) => {
      const lengthA = a.subdivisions.size + a.municipalities.size;
      const lengthB = b.subdivisions.size + b.municipalities.size;
      if (lengthA === lengthB) {
        return (a.country ?? '').localeCompare(b.country ?? '');
      }
      return lengthB - lengthA;
    });
}

export const langCodeToName = (code: string): string => {
  const primary = code.split('-')[0].toLowerCase();
  const lang = languages[primary as keyof typeof languages];
  return lang?.name ?? code.toUpperCase();
};

/**
 * Checks if a feed URL exists in the urls.direct_download column of the feeds_v2.csv file.
 * @param feedUrl The URL to check for existence.
 * @param csvUrl The CSV file URL (default: feeds_v2.csv from Mobility Database)
 * @returns Promise<boolean> true if exists, false otherwise
 */
export async function checkFeedUrlExistsInCsv(
  feedUrl: string,
  csvUrl = 'https://files.mobilitydatabase.org/feeds_v2.csv',
): Promise<string | null> {
  try {
    const response = await fetch(csvUrl);
    if (!response.ok) throw new Error('Failed to fetch CSV');
    const csvText = await response.text();
    const parsed = Papa.parse<FeedCsvRow>(csvText, { header: true });
    if (parsed.data == null || !Array.isArray(parsed.data)) return null;
    const rows = parsed.data;
    const match = rows.find((row) => row['urls.direct_download'] === feedUrl);
    return typeof match?.id === 'string' ? match.id : null;
  } catch (e) {
    return null;
  }
}

/**
 * FeedCsvRow interface with fields
 */
interface FeedCsvRow {
  'urls.direct_download'?: string;
  id: string;
}