Merge pull request #39 from nuthx/dev-0.0.11

Dev 0.0.11

Author: nuthx

.dockerignore

@@ -4,6 +4,7 @@ node_modules
 .env
 .env.local
 .vscode
+.idea
 .DS_Store
 database
 logs

.gitignore

@@ -28,15 +28,12 @@ yarn-debug.log*
 yarn-error.log*
 .pnpm-debug.log*
 
-# env files (can opt-in for committing if needed)
+# env
 .env*
 
 # vercel
 .vercel
 
-# typescript
-*.tsbuildinfo
-next-env.d.ts
-
-# VSCode
+# ide
 .vscode
+.idea

app/anime/[hash]/page.js

@@ -0,0 +1,44 @@
+import { prisma } from "@/lib/db";
+import { dynamicImport } from "@/lib/core/dynamic";
+import { sendResponse } from "@/lib/http/response";
+
+// Get anime description from source url
+// Params: hash, string, required
+export async function GET(request, { params }) {
+  try {
+    const hash = (await params).hash;
+
+    const anime = await prisma.anime.findUnique({
+      where: {
+        hash: hash,
+      },
+      include: {
+        rss: { select: { name: true } },
+      },
+    });
+
+    if (!anime) {
+      throw new Error("No anime found");
+    }
+
+    // Dynamic import
+    const { parserModule } = await dynamicImport(anime.source);
+
+    // Get source content from source url
+    const res = await fetch(anime.sourceUrl);
+    const html = await res.text();
+    const desc = parserModule.extractDescription(html);
+
+    return sendResponse(request, {
+      data: {
+        type: anime.source,
+        content: desc
+      }
+    });
+  } catch (error) {
+    return sendResponse(request, {
+      code: 500,
+      message: error.message
+    });
+  }
+}

app/anime/page.js

@@ -0,0 +1,82 @@
+import { prisma } from "@/lib/db";
+import { getAnimeDetailFromAnilist } from "@/lib/api/anilist";
+import { getAnimeDetailFromBangumi } from "@/lib/api/bangumi";
+import { sendResponse } from "@/lib/http/response";
+
+// Refresh anime info by manually input anilist or bangumi id
+// Body: {
+//   anilist_id: string, optional
+//   bangumi_id: string, optional
+// }
+export async function POST(request, { params }) {
+  try {
+    const data = await request.json();
+    const hash = (await params).hash;
+    let newAnilist = null;
+    let newBangumi = null;
+
+    const anime = await prisma.anime.findUnique({
+      where: { hash },
+    });
+
+    if (!anime) {
+      throw new Error("No anime found");
+    }
+
+    if (
+      (!data.anilist_id || data.anilist_id === anime.idAnilist) && 
+      (!data.bangumi_id || data.bangumi_id === anime.idBangumi)
+    ) {
+      throw new Error("No change");
+    }
+
+    if (data.anilist_id) {
+      newAnilist = await getAnimeDetailFromAnilist(data.anilist_id);
+    }
+
+    if (data.bangumi_id) {
+      newBangumi = await getAnimeDetailFromBangumi(data.bangumi_id);
+    }
+
+    if (newAnilist) {
+      await prisma.anime.update({
+        where: { hash },
+        data: {
+          titleJp: newAnilist?.title?.native,
+          titleEn: newAnilist?.title?.english,
+          titleRomaji: newAnilist?.title?.romaji,
+          coverAnilist: newAnilist?.coverImage?.extraLarge,
+          idAnilist: data.anilist_id
+        }
+      });
+    }
+
+    if (newBangumi) {
+      await prisma.anime.update({
+        where: { hash },
+        data: {
+          titleJp: newBangumi?.name,  // Japanese title in bangumi will replace the title in anilist
+          titleCn: newBangumi?.name_cn,
+          coverBangumi: newBangumi?.images?.large,
+          idBangumi: data.bangumi_id
+        }
+      });
+    }
+
+    return sendResponse(request, {
+      message: "Anime info updated successfully"
+    });
+  } catch (error) {
+    if (error.message === "No change") {
+      return sendResponse(request, {
+        code: 240,
+        message: error.message
+      });
+    }
+
+    return sendResponse(request, {
+      code: 500,
+      message: error.message
+    });
+  }
+}

app/api/anime/[hash]/desc/route.js

@@ -0,0 +1,36 @@
+import { prisma } from "@/lib/db";
+import { getTitleFirst } from "@/lib/title";
+import { sendResponse } from "@/lib/http/response";
+
+// Get anime detail by hash
+// Params: hash, string, required
+export async function GET(request, { params }) {
+  try {
+    const hash = (await params).hash;
+
+    const anime = await prisma.anime.findUnique({
+      where: {
+        hash: hash,
+      },
+      include: {
+        rss: { select: { name: true } },
+      },
+    });
+
+    if (!anime) {
+      throw new Error("No anime found");
+    }
+
+    return sendResponse(request, {
+      data: {
+        ...anime,
+        titleFirst: await getTitleFirst(anime)
+      }
+    });
+  } catch (error) {
+    return sendResponse(request, {
+      code: 500,
+      message: error.message
+    });
+  }
+}

app/api/anime/[hash]/reanalysis/route.js

@@ -1,4 +1,5 @@
-import { prisma, getConfig } from "@/lib/db";
+import { prisma } from "@/lib/db";
+import { getTitleFirst } from "@/lib/title";
 import { sendResponse } from "@/lib/http/response";
 
 // Get anime list with pagination
@@ -30,11 +31,7 @@ export async function GET(request) {
           pubDate: "desc"
         },
         include: {
-          rss: {
-            select: {
-              name: true
-            }
-          }
+          rss: { select: { name: true } }
         }
       }),
       // Get total count
@@ -67,26 +64,14 @@ export async function GET(request) {
       })
     ]);
 
-    // Add titleFirst to each anime item
-    const config = await getConfig();
-    const priorities = config.animeTitlePriority.split(",");
-    const animeWithTitleFirst = anime.map(item => ({
-      ...item,
-      titleFirst: priorities.reduce((title, p) => {
-        if (title) return title;
-        const titleMap = {
-          jp: item.titleJp,
-          romaji: item.titleRomaji,
-          cn: item.titleCn,
-          en: item.titleEn
-        };
-        return titleMap[p];
-      }, "") || item.titleParsed || item.titleRaw
-    }));
-
     return sendResponse(request, {
       data: {
-        anime: animeWithTitleFirst,
+        anime: await Promise.all(
+          anime.map(async (item) => ({
+            ...item,
+            titleFirst: await getTitleFirst(item)
+          }))
+        ),
         rss: {
           current: rss || "",
           list: rssList.map(r => r.name)

app/api/anime/[hash]/route.js

@@ -1,7 +1,7 @@
 import { prisma } from "@/lib/db";
-import { formatBytes } from "@/lib/bytes";
+import { formatBytes, formatEta } from "@/lib/format";
 import { sendResponse } from "@/lib/http/response";
-import { getQbittorrentVersion, getQbittorrentTorrents, manageQbittorrentTorrent } from "@/lib/api/qbittorrent";
+import { QB_STATE, getQbittorrentVersion, getQbittorrentTorrents, manageQbittorrentTorrent } from "@/lib/api/qbittorrent";
 
 // Get torrent list
 
@@ -27,10 +27,12 @@ export async function GET(request) {
           name: torrent.name,
           hash: torrent.hash,
           state: torrent.state,
+          state_class: Object.keys(QB_STATE).find(key => QB_STATE[key].includes(torrent.state)) || "stalled",
           progress: torrent.progress,
           eta: torrent.eta,
-          dlspeed: formatBytes(torrent.dlspeed),
-          upspeed: formatBytes(torrent.upspeed),
+          eta_dict: formatEta(torrent.eta),
+          dl_speed: formatBytes(torrent.dl_speed),
+          up_speed: formatBytes(torrent.up_speed),
           completed: formatBytes(torrent.completed),
           size: formatBytes(torrent.size),
           added_on: torrent.added_on,

app/api/anime/route.js

@@ -29,7 +29,7 @@ import { Pause, RefreshCcw, Trash2 } from "lucide-react";
 export default function Home() {
   const { t } = useTranslation();
 
-  const { data: torrentsData, error: torrentsError, isLoading: torrentsLoading, mutate: mutateTorrents } = useData(API.TORRENTS, null, { refreshInterval: 1000 });
+  const { data: torrentsData, error: torrentsError, isLoading: torrentsLoading, mutate: torrentsMutate } = useData(API.TORRENTS, null, { refreshInterval: 1000 });
 
   // Set page title
   useEffect(() => {
@@ -39,7 +39,7 @@ export default function Home() {
   const handleManage = async (action, downloader, hash) => {
     const result = await handleRequest("POST", API.TORRENTS, { action, downloader, hash }, t(`toast.failed.${action}`));
     if (result) {
-      mutateTorrents();
+      torrentsMutate();
     }
   };
 
@@ -74,8 +74,8 @@ export default function Home() {
               </div>
               <a className="font-medium">{item.name}</a>
               <div className="flex items-center gap-2">
-                <a className="w-1/6 text-sm text-muted-foreground">{t("downloads.d_speed")}: {item.dlspeed}/s</a>
-                <a className="w-1/6 text-sm text-muted-foreground">{t("downloads.u_speed")}: {item.upspeed}/s</a>
+                <a className="w-1/6 text-sm text-muted-foreground">{t("downloads.d_speed")}: {item.dl_speed}/s</a>
+                <a className="w-1/6 text-sm text-muted-foreground">{t("downloads.u_speed")}: {item.up_speed}/s</a>
                 {item.eta !== 8640000 && <a className="w-2/6 text-sm text-muted-foreground">
                   {t("downloads.eta")}: {Math.floor(item.eta/86400) > 0 && `${Math.floor(item.eta/86400)} ${t("downloads.d")} `}
                   {Math.floor((item.eta%86400)/3600) > 0 && `${Math.floor((item.eta%86400)/3600)} ${t("downloads.h")} `}