import { inject, Injectable } from '@angular/core'; import { HttpClient, HttpErrorResponse } from '@angular/common/http'; import { of, Subject } from 'rxjs'; import { catchError } from 'rxjs/operators'; import { MeTubeSocket } from './metube-socket.service'; import { Download, Status, State } from '../interfaces'; import { takeUntilDestroyed } from '@angular/core/rxjs-interop'; export interface AddDownloadPayload { url: string; downloadType: string; codec: string; quality: string; format: string; folder: string; customNamePrefix: string; playlistItemLimit: number; autoStart: boolean; splitByChapters: boolean; chapterTemplate: string; subtitleLanguage: string; subtitleMode: string; ytdlOptionsPresets: string[]; ytdlOptionsOverrides: string; } @Injectable({ providedIn: 'root' }) export class DownloadsService { private http = inject(HttpClient); private socket = inject(MeTubeSocket); loading = true; queue = new Map(); done = new Map(); queueChanged = new Subject(); doneChanged = new Subject(); customDirsChanged = new Subject>(); ytdlOptionsChanged = new Subject>(); configurationChanged = new Subject>(); updated = new Subject(); configuration: Record = {}; customDirs: Record = {}; constructor() { this.socket.fromEvent('all') .pipe(takeUntilDestroyed()) .subscribe((strdata: string) => { this.loading = false; const data: [[[string, Download]], [[string, Download]]] = JSON.parse(strdata); this.queue.clear(); data[0].forEach(entry => this.queue.set(...entry)); this.done.clear(); data[1].forEach(entry => this.done.set(...entry)); this.queueChanged.next(); this.doneChanged.next(); }); this.socket.fromEvent('added') .pipe(takeUntilDestroyed()) .subscribe((strdata: string) => { const data: Download = JSON.parse(strdata); this.queue.set(data.url, data); this.queueChanged.next(); }); this.socket.fromEvent('updated') .pipe(takeUntilDestroyed()) .subscribe((strdata: string) => { const data: Download = JSON.parse(strdata); const dl: Download | undefined = this.queue.get(data.url); data.checked = !!dl?.checked; data.deleting = !!dl?.deleting; this.queue.set(data.url, data); this.updated.next(); }); this.socket.fromEvent('completed') .pipe(takeUntilDestroyed()) .subscribe((strdata: string) => { const data: Download = JSON.parse(strdata); this.queue.delete(data.url); this.done.set(data.url, data); this.queueChanged.next(); this.doneChanged.next(); }); this.socket.fromEvent('canceled') .pipe(takeUntilDestroyed()) .subscribe((strdata: string) => { const data: string = JSON.parse(strdata); this.queue.delete(data); this.queueChanged.next(); }); this.socket.fromEvent('cleared') .pipe(takeUntilDestroyed()) .subscribe((strdata: string) => { const data: string = JSON.parse(strdata); this.done.delete(data); this.doneChanged.next(); }); this.socket.fromEvent('configuration') .pipe(takeUntilDestroyed()) .subscribe((strdata: string) => { const data = JSON.parse(strdata); console.debug("got configuration:", data); this.configuration = data; this.configurationChanged.next(data); }); this.socket.fromEvent('custom_dirs') .pipe(takeUntilDestroyed()) .subscribe((strdata: string) => { const data = JSON.parse(strdata); console.debug("got custom_dirs:", data); this.customDirs = data; this.customDirsChanged.next(data); }); this.socket.fromEvent('ytdl_options_changed') .pipe(takeUntilDestroyed()) .subscribe((strdata: string) => { const data = JSON.parse(strdata); this.ytdlOptionsChanged.next(data); }); } handleHTTPError(error: HttpErrorResponse) { const msg = error.error instanceof ErrorEvent ? error.error.message : (typeof error.error === 'string' ? error.error : (error.error?.msg || error.message || 'Request failed')); return of({ status: 'error', msg }); } public add(payload: AddDownloadPayload) { return this.http.post('add', { url: payload.url, download_type: payload.downloadType, codec: payload.codec, quality: payload.quality, format: payload.format, folder: payload.folder, custom_name_prefix: payload.customNamePrefix, playlist_item_limit: payload.playlistItemLimit, auto_start: payload.autoStart, split_by_chapters: payload.splitByChapters, chapter_template: payload.chapterTemplate, subtitle_language: payload.subtitleLanguage, subtitle_mode: payload.subtitleMode, ytdl_options_presets: payload.ytdlOptionsPresets, ytdl_options_overrides: payload.ytdlOptionsOverrides, }).pipe( catchError(this.handleHTTPError) ); } public getPresets() { return this.http.get<{ presets: string[] }>('presets').pipe( catchError(() => of({ presets: [] })) ); } public startById(ids: string[]) { return this.http.post('start', {ids: ids}); } public delById(where: State, ids: string[]) { const map = this[where]; if (map) { for (const id of ids) { const obj = map.get(id); if (obj) { obj.deleting = true; } } } return this.http.post('delete', {where: where, ids: ids}); } public startByFilter(where: State, filter: (dl: Download) => boolean) { const ids: string[] = []; this[where].forEach((dl: Download) => { if (filter(dl)) ids.push(dl.url) }); return this.startById(ids); } public delByFilter(where: State, filter: (dl: Download) => boolean) { const ids: string[] = []; this[where].forEach((dl: Download) => { if (filter(dl)) ids.push(dl.url) }); return this.delById(where, ids); } public cancelAdd() { return this.http.post('cancel-add', {}).pipe( catchError(this.handleHTTPError) ); } uploadCookies(file: File) { const formData = new FormData(); formData.append('cookies', file); return this.http.post<{ status: string; msg?: string }>('upload-cookies', formData).pipe( catchError(this.handleHTTPError) ); } deleteCookies() { return this.http.post<{ status: string; msg?: string }>('delete-cookies', {}).pipe( catchError(this.handleHTTPError) ); } getCookieStatus() { return this.http.get<{ status: string; has_cookies: boolean }>('cookie-status').pipe( catchError(this.handleHTTPError) ); } }