From 8f23240ea14f6eb0844956da097060b780811c0d Mon Sep 17 00:00:00 2001 From: Jip Fr Date: Sat, 11 Feb 2023 00:43:38 +0100 Subject: [PATCH] Get started on migration --- src/state/watched/context.tsx | 2 +- src/state/watched/store.ts | 2 +- src/views/search/HomeView.tsx | 219 +++++++++++++++++++++++++++++++++- 3 files changed, 219 insertions(+), 4 deletions(-) diff --git a/src/state/watched/context.tsx b/src/state/watched/context.tsx index eff5a5c1..62da316d 100644 --- a/src/state/watched/context.tsx +++ b/src/state/watched/context.tsx @@ -44,7 +44,7 @@ interface MediaItem { }; } -interface WatchedStoreItem { +export interface WatchedStoreItem { item: MediaItem; progress: number; percentage: number; diff --git a/src/state/watched/store.ts b/src/state/watched/store.ts index aada4131..075e9e43 100644 --- a/src/state/watched/store.ts +++ b/src/state/watched/store.ts @@ -1,7 +1,7 @@ import { versionedStoreBuilder } from "@/utils/storage"; export const VideoProgressStore = versionedStoreBuilder() - .setKey("video-progress") + .setKey("video-progress-v3") .addVersion({ version: 0, }) diff --git a/src/views/search/HomeView.tsx b/src/views/search/HomeView.tsx index b5a17843..d07bdd1f 100644 --- a/src/views/search/HomeView.tsx +++ b/src/views/search/HomeView.tsx @@ -6,11 +6,19 @@ import { getIfBookmarkedFromPortable, useBookmarkContext, } from "@/state/bookmark"; -import { useWatchedContext } from "@/state/watched"; +import { + useWatchedContext, + WatchedStoreData, + WatchedStoreItem, +} from "@/state/watched"; import { WatchedMediaCard } from "@/components/media/WatchedMediaCard"; import { EditButton } from "@/components/buttons/EditButton"; -import { useState } from "react"; +import { useCallback, useState } from "react"; import { useAutoAnimate } from "@formkit/auto-animate/react"; +import { VideoProgressStore } from "@/state/watched/store"; +import { searchForMedia } from "@/backend/metadata/search"; +import { DetailedMeta, getMetaFromId } from "@/backend/metadata/getmeta"; +import { MWMediaMeta, MWMediaType } from "@/backend/metadata/types"; function Bookmarks() { const { t } = useTranslation(); @@ -43,6 +51,30 @@ function Bookmarks() { ); } +interface OldMediaBase { + mediaId: number; + mediaType: MWMediaType; + percentage: number; + progress: number; + providerId: string; + title: string; + year: number; +} + +interface OldMovie extends OldMediaBase { + mediaType: MWMediaType.MOVIE; +} + +interface OldSeries extends OldMediaBase { + mediaType: MWMediaType.SERIES; + episodeId: number; + seasonId: number; +} + +interface OldData { + items: (OldMovie | OldSeries)[]; +} + function Watched() { const { t } = useTranslation(); const { getFilteredBookmarks } = useBookmarkContext(); @@ -55,6 +87,189 @@ function Watched() { (v) => !getIfBookmarkedFromPortable(bookmarks, v.item.meta) ); + /* + AAA + */ + const watchedLocalstorage = VideoProgressStore.get(); + const [watched, setWatchedReal] = useState( + watchedLocalstorage as WatchedStoreData + ); + + const setWatched = useCallback( + (data: any) => { + setWatchedReal((old) => { + let newData = data; + if (data.constructor === Function) { + newData = data(old); + } + watchedLocalstorage.save(newData); + return newData; + }); + }, + [setWatchedReal, watchedLocalstorage] + ); + + (async () => { + const oldData: OldData | null = localStorage.getItem("video-progress") + ? JSON.parse(localStorage.getItem("video-progress") || "") + : null; + + if (!oldData) return; + + const uniqueMedias: Record = {}; + oldData.items.forEach((item: any) => { + if (uniqueMedias[item.mediaId]) return; + uniqueMedias[item.mediaId] = item; + }); + + const yearsAreClose = (a: number, b: number) => { + return Math.abs(a - b) <= 1; + }; + + const mediaMetas: Record> = {}; + + Promise.all( + Object.values(uniqueMedias).map(async (item) => { + const year = Number(item.year.toString().split("-")[0]); + const data = await searchForMedia({ + searchQuery: `${item.title} ${year}`, + type: item.mediaType, + }); + const relevantItem = data.find((res) => + yearsAreClose(Number(res.year), year) + ); + if (!relevantItem) { + console.error("No item"); + return; + } + return { + id: item.mediaId, + data: relevantItem, + }; + }) + ).then(async (relevantItems) => { + console.log(relevantItems); + for (const item of relevantItems.filter(Boolean)) { + if (!item) continue; + + let keys: (string | null)[][] = [["0", "0"]]; + if (item.data.type === "series") { + const meta = await getMetaFromId(item.data.type, item.data.id); + if (!meta || !meta?.meta.seasons) return; + const seasonNumbers = [ + ...new Set( + oldData.items + .filter((watchedEntry: any) => watchedEntry.mediaId === item.id) + .map((watchedEntry: any) => watchedEntry.seasonId) + ), + ]; + const seasons = seasonNumbers + .map((num) => ({ + num, + season: meta.meta?.seasons?.[(num as number) - 1], + })) + .filter(Boolean); + keys = seasons + .map((season) => (season ? [season.num, season?.season?.id] : [])) + .filter((entry) => entry.length > 0); // Stupid TypeScript + } + + if (!mediaMetas[item.id]) mediaMetas[item.id] = {}; + await Promise.all( + keys.map(async ([key, id]) => { + if (!key) return; + mediaMetas[item.id][key] = await getMetaFromId( + item.data.type, + item.data.id, + id === "0" || id === null ? undefined : id + ); + }) + ); + } + + // We've got all the metadata you can dream of now + // Now let's convert stuff into the new format. + const newData: WatchedStoreData = JSON.parse(JSON.stringify(watched)); + + for (const oldWatched of oldData.items) { + if (oldWatched.mediaType === "movie") { + if (!mediaMetas[oldWatched.mediaId]["0"]?.meta) continue; + + const newItem: WatchedStoreItem = { + item: { + meta: mediaMetas[oldWatched.mediaId]["0"]?.meta as MWMediaMeta, + }, + progress: oldWatched.progress, + percentage: oldWatched.percentage, + watchedAt: Date.now(), // There was no watchedAt in V2 + }; + if ( + newData.items.find( + (item) => item.item.meta.id === newItem.item.meta.id + ) + ) + continue; + + oldData.items = oldData.items.filter( + (item) => JSON.stringify(item) !== JSON.stringify(oldWatched) + ); + newData.items.push(newItem); + } else if (oldWatched.mediaType === "series") { + // console.log(oldWatched); + // console.log(mediaMetas[oldWatched.mediaId][oldWatched.seasonId]); + + if (!mediaMetas[oldWatched.mediaId][oldWatched.seasonId]?.meta) + continue; + + const meta = mediaMetas[oldWatched.mediaId][oldWatched.seasonId] + ?.meta as MWMediaMeta; + + if (meta.type !== "series") return; + + // console.log(meta.seasonData); + const newItem: WatchedStoreItem = { + item: { + meta, + series: { + episode: Number(oldWatched.episodeId), + season: Number(oldWatched.seasonId), + seasonId: meta.seasonData.id, + episodeId: + meta.seasonData.episodes[Number(oldWatched.episodeId) - 1].id, + }, + }, + progress: oldWatched.progress, + percentage: oldWatched.percentage, + watchedAt: Date.now(), // There was no watchedAt in V2 + }; + + if ( + newData.items.find( + (item) => + item.item.meta.id === newItem.item.meta.id && + item.item.series?.episodeId === newItem.item.series?.episodeId + ) + ) + continue; + + oldData.items = oldData.items.filter( + (item) => JSON.stringify(item) !== JSON.stringify(oldWatched) + ); + newData.items.push(newItem); + } + } + + if (JSON.stringify(newData) !== JSON.stringify(watched)) { + localStorage.setItem("video-progress", JSON.stringify(oldData)); + setWatched(() => newData); + } + }); + })(); + + /* + AAA + */ + if (watchedItems.length === 0) return null; return (