✨ Add new admin routes for better queue management

Open-Webtoon-Reader · May 15, 2024 · a69973b · a69973b
1 parent e0c6a32
commit a69973b
Show file tree

Hide file tree

Showing 2 changed files with 58 additions and 4 deletions.
diff --git a/src/modules/webtoon/admin/admin.controller.ts b/src/modules/webtoon/admin/admin.controller.ts
@@ -1,8 +1,9 @@
-import {Body, Controller, Post} from "@nestjs/common";
+import {Body, Controller, Delete, Get, HttpCode, Post} from "@nestjs/common";
 import {ApiResponse, ApiTags} from "@nestjs/swagger";
 import {DownloadManagerService} from "../webtoon/download-manager.service";
 import {AddWebtoonToQueueDto} from "./models/dto/add-webtoon-to-queue.dto";
 import {HttpStatusCode} from "axios";
+import CachedWebtoonModel from "../webtoon/models/models/cached-webtoon.model";
 
 
 @Controller("admin")
@@ -14,16 +15,44 @@ export class AdminController{
     ){}
 
     @Post("queue")
-    @ApiResponse({status: HttpStatusCode.Ok, description: "Adds a webtoon to the download queue"})
+    @ApiResponse({status: HttpStatusCode.Created, description: "Adds a webtoon to the download queue"})
     @ApiResponse({status: HttpStatusCode.TooEarly, description: "Cache not loaded."})
     async addWebtoonToQueue(@Body() addWebtoonToQueueDto: AddWebtoonToQueueDto): Promise<void>{
         return this.downloadManagerService.addWebtoonToQueue(addWebtoonToQueueDto.name, addWebtoonToQueueDto.language);
     }
 
     @Post("update/all")
-    @ApiResponse({status: HttpStatusCode.Ok, description: "Updates all webtoons in the database"})
+    @ApiResponse({status: HttpStatusCode.Created, description: "Updates all webtoons in the database"})
     @ApiResponse({status: HttpStatusCode.TooEarly, description: "Cache not loaded."})
     async updateAllWebtoons(): Promise<void>{
         return this.downloadManagerService.updateAllWebtoons();
     }
+
+    @Get("current-download")
+    @ApiResponse({status: HttpStatusCode.Ok, description: "Returns the current download"})
+    @ApiResponse({status: HttpStatusCode.NotFound, description: "No download in progress"})
+    async getCurrentDownload(): Promise<CachedWebtoonModel>{
+        return this.downloadManagerService.getCurrentDownload();
+    }
+
+    @Get("queue")
+    @ApiResponse({status: HttpStatusCode.Ok, description: "Returns the current download queue"})
+    @ApiResponse({status: HttpStatusCode.NotFound, description: "No download in progress"})
+    async getQueue(): Promise<CachedWebtoonModel[]>{
+        return this.downloadManagerService.getDownloadQueue();
+    }
+
+    @Delete("current-download")
+    @HttpCode(HttpStatusCode.NoContent)
+    @ApiResponse({status: HttpStatusCode.NoContent, description: "Cancels the current download"})
+    async cancelCurrentDownload(): Promise<void>{
+        return this.downloadManagerService.skipCurrentDownload();
+    }
+
+    @Delete("queue")
+    @HttpCode(HttpStatusCode.NoContent)
+    @ApiResponse({status: HttpStatusCode.NoContent, description: "Clears the download queue"})
+    async clearQueue(): Promise<void>{
+        return this.downloadManagerService.clearDownloadQueue();
+    }
 }
diff --git a/src/modules/webtoon/webtoon/download-manager.service.ts b/src/modules/webtoon/webtoon/download-manager.service.ts
@@ -4,7 +4,7 @@ import {WebtoonParserService} from "./webtoon-parser.service";
 import CachedWebtoonModel from "./models/models/cached-webtoon.model";
 import EpisodeModel from "./models/models/episode.model";
 import EpisodeDataModel from "./models/models/episode-data.model";
-import {HttpException, Injectable} from "@nestjs/common";
+import {HttpException, Injectable, NotFoundException} from "@nestjs/common";
 import WebtoonModel from "./models/models/webtoon.model";
 import WebtoonDataModel from "./models/models/webtoon-data.model";
 import WebtoonQueue from "../../../common/utils/models/webtoon-queue";
@@ -71,10 +71,35 @@ export class DownloadManagerService{
             const startEpisode: number = await this.webtoonDatabase.getLastSavedEpisodeNumber(this.currentDownload.title, this.currentDownload.language);
             const epList: EpisodeModel[] = await this.webtoonParser.getEpisodes(this.currentDownload);
             for(let i = startEpisode; i < epList.length; i++){
+                if(!this.currentDownload)
+                    break;
                 const epImageLinks: string[] = await this.webtoonParser.getEpisodeLinks(this.currentDownload, epList[i]);
                 const episodeData: EpisodeDataModel = await this.webtoonDownloader.downloadEpisode(epList[i], epImageLinks);
+                if(!this.currentDownload)
+                    break;
                 await this.webtoonDatabase.saveEpisode(this.currentDownload, epList[i], episodeData);
             }
         }
     }
+
+    getCurrentDownload(): CachedWebtoonModel | undefined{
+        if(this.currentDownload)
+            return this.currentDownload;
+        throw new NotFoundException("No download in progress.");
+    }
+
+    getDownloadQueue(): CachedWebtoonModel[]{
+        if(!this.currentDownload)
+            throw new NotFoundException("No download in progress.");;
+        return [this.currentDownload, ...this.queue.getElements()];
+    }
+
+    skipCurrentDownload(): void{
+        this.currentDownload = undefined;
+    }
+
+    clearDownloadQueue(){
+        this.queue.clear();
+        this.currentDownload = undefined;
+    }
 }