iptvnator/libs/m3u-state/src/lib/effects.ts

import { inject, Injectable } from '@angular/core';
import { MatSnackBar } from '@angular/material/snack-bar';
import { Router } from '@angular/router';
import { EpgService } from '@iptvnator/epg/data-access';
import {
    normalizeEpgUrls,
    resolveM3uCatchupUrl,
} from '@iptvnator/shared/m3u-utils';
import { Actions, createEffect, ofType } from '@ngrx/effects';
import { Store } from '@ngrx/store';
import { StorageMap } from '@ngx-pwa/local-storage';
import { TranslateService } from '@ngx-translate/core';
import {
    EMPTY,
    filter,
    firstValueFrom,
    from,
    map,
    mergeMap,
    switchMap,
    tap,
    withLatestFrom,
} from 'rxjs';
import {
    DataService,
    PlaylistsService,
    SettingsStore,
} from '@iptvnator/services';
import {
    OPEN_MPV_PLAYER,
    OPEN_VLC_PLAYER,
    Channel,
    Playlist,
    PlaylistMeta,
    STORE_KEY,
    VideoPlayer,
} from '@iptvnator/shared/interfaces';
import {
    ChannelActions,
    EpgActions,
    FavoritesActions,
    PlaylistActions,
} from './actions';
import {
    selectActive,
    selectActivePlaylistId,
    selectChannels,
    selectFavorites,
} from './selectors';
import { resolveChannelEpgLookupKey } from './channel-epg-lookup.util';
import { buildExternalPlayerPayload } from './external-player-payload.util';
import { resolvePlaylistScopedEpgFetchPlan } from './playlist-scoped-epg-fetch.util';

@Injectable({ providedIn: 'any' })
export class PlaylistEffects {
    private actions$ = inject(Actions);
    private playlistsService = inject(PlaylistsService);
    private dataService = inject(DataService);
    private epgService = inject(EpgService);
    private router = inject(Router);
    private snackBar = inject(MatSnackBar);
    private storage = inject(StorageMap);
    private store = inject(Store);
    private translate = inject(TranslateService);
    private settingsStore = inject(SettingsStore);
    private readonly playlistScopedEpgFetchKeys = new Map<string, string>();

    updateFavorites$ = createEffect(
        () => {
            return this.actions$.pipe(
                ofType(FavoritesActions.updateFavorites),
                withLatestFrom(
                    this.store.select(selectFavorites),
                    this.store.select(selectActivePlaylistId)
                ),
                filter(([, , playlistId]) => !!playlistId),
                switchMap(([, favorites, playlistId]) =>
                    this.playlistsService.updateFavorites(playlistId, favorites)
                ),
                tap(() => {
                    this.snackBar.open(
                        this.translate.instant('CHANNELS.FAVORITES_UPDATED'),
                        undefined,
                        { duration: 2000 }
                    );
                })
            );
        },
        { dispatch: false }
    );

    setFavorites$ = createEffect(
        () => {
            return this.actions$.pipe(
                ofType(FavoritesActions.setFavorites),
                withLatestFrom(this.store.select(selectActivePlaylistId)),
                filter(([, playlistId]) => !!playlistId),
                switchMap(([action, playlistId]) =>
                    this.playlistsService.setFavorites(
                        playlistId,
                        action.channelIds
                    )
                )
            );
        },
        {
            dispatch: false,
        }
    );

    resolveActiveEpgProgram$ = createEffect(() => {
        return this.actions$.pipe(
            ofType(EpgActions.setActiveEpgProgram),
            withLatestFrom(this.store.select(selectActive)),
            map(([action, activeChannel]) => {
                const playbackUrl = activeChannel
                    ? resolveM3uCatchupUrl(activeChannel, action.program)
                    : null;

                return playbackUrl
                    ? EpgActions.setActivePlaybackUrl({ playbackUrl })
                    : EpgActions.resetActiveEpgProgram();
            })
        );
    });

    openArchivedPlayback$ = createEffect(
        () => {
            return this.actions$.pipe(
                ofType(EpgActions.setActivePlaybackUrl),
                withLatestFrom(this.store.select(selectActive)),
                tap(([action, activeChannel]) => {
                    void this.openWithConfiguredExternalPlayer(
                        action.playbackUrl,
                        activeChannel
                    );
                })
            );
        },
        { dispatch: false }
    );

    returnToLivePlayback$ = createEffect(
        () => {
            return this.actions$.pipe(
                ofType(EpgActions.returnToLivePlayback),
                withLatestFrom(this.store.select(selectActive)),
                filter(([, activeChannel]) => Boolean(activeChannel?.url)),
                tap(([, activeChannel]) => {
                    void this.openWithConfiguredExternalPlayer(
                        activeChannel?.url ?? '',
                        activeChannel
                    );
                })
            );
        },
        { dispatch: false }
    );

    setActiveChannel$ = createEffect(() => {
        return this.actions$.pipe(
            ofType(ChannelActions.setActiveChannel),
            // Skip the effect entirely when channel is falsy
            filter((action) => !!action.channel),
            map((action) => {
                const { channel } = action;

                // Use modern EPG service to get channel programs
                const channelId = resolveChannelEpgLookupKey(channel);
                if (channelId) {
                    this.epgService.getChannelPrograms(channelId);
                }

                void window.electron
                    ?.setUserAgent(
                        channel.http?.['user-agent'],
                        channel.http?.referrer,
                        channel.url
                    )
                    .catch((error: unknown) => {
                        console.warn(
                            '[PlaylistEffects] Failed to configure Electron request headers:',
                            error
                        );
                    });

                firstValueFrom(this.storage.get(STORE_KEY.Settings)).then(
                    (settings: any) => {
                        const shouldOpenExternalPlayer =
                            !settings?.openStreamOnDoubleClick ||
                            action.startPlayback === true;

                        if (
                            settings &&
                            Object.keys(settings).length > 0 &&
                            shouldOpenExternalPlayer &&
                            settings.player === VideoPlayer.MPV &&
                            channel.radio !== 'true'
                        ) {
                            this.dataService.sendIpcEvent(OPEN_MPV_PLAYER, {
                                url: channel.url,
                                title: channel.name ?? '',
                                'user-agent': channel.http['user-agent'],
                                referer: channel.http.referrer,
                                origin: channel.http.origin,
                            });
                        } else if (
                            settings &&
                            Object.keys(settings).length > 0 &&
                            shouldOpenExternalPlayer &&
                            settings.player === VideoPlayer.VLC &&
                            channel.radio !== 'true'
                        )
                            this.dataService.sendIpcEvent(OPEN_VLC_PLAYER, {
                                url: channel.url,
                                title: channel.name ?? '',
                                'user-agent': channel.http['user-agent'],
                                referer: channel.http.referrer,
                                origin: channel.http.origin,
                            });
                    }
                );

                return ChannelActions.setActiveChannelSuccess({
                    channel: action.channel,
                });
            })
        );
    });

    loadPlaylists$ = createEffect(() => {
        return this.actions$.pipe(
            ofType(PlaylistActions.loadPlaylists),
            switchMap(() =>
                this.playlistsService.getAllPlaylists().pipe(
                    tap((playlists) => {
                        this.fetchPlaylistScopedEpgForPlaylists(playlists);
                    }),
                    map((playlists) =>
                        PlaylistActions.loadPlaylistsSuccess({
                            playlists,
                        })
                    )
                )
            )
        );
    });

    private async openWithConfiguredExternalPlayer(
        playbackUrl: string,
        activeChannel: Channel | undefined | null
    ): Promise<void> {
        const payload = buildExternalPlayerPayload(activeChannel, playbackUrl);
        if (!payload) {
            return;
        }

        const settings: any = await firstValueFrom(
            this.storage.get(STORE_KEY.Settings)
        );
        if (!settings || Object.keys(settings).length === 0) {
            return;
        }

        if (settings.player === VideoPlayer.MPV) {
            this.dataService.sendIpcEvent(OPEN_MPV_PLAYER, payload);
            return;
        }

        if (settings.player === VideoPlayer.VLC) {
            this.dataService.sendIpcEvent(OPEN_VLC_PLAYER, payload);
        }
    }

    removePlaylist$ = createEffect(
        () => {
            return this.actions$.pipe(
                ofType(PlaylistActions.removePlaylist),
                switchMap(async (action) => {
                    this.playlistScopedEpgFetchKeys.delete(action.playlistId);
                    await firstValueFrom(
                        this.playlistsService.deletePlaylist(action.playlistId)
                    );
                })
            );
        },
        { dispatch: false }
    );

    updatePlaylist$ = createEffect(
        () => {
            return this.actions$.pipe(
                ofType(PlaylistActions.updatePlaylist),
                switchMap((action) =>
                    this.playlistsService.updatePlaylist(action.playlistId, {
                        ...action.playlist,
                        _id: action.playlistId,
                    }).pipe(
                        tap(() => {
                            this.fetchPlaylistScopedEpg(action.playlist, {
                                force: action.refreshEpg === true,
                            });
                        })
                    )
                )
            );
        },
        { dispatch: false }
    );

    parsePlaylist$ = createEffect(() => {
        return this.actions$.pipe(
            ofType(PlaylistActions.parsePlaylist),
            mergeMap((action) =>
                from(
                    this.playlistsService.handlePlaylistParsing(
                        action.uploadType,
                        action.playlist,
                        action.title,
                        action.path
                    )
                )
            ),
            map((playlist) =>
                PlaylistActions.addPlaylist({
                    playlist,
                })
            )
        );
    });

    addPlaylist$ = createEffect(
        () => {
            return this.actions$.pipe(
                ofType(
                    PlaylistActions.addPlaylist,
                    PlaylistActions.handleAddingPlaylistByUrl
                ),
                switchMap((action) => {
                    if ('isTemporary' in action && action.isTemporary) {
                        return EMPTY;
                    }
                    return this.playlistsService.addPlaylist(action.playlist).pipe(
                        tap(() => {
                            this.fetchPlaylistScopedEpg(action.playlist);
                            this.navigateToPlaylist(action.playlist);
                        })
                    );
                })
            );
        },
        { dispatch: false }
    );

    updatePlaylistMeta$ = createEffect(
        () => {
            return this.actions$.pipe(
                ofType(PlaylistActions.updatePlaylistMeta),
                switchMap((action) =>
                    this.playlistsService.updatePlaylistMeta(action.playlist).pipe(
                        tap(() => {
                            if (
                                this.hasPlaylistScopedEpgSourceChange(
                                    action.playlist
                                )
                            ) {
                                this.fetchPlaylistScopedEpg(action.playlist);
                            }
                        })
                    )
                )
            );
        },
        { dispatch: false }
    );

    updatePlaylistPositions$ = createEffect(
        () => {
            return this.actions$.pipe(
                ofType(PlaylistActions.updatePlaylistPositions),
                switchMap((action) =>
                    this.playlistsService.updatePlaylistPositions(
                        action.positionUpdates
                    )
                )
            );
        },
        { dispatch: false }
    );

    addManyPlaylists$ = createEffect(
        () => {
            return this.actions$.pipe(
                ofType(PlaylistActions.addManyPlaylists),
                switchMap((action) =>
                    this.playlistsService.addManyPlaylists(action.playlists)
                )
            );
        },
        { dispatch: false }
    );

    updateManyPlaylists$ = createEffect(
        () => {
            return this.actions$.pipe(
                ofType(PlaylistActions.updateManyPlaylists),
                switchMap((action) =>
                    this.playlistsService
                        .updateManyPlaylists(action.playlists)
                        .pipe(
                            tap(() => {
                                action.playlists.forEach((playlist) =>
                                    this.fetchPlaylistScopedEpg(playlist)
                                );
                            })
                        )
                )
            );
        },
        { dispatch: false }
    );

    setAdjacentChannelAsActive$ = createEffect(() => {
        return this.actions$.pipe(
            ofType(ChannelActions.setAdjacentChannelAsActive),
            withLatestFrom(
                this.store.select(selectChannels),
                this.store.select(selectActive)
            ),
            map(([action, channels, activeChannel]) => {
                let adjacentChannel;
                const index = channels.findIndex(
                    (channel) => channel.id === activeChannel?.id
                );
                if (action.direction === 'next') {
                    if (index === channels.length - 1)
                        adjacentChannel = activeChannel;
                    adjacentChannel = channels[index + 1];
                } else if (action.direction === 'previous') {
                    if (index === -1 || index === 0)
                        adjacentChannel = activeChannel;
                    adjacentChannel = channels[index - 1];
                }
                return ChannelActions.setActiveChannelSuccess({
                    channel: adjacentChannel!,
                });
            })
        );
    });

    private navigateToPlaylist(playlist: Playlist): void {
        if (playlist.serverUrl) {
            void this.router.navigate(['/workspace', 'xtreams', playlist._id]);
            return;
        }

        if (playlist.macAddress) {
            void this.router.navigate(['/workspace', 'stalker', playlist._id]);
            return;
        }

        void this.router.navigate(['/workspace', 'playlists', playlist._id]);
    }

    private fetchPlaylistScopedEpg(
        playlist: Pick<
            Playlist,
            '_id' | 'epgUrls' | 'macAddress' | 'serverUrl'
        >,
        options: { force?: boolean } = {}
    ): void {
        const plan = resolvePlaylistScopedEpgFetchPlan(
            playlist,
            this.getGlobalEpgUrls(),
            this.playlistScopedEpgFetchKeys.get(playlist._id),
            options
        );
        this.playlistScopedEpgFetchKeys.set(playlist._id, plan.key);

        if (!plan.shouldFetch) {
            return;
        }

        this.epgService.fetchEpg(plan.urls);
    }

    private hasPlaylistScopedEpgSourceChange(playlist: PlaylistMeta): boolean {
        return (
            Object.prototype.hasOwnProperty.call(playlist, 'epgUrls') ||
            Object.prototype.hasOwnProperty.call(playlist, 'detectedEpgUrls') ||
            Object.prototype.hasOwnProperty.call(playlist, 'manualEpgUrls') ||
            Object.prototype.hasOwnProperty.call(playlist, 'disabledEpgUrls')
        );
    }

    private fetchPlaylistScopedEpgForPlaylists(playlists: Playlist[]): void {
        const epgUrls = new Set<string>();
        const globalEpgUrls = this.getGlobalEpgUrls();

        for (const playlist of playlists) {
            const plan = resolvePlaylistScopedEpgFetchPlan(
                playlist,
                globalEpgUrls,
                this.playlistScopedEpgFetchKeys.get(playlist._id)
            );
            this.playlistScopedEpgFetchKeys.set(playlist._id, plan.key);

            if (!plan.shouldFetch) {
                continue;
            }

            for (const url of plan.urls) {
                epgUrls.add(url);
            }
        }

        if (epgUrls.size > 0) {
            this.epgService.fetchEpg(Array.from(epgUrls));
        }
    }

    private getGlobalEpgUrls(): string[] {
        return normalizeEpgUrls(this.settingsStore.getSettings().epgUrl ?? []);
    }
}