import { UrlExtractorInterface } from "./types";

export default class UrlExtractor implements UrlExtractorInterface {
  private extract<T extends string>(
    url: string,
    regex: RegExp,
    entityName: string,
    groups: T[]
  ): Record<T, string> {
    return groups.reduce((carry: { [key: string]: string }, item: string) => {
      const match = url.match(regex)?.groups?.[item];
      if (!match) {
        throw new Error(`non ${entityName} url received`);
      }
      carry[item] = match;
      return carry;
    }, {}) as Record<T, string>;
  }

  extractTrackId(url: string): number {
    const extracted = this.extract(
      url,
      /(https?:\/\/)?music\.yandex\.(ru|com|kz|by)\/album\/\d+\/track\/(?<id>\d+)/,
      "track",
      ["id"]
    );
    return Number(extracted.id);
  }

  extractAlbumId(url: string): number {
    const extracted = this.extract(
      url,
      /(https?:\/\/)?music\.yandex\.(ru|com|kz|by)\/album\/(?<id>\d+)/,
      "album",
      ["id"]
    );
    return Number(extracted.id);
  }

  extractArtistId(url: string): number {
    const extracted = this.extract(
      url,
      /(https?:\/\/)?music\.yandex\.(ru|com|kz|by)\/artist\/(?<id>\d+)/,
      "artist",
      ["id"]
    );
    return Number(extracted.id);
  }

  extractPlaylistId(url: string): { id: number; user: string } {
    const extracted = this.extract(
      url,
      /(https?:\/\/)?music\.yandex\.(ru|com|kz|by)\/users\/(?<user>[\w\d\-_\.\@]+)\/playlists\/(?<id>\d+)/,
      "playlist",
      ["id", "user"]
    );
    return { id: Number(extracted.id), user: extracted.user };
  }
}