import { get, OK } from "https://deno.land/x/kall@v0.1.0/mod.ts";
import { components as searchComponents } from "./nrk-search.ts";
import { components as catalogComponents } from "./nrk-catalog.ts";
import { z } from "zod";

type ArrayElement<A> = A extends readonly (infer T)[] ? T : never;
export type Serie = catalogComponents["schemas"]["SeriesViewModel"];
type _OriginalEpisode = catalogComponents["schemas"]["EpisodeHalResource"];
export type OriginalEpisode = _OriginalEpisode & { url: string };
export type SearchResultList = searchComponents["schemas"]["seriesResult"]["results"];
export type SearchResult = ArrayElement<SearchResultList> & {
  description?: string;
};

const manifestSchema = z.object({
  id: z.string(),
  playable: z.object({
    endSequenceStartTime: z.null(),
    duration: z.string(),
    assets: z.array(z.object({
      url: z.string(),
      format: z.string(),
      mimeType: z.string(),
      encrypted: z.boolean(),
    })),
  }),
});

/** Incomplete manifest types. */
type Manifest = z.infer<typeof manifestSchema>;

const nrkAPI = `https://psapi.nrk.no`;

async function withDownloadLink(
  episode: _OriginalEpisode,
): Promise<OriginalEpisode> {
  // getting stream link
  const [playbackStatus, playbackResponseRaw] = await get<Manifest>(
    `${nrkAPI}/playback/manifest/podcast/${episode.episodeId}`,
  );

  const playbackResponse = manifestSchema.parse(playbackResponseRaw);

  if (playbackStatus === OK && playbackResponse) {
    return { ...episode, url: playbackResponse.playable.assets[0].url };
  } else {
    throw `Error getting downloadLink for ${episode.episodeId}, serie: ${episode.originalTitle}. Status: ${playbackStatus}`;
  }
}

export const nrkRadio = {
  search: async (query: string): Promise<SearchResultList> => {
    const [status, response] = await get<
      searchComponents["schemas"]["searchresult"]
    >(`${nrkAPI}/radio/search/search?q=${query}`);
    if (status === OK && response) {
      return response.results.series?.results;
    }
    throw `Something went wrong with ${query} - got status ${status}`;
  },
  getSerieData: async (seriesId: string) => {
    const [
      [episodeStatus, episodeResponse],
      [seriesStatus, serieResponse],
    ] = await Promise.all([
      get<catalogComponents["schemas"]["EpisodesHalResource"]>(
        `${nrkAPI}/radio/catalog/podcast/${seriesId}/episodes`,
      ),
      get<catalogComponents["schemas"]["SeriesHalResource"]>(
        `${nrkAPI}/radio/catalog/podcast/${seriesId}`,
      ),
    ]);

    if (
      episodeStatus === OK && seriesStatus === OK &&
      serieResponse?.series && episodeResponse?._embedded.episodes?.length
    ) {
      const episodes = await Promise.all(
        episodeResponse._embedded.episodes.map((episode) => withDownloadLink(episode)),
      );
      return {
        ...serieResponse.series,
        episodes,
      };
    }
    throw `Error getting episodes for ${seriesId}: EpisodeStatus: ${episodeStatus}. SerieStatus: ${seriesStatus}`;
  },
  getEpisode: async (seriesId: string, episodeId: string): Promise<OriginalEpisode | null> => {
    const url = `${nrkAPI}/radio/catalog/podcast/${seriesId}/episodes/${episodeId}`;
    const [status, episode] = await get<OriginalEpisode>(url);
    if (status === OK && episode) {
      return episode;
    }
    throw `Error getting episode ${episodeId}. Status: ${status}. Series: ${seriesId}`;
  },
};