import { ScrapeMedia } from "@movie-web/providers"; import { MakeSlice } from "@/stores/player/slices/types"; import { SourceQuality, SourceSliceSource, selectQuality, } from "@/stores/player/utils/qualities"; import { useQualityStore } from "@/stores/quality"; import { ValuesOf } from "@/utils/typeguard"; export const playerStatus = { IDLE: "idle", SCRAPING: "scraping", PLAYING: "playing", SCRAPE_NOT_FOUND: "scrapeNotFound", PLAYBACK_ERROR: "playbackError", } as const; export type PlayerStatus = ValuesOf; export interface PlayerMetaEpisode { number: number; tmdbId: string; title: string; } export interface PlayerMeta { type: "movie" | "show"; title: string; tmdbId: string; imdbId?: string; releaseYear: number; poster?: string; episodes?: PlayerMetaEpisode[]; episode?: PlayerMetaEpisode; season?: { number: number; tmdbId: string; title: string; }; } export interface Caption { language: string; url?: string; srtData: string; } export interface CaptionListItem { language: string; url: string; needsProxy: boolean; } export interface SourceSlice { status: PlayerStatus; source: SourceSliceSource | null; sourceId: string | null; qualities: SourceQuality[]; currentQuality: SourceQuality | null; captionList: CaptionListItem[]; caption: { selected: Caption | null; asTrack: boolean; }; meta: PlayerMeta | null; setStatus(status: PlayerStatus): void; setSource( stream: SourceSliceSource, captions: CaptionListItem[], startAt: number ): void; switchQuality(quality: SourceQuality): void; setMeta(meta: PlayerMeta, status?: PlayerStatus): void; setCaption(caption: Caption | null): void; setSourceId(id: string | null): void; enableAutomaticQuality(): void; redisplaySource(startAt: number): void; } export function metaToScrapeMedia(meta: PlayerMeta): ScrapeMedia { if (meta.type === "show") { if (!meta.episode || !meta.season) throw new Error("missing show data"); return { title: meta.title, releaseYear: meta.releaseYear, tmdbId: meta.tmdbId, type: "show", imdbId: meta.imdbId, episode: meta.episode, season: meta.season, }; } return { title: meta.title, releaseYear: meta.releaseYear, tmdbId: meta.tmdbId, type: "movie", imdbId: meta.imdbId, }; } export const createSourceSlice: MakeSlice = (set, get) => ({ source: null, sourceId: null, qualities: [], captionList: [], currentQuality: null, status: playerStatus.IDLE, meta: null, caption: { selected: null, asTrack: false, }, setSourceId(id) { set((s) => { s.sourceId = id; }); }, setStatus(status: PlayerStatus) { set((s) => { s.status = status; }); }, setMeta(meta, newStatus) { set((s) => { s.meta = meta; s.interface.hideNextEpisodeBtn = false; if (newStatus) s.status = newStatus; }); }, setCaption(caption) { const store = get(); store.display?.setCaption(caption); set((s) => { s.caption.selected = caption; }); }, setSource( stream: SourceSliceSource, captions: CaptionListItem[], startAt: number ) { let qualities: string[] = []; if (stream.type === "file") qualities = Object.keys(stream.qualities); const qualityPreferences = useQualityStore.getState(); const loadableStream = selectQuality(stream, qualityPreferences.quality); set((s) => { s.source = stream; s.qualities = qualities as SourceQuality[]; s.currentQuality = loadableStream.quality; s.captionList = captions; }); const store = get(); store.redisplaySource(startAt); }, redisplaySource(startAt: number) { const store = get(); const quality = store.currentQuality; if (!store.source) return; const qualityPreferences = useQualityStore.getState(); const loadableStream = selectQuality(store.source, { automaticQuality: qualityPreferences.quality.automaticQuality, lastChosenQuality: quality, }); store.display?.load({ source: loadableStream.stream, startAt, automaticQuality: qualityPreferences.quality.automaticQuality, preferredQuality: qualityPreferences.quality.lastChosenQuality, }); }, switchQuality(quality) { const store = get(); if (!store.source) return; if (store.source.type === "file") { const selectedQuality = store.source.qualities[quality]; if (!selectedQuality) return; set((s) => { s.currentQuality = quality; }); store.display?.load({ source: selectedQuality, startAt: store.progress.time, automaticQuality: false, preferredQuality: quality, }); } else if (store.source.type === "hls") { store.display?.changeQuality(false, quality); } }, enableAutomaticQuality() { const store = get(); store.display?.changeQuality(true, null); }, });