import { faCheck, faCircleNotch, faExclamationTriangle, faInfoCircle, faTrash, } from "@fortawesome/free-solid-svg-icons"; import { FontAwesomeIcon } from "@fortawesome/react-fontawesome"; import React, { FunctionComponent, useCallback, useEffect, useMemo, useState, } from "react"; import { Button, Container, Form } from "react-bootstrap"; import { Column, TableUpdater } from "react-table"; import { AsyncButton, FileForm, LanguageSelector, MessageIcon, SimpleTable, useCloseModal, usePayload, } from ".."; import { useProfileBy, useProfileItemsToLanguages } from "../../@redux/hooks"; import { EpisodesApi, SubtitlesApi } from "../../apis"; import { Selector } from "../inputs"; import BaseModal, { BaseModalProps } from "./BaseModal"; enum State { Update, Valid, Warning, Error, } interface PendingSubtitle { form: FormType.UploadSubtitle; didCheck: boolean; instance?: Item.Episode; } type SubtitleState = { state: State; infos: string[]; }; type ProcessState = { [name: string]: SubtitleState; }; type EpisodeMap = { [name: string]: Item.Episode; }; interface SerieProps { episodes: readonly Item.Episode[]; } const SeriesUploadModal: FunctionComponent = ({ episodes, ...modal }) => { const series = usePayload(modal.modalKey); const [uploading, setUpload] = useState(false); const closeModal = useCloseModal(); const [pending, setPending] = useState([]); const [processState, setProcessState] = useState({}); const profile = useProfileBy(series?.profileId); const languages = useProfileItemsToLanguages(profile); const filelist = useMemo(() => pending.map((v) => v.form.file), [pending]); // Vaildate useEffect(() => { const states = pending.reduce((prev, info) => { const subState: SubtitleState = { state: State.Valid, infos: [], }; const { form, instance } = info; if (!info.didCheck) { subState.state = State.Update; } else if (!instance) { subState.infos.push("Season or episode info is missing"); subState.state = State.Error; } else { if ( instance.subtitles.find((v) => v.code2 === form.language) !== undefined ) { subState.infos.push("Overwrite existing subtitle"); subState.state = State.Warning; } } prev[form.file.name] = subState; return prev; }, {}); setProcessState(states); }, [pending]); const checkEpisodes = useCallback( async (list: PendingSubtitle[]) => { const names = list.map((v) => v.form.file.name); if (names.length > 0) { const results = await SubtitlesApi.info(names); const episodeMap = results.reduce((prev, curr) => { const ep = episodes.find( (v) => v.season === curr.season && v.episode === curr.episode ); if (ep) { prev[curr.filename] = ep; } return prev; }, {}); setPending((pd) => pd.map((v) => ({ ...v, didCheck: true, instance: episodeMap[v.form.file.name], })) ); } }, [episodes] ); const updateLanguage = useCallback( (lang: Nullable) => { if (lang) { const list = pending.map((v) => { const form = v.form; return { ...v, form: { ...form, language: lang.code2, hi: lang.hi ?? false, forced: lang.forced ?? false, }, }; }); setPending(list); } }, [pending] ); const setFiles = useCallback( (files: File[]) => { // At lease 1 language is required const lang = languages[0]; const list: PendingSubtitle[] = files.map((f) => { return { form: { file: f, language: lang.code2, hi: lang.hi ?? false, forced: lang.forced ?? false, }, didCheck: false, }; }); setPending(list); const states = files.reduce( (v, curr) => ({ ...v, [curr.name]: { state: State.Update, infos: [] }, }), {} ); setProcessState(states); checkEpisodes(list); }, [languages, checkEpisodes] ); const uploadSubtitles = useCallback(async () => { if (series === null) { return; } const { sonarrSeriesId: seriesid } = series; let uploadStates = pending.reduce((prev, curr) => { prev[curr.form.file.name] = { state: State.Update, infos: [] }; return prev; }, {}); setProcessState(uploadStates); let exception = false; for (const info of pending) { if (info.instance) { const { sonarrEpisodeId: episodeid } = info.instance; try { await EpisodesApi.uploadSubtitles(seriesid, episodeid, info.form); uploadStates = { ...uploadStates, [info.form.file.name]: { state: State.Valid, infos: [] }, }; } catch (error) { uploadStates = { ...uploadStates, [info.form.file.name]: { state: State.Error, infos: [] }, }; exception = true; } setProcessState(uploadStates); } } if (exception) { throw new Error("Error when uploading subtitles"); } }, [series, pending]); const canUpload = useMemo( () => pending.length > 0 && pending.every((v) => v.instance !== undefined), [pending] ); const tableShow = pending.length > 0; const columns = useMemo[]>( () => [ { id: "Icon", accessor: "instance", className: "text-center", Cell: ({ row, loose }) => { const { form: { file }, } = row.original; const name = file.name; const states = loose![1] as ProcessState; let icon = faCircleNotch; let color: string | undefined = undefined; let spin = false; let msgs: string[] = []; if (name in states) { const state = states[name]; msgs = state.infos; switch (state.state) { case State.Error: icon = faExclamationTriangle; color = "var(--danger)"; break; case State.Valid: icon = faCheck; color = "var(--success)"; break; case State.Warning: icon = faInfoCircle; color = "var(--warning)"; break; case State.Update: spin = true; break; default: break; } } return ( ); }, }, { Header: "File", accessor: (d) => d.form.file.name, }, { Header: "Episode", accessor: "instance", className: "vw-1", Cell: ({ value, loose, row, externalUpdate }) => { const uploading = loose![0] as boolean; const availables = loose![2] as Item.Episode[]; const options = availables.map>( (ep) => ({ label: `(${ep.season}x${ep.episode}) ${ep.title}`, value: ep, }) ); const change = useCallback( (ep: Nullable) => { if (ep) { const newInfo = { ...row.original }; newInfo.instance = ep; externalUpdate && externalUpdate(row, newInfo); } }, [row, externalUpdate] ); return ( ); }, }, { accessor: "form", Cell: ({ row, externalUpdate, loose }) => { const [uploading] = loose!; return ( ); }, }, ], [] ); const updateItem = useCallback>( (row, info?: PendingSubtitle) => { setPending((pd) => { const newPending = [...pd]; if (info) { newPending[row.index] = info; } else { newPending.splice(row.index, 1); } return newPending; }); }, [] ); const footer = (
0 ? languages[0] : undefined} onChange={updateLanguage} >
{ closeModal(); setFiles([]); }} > Upload
); return (
); }; export default SeriesUploadModal;